3
\�O �
@ s� d dl mZmZmZmZmZmZmZmZ d dl m
Z
m Z d dl Z d dl
Z
d dlZeefZdddddd d
d d d
dddg
ZG dd� de�ZG dd
� d
e j�ZG dd� de�ZG dd� de j�ZG dd� d�ZdS )� )�getweakrefcount� getweakrefs�ref�proxy�CallableProxyType� ProxyType�
ReferenceType�_remove_dead_weakref)�WeakSet�_IterationGuardNr r r r �WeakKeyDictionaryr r r �
ProxyTypes�WeakValueDictionaryr
�
WeakMethod�finalizec s@ e Zd ZdZddd�Z� fdd �Zd
d � Zd d
� Zej Z � Z
S )r � _func_ref�
_meth_type�_alive� __weakref__Nc s~ y|j }|j}W n( tk
r8 tdjt|���d �Y nX � �fdd�}tj| ||�}t||�|_t|�|_ d|_
t|��|S )Nz)argument should be a bound method, not {}c s& �� }|j r"d|_ � d k r"� |� d S )NF)r )�arg�self)�callback�self_wr� � /usr/lib64/python3.6/weakref.py�_cb3 s
z WeakMethod.__new__.<locals>._cbT) �__self__�__func__�AttributeError� TypeError�format�typer �__new__r r r )�cls�methr �obj�funcr r r )r r r r" , s
zWeakMethod.__new__c s2 t � j� }| j� }|d ks"|d kr&d S | j||�S )N)�super�__call__r r )r r% r& )� __class__r r r( B s
zWeakMethod.__call__c C s> t |t�r:| j s|j r"| |kS tj| |�o8| j|jkS dS )NF)�
isinstancer r r �__eq__r )r �otherr r r r+ I s
zWeakMethod.__eq__c C s> t |t�r:| j s|j r"| |k S tj| |�p8| j|jkS dS )NT)r* r r r �__ne__r )r r, r r r r- P s
zWeakMethod.__ne__)r r r r )N) �__name__�
__module__� __qualname__� __slots__r" r( r+ r- r �__hash__�
__classcell__r r )r) r r $ s
c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z e Z dd� Z
d*dd�Zdd� Zdd� ZeZd d � Zd d � Zd d!� Zd"d#� Zd+d$d%�Zd&d'� Zd(d)� ZdS ),r c st | s t d��| ^}} t| �dkr0t dt| � ��t|�tf� fdd� }||_g |_t� |_i |_� |j | |� d S )NzGdescriptor '__init__' of 'WeakValueDictionary' object needs an argument� z$expected at most 1 arguments, got %dc s4 |� }|d k r0|j r$|jj| j� n |� | j� d S )N)�
_iterating�_pending_removals�append�key)�wr�selfref�_atomic_removalr )�dr r �removem s
z,WeakValueDictionary.__init__.<locals>.remove)
r �lenr r �_remover6 �setr5 �data�update)�args�kwr r= r )r< r �__init__f s
z WeakValueDictionary.__init__c C s, | j }| j}x|r&|j� }t||� qW d S )N)r6 rA �popr )r �lr< r8 r r r �_commit_removals} s
z$WeakValueDictionary._commit_removalsc C s4 | j r| j� | j| � }|d kr,t|��n|S d S )N)r6 rH rA �KeyError)r r8 �or r r � __getitem__� s
z WeakValueDictionary.__getitem__c C s | j r| j� | j|= d S )N)r6 rH rA )r r8 r r r � __delitem__� s z WeakValueDictionary.__delitem__c C s | j r| j� t| j�S )N)r6 rH r> rA )r r r r �__len__� s zWeakValueDictionary.__len__c C s< | j r| j� y| j| � }W n tk
r2 dS X |d k S )NF)r6 rH rA rI )r r8 rJ r r r � __contains__� s z WeakValueDictionary.__contains__c C s d| j jt| �f S )Nz <%s at %#x>)r) r. �id)r r r r �__repr__� s z WeakValueDictionary.__repr__c C s& | j r| j� t|| j|�| j|< d S )N)r6 rH �KeyedRefr? rA )r r8 �valuer r r � __setitem__� s z WeakValueDictionary.__setitem__c C sF | j r| j� t� }x,| jj� D ] \}}|� }|d k r |||< q W |S )N)r6 rH r rA �items)r �newr8 r9 rJ r r r �copy� s zWeakValueDictionary.copyc C sZ ddl m} | jr| j� | j� }x2| jj� D ]$\}}|� }|d k r.|||||�< q.W |S )Nr )�deepcopy)rV rW r6 rH r) rA rT )r �memorW rU r8 r9 rJ r r r � __deepcopy__� s z WeakValueDictionary.__deepcopy__Nc C sL | j r| j� y| j| }W n tk
r0 |S X |� }|d krD|S |S d S )N)r6 rH rA rI )r r8 �defaultr9 rJ r r r �get� s zWeakValueDictionary.getc
c sV | j r| j� t| ��6 x.| jj� D ] \}}|� }|d k r$||fV q$W W d Q R X d S )N)r6 rH r rA rT )r �kr9 �vr r r rT � s
zWeakValueDictionary.itemsc
c sN | j r| j� t| ��. x&| jj� D ]\}}|� d k r$|V q$W W d Q R X d S )N)r6 rH r rA rT )r r\ r9 r r r �keys� s
zWeakValueDictionary.keysc c s6 | j r| j� t| �� | jj� E d H W d Q R X d S )N)r6 rH r rA �values)r r r r �
itervaluerefs� s
z!WeakValueDictionary.itervaluerefsc
c sN | j r| j� t| ��. x&| jj� D ]}|� }|d k r$|V q$W W d Q R X d S )N)r6 rH r rA r_ )r r9 r% r r r r_ � s
zWeakValueDictionary.valuesc C s< | j r| j� x(| jj� \}}|� }|d k r||fS qW d S )N)r6 rH rA �popitem)r r8 r9 rJ r r r ra � s zWeakValueDictionary.popitemc G s` | j r| j� y| jj|�� }W n tk
r8 d }Y nX |d krX|rN|d S t|��n|S d S )Nr )r6 rH rA rF rI )r r8 rC rJ r r r rF s
zWeakValueDictionary.popc C s` y| j | � }W n tk
r( d }Y nX |d krX| jr@| j� t|| j|�| j |< |S |S d S )N)rA rI r6 rH rQ r? )r r8 rZ rJ r r r �
setdefault s
z WeakValueDictionary.setdefaultc O s� | s t d��| ^}} t| �dkr0t dt| � ��| r<| d nd }|jrN|j� |j}|d k r�t|d�srti �|�}x&|j� D ]\}}t||j |�||< q|W t|�r�|j
|� d S )NzEdescriptor 'update' of 'WeakValueDictionary' object needs an argumentr4 z$expected at most 1 arguments, got %dr rT ) r r> r6 rH rA �hasattrr! rT rQ r? rB )rC �kwargsr �dictr< r8 rJ r r r rB s
zWeakValueDictionary.updatec C s | j r| j� t| jj� �S )N)r6 rH �listrA r_ )r r r r � valuerefs3 s
z WeakValueDictionary.valuerefs)N)N)r. r/ r0 rE rH rK rL rM rN rP rS rV �__copy__rY r[ rT r^ �__iter__r` r_ ra rF rb rB rg r r r r r Z s,
c s( e Zd ZdZdd� Z� fdd�Z� ZS )rQ r8 c C s t j| ||�}||_|S )N)r r" r8 )r! �obr r8 r r r r r" N s zKeyedRef.__new__c s t � j||� d S )N)r' rE )r rj r r8 )r) r r rE S s zKeyedRef.__init__)r8 )r. r/ r0 r1 r" rE r3 r r )r) r rQ B s
rQ c @ s� e Zd Zd*dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z e Z dd� Z
d+dd�Zdd� Zdd� Zd d � ZeZd d � Zd d!� Zd"d#� Zd$d%� Zd,d&d'�Zd-d(d)�ZdS ).r Nc C sF i | _ t| �fdd�}|| _g | _t� | _d| _|d k rB| j|� d S )Nc S s. |� }|d k r*|j r"|jj| � n|j| = d S )N)r5 r6 r7 rA )r\ r: r r r r r= d s
z*WeakKeyDictionary.__init__.<locals>.removeF)rA r r? r6 r@ r5 �
_dirty_lenrB )r re r= r r r rE b s zWeakKeyDictionary.__init__c C s> | j }| j}x,|r8y||j� = W q tk
r4 Y qX qW d S )N)r6 rA rF rI )r rG r< r r r rH s s z"WeakKeyDictionary._commit_removalsc s& | j � � fdd�| jD �| _d| _d S )Nc s g | ]}|� kr|�qS r r )�.0r\ )r< r r �
<listcomp>� s z5WeakKeyDictionary._scrub_removals.<locals>.<listcomp>F)rA r6 rk )r r )r< r �_scrub_removals� s z!WeakKeyDictionary._scrub_removalsc C s d| _ | jt|�= d S )NT)rk rA r )r r8 r r r rL � s z WeakKeyDictionary.__delitem__c C s | j t|� S )N)rA r )r r8 r r r rK � s z WeakKeyDictionary.__getitem__c C s( | j r| jr| j� t| j�t| j� S )N)rk r6 rn r> rA )r r r r rM � s zWeakKeyDictionary.__len__c C s d| j jt| �f S )Nz <%s at %#x>)r) r. rO )r r r r rP � s zWeakKeyDictionary.__repr__c C s || j t|| j�< d S )N)rA r r? )r r8 rR r r r rS � s z WeakKeyDictionary.__setitem__c C s8 t � }x,| jj� D ] \}}|� }|d k r|||< qW |S )N)r rA rT )r rU r8 rR rJ r r r rV � s zWeakKeyDictionary.copyc C sL ddl m} | j� }x2| jj� D ]$\}}|� }|d k r |||�||< q W |S )Nr )rW )rV rW r) rA rT )r rX rW rU r8 rR rJ r r r rY � s z WeakKeyDictionary.__deepcopy__c C s | j jt|�|�S )N)rA r[ r )r r8 rZ r r r r[ � s zWeakKeyDictionary.getc C s, y t |�}W n tk
r dS X || jkS )NF)r r rA )r r8 r9 r r r rN � s
z WeakKeyDictionary.__contains__c
c sH t | ��6 x.| jj� D ] \}}|� }|d k r||fV qW W d Q R X d S )N)r rA rT )r r9 rR r8 r r r rT � s
zWeakKeyDictionary.itemsc
c s< t | ��* x"| jD ]}|� }|d k r|V qW W d Q R X d S )N)r rA )r r9 r% r r r r^ � s
zWeakKeyDictionary.keysc
c s@ t | ��. x&| jj� D ]\}}|� d k r|V qW W d Q R X d S )N)r rA rT )r r9 rR r r r r_ � s
zWeakKeyDictionary.valuesc C s
t | j�S )N)rf rA )r r r r �keyrefs� s
zWeakKeyDictionary.keyrefsc C s4 d| _ x(| jj� \}}|� }|d k r||fS qW d S )NT)rk rA ra )r r8 rR rJ r r r ra � s zWeakKeyDictionary.popitemc G s d| _ | jjt|�f|�� S )NT)rk rA rF r )r r8 rC r r r rF � s zWeakKeyDictionary.popc C s | j jt|| j�|�S )N)rA rb r r? )r r8 rZ r r r rb � s z WeakKeyDictionary.setdefaultc K s` | j }|d k rJt|d�s$ti �|�}x$|j� D ]\}}||t|| j�< q.W t|�r\| j|� d S )NrT )rA rc r! rT r r? r> rB )r re rd r< r8 rR r r r rB � s
zWeakKeyDictionary.update)N)N)N)N)r. r/ r0 rE rH rn rL rK rM rP rS rV rh rY r[ rN rT r^ ri r_ ro ra rF rb rB r r r r r W s,
c @ s� e Zd Zf Zi ZdZej� ZdZ dZ
G dd� d�Z dd� Z ddd�Z
d d
� Zd d � Zed
d� �Zedd� �Zejdd� �Zdd� Zedd� �Zedd� �ZdS )r Fc @ s e Zd ZdZdS ) zfinalize._Info�weakrefr& rC rd �atexit�indexN)rp r& rC rd rq rr )r. r/ r0 r1 r r r r �_Info s rs c O sp | j s dd l}|j| j� dt_ | j� }t|| �|_||_||_ |pFd |_
d|_t | j �|_
|| j| < dt_d S )Nr T)�_registered_with_atexitrq �register� _exitfuncr rs r rp r& rC rd �next� _index_iterrr � _registry�_dirty)r r% r& rC rd rq �infor r r rE s
zfinalize.__init__Nc C s2 | j j| d �}|r.| j r.|j|j|jp*i �S d S )N)ry rF � _shutdownr& rC rd )r �_r{ r r r r( s zfinalize.__call__c C sH | j j| �}|o|j� }|d k rD| j j| d �rD||j|j|jp@i fS d S )N)ry r[ rp rF r& rC rd )r r{ r% r r r �detach& s zfinalize.detachc C s: | j j| �}|o|j� }|d k r6||j|j|jp2i fS d S )N)ry r[ rp r& rC rd )r r{ r% r r r �peek. s z
finalize.peekc C s
| | j kS )N)ry )r r r r �alive6 s zfinalize.alivec C s | j j| �}t|�o|jS )N)ry r[ �boolrq )r r{ r r r rq ; s zfinalize.atexitc C s | j j| �}|rt|�|_d S )N)ry r[ r� rq )r rR r{ r r r rq A s c C s^ | j j| �}|o|j� }|d kr6dt| �jt| �f S dt| �jt| �t|�jt|�f S d S )Nz<%s object at %#x; dead>z!<%s object at %#x; for %r at %#x>)ry r[ rp r! r. rO )r r{ r% r r r rP G s zfinalize.__repr__c C s2 dd� | j j� D �}|jdd� d� dd� |D �S )Nc S s g | ]\}}|j r||f�qS r )rq )rl �f�ir r r rm S s z-finalize._select_for_exit.<locals>.<listcomp>c S s
| d j S )Nr4 )rr )�itemr r r �<lambda>T s z+finalize._select_for_exit.<locals>.<lambda>)r8 c S s g | ] \}}|�qS r r )rl r� r� r r r rm U s )ry rT �sort)r# �Lr r r �_select_for_exitP s zfinalize._select_for_exitc C s� d}z�| j r�dd l}|j� r(d}|j� d }x\|d ks<tjrJ| j� }dt_|sPP |j� }y
|� W q. tk
r� t j
t j � � Y q.X q.W W d dt_ |r�|j
� X d S )NFr T)ry �gc� isenabled�disabler rz r� rF � Exception�sys�
excepthook�exc_infor| �enable)r# � reenable_gcr� �pendingr� r r r rv W s.
zfinalize._exitfunc)N)r. r/ r0 r1 ry r| � itertools�countrx rz rt rs rE r( r~ r �propertyr� rq �setterrP � classmethodr� rv r r r r r � s"
)�_weakrefr r r r r r r r � _weakrefsetr
r � collectionsr� r� r
�__all__r �MutableMappingr rQ r r r r r r �<module> s (
6 i