提交: 0c22ac7e0c1b3b8cb7bd6628492f388782e688ef - winazurestorage (git)

winazurestorageのフォーク

Commit MetaInfo

• 修订版: 0c22ac7e0c1b3b8cb7bd6628492f388782e688ef (tree)
• 时间: 2009-04-15 05:11:37
• 作者: Steve Marx <Steve.Marx@micr...>
• Commiter: Steve Marx

Log Message

adding enqueue and dequeue support

更改概述

• modified: winazurestorage.py

差异

--- a/winazurestorage.py
+++ b/winazurestorage.py
@@ -83,7 +83,7 @@ class SharedKeyCredentials(object):
         if not for_tables:
             string_to_sign += canonicalized_headers + NEW_LINE   # Canonicalized headers
         string_to_sign += canonicalized_resource                 # Canonicalized resource
-
+        
         request.add_header('Authorization', 'SharedKey ' + self._account + ':' + base64.encodestring(hmac.new(self._key, unicode(string_to_sign).encode("utf-8"), hashlib.sha256).digest()).strip())
         return request
 
@@ -126,6 +126,8 @@ class Storage(object):
 
 class TableEntity(object): pass
 
+class QueueMessage(): pass
+
 class QueueStorage(Storage):
     def __init__(self, host, account_name, secret_key, use_path_style_uris = None):
         super(QueueStorage, self).__init__(host, account_name, secret_key, use_path_style_uris)
@@ -148,6 +150,44 @@ class QueueStorage(Storage):
             return response.code
         except URLError, e:
             return e.code
+            
+    def put_message(self, queue_name, payload):
+        data = "<QueueMessage><MessageText>%s</MessageText></QueueMessage>" % base64.encodestring(payload)
+        req = RequestWithMethod("POST", "%s/%s/messages" % (self.get_base_url(), queue_name), data=data)
+        req.add_header("Content-Type", "application/xml")
+        req.add_header("Content-Length", len(data))
+        self._credentials.sign_request(req)
+        try:
+            response = urlopen(req)
+            return response.code
+        except URLError, e:
+            return e.code
+
+    def get_message(self, queue_name):
+        req = Request("%s/%s/messages" % (self.get_base_url(), queue_name))
+        self._credentials.sign_request(req)
+        response = urlopen(req)
+        dom = minidom.parseString(response.read())
+        messages = dom.getElementsByTagName("QueueMessage")
+        result = None
+        if len(messages) == 1:
+            message = messages[0]
+            result = QueueMessage()
+            result.id = message.getElementsByTagName("MessageId")[0].firstChild.data
+            result.pop_receipt = message.getElementsByTagName("PopReceipt")[0].firstChild.data
+            result.text = base64.decodestring(message.getElementsByTagName("MessageText")[0].firstChild.data)
+        return result
+
+    def delete_message(self, queue_name, message):
+        id = message.id
+        pop_receipt = message.pop_receipt
+        req = RequestWithMethod("DELETE", "%s/%s/messages/%s?popreceipt=%s" % (self.get_base_url(), queue_name, id, pop_receipt))
+        self._credentials.sign_request(req)
+        try:
+            response = urlopen(req)
+            return response.code
+        except URLError, e:
+            return e.code
 
 class TableStorage(Storage):
     '''Due to local development storage not supporting SharedKey authentication, this class