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