Edit File: identity.cpython-37.pyc
B ��4];( � @ sl d dl Z ddlmZ ddlmZ ddlmZ ddlmZ G dd� de�ZG d d � d e�Z G d d � d e�Z dS ) � N� )� attributes)�util� )�excc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd(d d �Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd S ))� IdentityMapc C s i | _ t� | _t�| �| _d S )N)�_dict�set� _modified�weakref�ref�_wr)�self� r �J/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/orm/identity.py�__init__ s zIdentityMap.__init__c C s | j �� S )N)r �keys)r r r r r s zIdentityMap.keysc C s t � �d S )N)�NotImplementedError)r �stater r r �replace s zIdentityMap.replacec C s t � �d S )N)r )r r r r r �add s zIdentityMap.addc C s | � |� dS )zUoptional inlined form of add() which can assume item isn't present in the mapN)r )r r �keyr r r �_add_unpresent s zIdentityMap._add_unpresentc C s t d��d S )Nz%IdentityMap uses add() to insert data)r )r �dict_r r r �update$ s zIdentityMap.updatec C s t d��d S )Nz(IdentityMap uses remove() to remove data)r )r r r r �clear' s zIdentityMap.clearc C s | j |_|jr| j�|� d S )N)r �_instance_dict�modifiedr r )r r r r r �_manage_incoming_state* s z"IdentityMap._manage_incoming_statec C s |` |jr| j�|� d S )N)r r r �discard)r r r r r �_manage_removed_state0 s z!IdentityMap._manage_removed_statec C s | j S )N)r )r r r r � _dirty_states5 s zIdentityMap._dirty_statesc C s t | j�S )z[return True if any InstanceStates present have been marked as 'modified'. )�boolr )r r r r �check_modified8 s zIdentityMap.check_modifiedc C s || kS )Nr )r r r r r �has_key? s zIdentityMap.has_keyc C s t d��d S )Nz(IdentityMap uses remove() to remove data)r )r r r r �popitemB s zIdentityMap.popitemc G s t d��d S )Nz(IdentityMap uses remove() to remove data)r )r r �argsr r r �popE s zIdentityMap.popNc C s t d��d S )Nz%IdentityMap uses add() to insert data)r )r r �defaultr r r � setdefaultH s zIdentityMap.setdefaultc C s t | j�S )N)�lenr )r r r r �__len__K s zIdentityMap.__len__c C s t � �d S )N)r )r r r r �copyN s zIdentityMap.copyc C s t d��d S )Nz%IdentityMap uses add() to insert data)r )r r �valuer r r � __setitem__Q s zIdentityMap.__setitem__c C s t d��d S )Nz(IdentityMap uses remove() to remove data)r )r r r r r � __delitem__T s zIdentityMap.__delitem__)N)�__name__� __module__� __qualname__r r r r r r r r r r! r# r$ r% r' r) r+ r, r. r/ r r r r r s& r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd$dd�Z dd� Z dd� Z dd� Z e jrpdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd S )%�WeakInstanceDictc C s& | j | }|�� }|d kr"t|��|S )N)r �obj�KeyError)r r r �or r r � __getitem__Y s z WeakInstanceDict.__getitem__c C sH y&|| j kr | j | }|�� }ndS W n tk r: dS X |d k S d S )NF)r r4 r5 )r r r r6 r r r � __contains__` s z WeakInstanceDict.__contains__c C s< |j | jkr4y| j|j |kS tk r0 dS X ndS d S )NF)r r r5 )r r r r r �contains_statel s z WeakInstanceDict.contains_statec C sd |j | jkrJy| j|j }W n tk r0 Y nX ||k rF| �|� nd S || j|j < | �|� d S )N)r r r5 r r )r r �existingr r r r u s zWeakInstanceDict.replacec C s� |j }|| jkrly| j| }W n tk r2 Y n:X ||k rh|�� }|d k rlt�dt�|�|j f ��ndS || j|< | �|� dS )NzZCan't attach instance %s; another instance with key %s is already present in this session.FT) r r r5 r4 �sa_exc�InvalidRequestError�orm_util� state_strr )r r r Zexisting_stater6 r r r r � s zWeakInstanceDict.addc C s || j |< | j|_d S )N)r r r )r r r r r r r � s z WeakInstanceDict._add_unpresentNc C sN || j kr|S y| j | }W n tk r0 |S X |�� }|d krF|S |S d S )N)r r5 r4 )r r r( r r6 r r r �get� s zWeakInstanceDict.getc C s> | � � }g }x,|D ]$}|�� }|d k r|�|j|f� qW |S )N)� all_statesr4 �appendr )r �values�resultr r- r r r �items� s zWeakInstanceDict.itemsc C s8 | � � }g }x&|D ] }|�� }|d k r|�|� qW |S )N)r@ r4 rA )r rB rC r r- r r r rB � s zWeakInstanceDict.valuesc C s t | �� �S )N)�iterr )r r r r �__iter__� s zWeakInstanceDict.__iter__c C s t | �� �S )N)rE rD )r r r r � iteritems� s zWeakInstanceDict.iteritemsc C s t | �� �S )N)rE rB )r r r r � itervalues� s zWeakInstanceDict.itervaluesc C s" t jr| j�� S t| j�� �S d S )N)r �py2kr rB �list)r r r r r@ � s zWeakInstanceDict.all_statesc C sB y| j |j }W n tk r$ Y nX ||kr>| j �|jd � d S )N)r r r5 r' )r r �str r r � _fast_discard� s z WeakInstanceDict._fast_discardc C s | � |� d S )N)� safe_discard)r r r r r r � s zWeakInstanceDict.discardc C sX |j | jkrTy| j|j }W n tk r0 Y n$X ||krT| j�|j d � | �|� d S )N)r r r5 r' r )r r rK r r r rM � s z WeakInstanceDict.safe_discardc C s dS )Nr r )r r r r �prune� s zWeakInstanceDict.prune)N)r0 r1 r2 r7 r8 r9 r r r r? rD rB rF r rI rG rH r@ rL r rM rN r r r r r3 X s$ r3 c @ s� e Zd ZdZejr"dd� Zdd� Zdd� Zdd � Z d d � Z d%d d�Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd S )&�StrongInstanceDicta& A 'strong-referencing' version of the identity map. .. deprecated 1.1:: The strong reference identity map is legacy. See the recipe at :ref:`session_referencing_behavior` for an event-based approach to maintaining strong identity references. c C s | j �� S )N)r rH )r r r r rH s z StrongInstanceDict.itervaluesc C s | j �� S )N)r rG )r r r r rG s z StrongInstanceDict.iteritemsc C s t | j�S )N)rE r )r r r r rF s zStrongInstanceDict.__iter__c C s | j | S )N)r )r r r r r r7 s z StrongInstanceDict.__getitem__c C s || j kS )N)r )r r r r r r8 s z StrongInstanceDict.__contains__Nc C s | j �||�S )N)r r? )r r r( r r r r? s zStrongInstanceDict.getc C s | j �� S )N)r rB )r r r r rB s zStrongInstanceDict.valuesc C s | j �� S )N)r rD )r r r r rD s zStrongInstanceDict.itemsc C s dd� | � � D �S )Nc S s g | ]}t �|��qS r )r �instance_state)�.0r6 r r r � <listcomp> s z1StrongInstanceDict.all_states.<locals>.<listcomp>)rB )r r r r r@ s z StrongInstanceDict.all_statesc C s |j | ko t�| |j �|kS )N)r r rP )r r r r r r9 s z!StrongInstanceDict.contains_statec C sX |j | jkr:| j|j }t�|�}||k r6| �|� nd S |�� | j|j < | �|� d S )N)r r r rP r r4 r )r r r: r r r r $ s zStrongInstanceDict.replacec C s` |j | kr>t�| j|j �|k r:t�dt�|�|j f ��dS |�� | j|j < | � |� dS d S )NzZCan't attach instance %s; another instance with key %s is already present in this session.FT) r r rP r r; r< r= r> r4 r )r r r r r r 0 s zStrongInstanceDict.addc C s |� � | j|< | j|_d S )N)r4 r r r )r r r r r r r ? s z!StrongInstanceDict._add_unpresentc C sH y| j |j }W n tk r$ Y n X t�|�|krD| j �|jd � d S )N)r r r5 r rP r' )r r r4 r r r rL D s z StrongInstanceDict._fast_discardc C s | � |� d S )N)rM )r r r r r r P s zStrongInstanceDict.discardc C sH |j | jkrD| j|j }t�|�}||krD| j�|j d � | �|� d S )N)r r r rP r' r )r r r4 rK r r r rM S s z StrongInstanceDict.safe_discardc C sX t | �}dd� | �� D �}t�� }|�| � | j�� | j�|� t|�| _|t | � S )z%prune unreferenced, non-dirty states.c S s g | ]}|j r|�� �qS r )r r4 )rQ �sr r r rR _ s z,StrongInstanceDict.prune.<locals>.<listcomp>) r* r@ r �WeakValueDictionaryr r r r" r )r Z ref_countZdirtyZkeepersr r r rN [ s zStrongInstanceDict.prune)N)r0 r1 r2 �__doc__r rI rH rG rF r7 r8 r? rB rD r@ r9 r r r rL r rM rN r r r r rO � s& rO ) r � r r r= r r; �objectr r3 rO r r r r �<module> s H