U TgG @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@seZdZddZdS)NamedcCs ||_dSNname)selfr r 9/tmp/pip-unpacked-wheel-1_7di8ic/twisted/persisted/aot.py__init__"szNamed.__init__N)__name__ __module__ __qualname__rr r r rr!src@seZdZddZdS)ClasscCs d|jS)Nz Class(%r)r r r r r getSource'szClass.getSourceNrrrrr r r rr&src@seZdZddZdS)FunctioncCs d|jS)Nz Function(%r)r rr r rr,szFunction.getSourceNrr r r rr+src@seZdZddZdS)ModulecCs d|jS)Nz Module(%r)r rr r rr1szModule.getSourceNrr r r rr0src@seZdZddZddZdS)InstanceMethodcCs@t|ts*t|ts*t|ts*td|||_||_||_dS)Nz$%s isn't an Instance, Ref, or Deref!) isinstanceRefInstanceDeref TypeErrorr klassinstance)r r rinstr r rr6s zInstanceMethod.__init__cCsd|j|jt|jS)Nz InstanceMethod({!r}, {!r}, {}))formatr rprettifyr rr r rrAs zInstanceMethod.getSourceNrrrrrr r r rr5s rc@s eZdZdS) _NoStateObjN)rrrr r r rr%Isr%c@s eZdZefddZddZdS)rcKsBt|tstd|||_|tk r2||_d|_n ||_d|_dS)Nz%s isn't a string!rr)rstrrr NoStateObjstate stateIsDict)r Z classNameZ __stateObj__r(r r rr]s  zInstance.__init__cCs|jr|j}n(t|jtr2t|jjtr2|jj}nd}|dk rzd|jdt|dWStk rd|jdt |dYSXd|jdt |jdS)Nz Instance(, )) r)r(rrobjdictrdictToKWNonFormattableDictr#)r Z stateDictr r rrhs zInstance.getSourceN)rrrr'rrr r r rr\s rc@s,eZdZddZddZddZddZd S) rcGs6t|dkr"|d|_|d|_n|s2d|_d|_dS)Nrr)lenrefnumr,)r argsr r rrys    z Ref.__init__cCs&|jrtd|d|j||_dS)NzError setting id , I already have )r2 ValueErrorr numr r rsetRefsz Ref.setRefcCs&|jrtd|d|j||_dS)NzError setting obj r4)r,r5)r r,r r rsetObjsz Ref.setObjcCs6|jdkrtd|jr,d|jt|jfSt|jS)Nz7Don't try to display me before setting an object on me!z Ref(%d, %s))r, RuntimeErrorr2r#rr r rrs z Ref.getSourceN)rrrrr8r9rr r r rrxs rc@s eZdZddZddZeZdS)rcCs ||_dSr r2r6r r rrszDeref.__init__cCs d|jS)Nz Deref(%d)r;rr r rrszDeref.getSourceN)rrrrr__repr__r r r rrsrc@seZdZddZddZdS)CopyregcCs||_||_dSr )loadfuncr()r r>r(r r rrszCopyreg.__init__cCsd|jdt|jdS)NzCopyreg(r*r+)r>r#r(rr r rrszCopyreg.getSourceNr$r r r rr=sr=cCstdt|S)zDPass me an AO, I'll return a nicely-formatted source representation.zapp = ) indentifyr#)aor r rrsrc@seZdZdZdS)r/z!A dictionary was not formattable.N)rrr__doc__r r r rr/sr/z[a-zA-Z_][a-zA-Z0-9_]*$cCsxg}t|}||D]P\}}t|ts:td|t|sPtd||d|dt |dqd |S)Nz%r ain't a stringz%r ain't an identifier =,) listitemssortrr&r/rmatchappendr#join)doutrGkvr r rr.s      r.cCs>t|dr|St|}|tkr*t|S|tkrdg}|D](\}}|dt|dt|dq@|t |rzdp|dd |S|t krd g}|D]}|d t|q|t |rd pd d |S|t kr$d g}|D]}|d t|q|t |rdpdd |St d|d|ddS)Nr{rBz: rDz }}rE[z %s,z ]](z )r+zUnsupported type z when trying to prettify .)hasattrrtype_SIMPLE_BUILTINSreprr-rGrKr#r1rLrFtupler)r,trNrOrPxr r rr#s0 "    r#c Csg}g}d|g}t|jD]`\}}\}}\}} } |dkrD||n|dkrT||dkrp|dt|q||qd|S)NrE)rSrUrQ)rTr+rRz )tokenizepoprKr1rL) srNstackl tokenTypeZ tokenStringZstartRowZ startColumnZendRowZ endColumnZ logicalLiner r rr?s$  r?cCs t|S)zG Pass me an Abstract Object Tree, and I'll unjelly it for you. ) AOTUnjellierunjelly)aotr r runjellyFromAOTsrhc Csjttttttttd}t|dr*| }n|}t |dd}t |||d|krZt |dSt d|dS)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! )rrrrrrrr=readzexecZappz'%s needs to define an 'app', it didn't!N)rrrrrrrr=rWricompileevalrhr5)Z stringOrFilenssourcecoder r runjellyFromSource s"     rpc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)rezWI handle the unjellying of an Abstract Object Tree. See AOTUnjellier.unjellyAO cCsi|_g|_g|_dSr ) referencesrb afterUnjellyrr r rr/szAOTUnjellier.__init__cCst}||d||S)zUnjelly a node, later.r)rZ_Defer unjellyInto)r noderMr r r unjellyLater7szAOTUnjellier.unjellyLatercCs.||}|||<t|tjr*||||S)zvUtility method for unjellying one object into another. This automates the handling of backreferences. ) unjellyAOrrNotKnown addDependant)r r,locr@or r rrs=s    zAOTUnjellier.unjellyIntocCs:t|tjr dg}||dn|g}|j||fdS)Nr)rrrwrxrrrK)r callableresultZ listResultr r r 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)rs__dict__)r r attrNamer@r r runjellyAttributeOszAOTUnjellier.unjellyAttributecCs&|j|t|}|tkr |S|tkrZg}|D]$}|d||t|d|q0|S|tkrg}t}|D]2}|dt||t|d|t j rnt j }qn||S|t kri}| D].\}}t |} || d||| d|q|S|j} | tkrt|jS| ttfks*t| tr6t|jS| tkr|j} t|j} ||j} | | jkr| dkr~t| | St| t j rt | | | St| | | Snt dnn| t!krt|j}||j"}|#|}t$|dr|%|j&|nt|t r||_n |'|_|S| t(kr||j)}|j*}|j+,|}|dkr\||j+|<nBt|t j r|-|||j+|<n|dkrnt.d|||f|S| t/kr|j*}|j+,|}|dkrt 0|}||j+|<|S|S| t1krt|j2}|3|j"4dd|}|St d |dS) 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 )r|Z_lr r rz(AOTUnjellier.unjellyAO..zUnsupported AOT type: %s)5rbrKrXrYrFrsr1r[rrrwZ_Tupler-rGZ_DictKeyAndValue __class__rrZ namedModuler rr issubclassZ namedObjectrrrvr r~getattrZ_InstanceMethodrrrr(__new__rWr}r __getstate__rr,r2rqgetZresolveDependantsr5rZ _Dereferencer=r>ruZ addCallback)r r@r\rcr]tuple_rMrOrPZkvdcZim_nameZim_classZim_selfrr(r!rzZrefkeyrefr7Zderr>r r rrvXs                                      zAOTUnjellier.unjellyAOc Csxz:dg}||d||jD]\}}||dq|dWStk rrtdtdtt|jYnXdS)Nr+Error jellying object! Stacktrace follows:: ) rsrr BaseExceptionrmsgrLmaprZrb)r r@rcfuncrPr r rrfs  zAOTUnjellier.unjellyN) rrrrArrursr}rrvrfr r r rre*s  hrecCs t|S)z-Convert an object to an Abstract Object Tree.) AOTJellierjelly)r,r r r jellyToAOTsrcCs.t|}|r"|t|dnt|SdS)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)r,filergr r r 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} )__self__rZ methodObjectr r r_classOfMethods rcCs|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} )__func__rr r r _funcOfMethods rcCs|jS)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} )rrr r r _selfOfMethods rc@s,eZdZddZddZddZddZd S) rcCsi|_d|_g|_dS)Nr)prepared_ref_idrbrr r rr szAOTJellier.__init__cCs||jt|<dS)zaI prepare an object for later referencing, by storing its id() and its _AORef in a cache.N)rid)r Zaorefobjectr r r prepareForRefszAOTJellier.prepareForRefc s0t}jt|tkr(nt|tjr^tt j t t tnt|tjrxtj nt|trtt n|tjkrtt nttjkrjt}|jr|j}njd_j}||t|Stfdd}|tkrDfddDn|t krft t!jn|t"kri}#D]\}}|||<q||n||t$j%krt$j%|\} } t&t | | n@t'dr|(n&t'dr|j)nt*d|j jd =S) z+I turn an object into an AOT and return it.rcs"ttj|dSr )r9rrqualr jellyToAO)r(r,retvalr r r _stateFromLsz(AOTJellier.jellyToAO.._stateFromcsg|]}|qSr )r).0rzrr r Tsz(AOTJellier.jellyToAO..rr~zUnsupported type: %s)+rXrbrKrZrYrtypes MethodTyperrrrrrrr ModuleTyperr FunctionTyperZ fullFuncNamerrr2rr8rrrrFr9r[rr-rGcopy_regdispatch_tabler=rWrr~r) r r,ZobjTypeZoldRefkeyrrMrOrPZ unpickleFuncr(r rrrs`                  zAOTJellier.jellyToAOcCsHz||}|WStk rBtdtd|jYnXdS)Nrr)rrrrrLrb)r r,r@r r rrps  zAOTJellier.jellyN)rrrrrrrr r r rr sXr)N)5rAcopyregrrerZtwisted.persistedrZtwisted.pythonrrZtwisted.python.compatr _tokenizerr_rrrrrr%r'boolbytesr&intfloatcomplexrXsliceEllipsisrYrrrr=r Exceptionr/rkrIr.r#r?rhrprerrrrrrr r r rsX       ")