File: //usr/lib/python3/dist-packages/twisted/persisted/__pycache__/aot.cpython-310.pyc
o
�b�F �
@ s� d Z ddlZddlZddlZddlmZ ddlmZ ddl m
Z
m Z ddl m
Z
G dd� d�ZG d d
� d
e�ZG d d � d e�ZG d
d� de�ZG dd� d�ZG dd� d�Ze� Zeeeeeeed�e ee �g Z G dd� d�Z G dd� d�Z G dd� d�Z!G dd� d�Z"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� d7�Z3dS )9z|
AOT: Abstract Object Trees
The source-code-marshallin'est abstract-object-serializin'est persister
this side of Marmalade!
� N)�generate_tokens)�crefutil)�log�reflect)�_constructMethodc @ � e Zd Zdd� ZdS )�Namedc C �
|| _ d S �N��name)�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.getSourceN�r 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 |t�st |t�st |t�std| ��|| _|| _|| _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 3 s ���
zInstanceMethod.__init__c C s d� | j| jt| j��S )Nz InstanceMethod({!r}, {!r},
{}))�formatr r$ �prettifyr% r r r r r > s
�zInstanceMethod.getSourceN�r 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 Zefdd�Zdd� ZdS )r! c K sD t |t�s td| ��|| _|tur|| _d| _d S || _d| _d S )Nz%s isn't a string!r � )r �strr# r$ �
NoStateObj�state� stateIsDict)r
� className� __stateObj__r. r r r r Z s
zInstance.__init__c C s� | j r| j}nt| jt�rt| jjt�r| jj}nd }|d urBz
d| j�dt|�� d�W S tyA d| j�dt |�� d� Y S w d| j�dt | j�� d�S )Nz Instance(�, �))
r/ r. r r �obj�dictr$ �dictToKW�NonFormattableDictr( )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+ )�len�refnumr4 )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< �
ValueError�r
�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�� d�S )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|t�s td| ��t�|�s(td| ��|�d|� dt |�� d�� qd�
|�S )Nz%r ain't a stringz%r ain't an identifier�
�=�,� ) �list�items�sortr r, r7 �r�match�appendr( �join)�d�outrS �k�vr r r r6 � s
r6 c C s4 t | d�r | �� 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 r�d
g}| D ] }|�d
t|� � qs|�t | �r�dp�d� 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_BUILTINS�reprr5 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 |j�D ]0\}}\}}\}} }
|dv r"|�|� n|dv r*|�� |dkr8|�dt|� � q
|�|� q
d�|�S )NrQ )r_ ra r] )r` r3 r^ � z )�tokenize�poprW r; rX ) �srZ �stack�l� tokenType� tokenString�startRow� startColumn�endRow� 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.
)� AOTUnjellier�unjelly)�aotr r r �unjellyFromAOT s r{ c C sf t tttttttd�}t| d�r| � � }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 �readz<source>�exec�appz'%s needs to define an 'app', it didn't!)r! r r r r r r" rI rc r| �compile�evalr{ r@ )� stringOrFile�ns�source�coder 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|tj�r|�||� |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 |tj�rd g}|�|d� n|g}| j�||f� d S )Nr+ )r r r� r� r� rW )r
�callable�result�
listResultr r r � callAfterD 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)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 r�t�|j�S |
ttfv s�t|
t�r�t�|j�S |
tu r�|j} t�|j�} | �|j�}
| | j v r�|
du r�t | | �S t|
t j
�r�t � | |
| �S t | | |
�S t d��|
t!u r�t�|j�}| �|j"�}|�#|�}t$|d�r�| �%|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 �<lambda>� s z(AOTUnjellier.unjellyAO.<locals>.<lambda>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� �get�resolveDependantsr@ r" � _DereferencerI rJ r� � addCallback)r
rL rh ro ri �tuple_rY r[ r\ �kvd�c�im_name�im_class�im_selfr$ r. r& r� �refkey�refrB �derrJ r r r r� U s�
�
�
�
����
� zAOTUnjellier.unjellyAOc
C sn z d g}| � |d|� | jD ]
\}}||d � q|d W S ty6 t�d� t�d�tt| 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.)�
AOTJellier�jelly)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
�aoref�objectr r r �
prepareForRef s zAOTJellier.prepareForRefc s t � �}�j�t� �� |tv r� �n�t|tj�r-tt � �j
t � t
� ����t� ����n�t|tj�r9t� j
��n�t|t �rFtt � � ���n�|tju rStt �� ���n�t� ��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 r�i }� �#� D ]\}}��|�|��|�< q��� |� n<|t$j%v r�t$j%| � �\} }
�� t&t �| ���|
��� n t'� d�r�|� �(� � nt'� 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
)rE r! r �qualr� � jellyToAO)r. �r4 �retvalr
r r �
_stateFromH s �z(AOTJellier.jellyToAO.<locals>._stateFromc s g | ]}� � |��qS r )r� )�.0r� r r r �
<listcomp>P s z(AOTJellier.jellyToAO.<locals>.<listcomp>� __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_reg�dispatch_tablerI rc r� r� r# ) r
r4 �objType�oldRef�keyr� rY r[ r\ � unpickleFuncr. r r� r r� sb �
�
�
zAOTJellier.jellyToAOc C s> z| � |�}|W S ty t�d� t�d�| 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- �bool�bytesr, �int�float�complexrd �slice�Ellipsisre r! r r" rI r � Exceptionr7 r rU r6 r( rK r{ r� rx r� r� r� r� r� r� r r r r �<module> sZ
�
" '