CasperSecurity
o
�b� � @ s� U d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z m Z e�� a
G dd� d�Zi Ze eef ed < d
d � Zd d
� ZG dd� de�Zdd� Zdd� Zddd�ZdS )z&
*Real* reloading support for Python.
� N)�reload)�
ModuleType)�Dict)�log�reflectc @ s, e Zd ZdZeZdd� Zdd� Zdd� ZdS ) � Sensitivez�
A utility mixin that's sensitive to rebuilds.
This is a mixin for classes (usually those which represent collections of
callbacks) to make sure that their code is up-to-date before running.
c C s | j t k }|S �N)� lastRebuild)�self�yn� r �8/usr/lib/python3/dist-packages/twisted/python/rebuild.py�needRebuildUpdate$ s
zSensitive.needRebuildUpdatec C s t � � | _d S r )�timer �r
r r r
�rebuildUpToDate( s zSensitive.rebuildUpToDatec C sX t |�}|tjkr
t|�S |tjkr%|jdu r t|j|j�S t|j|j�S t �
d� |S )z�
Get the latest version of an object.
This can handle just about anything callable; instances, functions,
methods, and classes.
Nzwarning returning anObject!) �type�types� FunctionType�latestFunction�
MethodType�__self__�getattr�im_class�__name__r �msg)r
�anObject�tr r r
�latestVersionOf+ s
zSensitive.latestVersionOfN)r �
__module__� __qualname__�__doc__r r r r r r r r
r s r �
_modDictIDMapc C s, t | j�}t�|�}|du r| S t|| j�S )z/
Get the latest version of a function.
N)�id� __globals__r" �getr r )�oldFunc�dictID�moduler r r
r B s
r c C sz t �| j�}t|| j�}dd� |jD �}|jdkr|S zt|�|_|W S ty< t|�}||jt|�t |j
�� Y S w )z,
Get the latest version of a class.
c S s g | ]}t |��qS r )� latestClass)�.0�baser r r
�
<listcomp>U s z latestClass.<locals>.<listcomp>�builtins) r � namedModuler r r � __bases__�tuple� TypeErrorr �dict�__dict__)�oldClassr( �newClass�newBases�ctorr r r
r) O s
�r) c @ s e Zd ZdZdS )� RebuildErrorzT
Exception raised when trying to rebuild a class whereas it's not possible.
N)r r r r! r r r r
r8 d s r8 c C s t | j�| _dS )z,
Updates an instance to be current.
N)r) � __class__r r r r
�updateInstancej s r: c C sF |dkrt d��t| � t�dt�| j�� d|� d�� t| |�}|S )z<
A getattr method to cause a class to be refreshed.
�__del__z Without this, Python segfaults.z(rebuilding stale z instance (z)))�AttributeErrorr: r r r �qualr9 r )r
�name�resultr r r
�__injectedgetattr__q s
r@ � c
C s� t � � at| d�r| jstd��|r t�dt| j�� d�� ddl m
} d|_ | j }| t
t|�< i }i }i }i }|rFt�dt| j�� d �� |�� D ]@\}} tt| �tj�rn| j| j u rmd
|| < |rmtj�d � tj�� qJt| t�r�| j| jkr�d
|| < |r�tj�d � tj�� qJ|�|� |�|� |j}
|� � }|� � }|� � }|r�t�d
� t�dt| j�� d�� t | � t � � |r�t�dt| j�� d �� |D ]3} t | | j�| u r�t�dt!�"| �� d�� q�|r�tj�d� tj�� d| _#| j �$� t%| _&| j| _q�|�r
ddl'} |D ]0}
t | |
j�}||
u �r&t�d�(t!�"|
��� �q | �)|
�D ]}t |dd�|
u �r9||_*�q+�q |�rQt�d
� t�dt| j�� d �� d}t+j,�� D ]{\}}|d
}|| k�sj|du �rl�qXt|d��st�qXt|d��r|�qXd}|j �� D ]6\}} zt-| � W n t.�y� Y �q�w |
| ��r�|�r�tj�d � tj�� t/| �}d
}t0|||� �q�|�r�|�s�|d dk�r�tj�d � tj�� �qXd |_ |�r�t�d
� t�d t| j�� d �� | S ) zO
Reload a module and do as much as possible to replace its references.
�ALLOW_TWISTED_REBUILDz I am not allowed to be rebuilt.z Rebuilding z...r )�
componentsTz (scanning z): rA �f�o� z (reload �)z (cleaning zWARNING: class z not replaced by reload!�xr Nz-WARNING: new-class {} not replaced by reload!r9 z (fixing �__file__�
__bundle__�
�.Fz Rebuilt )1r r �hasattrrB � RuntimeErrorr r �strr �twisted.pythonrC �ALLOW_DUPLICATESr3 r" r# �items�
issubclassr r r r$ �logfile�write�flush�
isinstancer �update� __contains__�keysr � linecache�
clearcacher r r= r/ �clearr@ � __getattr__�gc�format�
get_referrersr9 �sys�modules�hash� Exceptionr �setattr)r( �doLogrC �d�
newclasses�classes� functions�values�k�v�
fromOldModule�clazzr_ �nclass�ga�r�modcount�mk�mod�changed�nvr r r
�rebuild} s�
�
�
����
�
�
�
ry )rA )r! r[ rb r r �impr r �typingr rP r r r r r" �int�__annotations__r r) re r8 r: r@ ry r r r r
�<module> s$ %