o wvXj< @sRdZddlmZmZddlZddlZddlZddlZzddl Z Wn e y+dZ Ynwddl m Z ddlmZddlmZddlmZmZmZmZdd lmZmZdd lmZdd lmZmZmZd Z e!d dgZ"e!ddgZ#ej$%dr{ddZ&nej'Z&e peZej(Z)e*ej+,ddZ-e ddZ.ee/feffddZ0e)fddZ1ddZ2Gddde3Z4e4Z5e5j6Z6e5j7Z7e5j8Z8e5j9Z9d d!Z:d"d#Z;d$d%Zne1Z>d)d*Z?d+d,Z@e;e?e<e@e5Ad-e;e?esr5c@sfeZdZdZddZ dddZddZd d Zd d Zd dZ dddZ dde fddZ ddZ dS)SerializerRegistryz2The registry keeps track of serialization methods.cCs6i|_i|_d|_d|_d|_t|_i|_i|_dSr/) _encoders _decoders_default_encode_default_content_type_default_content_encodingset_disabled_content_types type_to_name name_to_type)selfrrr __init__Es zSerializerRegistry.__init__utf-8cCs<|r t||||j|<|r||j|<||j|<||j|<dS)aRegister a new encoder/decoder. Arguments: name (str): A convenience name for the serialization method. encoder (callable): A method that will be passed a python data structure and should return a string representing the serialized data. If :const:`None`, then only a decoder will be registered. Encoding will not be possible. decoder (Callable): A method that will be passed a string representing serialized data and should return a python data structure. If :const:`None`, then only an encoder will be registered. Decoding will not be possible. content_type (str): The mime-type describing the serialized structure. content_encoding (str): The content encoding (character set) that the `decoder` method will be returning. Will usually be `utf-8`, `us-ascii`, or `binary`. N)r$r7r8r>r?)r@namer'decoderr%r&rrr rOs   zSerializerRegistry.registercC"d|vr |j|}|j|dSN/)r?r=discardr@rCrrr enablep zSerializerRegistry.enablecCrErF)r?r=addrIrrr disableurKzSerializerRegistry.disablecCsfz$|j|}|j|d|j|d|j|d|j|dWdSty2td|w)zUnregister registered encoder/decoder. Arguments: name (str): Registered serialization method name. Raises: SerializerNotInstalled: If a serializer by that name cannot be found. Nz$No encoder/decoder installed for {0})r?r8popr7r>KeyErrorr format)r@rCr%rrr rzs  zSerializerRegistry.unregistercCs:z|j|\|_|_|_WdStytd|w)aSet the default serialization method used by this library. Arguments: name (str): The name of the registered serialization method. For example, `json` (default), `pickle`, `yaml`, `msgpack`, or any custom methods registered using :meth:`register`. Raises: SerializerNotInstalled: If the serialization method requested is not available. No encoder installed for {0}N)r7r:r;r9rOr rPrIrrr _set_default_serializers   z*SerializerRegistry._set_default_serializerNcCs|dkrt|S|r|j|std||s#t|tr#dd|fS|sJt|trJtt dd | d}Wdn1s@wYd d|fS|rU|j|\}}}n |j }|j }|j }tt  ||}Wdn1sqwY|||fS) aEncode data. Serialize a data structure into a string suitable for sending as an AMQP message body. Arguments: data (List, Dict, str): The message data to send. serializer (str): An optional string representing the serialization method you want the data marshalled into. (For example, `json`, `raw`, or `pickle`). If :const:`None` (default), then json will be used, unless `data` is a :class:`str` or :class:`unicode` object. In this latter case, no serialization occurs as it would be unnecessary. Note that if `serializer` is specified, then that serialization method will be used even if a :class:`str` or :class:`unicode` object is passed in. Returns: Tuple[str, str, str]: A three-item tuple containing the content type (e.g., `application/json`), content encoding, (e.g., `utf-8`) and a string containing the serialized data. Raises: SerializerNotInstalled: If the serialization method requested is not available. rawrQrrrr,rBNz text/plain) raw_encoder7getr rP isinstancerrr.r encoder9r:r;)r@data serializerpayloadr%r&r'rrr rs.       zSerializerRegistry.dumpsFcCs|rt|nd}|dur||vr||vr||dn ||jvr(|s(||d|p+d}|rr|j|}|rPtt ||WdS1sKwY|tvrrt |t srtt t ||WdS1smwY|S)aDecode serialized data. Deserialize a data stream as serialized using `dumps` based on `content_type`. Arguments: data (bytes, buffer, str): The message data to deserialize. content_type (str): The content-type of the data. (e.g., `application/json`). content_encoding (str): The content-encoding of the data. (e.g., `utf-8`, `binary`, or `us-ascii`). accept (Set): List of content-types to accept. Raises: ContentDisallowed: If the content-type is not accepted. Returns: Any: The unserialized data. rN untrusteddisabledrB) r_for_untrusted_contentr=lowerr8rVr.r SKIP_DECODErWrr!)r@rYr%r&acceptforce_trusted_contentdecoderrr rs0         zSerializerRegistry.loadsc Cs td|t|j|||S)Nz/Refusing to deserialize {0} content of type {1})r rPr5r>rV)r@ctypewhyrrr r^s z)SerializerRegistry._for_untrusted_content)rBr/)__name__ __module__ __qualname____doc__rArrJrMrrRrTRUSTED_CONTENTrr^rrrr r6Bs ! @ .r6cCs\d}|}t|tr'd}ttdd ||}Wdn1s!wYnd}|||fS)zSpecial case serializer.rrBrrTNr)rWrr.r rX)rYr%r[r&rrr rUs   rUcCs(ddlm}tjd|j|jddddS)z2Register a encoder/decoder for JSON serialization.r)jsonrlapplication/jsonrBr%r&N) kombu.utilsrlregistryrrr)_jsonrrr register_json,s  rrc CsTzddl}tjd|j|jdddWdSty)dd}tdd|dYdSw) zRegister a encoder/decoder for YAML serialization. It is slower than JSON, but allows for more data types to be serialized. Useful if you need to send data such as dates rNyamlapplication/x-yamlrBrnc_td)zRaise SerializerNotInstalled. Used in case a client receives a yaml message, but yaml isn't installed. z9No decoder installed for YAML. Install the PyYAML libraryr argskwargsrrr not_availableCsz$register_yaml..not_available)rsrpr safe_dump safe_load ImportError)rsrzrrr register_yaml5s r~cCs tt|Sr/)r2rr0rrr unpickleP rcCs&tjfdd}tjd|tddddS)zpRegister pickle serializer. The fastest serialization method, but restricts you to python clients. cSs ||tdS)N)protocol)pickle_protocol)objdumperrrr pickle_dumps]rz%register_pickle..pickle_dumpsrapplication/x-python-serializerrnN)rrrprr)rrrr register_pickleWs   rc sd}}z(ddl}|jdkr#ddlmmfdd}fdd}nd d }|}}Wnttfy?d d }|}}Ynwtjd ||ddddS)zNRegister msgpack serializer. See Also: https://msgpack.org/. Nr)rpackbunpackbc |ddS)NT) use_bin_typerr)rrr packqrzregister_msgpack..packcr)NF)rSrr)rrr unpacktrz register_msgpack..unpackc_ru)Nz(msgpack requires msgpack-python >= 0.4.0rvrwrrr version_mismatchwz*register_msgpack..version_mismatchc_ru)NzKNo decoder installed for msgpack. Please install the msgpack-python libraryrvrwrrr rz|rz'register_msgpack..not_availablemsgpackapplication/x-msgpackrrn)rversionrrr} ValueErrorrpr)rrrrrzrrr register_msgpackes&    rrl)rlrrsrrmrtrrc CsN|turgdn|}|dur#|D]}zt|Wqty"YqwdSdS)zEnable serializers that are considered to be unsafe. Note: Will enable ``pickle``, ``yaml`` and ``msgpack`` by default, but you can also specify a list of serializers (by name or content type) to enable. )rrsrN)NOTSETrprJrO)choiceschoicerrr enable_insecure_serializerss rcCsL|turdgn|}tjD]}t|q |dur"|D] }t|qdSdS)aADisable untrusted serializers. Will disable all serializers except ``json`` or you can specify a list of deserializers to allow. Note: Producers will still be able to serialize data in these formats, but consumers will not accept incoming data using the untrusted content types. rlN)rrpr8rMrJ)allowedrCrrr disable_insecure_serializerss   rzkombu.serializerscs,stjn|durfdd|DS|S)Ncs h|] }d|vr |n|qS)rGr).0nr?rr s z)prepare_accept_content..)rpr?) content_typesr?rrr prepare_accept_contentsrr/)Jrj __future__rrrosr(rpypicklecPicklecpickler} collectionsr contextlibrior exceptionsr r r r fiver r utils.compatrutils.encodingrrr__all__ frozensetr`rkplatform startswithr!rdr1 pickle_loadintenvironrVrr$ Exceptionr.r2r5objectr6rprrrrrUrrr~ version_inforrrrR _setupfunsrrreprxrCrrrrr s            W   $