3jS dZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z GddZ e Z e jZ e jZe jZGddZGd d eZej%ejGd d eZej%ejej%ej(ej%ej*Gd deZej%eej0dGddeZej%ej4GddZGddeZej%ej:GddeZej>Z ej%e jBGddeZ"e"j%ejFy)a  Custom handlers may be created to handle other objects. Each custom handler must derive from :class:`jsonpickle.handlers.BaseHandler` and implement ``flatten`` and ``restore``. A handler can be bound to other types by calling :func:`jsonpickle.handlers.register`. N)utilc(eZdZdZddZddZdZy)Registryc i|_i|_yN) _handlers_base_handlersselfs @/DATA/.local/lib/python3.12/site-packages/jsonpickle/handlers.py__init__zRegistry.__init__s Nc|jj|}|Gtj|r2|jj D]\}}t ||s|cS||S|S)a\ :param cls_or_name: the type or its fully qualified name :param default: default value, if a matching handler is not found Looks up a handler by type reference or its fully qualified name. If a direct match is not found, the search is performed over all handlers registered with base=True. )r getris_typer items issubclass)r cls_or_namedefaulthandlercls base_handlers r rz Registry.getsh..$$[1 ?t||K8%)%8%8%>%>%@!\k3/''&A"/w6w6rc| fd}|Stjstd|xjtj<j<r|j <yy)a,Register the a custom handler for a class :param cls: The custom object class to handle :param handler: The custom handler class (if None, a decorator wrapper is returned) :param base: Indicates whether the handler should be registered for all subclasses This function can be also used as a decorator by omitting the `handler` argument:: @jsonpickle.handlers.register(Foo, base=True) class FooHandler(jsonpickle.handlers.BaseHandler): pass Nc0j||S)N)rbase)register) handler_clsrrr s r _registerz$Registry.register.._registerAs c;T B""rz is not a class/type)rr TypeErrorr importable_namer )r rrrrs`` ` r rzRegistry.register.sy" ? # ||C sg%9:; ;KRQt++C01DNN34G '.D   $ rc|jj|d|jjtj|d|jj|dyr)r poprr!r r rs r unregisterzRegistry.unregisterOsK 3% 4//4d; T*rr)NF)__name__ __module__ __qualname__rrrr%rr rrs!7$/B+rrc4eZdZdZdZdZedZdZy) BaseHandlerc||_y)z Initialize a new handler to handle a registered type. :Parameters: - `context`: reference to pickler/unpickler Ncontextr r.s r rzBaseHandler.__init__\s  rc2td|jz)a: Flatten `obj` into a json-friendly form and write result to `data`. :param object obj: The object to be serialized. :param dict data: A partially filled dictionary which will contain the json-friendly representation of `obj` once this method has finished. z"You must implement flatten() in %sNotImplementedError __class__r objdatas r flattenzBaseHandler.flattenfs""F"WXXrc2td|jz)z} Restore an object of the registered type from the json-friendly representation `obj` and return it. z"You must implement restore() in %sr1)r r5s r restorezBaseHandler.restoreqs ""F"WXXrc2tj|||S)z Register this handler for the given class. Suitable as a decorator, e.g.:: @MyCustomHandler.handles class MyCustomClass: def __reduce__(self): ... )registryrr$s r handleszBaseHandler.handlesxs #t$ rc||_|S)zThis permits registering either Handler instances or classes :Parameters: - `context`: reference to pickler/unpickler r-r/s r __call__zBaseHandler.__call__s   rN) r&r'r(rr7r9 classmethodr<r>r)rr r+r+[s, YY  rr+ceZdZdZdZdZy) ArrayHandlerz'Flatten and restore array.array objectsc|j|d<|jj|jd|d<|S)NtypecodeFresetvalues)rCr.r7tolistr4s r r7zArrayHandler.flattens8<<Z--cjjl%-HX rc|d}|jj|dd}|dk(r|Dcgc] }t|}}tj||Scc}w)NrCrFFrDc)r.r9bytesarray)r r6rCrFxs r r9zArrayHandler.restores^ #%%d8nE%B s?(./1eAhF/{{8V,,0sANr&r'r(__doc__r7r9r)rr rArAs1 -rrAceZdZdZdZdZy)DatetimeHandlerzCustom handler for datetime objects Datetime objects use __reduce__, and they generate binary strings encoding the payload. This handler encodes that payload to reconstruct the object. c`|j}|js+t|dr|j}|St |}|S|j \}}|j }tj|d}|g|ddD cgc] } || dc} z}||d|f|d<|Scc} w)N isoformatrrFrD __reduce__) r. unpicklablehasattrrRstrrSr7r b64encode) r r5r6picklerresultrargsr7payloadis r r7zDatetimeHandler.flattens,,""sK(MSMNN$ T//..a)yT!"XFXGAU3XFF%c7>\ GsB+c |d\}}|j}|j}||d}tj|d}|ft |ddDcgc] }||dc}z}|j |g|Scc}w)NrSFrDrr)r.r9r b64decodetuple__new__) r r6rrZ unpicklerr9valuer\paramss r r9zDatetimeHandler.restores& TLL ##c'tAw'EDH"MHq71E#:H"MNNs{{3(((#NsA> NrMr)rr rPrPs )rrPceZdZdZdZdZy) RegexHandlerz1Flatten _sre.SRE_Pattern (compiled regex) objectsc$|j|d<|SNpattern)rhr4s r r7zRegexHandler.flattens++Y rc2tj|dSrg)recompiler r6s r r9zRegexHandler.restoreszz$y/**rNrMr)rr reres;+rreceZdZdZdZdZy) QueueHandlerzOpaquely serializes Queue objects Queues contains mutex and condition variables which cannot be serialized. Construct a new Queue instance when restoring. c|Srr)r4s r r7zQueueHandler.flattens rc*tjSr)queueQueuerls r r9zQueueHandler.restores{{}rNrMr)rr rorosrroc:eZdZdZdZej fdZdZy) CloneFactoryzASerialization proxy for collections.defaultdict's default_factoryc||_yrexemplar)r rxs r rzCloneFactory.__init__s   rc&||jS)z>Create new instances by making copies of the provided exemplarrw)r clones r r>zCloneFactory.__call__sT]]##rc<dt|dd|jdS)Nz)idrxr s r __repr__zCloneFactory.__repr__s!+BtHQr}r)rr rurusK!"YY$LrruceZdZdZdZdZy) UUIDHandlerzSerialize uuid.UUID objectsc$|j|d<|SNhex)rr4s r r7zUUIDHandler.flattensggU  rc2tj|dSr)uuidUUIDrls r r9zUUIDHandler.restoresyye%%rNrMr)rr rrs%&rrceZdZdZdZdZy) LockHandlerz Serialize threading.Lock objectsc,|j|d<|S)Nlocked)rr4s r r7zLockHandler.flatten sX rcrtj}|jddr|j|S)NrF) threadingLockracquire)r r6locks r r9zLockHandler.restore s*~~ 88He $ LLN rNrMr)rr rrs*rrceZdZdZdZdZy) TextIOHandlerz>Serialize file descriptors as None because we cannot roundtripcyrr)r4s r r7zTextIOHandler.flattensrctd)z>Restore should never get called because flatten() returns Nonez+Restoring IO.TextIOHandler is not supported)AssertionErrorrls r r9zTextIOHandler.restoresJKKrNrMr)rr rrsHLrr)$rNrKr~datetimeiorrrjrrrmrrr;rr%rr+rAr<rPdatetimeretyperkrorsrurrrr_lockr3r TextIOWrapperr)rr rs   ;+;+| :     ll22j -; -  U[[!)k)D))* & &+;+ T*"**R.)* ;  U[[! L L&+& DII +    EOO$LKLb&&'r