mns/mns_client.py (564 lines of code) (raw):
#coding=utf-8
# Copyright (C) 2015, Alibaba Cloud Computing
#Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
#The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
#THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
import time
import hashlib
import hmac
import base64
import string
import platform
from . import pkg_info
from .mns_xml_handler import *
from .mns_exception import *
from .mns_request import *
from .mns_tool import *
from .mns_http import *
from .auth import *
#from mns.mns_xml_handler import *
#from mns.mns_exception import *
#from mns.mns_request import *
#from mns.mns_tool import *
#from mns.mns_http import *
URISEC_QUEUE = "queues"
URISEC_MESSAGE = "messages"
URISEC_TOPIC = "topics"
URISEC_SUBSCRIPTION = "subscriptions"
class MNSClient(object):
#__metaclass__ = type
def __init__(self, host, access_id="", access_key="", version="2015-06-06", security_token="",
logger=None, credentials_provider=None):
self.host, self.is_https = self.process_host(host)
if credentials_provider:
self.credentials_provider = credentials_provider
else:
self.credentials_provider = StaticCredentialsProvider(access_id, access_key, security_token)
self.version = version
self.logger = logger
self.http = MNSHttp(self.host, logger=logger, is_https=self.is_https)
if self.logger:
self.logger.info("InitClient Host:%s Version:%s" % (host, version))
def set_log_level(self, log_level):
if self.logger:
MNSLogger.validate_loglevel(log_level)
self.logger.setLevel(log_level)
self.http.set_log_level(log_level)
def close_log(self):
self.logger = None
self.http.close_log()
def set_connection_timeout(self, connection_timeout):
self.http.set_connection_timeout(connection_timeout)
def set_keep_alive(self, keep_alive):
self.http.set_keep_alive(keep_alive)
def close_connection(self):
self.http.conn.close()
#===============================================queue operation===============================================#
def set_account_attributes(self, req, resp):
#check parameter
SetAccountAttributesValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/?accountmeta=true")
req_inter.data = SetAccountAttrEncoder.encode(req)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
def get_account_attributes(self, req, resp):
#make request internal
req_inter = RequestInternal(req.method, "/?accountmeta=true")
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
account_attr = GetAccountAttrDecoder.decode(resp_inter.data, req_inter.get_req_id())
resp.logging_bucket = account_attr["LoggingBucket"]
if self.logger:
self.logger.info("GetAccountAttributes RequestId:%s LoggingBucket:%s" % (resp.get_requestid(), resp.logging_bucket))
def create_queue(self, req, resp):
#check parameter
CreateQueueValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s" % (URISEC_QUEUE, req.queue_name))
req_inter.data = QueueEncoder.encode(req)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
lheader = self.lower_header(resp.header)
resp.queue_url = lheader["location"]
if self.logger:
self.logger.info("CreateQueue RequestId:%s QueueName:%s QueueURL:%s" % \
(resp.get_requestid(), req.queue_name, resp.queue_url))
def delete_queue(self, req, resp):
#check parameter
DeleteQueueValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s" % (URISEC_QUEUE, req.queue_name))
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if self.logger:
self.logger.info("DeleteQueue RequestId:%s QueueName:%s" % (resp.get_requestid(), req.queue_name))
def list_queue(self, req, resp):
#check parameter
ListQueueValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s" % URISEC_QUEUE)
if req.prefix != u"":
req_inter.header["x-mns-prefix"] = req.prefix
if req.ret_number != -1:
req_inter.header["x-mns-ret-number"] = str(req.ret_number)
if req.marker != u"":
req_inter.header["x-mns-marker"] = str(req.marker)
if req.with_meta:
req_inter.header["x-mns-with-meta"] = u"true"
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
resp.queueurl_list, resp.next_marker, resp.queuemeta_list = ListQueueDecoder.decode(resp_inter.data, req.with_meta, req_inter.get_req_id())
if self.logger:
firstQueueURL = "" if resp.queueurl_list == [] else resp.queueurl_list[0]
lastQueueURL = "" if resp.queueurl_list == [] else resp.queueurl_list[len(resp.queueurl_list)-1]
self.logger.info("ListQueue RequestId:%s Prefix:%s RetNumber:%s Marker:%s QueueCount:%s FirstQueueURL:%s LastQueueURL:%s NextMarker:%s" % \
(resp.get_requestid(), req.prefix, req.ret_number, req.marker, \
len(resp.queueurl_list), firstQueueURL, lastQueueURL, resp.next_marker))
def set_queue_attributes(self, req, resp):
#check parameter
SetQueueAttrValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s?metaoverride=true" % (URISEC_QUEUE, req.queue_name))
req_inter.data = QueueEncoder.encode(req, False)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if self.logger:
self.logger.info("SetQueueAttributes RequestId:%s QueueName:%s" % (resp.get_requestid(), req.queue_name))
def get_queue_attributes(self, req, resp):
#check parameter
GetQueueAttrValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s" % (URISEC_QUEUE, req.queue_name))
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
queue_attr = GetQueueAttrDecoder.decode(resp_inter.data, req_inter.get_req_id())
resp.active_messages = int(queue_attr["ActiveMessages"])
resp.create_time = int(queue_attr["CreateTime"])
resp.delay_messages = int(queue_attr["DelayMessages"])
resp.delay_seconds = int(queue_attr["DelaySeconds"])
resp.inactive_messages = int(queue_attr["InactiveMessages"])
resp.last_modify_time = int(queue_attr["LastModifyTime"])
resp.maximum_message_size = int(queue_attr["MaximumMessageSize"])
resp.message_retention_period = int(queue_attr["MessageRetentionPeriod"])
resp.queue_name = queue_attr["QueueName"]
resp.visibility_timeout = int(queue_attr["VisibilityTimeout"])
resp.polling_wait_seconds = int(queue_attr["PollingWaitSeconds"])
resp.logging_enabled = True if queue_attr["LoggingEnabled"].lower() == "true" else False
if self.logger:
self.logger.info("GetQueueAttributes RequestId:%s QueueName:%s" % (resp.get_requestid(), req.queue_name))
def send_message(self, req, resp):
#check parameter
SendMessageValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, uri = "/%s/%s/%s" % (URISEC_QUEUE, req.queue_name, URISEC_MESSAGE))
req_inter.data = MessageEncoder.encode(req)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
resp.message_id, resp.message_body_md5, resp.receipt_handle = SendMessageDecoder.decode(resp_inter.data, req_inter.get_req_id())
if self.logger:
self.logger.info("SendMessage RequestId:%s QueueName:%s Priority:%s DelaySeconds:%s MessageId:%s MessageBodyMD5:%s" % \
(resp.get_requestid(), req.queue_name, req.priority, \
req.delay_seconds, resp.message_id, resp.message_body_md5))
def batch_send_message(self, req, resp):
#check parameter
BatchSendMessageValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, uri = "/%s/%s/%s" % (URISEC_QUEUE, req.queue_name, URISEC_MESSAGE))
req_inter.data = MessagesEncoder.encode(req.message_list, req.base64encode)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp, BatchSendMessageDecoder)
if resp.error_data == "":
resp.message_list = BatchSendMessageDecoder.decode(resp_inter.data, req_inter.get_req_id())
if self.logger:
self.logger.info("BatchSendMessage RequestId:%s QueueName:%s MessageCount:%s MessageInfo\n%s" % \
(resp.get_requestid(), req.queue_name, len(req.message_list), \
"\n".join(["MessageId:%s MessageBodyMD5:%s" % (msg.message_id, msg.message_body_md5) for msg in resp.message_list])))
def receive_message(self, req, resp):
#check parameter
ReceiveMessageValidator.validate(req)
#make request internal
req_url = "/%s/%s/%s" % (URISEC_QUEUE, req.queue_name, URISEC_MESSAGE)
if req.wait_seconds != -1:
req_url += "?waitseconds=%s" % req.wait_seconds
req_inter = RequestInternal(req.method, req_url)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
data = RecvMessageDecoder.decode(resp_inter.data, req.base64decode, req_inter.get_req_id())
self.make_recvresp(data, resp)
if self.logger:
self.logger.info("ReceiveMessage RequestId:%s QueueName:%s WaitSeconds:%s MessageId:%s MessageBodyMD5:%s NextVisibilityTime:%s ReceiptHandle:%s EnqueueTime:%s DequeueCount:%s" % \
(resp.get_requestid(), req.queue_name, req.wait_seconds, resp.message_id, \
resp.message_body_md5, resp.next_visible_time, resp.receipt_handle, resp.enqueue_time, resp.dequeue_count))
def batch_receive_message(self, req, resp):
#check parameter
BatchReceiveMessageValidator.validate(req)
#make request internal
req_url = "/%s/%s/%s?numOfMessages=%s" % (URISEC_QUEUE, req.queue_name, URISEC_MESSAGE, req.batch_size)
if req.wait_seconds != -1:
req_url += "&waitseconds=%s" % req.wait_seconds
req_inter = RequestInternal(req.method, req_url)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
resp.message_list = BatchRecvMessageDecoder.decode(resp_inter.data, req.base64decode, req_inter.get_req_id())
if self.logger:
self.logger.info("BatchReceiveMessage RequestId:%s QueueName:%s WaitSeconds:%s BatchSize:%s MessageCount:%s \
MessagesInfo\n%s" % (resp.get_requestid(), req.queue_name, req.wait_seconds, req.batch_size, len(resp.message_list),\
"\n".join(["MessageId:%s MessageBodyMD5:%s NextVisibilityTime:%s ReceiptHandle:%s EnqueueTime:%s DequeueCount:%s" % \
(msg.message_id, msg.message_body_md5, msg.next_visible_time, msg.receipt_handle, msg.enqueue_time, msg.dequeue_count) for msg in resp.message_list])))
def delete_message(self, req, resp):
#check parameter
DeleteMessageValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s/%s?ReceiptHandle=%s" %
(URISEC_QUEUE, req.queue_name, URISEC_MESSAGE, req.receipt_handle))
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if self.logger:
self.logger.info("DeleteMessage RequestId:%s QueueName:%s ReceiptHandle:%s" % \
(resp.get_requestid(), req.queue_name, req.receipt_handle))
def batch_delete_message(self, req, resp):
#check parameter
BatchDeleteMessageValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s/%s" % (URISEC_QUEUE, req.queue_name, URISEC_MESSAGE))
req_inter.data = ReceiptHandlesEncoder.encode(req.receipt_handle_list)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp, BatchDeleteMessageDecoder)
if self.logger:
self.logger.info("BatchDeleteMessage RequestId:%s QueueName:%s ReceiptHandles\n%s" % \
(resp.get_requestid(), req.queue_name, "\n".join(req.receipt_handle_list)))
def peek_message(self, req, resp):
#check parameter
PeekMessageValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s/%s?peekonly=true" %
(URISEC_QUEUE, req.queue_name, URISEC_MESSAGE))
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
data = PeekMessageDecoder.decode(resp_inter.data, req.base64decode, req_inter.get_req_id())
self.make_peekresp(data, resp)
if self.logger:
self.logger.info("PeekMessage RequestId:%s QueueName:%s MessageInfo \
MessageId:%s BodyMD5:%s EnqueueTime:%s DequeueCount:%s" % \
(resp.get_requestid(), req.queue_name, resp.message_id, resp.message_body_md5,\
resp.enqueue_time, resp.dequeue_count))
def batch_peek_message(self, req, resp):
#check parameter
BatchPeekMessageValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s/%s?peekonly=true&numOfMessages=%s" %
(URISEC_QUEUE, req.queue_name, URISEC_MESSAGE, req.batch_size))
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
resp.message_list = BatchPeekMessageDecoder.decode(resp_inter.data, req.base64decode, req_inter.get_req_id())
if self.logger:
self.logger.info("BatchPeekMessage RequestId:%s QueueName:%s BatchSize:%s MessageCount:%s MessageInfo\n%s" % \
(resp.get_requestid(), req.queue_name, req.batch_size, len(resp.message_list), \
"\n".join(["MessageId:%s BodyMD5:%s EnqueueTime:%s DequeueCount:%s" % \
(msg.message_id, msg.message_body_md5, msg.enqueue_time, msg.dequeue_count) for msg in resp.message_list])))
def change_message_visibility(self, req, resp):
#check parameter
ChangeMsgVisValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s/%s?ReceiptHandle=%s&VisibilityTimeout=%d" %
(URISEC_QUEUE, req.queue_name, URISEC_MESSAGE, req.receipt_handle, req.visibility_timeout))
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
resp.receipt_handle, resp.next_visible_time = ChangeMsgVisDecoder.decode(resp_inter.data, req_inter.get_req_id())
if self.logger:
self.logger.info("ChangeMessageVisibility RequestId:%s QueueName:%s ReceiptHandle:%s VisibilityTimeout:%s NewReceiptHandle:%s NextVisibleTime:%s" % \
(resp.get_requestid(), req.queue_name, req.receipt_handle, req.visibility_timeout,\
resp.receipt_handle, resp.next_visible_time))
#===============================================topic operation===============================================#
def create_topic(self, req, resp):
#check parameter
CreateTopicValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s" % (URISEC_TOPIC, req.topic_name))
req_inter.data = TopicEncoder.encode(req)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
resp.topic_url = self.lower_header(resp.header)["location"]
if self.logger:
self.logger.info("CreateTopic RequestId:%s TopicName:%s TopicURl:%s" % \
(resp.get_requestid(), req.topic_name, resp.topic_url))
def delete_topic(self, req, resp):
#check parameter
DeleteTopicValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s" % (URISEC_TOPIC, req.topic_name))
self.build_header(req, req_inter)
#send reqeust
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if self.logger:
self.logger.info("DeleteTopic RequestId:%s TopicName:%s" % (resp.get_requestid(), req.topic_name))
def list_topic(self, req, resp):
#check parameter
ListTopicValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s" % URISEC_TOPIC)
if req.prefix != "":
req_inter.header["x-mns-prefix"] = req.prefix
if req.ret_number != -1:
req_inter.header["x-mns-ret-number"] = str(req.ret_number)
if req.marker != "":
req_inter.header["x-mns-marker"] = str(req.marker)
if req.with_meta:
req_inter.header["x-mns-with-meta"] = "true"
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
resp.topicurl_list, resp.next_marker, resp.topicmeta_list = ListTopicDecoder.decode(resp_inter.data, req.with_meta, req_inter.get_req_id())
first_topicurl = "" if len(resp.topicurl_list) == 0 else resp.topicurl_list[0]
last_topicurl = "" if len(resp.topicurl_list) == 0 else resp.topicurl_list[len(resp.topicurl_list)-1]
if self.logger:
self.logger.info("ListTopic RequestId:%s Prefix:%s RetNumber:%s Marker:%s TopicCount:%s FirstTopicURL:%s LastTopicURL:%s NextMarker:%s" % \
(resp.get_requestid(), req.prefix, req.ret_number, req.marker,\
len(resp.topicurl_list), first_topicurl, last_topicurl, resp.next_marker))
def set_topic_attributes(self, req, resp):
#check parameter
SetTopicAttrValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s?metaoverride=true" % (URISEC_TOPIC, req.topic_name))
req_inter.data = TopicEncoder.encode(req)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if self.logger:
self.logger.info("SetTopicAttributes RequestId:%s TopicName:%s" % (resp.get_requestid(), req.topic_name))
def get_topic_attributes(self, req, resp):
#check parameter
GetTopicAttrValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s" % (URISEC_TOPIC, req.topic_name))
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
topic_attr = GetTopicAttrDecoder.decode(resp_inter.data, req_inter.get_req_id())
resp.message_count = int(topic_attr["MessageCount"])
resp.create_time = int(topic_attr["CreateTime"])
resp.last_modify_time = int(topic_attr["LastModifyTime"])
resp.maximum_message_size = int(topic_attr["MaximumMessageSize"])
resp.message_retention_period = int(topic_attr["MessageRetentionPeriod"])
resp.topic_name = topic_attr["TopicName"]
resp.logging_enabled = True if topic_attr["LoggingEnabled"].lower() == "true" else False
if self.logger:
self.logger.info("GetTopicAttributes RequestId:%s TopicName:%s" % (resp.get_requestid(), req.topic_name))
def publish_message(self, req, resp):
#check parameter
PublishMessageValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, uri = "/%s/%s/%s" % (URISEC_TOPIC, req.topic_name, URISEC_MESSAGE))
req_inter.data = TopicMessageEncoder.encode(req)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
resp.message_id, resp.message_body_md5 = PublishMessageDecoder.decode(resp_inter.data, req_inter.get_req_id())
if self.logger:
self.logger.info("PublishMessage RequestId:%s TopicName:%s MessageId:%s MessageBodyMD5:%s" % \
(resp.get_requestid(), req.topic_name, resp.message_id, resp.message_body_md5))
def subscribe(self, req, resp):
#check parameter
SubscribeValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, uri="/%s/%s/%s/%s" % (URISEC_TOPIC, req.topic_name, URISEC_SUBSCRIPTION, req.subscription_name))
req_inter.data = SubscriptionEncoder.encode(req)
self.build_header(req, req_inter)
#send request
req_inter.data = req_inter.data.decode('utf-8')
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
lheader = self.lower_header(resp.header)
resp.subscription_url = lheader["location"]
if self.logger:
self.logger.info("Subscribe RequestId:%s TopicName:%s SubscriptionName:%s SubscriptionURL:%s" % \
(resp.get_requestid(), req.topic_name, req.subscription_name, resp.subscription_url))
def unsubscribe(self, req, resp):
#check parameter
UnsubscribeValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s/%s/%s" % (URISEC_TOPIC, req.topic_name, URISEC_SUBSCRIPTION, req.subscription_name))
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if self.logger:
self.logger.info("Unsubscribe RequestId:%s TopicName:%s SubscriptionName:%s" % (resp.get_requestid(), req.topic_name, req.subscription_name))
def list_subscription_by_topic(self, req, resp):
#check parameter
ListSubscriptionByTopicValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s/%s" % (URISEC_TOPIC, req.topic_name, URISEC_SUBSCRIPTION))
if req.prefix != "":
req_inter.header["x-mns-prefix"] = req.prefix
if req.ret_number != -1:
req_inter.header["x-mns-ret-number"] = str(req.ret_number)
if req.marker != "":
req_inter.header["x-mns-marker"] = req.marker
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
resp.subscriptionurl_list, resp.next_marker = ListSubscriptionByTopicDecoder.decode(resp_inter.data, req_inter.get_req_id())
if self.logger:
first_suburl = "" if len(resp.subscriptionurl_list) == 0 else resp.subscriptionurl_list[0]
last_suburl = "" if len(resp.subscriptionurl_list) == 0 else resp.subscriptionurl_list[len(resp.subscriptionurl_list)-1]
self.logger.info("ListSubscriptionByTopic RequestId:%s TopicName:%s Prefix:%s RetNumber:%s \
Marker:%s SubscriptionCount:%s FirstSubscriptionURL:%s LastSubscriptionURL:%s" % \
(resp.get_requestid(), req.topic_name, req.prefix, req.ret_number, \
req.marker, len(resp.subscriptionurl_list), first_suburl, last_suburl))
def set_subscription_attributes(self, req, resp):
#check parameter
SetSubscriptionAttrValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s/%s/%s?metaoverride=true" % (URISEC_TOPIC, req.topic_name, URISEC_SUBSCRIPTION, req.subscription_name))
req_inter.data = SubscriptionEncoder.encode(req, set=True)
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if self.logger:
self.logger.info("SetSubscriptionAttributes RequestId:%s TopicName:%s SubscriptionName:%s" % \
(resp.get_requestid(), req.topic_name, req.subscription_name))
def get_subscription_attributes(self, req, resp):
#check parameter
GetSubscriptionAttrValidator.validate(req)
#make request internal
req_inter = RequestInternal(req.method, "/%s/%s/%s/%s" % (URISEC_TOPIC, req.topic_name, URISEC_SUBSCRIPTION, req.subscription_name))
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
subscription_attr = GetSubscriptionAttrDecoder.decode(resp_inter.data, req_inter.get_req_id())
resp.topic_owner = subscription_attr["TopicOwner"]
resp.topic_name = subscription_attr["TopicName"]
resp.subscription_name = subscription_attr["SubscriptionName"]
resp.endpoint = subscription_attr["Endpoint"]
resp.filter_tag = subscription_attr["FilterTag"] if "FilterTag" in subscription_attr.keys() else ""
resp.notify_strategy = subscription_attr["NotifyStrategy"]
resp.notify_content_format = subscription_attr["NotifyContentFormat"]
resp.create_time = int(subscription_attr["CreateTime"])
resp.last_modify_time = int(subscription_attr["LastModifyTime"])
if self.logger:
self.logger.info("GetSubscriptionAttributes RequestId:%s TopicName:%s SubscriptionName:%s" % \
(resp.get_requestid(), req.topic_name, req.subscription_name))
def open_service(self, req, resp):
#make request internal
req_inter = RequestInternal(req.method, "/commonbuy/openservice")
self.build_header(req, req_inter)
#send request
resp_inter = self.http.send_request(req_inter)
#handle result, make response
resp.status = resp_inter.status
resp.header = resp_inter.header
self.check_status(req_inter, resp_inter, resp)
if resp.error_data == "":
resp_attr = OpenServiceDecoder.decode(resp_inter.data, req_inter.get_req_id())
resp.oder_id = resp_attr["OrderId"]
if self.logger:
self.logger.info("OpenServiceAttributes RequestId:%s" % resp.get_requestid())
###################################################################################################
#----------------------internal-------------------------------------------------------------------#
def build_header(self, req, req_inter):
if req.request_id is not None:
req_inter.header["x-mns-user-request-id"] = req.request_id
if self.http.is_keep_alive():
req_inter.header["Connection"] = "Keep-Alive"
if req_inter.data != "":
#req_inter.header["content-md5"] = base64.b64encode(hashlib.md5(req_inter.data).hexdigest())
#req_inter.header["content-md5"] = base64.b64encode(hashlib.md5(req_inter.data.encode("utf-8")).hexdigest().encode("utf-8")).decode("utf-8")
req_inter.header["content-md5"] = base64.b64encode(hashlib.md5(req_inter.data).hexdigest().encode("utf-8")).decode("utf-8")
req_inter.header["content-type"] = "text/xml;charset=UTF-8"
req_inter.header["x-mns-version"] = self.version
req_inter.header["host"] = self.host
req_inter.header["date"] = time.strftime("%a, %d %b %Y %H:%M:%S GMT", time.gmtime())
req_inter.header["user-agent"] = "aliyun-sdk-python/%s(%s/%s/%s;%s)" % \
(pkg_info.version, platform.system(), platform.release(), platform.machine(), platform.python_version())
credential = self.credentials_provider.get_credentials()
req_inter.header["Authorization"] = self.get_signature(req_inter.method, req_inter.header, req_inter.uri, credential)
if credential.get_security_token() != "":
req_inter.header["security-token"] = credential.get_security_token()
def get_signature(self, method, headers, resource, credential):
content_md5 = self.get_element('content-md5', headers)
content_type = self.get_element('content-type', headers)
date = self.get_element('date', headers)
canonicalized_resource = resource
canonicalized_mns_headers = ""
if len(headers) > 0:
x_header_list = headers.keys()
#x_header_list.sort()
x_header_list = sorted(x_header_list)
for k in x_header_list:
if k.startswith('x-mns-'):
canonicalized_mns_headers += k + ":" + headers[k] + "\n"
string_to_sign = "%s\n%s\n%s\n%s\n%s%s" % (method, content_md5, content_type, date, canonicalized_mns_headers, canonicalized_resource)
#hmac only support str in python2.7
#tmp_key = self.access_key.encode('utf-8') if isinstance(self.access_key, unicode) else self.access_key
tmp_key = credential.get_access_key_secret().encode('utf-8')
h = hmac.new(tmp_key, string_to_sign.encode('utf-8'), hashlib.sha1)
signature = base64.b64encode(h.digest())
signature = "MNS " + credential.get_access_key_id() + ":" + signature.decode('utf-8')
return signature
def get_element(self, name, container):
if name in container:
return container[name]
else:
return ""
def check_status(self, req_inter, resp_inter, resp, decoder=ErrorDecoder):
if resp_inter.status >= 200 and resp_inter.status < 400:
resp.error_data = ""
else:
resp.error_data = resp_inter.data
if resp_inter.status >= 400 and resp_inter.status <= 600:
excType, excMessage, reqId, hostId, subErr = decoder.decodeError(resp.error_data, req_inter.get_req_id())
if reqId is None:
reqId = resp.header["x-mns-request-id"]
raise MNSServerException(excType, excMessage, reqId, hostId, subErr)
else:
raise MNSClientNetworkException("UnkownError", resp_inter.data, req_inter.get_req_id())
def make_recvresp(self, data, resp):
resp.dequeue_count = int(data["DequeueCount"])
resp.enqueue_time = int(data["EnqueueTime"])
resp.first_dequeue_time = int(data["FirstDequeueTime"])
resp.message_body = data["MessageBody"]
resp.message_id = data["MessageId"]
resp.message_body_md5 = data["MessageBodyMD5"]
resp.next_visible_time = int(data["NextVisibleTime"])
resp.receipt_handle = data["ReceiptHandle"]
resp.priority = int(data["Priority"])
def make_peekresp(self, data, resp):
resp.dequeue_count = int(data["DequeueCount"])
resp.enqueue_time = int(data["EnqueueTime"])
resp.first_dequeue_time = int(data["FirstDequeueTime"])
resp.message_body = data["MessageBody"]
resp.message_id = data["MessageId"]
resp.message_body_md5 = data["MessageBodyMD5"]
resp.priority = int(data["Priority"])
def process_host(self, host):
if host.startswith("http://"):
if host.endswith("/"):
host = host[:-1]
host = host[len("http://"):]
return host, False
elif host.startswith("https://"):
if host.endswith("/"):
host = host[:-1]
host = host[len("https://"):]
return host, True
else:
raise MNSClientParameterException("InvalidHost", "Only support http prototol. Invalid host:%s" % host)
@staticmethod
def lower_header(header):
lower_header = dict()
for k, v in header.items():
k = k.lower()
lower_header[k] = v
return lower_header