File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/orm/__pycache__/identity.cpython-310.pyc
o
���g!$ � @ s d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d lm
Z
d d
lm Z d d lm Z d d lm
Z
d d
lmZ d dlZddlmZ ddlmZ e
rlddlmZ ddlmZ eded�Zeded�ZG dd� d�Z G dd� de �Z d$d"d#�Z dS )%� )� annotations)�Any)�cast)�Dict)�Iterable)�Iterator)�List)�NoReturn)�Optional)�Set)�Tuple)�
TYPE_CHECKING)�TypeVarN� )�util� )�exc)�_IdentityKeyType)�
InstanceState�_T)�bound�_Oc @ s� e Zd ZU ded< ded< ded< dCd d
�ZdCd d �ZdDdd�ZdEdd�ZdFdd�ZdGdd�Z dHd d �Z
dIdJd"d#�Z dKd%d&�Z dLd(d)�Z
dMd+d,�ZdNd.d/�ZdEd0d1�ZdGd2d3�ZdOd4d5�ZdGd6d7�ZdGd8d9�ZdPd:d;�ZdQd<d=�ZdFd>d?�ZdRdAdB�Zd S )S� IdentityMapzweakref.ref[IdentityMap]�_wrz Dict[_IdentityKeyType[Any], Any]�_dict�Set[InstanceState[Any]]� _modified�return�Nonec C s i | _ t� | _t�| �| _d S �N)r �setr �weakref�refr ��self� r% �B/usr/local/lib/python3.10/dist-packages/sqlalchemy/orm/identity.py�__init__+ s zIdentityMap.__init__c C s
t | _d S r )�_killed�_add_unpresentr# r% r% r& �_kill0 �
zIdentityMap._kill�List[InstanceState[Any]]c C � t � �r ��NotImplementedErrorr# r% r% r& �
all_states3 � zIdentityMap.all_states�state�InstanceState[Any]�boolc C r- r r. �r$ r2 r% r% r& �contains_state6 r1 zIdentityMap.contains_state�key�_IdentityKeyType[Any]c C r- r r. �r$ r7 r% r% r& � __contains__9 r1 zIdentityMap.__contains__c C r- r r. r5 r% r% r& � safe_discard< r1 zIdentityMap.safe_discard�_IdentityKeyType[_O]r c C r- r r. r9 r% r% r& � __getitem__? r1 zIdentityMap.__getitem__N�default� Optional[_O]c C r- r r. )r$ r7 r> r% r% r& �getB � zIdentityMap.get�Optional[InstanceState[_O]]c C r- r r. r9 r% r% r& �fast_get_stateG rA zIdentityMap.fast_get_state� Iterable[_IdentityKeyType[Any]]c C s
| j �� S r )r �keysr# r% r% r& rE L r+ zIdentityMap.keys�Iterable[object]c C r- r r. r# r% r% r& �valuesO r1 zIdentityMap.values�InstanceState[_O]c C r- r r. r5 r% r% r& �replaceR r1 zIdentityMap.replacec C r- r r. r5 r% r% r& �addU r1 zIdentityMap.addc C r- r r. r5 r% r% r& �
_fast_discardX r1 zIdentityMap._fast_discardc C s | � |� dS )zUoptional inlined form of add() which can assume item isn't present
in the mapN)rJ �r$ r2 r7 r% r% r& r) [ s zIdentityMap._add_unpresentc C s" | j |_|jr| j�|� d S d S r )r �_instance_dict�modifiedr rJ r5 r% r% r& �_manage_incoming_stateb s �z"IdentityMap._manage_incoming_statec C s |` |jr
| j�|� d S d S r )rM rN r �discardr5 r% r% r& �_manage_removed_stateh s �z!IdentityMap._manage_removed_statec C s | j S r )r r# r% r% r& �
_dirty_statesm r1 zIdentityMap._dirty_statesc C �
t | j�S )z[return True if any InstanceStates present have been marked
as 'modified'.
)r4 r r# r% r% r& �check_modifiedp s
zIdentityMap.check_modifiedc C s || v S r r% r9 r% r% r& �has_keyw s zIdentityMap.has_key�intc C rS r )�lenr r# r% r% r& �__len__z r+ zIdentityMap.__len__)r r �r r, �r2 r3 r r4 �r7 r8 r r4 �r2 r3 r r �r7 r< r r r �r7 r< r> r? r r? �r7 r< r rB )r rD )r rF )r2 rH r rB �r2 r3 r7 r8 r r )r r )r r4 )r rV )�__name__�
__module__� __qualname__�__annotations__r' r* r0 r6 r: r; r= r@ rC rE rG rI rJ rK r) rO rQ rR rT rU rX r% r% r% r& r % s4
�
r c @ s� e Zd ZU ded< d3dd�Zd4d d �Zd5dd�Zd6dd�Zd5dd�Zd7dd�Z d8dd�Z
d9d:d d �Z d;d"d#�Z d<d%d&�Z
d=d(d)�Zd>d+d,�Zd?d-d.�Zd?d/d0�Zd?d1d2�Zd S )@�WeakInstanceDictz/Dict[_IdentityKeyType[Any], InstanceState[Any]]r r7 r< r r c C s, t d| j| �}|�� }|d u rt|��|S �NrH )r r �obj�KeyError�r$ r7 r2 �or% r% r& r= � s
z WeakInstanceDict.__getitem__r8 r4 c C sD z|| j v r| j | }|�� }nW dS W |d uS ty! Y dS w �NF)r rg rh ri r% r% r& r: � s
� ��z WeakInstanceDict.__contains__r2 r3 c C sJ |j | jv r#tr|j d usJ �z | j|j |u W S ty" Y dS w dS rk )r7 r r
rh r5 r% r% r& r6 � s �z WeakInstanceDict.contains_state� Optional[InstanceState[Any]]c C s| |j d usJ �|j | jv r/z
| j|j }}W n ty" d }Y nw ||ur-| �|� nd S d }|| j|j < | �|� |S r )r7 r rh rQ rO )r$ r2 �existing�existing_non_noner% r% r& rI � s �
zWeakInstanceDict.replacec C s� |j }|d us J �|| jv r;z| j| }W n ty Y n w ||ur9|�� }|d ur8t�dt�|�|j f ��ndS || j|< | �|� dS )NzZCan't attach instance %s; another instance with key %s is already present in this session.FT) r7 r rh rg �sa_exc�InvalidRequestError�orm_util� state_strrO )r$ r2 r7 �existing_staterj r% r% r& rJ � s*
����
zWeakInstanceDict.addr c C s || j |< | j|_d S r )r r rM rL r% r% r& r) � s
z WeakInstanceDict._add_unpresentrB c C s | j �|�S r )r r@ r9 r% r% r& rC � s z WeakInstanceDict.fast_get_stateNr> r? c C sR || j vr|S z
td| j | �}W n ty | Y S w |�� }|d u r'|S |S rf )r r rh rg )r$ r7 r> r2 rj r% r% r& r@ � s
�zWeakInstanceDict.get�6List[Tuple[_IdentityKeyType[Any], InstanceState[Any]]]c C sJ | � � }g }|D ]}|�� }|j}|d usJ �|d ur"|�||f� q|S r )r0 rg r7 �append)r$ rG �resultr2 �valuer7 r% r% r& �items� s �zWeakInstanceDict.items� List[object]c C s4 | � � }g }|D ]}|�� }|d ur|�|� q|S r )r0 rg ru )r$ rG rv r2 rw r% r% r& rG � s
�zWeakInstanceDict.values� Iterator[_IdentityKeyType[Any]]c C s t | �� �S r )�iterrE r# r% r% r& �__iter__ s zWeakInstanceDict.__iter__r, c C s t | j�� �S r )�listr rG r# r% r% r& r0 � zWeakInstanceDict.all_statesc C sT |j }|d us J �z| j| }W n
ty Y d S w ||u r(| j�|d � d S d S r )r7 r rh �pop�r$ r2 r7 �str% r% r& rK s ��z WeakInstanceDict._fast_discardc C s | � |� d S r )r; r5 r% r% r& rP r~ zWeakInstanceDict.discardc C sl |j }|| jv r2|d usJ �z| j| }W n
ty Y d S w ||u r4| j�|d � | �|� d S d S d S r )r7 r rh r rQ r� r% r% r& r; s
��z WeakInstanceDict.safe_discardr] r[ rZ )r2 r3 r rl r` r_ r r^ )r rt )r ry )r rz rY r\ )ra rb rc rd r= r: r6 rI rJ r) rC r@ rx rG r| r0 rK rP r; r% r% r% r& re ~ s$
�
re r2 r3 r7 r8 r r c C s t jdt�| � dd��)Nz�Object %s cannot be converted to 'persistent' state, as this identity map is no longer valid. Has the owning Session been closed?�lkrp)�code)ro rp rq rr )r2 r7 r% r% r& r( & s ��r( )r2 r3 r7 r8 r r ) �
__future__r �typingr r r r r r r r
r r r
r r! � r rq r ro �_typingr r2 r r �objectr r re r( r% r% r% r& �<module> s2 Y )