U W¨+d¤&ã@s”dZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z Gdd „d e ƒZ Gd d „d e ƒZGd d „d eƒZGdd„deƒZdS)aX SQS Message A Message represents the data stored in an SQS queue. The rules for what is allowed within an SQS Message are here: http://docs.amazonwebservices.com/AWSSimpleQueueService/2008-01-01/SQSDeveloperGuide/Query_QuerySendMessage.html So, at it's simplest level a Message just needs to allow a developer to store bytes in it and get the bytes back out. However, to allow messages to have richer semantics, the Message class must support the following interfaces: The constructor for the Message class must accept a keyword parameter "queue" which is an instance of a boto Queue object and represents the queue that the message will be stored in. The default value for this parameter is None. The constructor for the Message class must accept a keyword parameter "body" which represents the content or body of the message. The format of this parameter will depend on the behavior of the particular Message subclass. For example, if the Message subclass provides dictionary-like behavior to the user the body passed to the constructor should be a dict-like object that can be used to populate the initial state of the message. The Message class must provide an encode method that accepts a value of the same type as the body parameter of the constructor and returns a string of characters that are able to be stored in an SQS message body (see rules above). The Message class must provide a decode method that accepts a string of characters that can be stored (and probably were stored!) in an SQS message and return an object of a type that is consistent with the "body" parameter accepted on the class constructor. The Message class must provide a __len__ method that will return the size of the encoded message that would be stored in SQS based on the current state of the Message object. The Message class must provide a get_body method that will return the body of the message in the same format accepted in the constructor of the class. The Message class must provide a set_body method that accepts a message body in the same format accepted by the constructor of the class. This method should alter to the internal state of the Message object to reflect the state represented in the message body parameter. The Message class must provide a get_body_encoded method that returns the current body of the message in the format in which it would be stored in SQS. éN)ÚStringIO)Úsix)Ú Attributes)ÚMessageAttributes)ÚSQSDecodeErrorc@sreZdZdZddd„Zdd„Zdd „Zd d „Zd d „Zdd„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„ZdS)Ú RawMessagea Base class for SQS messages. RawMessage does not encode the message in any way. Whatever you store in the body of the message is what will be written to SQS and whatever is returned from SQS is stored directly into the body of the message. NÚcCs@||_| |¡d|_d|_d|_t|ƒ|_t|ƒ|_d|_ dS©N) ÚqueueÚset_bodyÚidÚreceipt_handleÚmd5rÚ attributesrÚmessage_attributesÚmd5_message_attributes)Úselfr Úbody©rú4/tmp/pip-unpacked-wheel-dlxw5sjy/boto/sqs/message.pyÚ__init__Ts   zRawMessage.__init__cCst| |j¡ƒSr )ÚlenÚencodeÚ_body©rrrrÚ__len__^szRawMessage.__len__cCs |dkr|jS|dkr|jSdS)NÚ AttributeZMessageAttribute)rr)rÚnameÚattrsÚ connectionrrrÚ startElementas zRawMessage.startElementcCsd|dkr| |¡nL|dkr$||_n<|dkr4||_n,|dkrD||_n|dkrT||_n t|||ƒdS)NZBodyZ MessageIdZ ReceiptHandleZ MD5OfBodyZMD5OfMessageAttributes)r r r rrÚsetattr)rrÚvaluerrrrÚ endElemenths zRawMessage.endElementcCs| | | ¡¡¡dSr )r ÚdecodeÚget_body)rrrrrÚendNodevszRawMessage.endNodecCs|S)z8Transform body object into serialized byte array format.r©rr"rrrryszRawMessage.encodecCs|S)z/Transform seralized byte array into any object.rr'rrrr$}szRawMessage.decodecCs ||_dS)z@Override the current body for this object, using decoded format.N©r)rrrrrr szRawMessage.set_bodycCs|jSr r(rrrrr%…szRawMessage.get_bodycCs| | ¡¡S)zí This method is really a semi-private method used by the Queue.write method when writing the contents of the message to SQS. You probably shouldn't need to call this method in the normal course of events. )rr%rrrrÚget_body_encodedˆszRawMessage.get_body_encodedcCs|jr|j |¡SdSr )r Zdelete_messagerrrrÚdeleteszRawMessage.deletecCs |jr|jj |j|j|¡dSr )r rZchange_message_visibilityr )rZvisibility_timeoutrrrÚchange_visibility”s  þzRawMessage.change_visibility)Nr)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr r#r&rr$r r%r)r*r+rrrrrLs rc@s eZdZdZdd„Zdd„ZdS)ÚMessageaz The default Message class used for SQS queues. This class automatically encodes/decodes the message body using Base64 encoding to avoid any illegal characters in the message body. See: https://forums.aws.amazon.com/thread.jspa?threadID=13067 for details on why this is a good idea. The encode/decode is meant to be transparent to the end-user. cCs&t|tjƒs| d¡}t |¡ d¡S©Núutf-8)Ú isinstancerÚ binary_typerÚbase64Ú b64encoder$r'rrrr¦s  zMessage.encodecCs<zt | d¡¡ d¡}Wntj d¡|YSX|S©Nr2úUnable to decode message)r5Ú b64decoderr$ÚbotoÚlogÚwarningr'rrrr$«s   zMessage.decodeN)r,r-r.r/rr$rrrrr0šs r0cs|eZdZdZd‡fdd„ Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z dd„Zddd„Z‡ZS)Ú MHMessageaX The MHMessage class provides a message that provides RFC821-like headers like this: HeaderName: HeaderValue The encoding/decoding of this is handled automatically and after the message body has been read, the message instance can be treated like a mapping object, i.e. m['HeaderName'] would return 'HeaderValue'. Ncs*|dks|dkri}tt|ƒ ||¡dS)Nr)Úsuperr=r)rr rZ xml_attrs©Ú __class__rrr¿szMHMessage.__init__cCs|z`i}t|ƒ}| ¡}|r^| d¡}|d|…}||dd… ¡}| ¡|| ¡<| ¡}qWntd|ƒ‚YnX|S)Nú:rér8)rÚreadlineÚfindÚstripr)rr"ÚmsgÚfpÚlineÚdelimÚkeyrrrr$Äs  zMHMessage.decodecCs.d}| ¡D]}|d|d|df}q |S)Nrz%s: %s rrB)Úitems)rr"ÚsÚitemrrrrÓs zMHMessage.encodecCs ||jkSr r(©rrJrrrÚ __contains__ÙszMHMessage.__contains__cCs ||jkr|j|St|ƒ‚dSr )rÚKeyErrorrNrrrÚ __getitem__Üs  zMHMessage.__getitem__cCs||j|<| |j¡dSr )rr )rrJr"rrrÚ __setitem__âs zMHMessage.__setitem__cCs |j ¡Sr )rÚkeysrrrrrSæszMHMessage.keyscCs |j ¡Sr )rÚvaluesrrrrrTészMHMessage.valuescCs |j ¡Sr )rrKrrrrrKìszMHMessage.itemscCs ||jkSr r(rNrrrÚhas_keyïszMHMessage.has_keycCs|j |¡| |j¡dSr )rÚupdater )rÚdrrrrVòs zMHMessage.updatecCs|j ||¡Sr )rÚget)rrJÚdefaultrrrrXösz MHMessage.get)NNN)N)r,r-r.r/rr$rrOrQrRrSrTrKrUrVrXÚ __classcell__rrr?rr=³s r=cs,eZdZdZ‡fdd„Z‡fdd„Z‡ZS)ÚEncodedMHMessageaN The EncodedMHMessage class provides a message that provides RFC821-like headers like this: HeaderName: HeaderValue This variation encodes/decodes the body of the message in base64 automatically. The message instance can be treated like a mapping object, i.e. m['HeaderName'] would return 'HeaderValue'. csBzt | d¡¡ d¡}Wntd|ƒ‚YnXtt|ƒ |¡Sr7)r5r9rr$rr>r[r'r?rrr$s zEncodedMHMessage.decodecs&tt|ƒ |¡}t | d¡¡ d¡Sr1)r>r[rr5r6r$r'r?rrr szEncodedMHMessage.encode)r,r-r.r/r$rrZrrr?rr[ùs r[)r/r5r:Z boto.compatrrZboto.sqs.attributesrZboto.sqs.messageattributesrZboto.exceptionrÚobjectrr0r=r[rrrrÚs,     NF