o
bF
@ s d Z ddlZddlZddlZddlmZ ddlmZ ddl m
Z
m Z ddl m
Z
G dd dZG d d
d
eZG d d d eZG d
d deZG dd dZG dd dZe Zeeeeeeede ee g Z G dd dZ G dd dZ G dd dZ!G dd dZ"dd Z#G d d d e$Z%e&d Z'd d! Z(d"d# Z)d$d% Z*d&d' Z+d(d) Z,G d*d+ d+Z-d,d- Z.d8d.d/Z/d0d1 Z0d2d3 Z1d4d5 Z2G d6d7 d7Z3dS )9z|
AOT: Abstract Object Trees
The source-code-marshallin'est abstract-object-serializin'est persister
this side of Marmalade!
N)generate_tokens)crefutil)logreflect)_constructMethodc @ e Zd Zdd ZdS )Namedc C
|| _ d S Nname)selfr r 7/usr/lib/python3/dist-packages/twisted/persisted/aot.py__init__
zNamed.__init__N)__name__
__module__ __qualname__r r r r r r r c @ r )Classc C
d| j S )Nz Class(%r)r r
r r r getSource$ r zClass.getSourceNr r r r r r r r r # r r c @ r )Functionc C r )Nz Function(%r)r r r r r r ) r zFunction.getSourceNr r r r r r ( r r c @ r )Modulec C r )Nz
Module(%r)r r r r r r . r zModule.getSourceNr r r r r r - r r c @ e Zd Zdd Zdd ZdS )InstanceMethodc C s@ t |tst |tst |tstd| || _|| _|| _d S )Nz$%s isn't an Instance, Ref, or Deref!)
isinstanceRefInstanceDeref TypeErrorr klassinstance)r
r r$ instr r r r 3 s
zInstanceMethod.__init__c C s d | j| jt| jS )Nz InstanceMethod({!r}, {!r},
{}))formatr r$ prettifyr% r r r r r > s
zInstanceMethod.getSourceNr r r r r r r r r r 2 s r c @ s e Zd ZdS ) _NoStateObjN)r r r r r r r r* F s r* c @ s e Zd ZefddZdd ZdS )r! c K sD t |ts td| || _|tur|| _d| _d S || _d| _d S )Nz%s isn't a string!r )r strr# r$
NoStateObjstate stateIsDict)r
className __stateObj__r. r r r r Z s
zInstance.__init__c C s | j r| j}nt| jtrt| jjtr| jj}nd }|d urBz
d| jdt| dW S tyA d| jdt | d Y S w d| jdt | j dS )Nz Instance(, ))
r/ r. r r objdictr$ dictToKWNonFormattableDictr( )r
stateDictr r r r e s
zInstance.getSourceN)r r r r- r r r r r r r! Y s r! c @ , e Zd Zdd Zdd Zdd Zdd Zd S )
r c G s< t |dkr|d | _|d | _d S |s d | _d | _d S d S )N r r+ )lenrefnumr4 )r
argsr r r r v s
z Ref.__init__c C & | j rtd| d| j || _ d S )NzError setting id , I already have )r<
ValueErrorr
numr r r setRef
z
Ref.setRefc C r> )NzError setting obj r? )r4 r@ )r
r4 r r r setObj rD z
Ref.setObjc C s6 | j d u r td| jrd| jt| j f S t| j S )Nz7Don't try to display me before setting an object on me!z
Ref(%d,
%s))r4 RuntimeErrorr< r( r r r r r s
z
Ref.getSourceN)r r r r rC rE r r r r r r u s
r c @ s e Zd Zdd Zdd ZeZdS )r" c C r r
r< rA r r r r r zDeref.__init__c C r )Nz Deref(%d)rG r r r r r r zDeref.getSourceN)r r r r r __repr__r r r r r" s r" c @ r )Copyregc C s || _ || _d S r
)loadfuncr. )r
rJ r. r r r r s
zCopyreg.__init__c C s d| j dt| j dS )NzCopyreg(r2 r3 )rJ r( r. r r r r r s zCopyreg.getSourceNr) r r r r rI s rI c C s t dt| S )zDPass me an AO, I'll return a nicely-formatted source representation.zapp = ) indentifyr( )aor r r r s r c @ s e Zd ZdZdS )r7 z!A dictionary was not formattable.N)r r r __doc__r r r r r7 s r7 z[a-zA-Z_][a-zA-Z0-9_]*$c C sx g }t | }| |D ](\}}t|ts td| t|s(td| |d| dt | d qd
|S )Nz%r ain't a stringz%r ain't an identifier
=, ) listitemssortr r, r7 rmatchappendr( join)doutrS kvr r r r6 s
r6 c C s4 t | dr | S t| }|tv rt| S |tu rEdg}| D ]\}}|dt| dt| d q |t | r=dp>d d
|S |t u rjd g}| D ] }|d
t| qN|t | rbd pcd d
|S |t u rd
g}| D ] }|d
t| qs|t | rdpd d
|S t
d| d| d)Nr {rN z: rP z
}}rQ [z
%s,z
]](z
)r3 zUnsupported type z when trying to prettify .)hasattrr type_SIMPLE_BUILTINSreprr5 rS rW r( r; rX rR tupler# )r4 trZ r[ r\ xr r r r( s0
"
r( c C s g }g }d| g}t |jD ]0\}}\}}\}} }
|dv r"|| n|dv r*| |dkr8|dt| q
|| q
d|S )NrQ )r_ ra r] )r` r3 r^ z )tokenizepoprW r; rX ) srZ stackl tokenType tokenStringstartRow startColumnendRow endColumn logicalLiner r r rK s$
rK c C t | S )zG
Pass me an Abstract Object Tree, and I'll unjelly it for you.
) AOTUnjellierunjelly)aotr r r unjellyFromAOT s r{ c C sf t tttttttd}t| dr| }n| }t
|dd}t ||| d|v r-t |d S t
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! r r r r r r" rI readzexecappz'%s needs to define an 'app', it didn't!)r! r r r r r r" rI rc r| compileevalr{ r@ ) stringOrFilenssourcecoder r r unjellyFromSource s"
r c @ sH e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dS )rx zWI handle the unjellying of an Abstract Object Tree.
See AOTUnjellier.unjellyAO
c C s i | _ g | _g | _d S r
)
referencesrn afterUnjellyr r r r r , s
zAOTUnjellier.__init__c C s t }| |d| |S )zUnjelly a node, later.r )r _Defer unjellyInto)r
noderY r r r unjellyLater4 s zAOTUnjellier.unjellyLaterc C s. | |}|||< t|tjr||| |S )zvUtility method for unjellying one object into another.
This automates the handling of backreferences.
) unjellyAOr r NotKnown addDependant)r
r4 locrL or r r r : s
zAOTUnjellier.unjellyIntoc C s: t |tjrd g}||d n|g}| j||f d S )Nr+ )r r r r r rW )r
callableresult
listResultr r r callAfterD s
zAOTUnjellier.callAfterc C s | |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__)r
r% attrNamerL r r r unjellyAttributeL s z AOTUnjellier.unjellyAttributec C s | j | t|}|tv r|S |tu r-g }|D ]}|d | |t|d | q|S |tu rUg }t}|D ]}|d t| |t|d |t j
rPt j }q7||S |t u ryi }|
D ]\}}t |} | | d| | | d| q_|S |j}
|
tu rt|jS |
ttfv st|
trt|jS |
tu r|j} t|j} | |j}
| | j v r|
du rt | | S t|
t j
rt | |
| S t | | |
S t d|
t!u rt|j}| |j"}|#|}t$|dr| %|j&| |S ||_ |S |
t'u r;| |j(}|j)}| j*+|}|du r|| j*|< |S t|t j
r*|,| || j*|< |S |du r2 |S t-d|||f |
t.u r\|j)}| j*+|}|du rZt /|}|| j*|< |S |S |
t0u rut|j1}| 2|j"3dd |}|S t d | )
zaUnjelly an Abstract Object and everything it contains.
I return the real object.
Nr+ r zinstance method changed __setstate__z1Multiple references with the same ID: %s, %s, %s!c S s || S r
r )r _lr r r s z(AOTUnjellier.unjellyAO..zUnsupported AOT type: %s)4rn rW rd re rR r r; rg r r r _Tupler5 rS _DictKeyAndValue __class__r r namedModuler r r
issubclass namedObjectr r$ r r% r getattr_InstanceMethodr r# r! r. __new__rc r r r r4 r< r getresolveDependantsr@ r" _DereferencerI rJ r addCallback)r
rL rh ro ri tuple_rY r[ r\ kvdcim_nameim_classim_selfr$ r. r& r refkeyrefrB derrJ r r r r U s
zAOTUnjellier.unjellyAOc
C sn z d g}| |d| | jD ]
\}}||d q|d W S ty6 td tdtt| j w )Nr +Error jellying object! Stacktrace follows::
) r r
BaseExceptionr msgrX maprf rn )r
rL ro funcr\ r r r ry s
zAOTUnjellier.unjellyN) r r r rM r r r r r r ry r r r r rx ' s
frx c C rw )z-Convert an object to an Abstract Object Tree.)
AOTJellierjelly)r4 r r r
jellyToAOT s r c C s, t | }|r|t|d dS t|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)r writer encode)r4 filerz r r r
jellyToSource s r c C s | j jS )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__r methodObjectr r r _classOfMethod s
r c C | j S )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__r r r r
_funcOfMethod
r c C r )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}
)r r r r r
_selfOfMethod r r c @ r9 )
r c C s i | _ d| _g | _d S )Nr )prepared_ref_idrn r r r r r s
zAOTJellier.__init__c C s || j t|< dS )zaI prepare an object for later referencing, by storing its id()
and its _AORef in a cache.N)r id)r
aorefobjectr r r
prepareForRef s zAOTJellier.prepareForRefc s t }jt |tv r nt|tjr-tt j
t t
t nt|tjr9t j
nt|t rFtt n|tju rStt nt jv r}jt }|jrk|j}t|S jd _j}|| t|S t fdd} |t u r fdd D nk|t u r t t!j n[|t"u ri } # D ]\}}|||< q | n<|t$j%v rt$j%| \} }
t&t | |
n t' dr| ( nt' dr| j) nt*d|j
jd = S )
z+I turn an object into an AOT and return it.r+ c s" tt j| d S r
)rE r! r qualr jellyToAO)r. r4 retvalr
r r
_stateFromH s z(AOTJellier.jellyToAO.._stateFromc s g | ]} |qS r )r ).0r r r r
P s z(AOTJellier.jellyToAO.. __getstate__r zUnsupported type: %s)+rd rn rW rf re r types
MethodTyper r r r r r r r
ModuleTyper r FunctionTyper fullFuncNamer r r< r rC r" r r rR rE rg r r5 rS copy_regdispatch_tablerI rc r r r# ) r
r4 objTypeoldRefkeyr rY r[ r\ unpickleFuncr. r r r r sb
zAOTJellier.jellyToAOc C s> z| |}|W S ty td td| j w )Nr r )r r r r rX rn )r
r4 rL r r r r l s
zAOTJellier.jellyN)r r r r r r r r r r r r s
Yr r
)4rM copyregr rer rk r twisted.persistedr twisted.pythonr r twisted.python.compatr r r r r r r* r- boolbytesr, intfloatcomplexrd sliceEllipsisre r! r r" rI r Exceptionr7 r rU r6 r( rK r{ r rx r r r r r r r r r r sZ
" '