o iG @sdZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl m ZGdd d ZGd d d eZGd d d eZGdddeZGdddZGdddZeZeeeeeeedeeeg ZGdddZ GdddZ!GdddZ"GdddZ#ddZ$Gddde%Z&e'd Z(d!d"Z)d#d$Z*d%d&Z+d'd(Z,d)d*Z-Gd+d,d,Z.d-d.Z/d9d/d0Z0d1d2Z1d3d4Z2d5d6Z3Gd7d8d8Z4dS):z| AOT: Abstract Object Trees The source-code-marshallin'est abstract-object-serializin'est persister this side of Marmalade! N)crefutil)logreflect)_constructMethod)generate_tokensc@eZdZddZdS)NamedcC ||_dSNname)selfr r[C:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/persisted/aot.py__init__" zNamed.__init__N)__name__ __module__ __qualname__rrrrrr ! r c@r)ClasscC d|jS)Nz Class(%r)r rrrr getSource'rzClass.getSourceNrrrrrrrrr&rrc@r)FunctioncCr)Nz Function(%r)r rrrrr,rzFunction.getSourceNrrrrrr+rrc@r)ModulecCr)Nz Module(%r)r rrrrr1rzModule.getSourceNrrrrrr0rrc@eZdZddZddZdS)InstanceMethodcCs@t|tst|tst|tstd|||_||_||_dS)Nz$%s isn't an Instance, Ref, or Deref!) isinstanceRefInstanceDeref TypeErrorr klassinstance)rr r%instrrrr6s  zInstanceMethod.__init__cCsd|j|jt|jS)Nz InstanceMethod({!r}, {!r}, {}))formatr r%prettifyr&rrrrrAs zInstanceMethod.getSourceNrrrrrrrrrr5s rc@s eZdZdS) _NoStateObjN)rrrrrrrr+Isr+c@s eZdZefddZddZdS)r"cKsDt|ts td|||_|tur||_d|_dS||_d|_dS)Nz%s isn't a string!rr)r strr$r% NoStateObjstate stateIsDict)rZ classNameZ __stateObj__r.rrrr]s    zInstance.__init__cCs|jr|j}nt|jtrt|jjtr|jj}nd}|durBz d|jdt|dWStyAd|jdt |dYSwd|jdt |jdS)Nz Instance(, )) r/r.r r!objdictr%dictToKWNonFormattableDictr))rZ stateDictrrrrhs  zInstance.getSourceN)rrrr-rrrrrrr"\s  r"c@,eZdZddZddZddZddZd S) r!cGs<t|dkr|d|_|d|_dS|sd|_d|_dSdS)Nrr)lenrefnumr2)rargsrrrrys   z Ref.__init__cC&|jrtd|d|j||_dS)NzError setting id , I already have )r9 ValueErrorrnumrrrsetRef z Ref.setRefcCr;)NzError setting obj r<)r2r=)rr2rrrsetObjrAz Ref.setObjcCs6|jdur td|jrd|jt|jfSt|jS)Nz7Don't try to display me before setting an object on me!z Ref(%d, %s))r2 RuntimeErrorr9r)rrrrrs  z Ref.getSourceN)rrrrr@rBrrrrrr!xs   r!c@s eZdZddZddZeZdS)r#cCr r r9r>rrrrrzDeref.__init__cCr)Nz Deref(%d)rDrrrrrrzDeref.getSourceN)rrrrr__repr__rrrrr#sr#c@r)CopyregcCs||_||_dSr )loadfuncr.)rrGr.rrrrs zCopyreg.__init__cCsd|jdt|jdS)NzCopyreg(r0r1)rGr)r.rrrrrszCopyreg.getSourceNr*rrrrrFs rFcCstdt|S)zDPass me an AO, I'll return a nicely-formatted source representation.zapp = N) indentifyr))aorrrrsrc@seZdZdZdS)r5z!A dictionary was not formattable.N)rrr__doc__rrrrr5sr5z[a-zA-Z_][a-zA-Z0-9_]*$cCsxg}t|}||D](\}}t|tstd|t|s(td||d|dt |dqd |S)Nz%r ain't a stringz%r ain't an identifier =,) listitemssortr r,r5rmatchappendr)join)doutrPkvrrrr4s       r4cCs4t|dr |St|}|tvrt|S|turEdg}|D]\}}|dt|dt|dq |t |r=dp>dd |S|t urjd g}|D] }|d t|qN|t |rbd pcd d |S|t urd g}|D] }|d t|qs|t |rdpdd |St d|d|d)Nr{rKz: rMz }}rN[z %s,z ]](z )r1zUnsupported type z when trying to prettify .)hasattrrtype_SIMPLE_BUILTINSreprr3rPrTr)r8rUrOtupler$)r2trWrXrYxrrrr)s0 "   r)c Csg}g}d|g}t|jD]0\}}\}}\}} } |dvr"||n|dvr*||dkr8|dt|q ||q d|S)NrN)r\r^rZ)r]r1r[z )tokenizepoprTr8rU) srWstackl tokenTypeZ tokenStringZstartRowZ startColumnZendRowZ endColumnZ logicalLinerrrrHs$   rHcC t|S)zG Pass me an Abstract Object Tree, and I'll unjelly it for you. N) AOTUnjellierunjelly)aotrrrunjellyFromAOTs rrc Csfttttttttd}t|dr| }n|}t |dd}t |||d|vr-t |dSt d|)z Pass me a string of code or a filename that defines an 'app' variable (in terms of Abstract Objects!), and I'll execute it and unjelly the resulting AOT for you, returning a newly unpersisted Application object! )r"rrrrr!r#rFreadzexecZappz'%s needs to define an 'app', it didn't!N)r"rrrrr!r#rFr`rscompileevalrrr=)Z stringOrFilenssourcecoderrrunjellyFromSource s"     rzc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)rozWI handle the unjellying of an Abstract Object Tree. See AOTUnjellier.unjellyAO cCsi|_g|_g|_dSr ) referencesrk afterUnjellyrrrrr/s zAOTUnjellier.__init__cCst}||d||S)zUnjelly a node, later.rN)rZ_Defer unjellyInto)rnoderVrrr unjellyLater7szAOTUnjellier.unjellyLatercCs.||}|||<t|tjr||||S)zvUtility method for unjellying one object into another. This automates the handling of backreferences. N) unjellyAOr rNotKnown addDependant)rr2locrIorrrr}=s   zAOTUnjellier.unjellyIntocCs:t|tjrdg}||dn|g}|j||fdS)Nr)r rrrr|rT)rcallableresultZ listResultrrr callAfterGs zAOTUnjellier.callAftercCs||j||dS)zUtility method for unjellying into instances of attributes. Use this rather than unjellyAO unless you like surprising bugs! Alternatively, you can use unjellyInto on your instance's __dict__. N)r}__dict__)rr&attrNamerIrrrunjellyAttributeOszAOTUnjellier.unjellyAttributecCs|j|t|}|tvr|S|tur-g}|D]}|d||t|d|q|S|turUg}t}|D]}|dt||t|d|t j rPt j }q7||S|t uryi}| D]\}}t |} || d||| d|q_|S|j} | turt|jS| ttfvst| trt|jS| tur|j} t|j} ||j} | | jvr| durt| | St| t j rt | | | St| | | St d| t!urt|j}||j"}|#|}t$|dr|%|j&||St|t r||_|S|'|_|S| t(urH||j)}|j*}|j+,|}|dur$||j+|<|St|t j r7|-|||j+|<|S|dur? |St.d|||f| t/uri|j*}|j+,|}|durgt 0|}||j+|<|S|S| t1urt|j2}|3|j"4dd|}|St d |) zaUnjelly an Abstract Object and everything it contains. I return the real object. Nrrzinstance method changed __setstate__z1Multiple references with the same ID: %s, %s, %s!cSs||Sr r)rZ_lrrrsz(AOTUnjellier.unjellyAO..zUnsupported AOT type: %s)5rkrTrarbrOr}r8rdr rrZ_Tupler3rPZ_DictKeyAndValue __class__rrZ namedModuler rr issubclassZ namedObjectrr%rr&rgetattrZ_InstanceMethodrr$r"r.__new__r`rr __getstate__r!r2r9r{getZresolveDependantsr=r#Z _DereferencerFrGrZ addCallback)rrIrerlrftuple_rVrXrYZkvdcZim_nameZim_classZim_selfr%r.r'rZrefkeyrefr?ZderrGrrrrXs                                     zAOTUnjellier.unjellyAOc Csnzdg}||d||jD] \}}||dq|dWSty6tdtdtt|jw)Nr+Error jellying object! Stacktrace follows:: ) r}r| BaseExceptionrmsgrUmaprcrk)rrIrlfuncrYrrrrps   zAOTUnjellier.unjellyN) rrrrJrrr}rrrrprrrrro*s  hrocCrn)z-Convert an object to an Abstract Object Tree.N) AOTJellierjelly)r2rrr jellyToAOTs rcCs,t|}|r|t|ddSt|S)z Pass me an object and, optionally, a file object. I'll convert the object to an AOT either return it (if no file was specified) or write it to the file. zutf-8N)rwriterencode)r2filerqrrr jellyToSourcesrcCs|jjS)z Get the associated class of the given method object. @param methodObject: a bound method @type methodObject: L{types.MethodType} @return: a class @rtype: L{type} N)__self__rZ methodObjectrrr_classOfMethods rcC|jS)z Get the associated function of the given method object. @param methodObject: a bound method @type methodObject: L{types.MethodType} @return: the function implementing C{methodObject} @rtype: L{types.FunctionType} N)__func__rrrr _funcOfMethod rcCr)z Get the object that a bound method is bound to. @param methodObject: a bound method @type methodObject: L{types.MethodType} @return: the C{self} passed to C{methodObject} @rtype: L{object} N)rrrrr _selfOfMethodrrc@r6) rcCsi|_d|_g|_dS)Nr)prepared_ref_idrkrrrrr s zAOTJellier.__init__cCs||jt|<dS)zaI prepare an object for later referencing, by storing its id() and its _AORef in a cache.N)rid)rZaorefobjectrrr prepareForRefszAOTJellier.prepareForRefc s t}jt|tvrnt|tjr-tt j t t tnt|tjr9tj nt|trFtt n|tjurStt ntjvr}jt}|jrk|j}t|Sjd_j}||t|Stfdd}|turfddDnk|t urt t!jn[|t"uri}#D]\}}|||<q|n<|t$j%vrt$j%|\} } t&t | | nt'dr|(nt'dr|j)nt*d|j jd =S) z+I turn an object into an AOT and return it.rcs"ttj|dSr )rBr"rqualr jellyToAO)r.r2retvalrrr _stateFromLsz(AOTJellier.jellyToAO.._stateFromcsg|]}|qSr)r).0rrrr Tsz(AOTJellier.jellyToAO..rrzUnsupported type: %sN)+rarkrTrcrbrtypes MethodTyperrrrrrrr ModuleTyperr FunctionTyperZ fullFuncNamerrr9rr@r#r!rrOrBrdrr3rPcopy_regdispatch_tablerFr`rrr$) rr2ZobjTypeZoldRefkeyrrVrXrYZ unpickleFuncr.rrrrsb                zAOTJellier.jellyToAOcCs>z||}|WStytdtd|jw)Nrr)rrrrrUrk)rr2rIrrrrps   zAOTJellier.jellyN)rrrrrrrrrrrr s  Xrr )5rJcopyregrrerZtwisted.persistedrZtwisted.pythonrrZtwisted.python.compatr _tokenizerrhr rrrrr+r-boolbytesr,intfloatcomplexrasliceEllipsisrbr"r!r#rFr Exceptionr5rurRr4r)rHrrrzrorrrrrrrrrrsZ        ")