File: //usr/lib/python3/dist-packages/twisted/persisted/__pycache__/aot.cpython-38.pyc
U
��W[�H �
@ sL d Z ddlmZmZ ddlZddlZzddlmZ W n ek
rT ddlmZ Y nX z ddl Z W n ddl
Z Y nX ddl m Z m
Z
ddlmZ ddlmZmZmZ G d d
� d
�ZG d d � d e�ZG d
d� de�ZG dd� de�ZG dd� d�ZG dd� d�Ze� Zee ee e e e d�e!e e"�g Z#ze#�$e%� W n e&k
�rJ Y nX G dd� d�Z'G dd� d�Z(G dd� d�Z)G dd � d �Z*d d � Z+G d d � d e,�Z-e�.d!�Z/d"d#� Z0d$d%� Z1d&d'� Z2d(d)� Z3d*d+� Z4G d,d-� d-�Z5d.d/� Z6d;d0d1�Z7zdd2lm8Z9m:Z; W n ek
�r dZ9dZ;Y nX d3d4� Z<d5d6� Z=d7d8� Z>G d9d:� d:�Z?dS )<z|
AOT: Abstract Object Trees
The source-code-marshallin'est abstract-object-serializin'est persister
this side of Marmalade!
� )�division�absolute_importN)�generate_tokens)�tokenize)�reflect�log)�crefutil)�unicode�_PY3�_constructMethodc @ s e Zd Zdd� ZdS )�Namedc C s
|| _ d S �N��name)�selfr � r �7/usr/lib/python3/dist-packages/twisted/persisted/aot.py�__init__' s zNamed.__init__N)�__name__�
__module__� __qualname__r r r r r r & s r c @ s e Zd Zdd� ZdS )�Classc C s
d| j S )Nz Class(%r)r �r r r r � getSource+ s zClass.getSourceN�r r r r r r r r r * s r c @ s e Zd Zdd� ZdS )�Functionc C s
d| j S )Nz Function(%r)r r r r r r / s zFunction.getSourceNr r r r r r . s r c @ s e Zd Zdd� ZdS )�Modulec C s
d| j S )Nz
Module(%r)r r r r r r 3 s zModule.getSourceNr r r r r r 2 s r c @ s e Zd Zdd� Zdd� ZdS )�InstanceMethodc C s@ t |t�s*t |t�s*t |t�s*td| ��|| _|| _|| _d S )Nz$%s isn't an Instance, Ref, or Deref!)�
isinstance�Ref�Instance�Deref� TypeErrorr �klass�instance)r r r# �instr r r r 8 s
zInstanceMethod.__init__c C s d| j | jt| j�f S )Nz InstanceMethod(%r, %r,
%s))r r# �prettifyr$ r r r r r ? s zInstanceMethod.getSourceN�r r r r r r r r r r 7 s r c @ s e Zd ZdS )� _NoStateObjN)r r r r r r r r( C s r( c @ s e Zd Zefdd�Zdd� ZdS )r c K sB t |t�std| ��|| _|tk r2|| _d| _n || _d| _d S )Nz%s isn't a string!r � )r �strr" r# �
NoStateObj�state� stateIsDict)r Z classNameZ __stateObj__r, r r r r R s
zInstance.__init__c C s� | j r| j}n(t| jt�r2t| jjt�r2| jj}nd }|d k rzzd| jt|�f W S tk
rx d| jt |�f Y S X d| jt | j�f S )NzInstance(%r, %s))
r- r, r r �obj�dictr# �dictToKW�NonFormattableDictr& )r Z stateDictr r r r ] s
zInstance.getSourceN)r r r r+ r r r r r r r Q s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r c G s6 t |�dkr"|d | _|d | _n|s2d | _d | _d S )N� r r) )�len�refnumr. )r �argsr r r r n s
z Ref.__init__c C s" | j rtd|| j f ��|| _ d S )Nz&Error setting id %s, I already have %s)r4 �
ValueError�r �numr r r �setRefw s z
Ref.setRefc C s" | j rtd|| j f ��|| _ d S )Nz'Error setting obj %s, I already have %s)r. r6 )r r. r r r �setObj| s z
Ref.setObjc C s6 | j d krtd��| jr,d| jt| j �f S t| j �S )Nz7Don't try to display me before setting an object on me!z
Ref(%d,
%s))r. � RuntimeErrorr4 r& r r r r r � s
z
Ref.getSourceN)r r r r r9 r: r r r r r r l s r c @ s e Zd Zdd� Zdd� ZeZdS )r! c C s
|| _ d S r
�r4 r7 r r r r � s zDeref.__init__c C s
d| j S )Nz Deref(%d)r< r r r r r � s zDeref.getSourceN)r r r r r �__repr__r r r r r! � s r! c @ s e Zd Zdd� Zdd� ZdS )�Copyregc C s || _ || _d S r
)�loadfuncr, )r r? r, r r r r � s zCopyreg.__init__c C s d| j t| j�f S )NzCopyreg(%r, %s))r? r& r, r r r r r � s zCopyreg.getSourceNr' r r r r r> � s r> 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 )r1 z&A dictionary was not formattable.
N)r r r �__doc__r r r r r1 � s r1 z[a-zA-Z_][a-zA-Z0-9_]*$c C sr g }t | �� �}|�� |D ]J\}}t|t�s:td| ��t�|�sPtd| ��|�d|t |�f � q d�
|�S )Nz%r ain't a stringz%r ain't an identifierz
%s=%s,� ) �list�items�sortr r* r1 �r�match�appendr& �join)�d�outrE �k�vr r r r0 � s
�r0 c C s2 t | d�r| �� S t| �}|tkr*t| �S |tkr�dg}| �� D ]"\}}|�dt|�t|�f � q@|�t | �rtdpvd� d�
|�S |t kr�dg}| D ]}|�dt|� � q�|�t | �r�d p�d
� d�
|�S |t k�r d g}| D ]}|�dt|� � q�|�t | ��rd �pd
� d�
|�S t
d|| f ��d S )Nr �{z
%s: %s,z
}�}rC �[z
%s,z
]�]�(z
)�)z/Unsupported type %s when trying to prettify %s.)�hasattrr �type�_SIMPLE_BUILTINS�reprr/ rE rI r& r3 rJ rD �tupler" )r. �trL rM rN �xr r r r& � s0
r& c C s� g }g }d| g}t |j�D ]`\}}\}}\}} }
|dkrD|�|� n|dkrT|�� |dkrp|�dt|� � q|�|� qd�|�S )NrC )rQ rS rO )rR rT rP � z )r �poprI r3 rJ ) �srL �stack�lZ tokenTypeZ tokenStringZstartRowZ startColumnZendRowZ endColumnZ logicalLiner r r r@ � s � r@ c C s t � �| �S )zG
Pass me an Abstract Object Tree, and I'll unjelly it for you.
)� AOTUnjellier�unjelly)�aotr r r �unjellyFromAOT� s rd c C sj t tttttttd�}t| d�r*| � � }n| }t
|dd�}t |||� d|krZt |d �S t
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!
)r r r r r r r! r> �readz<source>�execZappz'%s needs to define an 'app', it didn't!N)r r r r r r r! r> rU re �compile�evalrd r6 )Z stringOrFile�ns�source�coder r r �unjellyFromSource� s" �
rl c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )ra zWI handle the unjellying of an Abstract Object Tree.
See AOTUnjellier.unjellyAO
c C s i | _ g | _g | _d S r
)�
referencesr_ � afterUnjellyr r r r r s zAOTUnjellier.__init__c C s t �� }| �|d|� |S )z Unjelly a node, later.
r )r Z_Defer� unjellyInto)r ZnoderK r r r � unjellyLater# s zAOTUnjellier.unjellyLaterc C s. | � |�}|||< t|tj�r*|�||� |S )zvUtility method for unjellying one object into another.
This automates the handling of backreferences.
)� unjellyAOr r �NotKnown� addDependant)r r. ZlocrA �or r r ro * s
zAOTUnjellier.unjellyIntoc C s: t |tj�r d g}|�|d� n|g}| j�||f� d S )Nr) )r r rr rs rn rI )r �callable�resultr` r r r � callAfter4 s
zAOTUnjellier.callAfterc C s | � |j||� dS )z�Utility 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)ro �__dict__)r r$ ZattrNamerA r r r �unjellyAttribute< s z AOTUnjellier.unjellyAttributec C s, | j �|� t|�}|tkr |S |tkrZg }|D ]$}|�d� | �|t|�d |� q0|S |tkr�g }t}|D ]2}|�d� t| �|t|�d |�t j
�rnt j }qn||�S |t k� r�i }|�
� D ].\}}t �|�} | �| d|� | �| d|� q�|S |j}
|
tk�rt�|j�S |
ttfk�s*t|
t��r6t�|j�S |
tk�r�|j} t�|j�} | �|j�}
| | j k�r�|
dk�r~t | | �S t|
t j
��r�t � | |
| �S t | | |
�S nt d���nl|
t!k�r t�|j�}| �|j"�}t#|d��r�|�$|�}nt%|�}t#|d��r| �&|j'|� n||_ |S |
t(k�r�| �|j)�}|j*}| j+�,|�}|dk�rZ|| j+|< nBt|t j
��r~|�-|� || j+|< n |dk�r�nt.d|||f ��|S |
t/k�r�|j*}| j+�,|�}|dk�r�t �0|�}|| j+|< |S |S |
t1k�rt�|j2�}| �3|j"��4dd � |�}|S t d
| ��| j d = dS ) zaUnjelly an Abstract Object and everything it contains.
I return the real object.
Nr) r zinstance method changed�__new__� __setstate__z1Multiple references with the same ID: %s, %s, %s!c S s || � S r
r )rv Z_lr r r �<lambda>� � z(AOTUnjellier.unjellyAO.<locals>.<lambda>zUnsupported AOT type: %s���)5r_ rI rV rW rD ro r3 rY r r rr Z_Tupler/ rE Z_DictKeyAndValue� __class__r r Z namedModuler r r �
issubclassZ namedObjectr r# rq r$ rx �getattrZ_InstanceMethodr r" r r, rU rz �_OldStyleInstancerw r{ r r. r4 rm �getZresolveDependantsr6 r! Z _Dereferencer> r? rp Z addCallback)r rA rZ r` r[ Ztuple_rK rM rN Zkvd�cZim_name�im_class�im_selfr# r, r% rt Zrefkey�refr8 Zderr? r r r rq E s�
� zAOTUnjellier.unjellyAOc
C sp z:d g}| � |d|� | jD ]\}}||d � q |d W S t�d� t�d�tt| j��� � Y nX d S )Nr �+Error jellying object! Stacktrace follows::�
)ro rn r �msgrJ �maprX r_ )r rA r` �funcrN r r r rb � s
zAOTUnjellier.unjellyN) r r r rB r rp ro rw ry rq rb r r r r ra s
hra c C s t � �| �S )z-Convert an object to an Abstract Object Tree.)�
AOTJellier�jelly)r. r r r �
jellyToAOT� s r� c C s. t | �}|r"|�t|��d�� nt|�S dS )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)r. �filerc r r r �
jellyToSource� s r� )� ClassType� InstanceTypec C s t r | jjS | 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{types.ClassType} or L{type}
)r
�__self__r r� �Z methodObjectr r r �_classOfMethod� s
r� c C s t r
| jS | 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}
)r
�__func__Zim_funcr� r r r �
_funcOfMethod� s
r� c C s t r
| jS | 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}
)r
r� r� r� r r r �
_selfOfMethod� s
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r� c C s i | _ d| _g | _d S )Nr )�prepared�_ref_idr_ 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 Zaoref�objectr r r �
prepareForRef s zAOTJellier.prepareForRefc sH t � �}�j�t� �� |tkr(� ��n|tjkr\tt� �j t
� t � ����
t� �����n�|tjkrtt� j ���n�|tkr�tt
� � ����n�t|t �r�tt
� � ����n�|tjkr�tt
�� ����nvt� ��jk�r�jt� � }|jr�|j}n �jd �_�j}|�|� t |�S t � �� ��fdd�}�� �� � |t k�r\�� �fdd�� D �� n�|t!k�r~�� t!t"�j
� ��� n�|t#k�r�i }� �$� D ] \}}��
|�|��
|�< �q��� |� n||t%j&k�r�t%j&| � �\} }
�� t't
�| ���
|
��� n@t(� d��r|� �)� � n&t(� d��r.|� 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
)r: r r �qualr � jellyToAO)r, �r. Zretvalr r r �
_stateFromI s �z(AOTJellier.jellyToAO.<locals>._stateFromc s g | ]}� � |��qS r )r� )�.0rt r r r �
<listcomp>O s z(AOTJellier.jellyToAO.<locals>.<listcomp>� __getstate__rx zUnsupported type: %sr~ ),rV r_ rI rX rW �types�
MethodTyper r� r r r� r� r� r� �
ModuleTyper �_OldStyleClassr r� � FunctionTyper Z fullFuncNamer� r� r4 r� r9 r! r r� rD r: rY r� r/ rE �copy_reg�dispatch_tabler> rU r� rx r" ) r r. ZobjTypeZoldRef�keyr� rK rM rN Z unpickleFuncr, r r� r r� sb
�
� zAOTJellier.jellyToAOc C s@ z| � |�}|W S t�d� t�d�| j�� � Y nX d S )Nr� r� )r� r r� rJ r_ )r r. rA r r r r� j s
zAOTJellier.jellyN)r r r r r� r� r� r r r r r� s Vr� )N)@rB Z
__future__r r r� �rer r � ImportErrorr� �copyregZtwisted.pythonr r Ztwisted.persistedr Ztwisted.python.compatr r
r r r r r r r( r+ �bool�bytes�int�float�complexrV �slice�EllipsisrW rI Zlong� NameErrorr r r! r> r � Exceptionr1 rg rG r0 r& r@ rd rl ra r� r� r� r� r� r� r� r� r� r� r r r r �<module> sv �
! '