B |bto@sddlmZmZmZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZddlmZd(d d Zd d ZddZGdddeZGdddeZddZddZd)ddZddZd*ddZGdddZd d!Zd"d#Zd$d%ZGd&d'd'eZdS)+)absolute_importdivisionunicode_literalsN)compaterrorshandlerstagsutil)json) numeric_typesFTignorec Cs`t|tr|}nt|s(td|p.t}|pDt|||||d}| |} |j | ||dS)aConvert a JSON string into a Python object. The keyword argument 'keys' defaults to False. If set to True then jsonpickle will decode non-string dictionary keys into python objects via the jsonpickle protocol. The keyword argument 'classes' defaults to None. If set to a single class, or a sequence (list, set, tuple) of classes, then the classes will be made available when constructing objects. This can be used to give jsonpickle access to local classes that are not available through the global module import scope. The keyword argument 'safe' defaults to False. If set to True, eval() is avoided, but backwards-compatible (pre-0.7.0) deserialization of repr-serialized objects is disabled. The keyword argument 'backend' defaults to None. If set to an instance of jsonpickle.backend.JSONBackend, jsonpickle will use that backend for deserialization. The keyword argument 'v1_decode' defaults to False. If set to True it enables you to decode objects serialized in jsonpickle v1. Please do not attempt to re-encode the objects in the v1 format! Version 2's format fixes issue #255, and allows dictionary identity to be preserved through an encode/decode cycle. The keyword argument 'on_missing' defaults to 'ignore'. If set to 'error', it will raise an error if the class it's decoding is not found. If set to 'warn', it will warn you in said case. If set to a non-awaitable function, it will call said callback function with the class name (a string) as the only parameter. Strings passed to on_missing are lowercased automatically. >>> decode('"my string"') == 'my string' True >>> decode('36') 36 zHUnpickler.on_missing must be a string or a function! It will be ignored!)keysbackendsafe v1_decode on_missing)resetclasses) isinstancestrlowerr is_functionwarningswarnr Unpicklerdecoderestore) stringrcontextrrrrrrdatar!3lib/python3.7/site-packages/jsonpickle/unpickler.pyrs3    rcCs*yt||dStk r$dSXdS)zBWorkaround unreliable hasattr() availability on sqlalchemy objectsTFN)object__getattribute__AttributeError)objattrr!r!r" _safe_hasattrXs  r(cCst|tjo|tjS)zc Cs|r"y||Stk r YnX|d}xtt|dddD]}td|d|}yRt|tj |}x:||dD]*}yt ||}Wqt k rwYqXqW|St t t fk rwBYqBXqBWdS)zLoads the module and returns the class. >>> cls = loadclass('datetime.datetime') >>> cls.__name__ 'datetime' >>> loadclass('does.not.exist') >>> loadclass('builtins.int')() 0 .rrN)KeyErrorsplitrangelenr Zuntranslate_module_namejoin __import__sysmodulesgetattrr% ImportError ValueError)Zmodule_and_namernamesZup_tomoduler& class_namer!r!r" loadclasss(    rOcCst|tko||kS)zHelper class that tests to see if the obj is a dictionary and contains a particular key/tag. >>> obj = {'test': 1} >>> has_tag(obj, 'test') True >>> has_tag(obj, 'fail') False >>> has_tag(42, 'fail') False )typedict)r&tagr!r!r"has_tagsrScCst|tjrtdt|tjr*|tjSt|tjr@|tjSy|tj}|tj}Wntk rlgSXt ||d}|sgSt |drt |j t |kr|SgS)z'Return arguments suitable for __new__()z+__newargs_ex__ returns both args and kwargs)r_fields) rSr NEWARGSEXrKZNEWARGSZINITARGSSEQOBJECTrArOhasattrrDrT)r&rZseq_listZobj_dicttyperefr!r!r"getargss$        rZc@seZdZdZdS)_trivialclassicz? A trivial class that can be instantiated with no args N)r2r3r4r5r!r!r!r"r[sr[cCst}||_|S)z Implement the mandated strategy for dealing with classic classes which cannot be instantiated without __getinitargs__ because they take parameters )r[ __class__)clsr.r!r!r"make_blank_classicsr^cCsD|d\}}t}|}d|kr0|ddd}t|||<t|S)zReturns an instance of the object from the object's repr() string. It involves the dynamic specification of code. >>> obj = loadrepr('datetime/datetime.datetime.now()') >>> obj.__class__.__name__ 'datetime' /r?rr)rBlocalsrFeval)ZreprstrrMZevalstrZmylocalsZ localnamer!r!r"loadreprs  rbcCs||kS)zHelper class that tests to see if the obj is a dictionary and contains a particular key/tag. >>> obj = {'test': 1} >>> has_tag(obj, 'test') True >>> has_tag(obj, 'fail') False >>> has_tag(42, 'fail') False r!)r&rRr!r!r" has_tag_dictsrcc@seZdZdHddZddZdd Zd d ZdId dZddZddZ ddZ ddZ ddZ ddZ ddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2ZdJd3d4Zd5d6Zd7d8ZdKd:d;Zdd?Zd@dAZ dBdCZ!dDdEZ"dFdGZ#dS)LrNFr cCs.|pt|_||_||_||_||_|dS)N)r rrrrrr)r/rrrrrr!r!r"r0&s  zUnpickler.__init__cCs(i|_g|_i|_g|_g|_i|_dS)z#Resets the object's internal state.N) _namedict _namestack _obj_to_idxr8_proxies_classes)r/r!r!r"r1s zUnpickler.resetcCs.x"|jD]\}}}}||||qWg|_dS)z2Replace proxies with their corresponding instancesN)rg)r/r&r'r;methodr!r!r" _swap_proxiesAszUnpickler._swap_proxiescCs0t|tttttfsdd}n ||}||S)NcSs|S)Nr!)xr!r!r"rLsz#Unpickler._restore..restore)rrlistrQsettuple _restore_tags)r/r&rr!r!r"_restoreGs  zUnpickler._restoreTcCs4|r ||r||||}|r0||S)a#Restores a flattened object to its original python state. Simply returns any of the basic builtin types >>> u = Unpickler() >>> u.restore('hello world') == 'hello world' True >>> u.restore({'key': 'value'}) == {'key': 'value'} True )rregister_classesrprj)r/r&rrvaluer!r!r"rSs   zUnpickler.restorecCs>t|tttfr*x(|D]}||qWn||jt|<dS)zqRegister one or more classes :param classes: sequence of classes or a single class to register N)rrlrnrmrqrhr Zimportable_name)r/rr]r!r!r"rqhs zUnpickler.register_classescCst|tjdS)Nzutf-8)r Z b64decoder B64encode)r/r&r!r!r"_restore_base64tszUnpickler._restore_base64cCst|tjdS)Nzutf-8)r Z b85decoder B85rt)r/r&r!r!r"_restore_base85wszUnpickler._restore_base85cCst|tjdS)Nzutf-8)quopriZ decodestringr BYTESrt)r/r&r!r!r"_restore_quopri{szUnpickler._restore_quopricCsdd|jS)aCalculates the name of the current location in the JSON stack. This is called as jsonpickle traverses the object structure to create references to previously-traversed objects. This allows cyclical data structures such as doubly-linked lists. jsonpickle ensures that duplicate python references to the same object results in only a single JSON object definition and special reference tags to represent each reference. >>> u = Unpickler() >>> u._namestack = [] >>> u._refname() == '/' True >>> u._namestack = ['a'] >>> u._refname() == '/a' True >>> u._namestack = ['a', 'b'] >>> u._refname() == '/a/b' True r_)rEre)r/r!r!r"_refname~szUnpickler._refnamecCsZt|}y|j|Wn>tk rTt|j|j|<|j|||j|<YnX|S)N)idrfrArDr8appendrdr{)r/r&Zobj_idr!r!r"_mkrefs zUnpickler._mkrefcsVgfdd|D}|tfddtD}j|S)Ncsg|]}|qSr!)rp).0v)r/r!r" sz+Unpickler._restore_list..cs&g|]\}}t|tr||fqSr!)rr-)rr=rr)riparentr!r"rs)r~extendr> enumeraterg)r/r&ZchildrenZproxiesr!)rirr/r" _restore_lists     zUnpickler._restore_listcCst||tjS)N)iterrr ITERATOR)r/r&r!r!r"_restore_iteratorszUnpickler._restore_iteratorcCsHt|}t|}|j|}||j|<|j|=||j|<||j|<dS)N)r|rfr8rdr{)r/r;r.Zproxy_idZ instance_idZinstance_indexr!r!r"_swaprefs   zUnpickler._swaprefcCs,t}||tt|j|tj}t|dkrJ|dgdt||\}}}}}|tj ksrt |dddkr|d} t | t s|| } | j | f|dd} n||} |ry| |Wntk ry.x"| jD]\} } || | qW|| _Wntk ry(x"|D]\} } t| | | q"WWnZtk r|\} }| rj| j| |rx"|D]\} } t| | | qzWYnXYnXYnX|ry| |Wn0tk rx|D]}| |qWYnX|rx|D]\} } | | | qW|| ||| | S)z Supports restoring with all elements of __reduce__ as per pep 307. Assumes that iterator items (the last two) are represented as lists as per pickler implementation. Nr2 __newobj__rr)r-r~rlmaprpr REDUCErDrZNEWOBJrIrrP__new__ __setstate__r%__dict__items setdefaultr: Exceptionupdater} __setitem__rr)r/r&r;Z reduce_valfargsstateZ listitemsZ dictitemsr]Zstage1krZ dict_stateZ slots_staterkr!r!r"_restore_reducesV      $   zUnpickler._restore_reducecCs6y|tj}|j|Stk r0t|j|SXdS)N)r IDr8 IndexErrorr6)r/r&r=r!r!r" _restore_ids   zUnpickler._restore_idcCs|j|tjS)N)rdr1r REF)r/r&r!r!r" _restore_refszUnpickler._restore_refcCs$t|tj|jd}|dkr |S|S)N)r)rOr TYPErh)r/r&rYr!r!r" _restore_typeszUnpickler._restore_typecCs"|jr dSt|tj}||S)N)rrbr REPRr~)r/r&r!r!r" _restore_repr szUnpickler._restore_reprcCs2y |d}Wntk r dSX|d=||S)Ndefault_factory)rArp)r/r&rr!r!r" _loadfactorys  zUnpickler._loadfactorycCsZ|jdkr nJ|jdkr&td|n0|jdkr@td|nt|jrV||dS)Nr rz,Unpickler._restore_object could not find %s!errorz+Unpickler.restore_object could not find %s!)rrrrZClassNotFoundErrorr r)r/rNr!r!r"_process_missings     zUnpickler._process_missingcCs.t|r*t|ttjd|j|ddd}|S)zRestore a possibly pickled keyNTF)rrrr)r,rrDr r*r)r/r+r!r!r"_restore_pickled_key)szUnpickler._restore_pickled_keycCs|jr|j}ndd}|S)zReturn a callable that restores keys This function is responsible for restoring non-string keys when we are decoding with `keys=True`. cSs|S)Nr!)r+r!r!r" restore_keyEsz.Unpickler._restore_key_fn..restore_key)rr)r/rr!r!r"_restore_key_fn5s zUnpickler._restore_key_fnc CsJ|}t}i}x"t|D]\}}|r8|tjkr8qt|trL|} n|} |j | ||}| |} t |st |ry"|dkrt||| n| ||<Wn0tk r|dkr| ||<|j wYnXn0|dst||| nt|d|jj|| t| tr(|j ||| |f|j qW|rF||}|S)Nr___)rr<r rr RESERVEDrr __str__rer}rpZ is_noncomplexZis_dictionary_subclassr: TypeErrorpopr)r\r2r-rg) r/r&r.ignorereservedrriZdeferredrrstr_krrr!r!r"_restore_from_dictJs>          zUnpickler._restore_from_dictcCs||tj}t|to2t|dko2t|dt}|oDt|dt}t|dr\||nft|trx|j ||dd}nJ|r|j |d|dd}|r|j |d|dd}nt|dst|ds|}|S) NrrrF)r__getnewargs____getnewargs_ex__) rpr STATErrnrDrQrXrr)r/r&r.rZ has_slotsZhas_slots_and_dictr!r!r"_restore_state|s $    zUnpickler._restore_statecCs|||}t|tjrvt|drHxR|tjD]}|||q.Wn.t|drvx"|tjD]}|||q^Wt|tjr| ||}|S)Nr}add) rrSr rVrXr}rprrr)r/r&r.rr!r!r""_restore_object_instance_variabless      z,Unpickler._restore_object_instance_variablesrc Cst}||||}t|tjr6|tj\}}nt||jd}i}|rV||}|rd||}t |t pxt |dd }yP|st |dr|r|j ||f||} || _q|j |f||} n t |} Wntk rd}YnX|rDy ||} WnJtk rBy t|} Wn&tk r<||||SXYnX|| ||| t | trj| S||| } t| drt | jtr| j| _| S)N)rZ__meta__rTr)r-r~rrSr rUrZrhrprrPrIrXrrr#rr^rrrrrnrr(r1) r/r&r]rNr;factoryrkwargsZ is_oldstyler.r!r!r"_restore_object_instancesN               z"Unpickler._restore_object_instancecCs|tj}t||jd}t|t|}|dk rjt}|||||}| || |||S|dkr|||S| |||S)N)r) r rWrOrhrr1r-r~rrrr)r/r&rNr]Zhandlerr;r.r!r!r"_restore_objects     zUnpickler._restore_objectcCst|tj|jdS)N)r)rOr FUNCTIONrh)r/r&r!r!r"_restore_functionszUnpickler._restore_functioncsfdd|tjDS)Ncsh|]}|qSr!)rp)rr)r/r!r" sz)Unpickler._restore_set..)r SET)r/r&r!)r/r" _restore_setszUnpickler._restore_setcCsFi}|js|||jrx\t|D]N\}}t|r8q&t|trL|}n|}|j || |||<|j q&Wxt|D]b\}}t|sq|j || |}| |||<}t|tr|j |||tf|j qWnVxTt|D]F\}}t|tr|}n|}|j || |||<|j qW|S)N)rr~rr rr,rr rrer}rprrr-rgr>)r/r&r rrrresultr!r!r" _restore_dicts<          zUnpickler._restore_dictcCsy.tjt|ks,t|ttfkr,dd}|SWntk rBYnXt|tkrjt|tjrh|j }qt|tj r||j }nt|tj r|j }nt|tjr|j}nt|tjr|j}nt|tjr|j}nt|tjr|j}nt|tjr|j}ntt|tjr |j}n^t|tjr |j}nHt|tjr6|j}n2t|tjrL|j}nt|tj rb|j!}n|j"}nt#$|r~|j%}ndd}|S)NcSs|S)Nr!)rkr!r!r"r0sz(Unpickler._restore_tags..restorecSs|S)Nr!)rkr!r!r"rWs)&r rrmrPrlrQrrcTUPLE_restore_tuplerrrsrurvrwrrrrrWrrrrrrrryrzrrrrrr Zis_listr)r/r&rr!r!r"ro,sL          zUnpickler._restore_tagscstfdd|tjDS)Ncsg|]}|qSr!)rp)rr)r/r!r"r]sz,Unpickler._restore_tuple..)rnr r)r/r&r!)r/r"r\szUnpickler._restore_tuple)NFFFr )TN)T)r)$r2r3r4r0rrjrprrqrurwrzr{r~rrrrrrrrrrrrrrrrrrrrrorr!r!r!r"r%sB     ?   2 ;00r)NNFTFNFr )N)N) Z __future__rrrrxrGrrrrrr r rr r rr(r,r#r-r6r<r>rOrSrZr[r^rbrcrr!r!r!r"s8   = '  (