o DfV<@sdZddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZmZmZmZdd lmZdd lmZmZd Zed d gZeddgZejdr_ddZnejZej Z!e"ej#$ddZ%eddZ&e e'feffddZ(e!fddZ)ddZ*GdddZ+e+Z,e,j-Z-e,j.Z.e,j/Z/e,j0Z0dd Z1d!d"Z2d#d$Z3d%d&Z4d'd(Z5d)d*Z6e2e5e3e6e,7d+e8Z9e9fd,d-Z:e9fd.d/Z;e;ed0D] \Zge=Rqd3d1d2Z?dS)4zSerialization utilities.) annotationsN) namedtuple)contextmanagerBytesIO)ContentDisallowed DecodeError EncodeErrorSerializerNotInstalledreraise) entrypoints) bytes_to_str str_to_bytes)pickleloadsdumpsregister unregisterbinaryz ascii-8bitapplication/datazapplication/textjavacCst||dS)Nr)codecs getdecoder)tcodingrL/home/ubuntu/webapp/venv/lib/python3.10/site-packages/kombu/serialization.py_decodesrPICKLE_PROTOCOLcodec) content_typecontent_encodingencoderc cs\zdVWdS|y|y-}zt|||tdWYd}~dSd}~ww)N)r sysexc_info)wrapperincludeexcludeexcrrr_reraise_errors%s  &r,cCs |t|SNr)sloadrrr pickle_loads0s r0cCs|r |d|dS|S)Nz ()r)firstsecondrrrparenthesize_alias5sr4c@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__<s 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!r6r7r=r>)r?namer$decoderr"r#rrrrFs   zSerializerRegistry.registercC"d|vr |j|}|j|dSN/)r>r<discardr?rBrrrenableh zSerializerRegistry.enablecCrDrE)r>r<addrHrrrdisablemrJzSerializerRegistry.disablecCsfz$|j|}|j|d|j|d|j|d|j|dWdSty2td|w)aUnregister registered encoder/decoder. Arguments: --------- name (str): Registered serialization method name. Raises ------ SerializerNotInstalled: If a serializer by that name cannot be found. N!No encoder/decoder installed for )r>r7popr6r=KeyErrorr )r?rBr"rrrrrs  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 N)r6r9r:r8rOr rHrrr_set_default_serializers  z*SerializerRegistry._set_default_serializerNcCs|dkrt|S|r|j|std||s#t|tr#dd|fS|sJt|trJttdd | 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. rawrPrrrr*rANz text/plain) raw_encoder6getr isinstancebytesstrr,r encoder8r9r:)r?data serializerpayloadr"r#r$rrrrs."       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 untrusteddisabledrA) r_for_untrusted_contentr<lowerr7rUr,r SKIP_DECODErVrXr)r?rZr"r#acceptforce_trusted_contentdecoderrrrs0         zSerializerRegistry.loadsc Cs td|t|j|||S)Nz-Refusing to deserialize {} content of type {})rformatr4r=rU)r?ctypewhyrrrr_s z)SerializerRegistry._for_untrusted_content)rAr-)__name__ __module__ __qualname____doc__r@rrIrLrrQrTRUSTED_CONTENTrr_rrrrr59s " C 1r5cCs\d}|}t|tr'd}ttdd ||}Wdn1s!wYnd}|||fS)zSpecial case serializer.rrArrSNr)rVrXr,r rY)rZr"r\r#rrrrT!s   rTcCs(ddlm}tjd|j|jddddS)z2Register a encoder/decoder for JSON serialization.r)jsonrnzapplication/jsonrAr"r#N) kombu.utilsrnregistryrrr)_jsonrrr register_json.s  rsc 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 rNyamlzapplication/x-yamlrAroc_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_availableEsz$register_yaml..not_available)rtrqr safe_dump safe_load ImportError)rtrzrrr register_yaml7s r~cCs tt|Sr-)r0rr.rrrunpickleP 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_dumpsZrz%register_pickle..pickle_dumpsrzapplication/x-python-serializerroN)rrrqrr)rrrrregister_pickleTs   rc sd}}z(ddl}|jdkr#ddlmmfdd}fdd}nd d }|}}Wnttfy?d d }|}}Ynwtjd ||ddddS)zZRegister msgpack serializer. See Also -------- https://msgpack.org/. Nr)rr packbunpackbc |ddS)NT) use_bin_typerr)rrrpackorzregister_msgpack..packcr)NF)rRrr)rrrunpackrrz register_msgpack..unpackc_ru)Nz(msgpack requires msgpack-python >= 0.4.0rvrwrrrversion_mismatchuz*register_msgpack..version_mismatchc_ru)NzKNo decoder installed for msgpack. Please install the msgpack-python libraryrvrwrrrrzzrz'register_msgpack..not_availablemsgpackzapplication/x-msgpackrro)rversionrrr} ValueErrorrqr)rrrrrzrrrregister_msgpackbs&    rrnc 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. )rrtrN)NOTSETrqrIrO)choiceschoicerrrenable_insecure_serializerss  rcCsL|turdgn|}tjD]}t|q |dur"|D] }t|qdSdS)aJDisable 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. rnN)rrqr7rLrI)allowedrBrrrdisable_insecure_serializerss   rzkombu.serializersc s\stjn|dur,z fdd|DWSty+}z td|jdd}~ww|S)zReplace aliases of content_types with full names from registry. Raises ------ SerializerNotInstalled: If the serialization method requested is not available. Ncs h|] }d|vr |n|qS)rFr).0nr>rr s z)prepare_accept_content..rMr)rqr>rOr rx) content_typesr>errrprepare_accept_contentsrr-)@rl __future__rrosrr& collectionsr contextlibrior exceptionsrr r r r utils.compatr utils.encodingrr__all__ frozensetrarmplatform startswithrrer/ pickle_loadintenvironrUrr! Exceptionr,r0r4r5rqrrrrrTrsr~rrrrQobjectrrreprxrBrrrrrsf           b  %