@@ -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 winazurestorage.py 132c7dcdcc51549576cc6f2a1dfbc35a75339b9e Steve Marx Steve.Marx@microsoft.com http://github.com/sriramk/winazurestorage/commit/0c22ac7e0c1b3b8cb7bd6628492f388782e688ef 0c22ac7e0c1b3b8cb7bd6628492f388782e688ef 2009-04-14T13:11:37-07:00 2009-04-14T13:11:37-07:00 adding enqueue and dequeue support 626bee8349da677696f7c2fde1fba07e5dd2291b Steve Marx Steve.Marx@microsoft.com