File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/orm/__pycache__/path_registry.cpython-310.pyc
o
���g<e � @ s� U d Z ddlmZ ddlmZ ddlmZ ddlZddlZddl m
Z
ddl m Z ddl m Z dd l m
Z
dd
l mZ dd l mZ dd l mZ dd
l mZ ddl mZ ddl mZ ddl mZ ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl m Z er�ddlm Z ddl!m"Z" ddl#m$Z$ dd l%m&Z& dd lm'Z' dd l m(Z( dd l)m*Z* dd l+m,Z, dd!l-m.Z. dd"l-m/Z/ dOd'd(�Z0dPd*d+�Z1n
e�2d(�Z0e�2d+�Z1ee
Z3e4Z5ee5d,d-f Z6ee6d.f Z7ed, Z8eed-e4f Z9e�:e;�Z<dQd0d1�Z=d2Z>d3e?d4< d5Z@G d6d$� d$e �ZAG d7d8� d8eA�ZBG d9d:� d:eB�ZCeC� eA_DG d;d<� d<ejEe e4�ZFG d=d>� d>eA�ZGG d?d@� d@eA�ZHG dAdB� dBeB�ZIG dCdD� dDeI�ZJG dEdF� dFe e
e
f �ZKG dGdH� dHeI�ZLe�rqdPdIdJ�ZMdRdLdM�ZNdS e�2d+�ZMe�2dN�ZNdS )Sz@Path tracking utilities, representing mapper graph traversals.
� )� annotations)�reduce)�chainN)�Any)�cast)�Dict)�Iterator)�List)�Optional)�overload)�Sequence)�Tuple)�
TYPE_CHECKING)�Union� )�base)�insp_is_mapper_property� )�exc)�util)�visitors)� HasCacheKey)�_InternalEntityType)�StrategizedProperty)�Mapper)�RelationshipProperty)� AliasedInsp)�_CacheKeyTraversalType)�
BindParameter)�anon_map)� _LiteralStar)� TypeGuard�path� PathRegistry�return�TypeGuard[RootRegistry]c C � d S �N� �r" r( r( �G/usr/local/lib/python3.10/dist-packages/sqlalchemy/orm/path_registry.py�is_root0 � r+ �!TypeGuard[AbstractEntityRegistry]c C r& r' r( r) r( r( r* � is_entity2 r, r. �_InternalEntityType[Any]�StrategizedProperty[Any].�_SerializedPathc C s
t �| �S r' )r# � deserializer) r( r( r* �_unreduce_pathP �
r3 �*r �_WILDCARD_TOKEN� _sa_defaultc @ s� e Zd ZU dZdZdZdZdZdZdZ de
d< de
d< de
d< d e
d
< d e
d < de j j
fgZd
e
d< dgdd�Zdgdd�Zedhdd��Zdid d �Zdjd#d$�Zdjd%d&�Z 'dkdld)d*�Zdmd+d,�Zdmd-d.�Zednd2d3��Zedod5d3��Zedpd7d3��Zedqd9d3��Zedrd<d3��Zdsd?d3�Zedmd@dA��ZdtdCdD�Z dudGdH�Z dvdIdJ�Z dwdKdL�Z e dxdNdO��Z!e dydPdQ��Z"dzdRdS�Z#e d{dTdU��Z$ee d|dWdX���Z%ee d}d[dX���Z%e d~d\dX��Z%e dd^d_��Z&d�d`da�Z'd�dcdd�Z(d�dedf�Z)d'S )�r# a� Represent query load paths and registry functions.
Basically represents structures like:
(<User mapper>, "orders", <Order mapper>, "items", <Item mapper>)
These structures are generated by things like
query options (joinedload(), subqueryload(), etc.) and are
used to compose keys stored in the query._attributes dictionary
for various options.
They are then re-composed at query compile/result row time as
the query is formed and as rows are fetched, where they again
serve to compose keys to look up options in the context.attributes
dictionary, which is copied from query._attributes.
The path structure has a limited amount of caching, where each
"root" ultimately pulls from a fixed registry associated with
the first mapper, that also contains elements for each of its
property keys. However paths longer than two elements, which
are the exception rather than the rule, are generated on an
as-needed basis.
r( F�bool� is_unnatural�_PathRepresentationr" � natural_pathzOptional[PathRegistry]�parent� RootRegistry�rootr �_cache_key_traversal�otherr r$ c C s> z |d uo
| j |jkW S ty t�dt|� � Y dS w )N�1Comparison of PathRegistry to %r is not supportedF�r" �_path_for_compare�AttributeErrorr �warn�type��selfr@ r( r( r* �__eq__� � ���zPathRegistry.__eq__c C s> z |d u p
| j |jkW S ty t�dt|� � Y dS w )NrA TrB rG r( r( r* �__ne__� rJ zPathRegistry.__ne__� Optional[_PathRepresentation]c C s | j S r' r) �rH r( r( r* rC � s z PathRegistry._path_for_compare�index�intr/ c C s
| j | S r' r) )rH rN r( r( r* � odd_element� r4 zPathRegistry.odd_element�
attributes�Dict[Any, Any]�key�value�Nonec C s" t �d|| |� |||| jf< d S )Nz set '%s' on path '%s' to '%s')�log�debugr; �rH rQ rS rT r( r( r* �set� s zPathRegistry.setc C s& t �d|| |� |�|| jf|� d S )Nz$setdefault '%s' on path '%s' to '%s')rV rW �
setdefaultr; rX r( r( r* rZ � s zPathRegistry.setdefaultN�
Optional[Any]c C s || j f}||v r
|| S |S r' )r; rX r( r( r* �get� s
zPathRegistry.getc C �
t | j�S r' ��lenr" rM r( r( r* �__len__� r4 zPathRegistry.__len__c C s t | �S r' )�idrM r( r( r* �__hash__� � zPathRegistry.__hash__�entity�
_StrPathToken�
TokenRegistryc C r& r' r( �rH rd r( r( r* � __getitem__� � zPathRegistry.__getitem__�_PathElementTypec C r& r' r( rg r( r( r* rh � ri �slicec C r& r' r( rg r( r( r* rh � ri �AbstractEntityRegistryc C r& r' r( rg r( r( r* rh � � r0 � PropRegistryc C r& r' r( rg r( r( r* rh � rm �TUnion[_StrPathToken, int, slice, _InternalEntityType[Any], StrategizedProperty[Any]]�aUnion[TokenRegistry, _PathElementType, _PathRepresentation, PropRegistry, AbstractEntityRegistry]c C s t � �r' )�NotImplementedErrorrg r( r( r* rh � s c C r] r' r^ rM r( r( r* �length� �
zPathRegistry.length�OIterator[Tuple[_InternalEntityType[Any], Union[str, StrategizedProperty[Any]]]]c c sH � t t| j�}t t|�}tdt|�d�D ]
}|| ||d fV qd S )Nr r r )r �_OddPathRepresentationr" �_EvenPathRepresentation�ranger_ )rH �odd_path� even_path�ir( r( r* �pairs� s �
�zPathRegistry.pairs�mapper� Mapper[Any]c sF t t| j�� � fdd�tdt� �d�D �D ] }|j�|�r dS qdS )Nc � g | ]}� | �qS r( r( ��.0rz ��_m_pathr( r* �
<listcomp>� � z0PathRegistry.contains_mapper.<locals>.<listcomp>r r TF)r ru r" rw r_ r| �isa)rH r| � path_mapperr( r� r* �contains_mapper� s " �z PathRegistry.contains_mapperc C s || j f|v S r' r) )rH rQ rS r( r( r* �contains� � zPathRegistry.containsc C s t | �� ffS r' )r3 � serializerM r( r( r* �
__reduce__� r� zPathRegistry.__reduce__r1 c
sv t t|�� t t|��tttdd� � fdd�tdt� �d�D �D ��tdd� �fdd�td t��d�D �D ��d
��S ) Nc s s* � | ]}|j s
|jr
|jnt|�V qd S r' )� is_mapper�is_aliased_class�class_�str)r� �mr( r( r* � <genexpr> s
� �
�z/PathRegistry._serialize_path.<locals>.<genexpr>c r~ r( r( r r� r( r* r� r� z0PathRegistry._serialize_path.<locals>.<listcomp>r r c s s& � | ]}t |�r |jnt|�V qd S r' )r rS r� )r� �pr( r( r* r� s
� �
�c r~ r( r( r )�_p_pathr( r* r� r� r r' )r ru rv �list�zip�tuplerw r_ )�clsr" r( )r� r� r* �_serialize_path� s
� ����z PathRegistry._serialize_pathc sP d
dd��ddd�� t t� �fd d
�|D �� �}|r&|d d u r&|d d � }|S )N�mclsr r$ c S s" | t jvr tj| dd�S t j| S �NT)� configure)� PathToken�_intern�orm_base�_inspect_mapped_class)r� r( r( r* �_deserialize_mapper_token s
���zAPathRegistry._deserialize_path.<locals>._deserialize_mapper_tokenrS c S sD |d u rd S |t jv rt j| S tj| dd�}|d us J �|j| S r� )r� r� r� r� �attrs)r� rS �mpr( r( r* �_deserialize_key_token s
z>PathRegistry._deserialize_path.<locals>._deserialize_key_tokenc s" g | ]
\}}�|�� ||�f�qS r( r( )r� r� rS �r� r� r( r* r� % s ���z2PathRegistry._deserialize_path.<locals>.<listcomp>���r )r� r r$ r )r� r rS r r$ r )r� r �r� r" r� r( r� r* �_deserialize_path s
��� z PathRegistry._deserialize_pathc C s | j }| �|�S r' )r" r� )rH r" r( r( r* r� 2 s
zPathRegistry.serializec C s |d usJ �| � |�}| �|�S r' )r� �coercer� r( r( r* r2 6 s
zPathRegistry.deserialize�CachingEntityRegistryc C r& r' r( �r� r| r( r( r* �
per_mapper< � zPathRegistry.per_mapper�AliasedInsp[Any]�SlotsEntityRegistryc C r& r' r( r� r( r( r* r� @ r� c C s |j r t| j|�S t| j|�S r' )r� r� r> r� r� r( r( r* r� D s �rawc C s ddd�}t ||| j�S ) N�prevr# �next_rj r$ c S � | | S r' r( �r� r� r( r( r* �_redO rc z!PathRegistry.coerce.<locals>._red�r� r# r� rj r$ r# )r r> )r� r� r� r( r( r* r� M s
zPathRegistry.coercec C s ddd�}t ||j| �S ) Nr� r# r� rj r$ c S r� r' r( r� r( r( r* r� V rc z"PathRegistry.__add__.<locals>._redr� )r r" )rH r@ r� r( r( r* �__add__U s
zPathRegistry.__add__r� c C s dd� dd� | jD ��� d�S )Nz ORM Path[z -> c s s � | ]}t |�V qd S r' �r� )r� �elemr( r( r* r� \ s � z'PathRegistry.__str__.<locals>.<genexpr>�])�joinr" rM r( r( r* �__str__[ s zPathRegistry.__str__c C s | j j� d| j�d�S )N�(�))� __class__�__name__r" rM r( r( r* �__repr__^ s zPathRegistry.__repr__)r@ r r$ r8 �r$ rL )rN rO r$ r/ )rQ rR rS r rT r r$ rU r' )rQ rR rS r rT r[ r$ r )r$ rO )rd re r$ rf )rd rO r$ rj )rd rk r$ r: )rd r/ r$ rl )rd r0 r$ rn )rd ro r$ rp )r$ rt )r| r} r$ r8 )rQ rR rS r r$ r8 )r$ r )r" r: r$ r1 )r" r1 r$ r: )r$ r1 �r" r1 r$ r# )r| r} r$ r� )r| r� r$ r� )r| r/ r$ rl )r� r: r$ r# )r@ r# r$ r# )r$ r� )*r� �
__module__� __qualname__�__doc__� __slots__�is_tokenr+ �
has_entity� is_propertyr. �__annotations__r �ExtendedInternalTraversal�dp_has_cache_key_listr? rI rK �propertyrC rP rY rZ r\ r` rb r rh rr r{ r� r� r� � classmethodr� r� r� r2 r� r� r� r� r� r( r( r( r* r# X s~
�
�
#
c @ s, e Zd ZU dZded< ded< d d d
�Zd S )
� CreatesTokenr( r8 r� r+ �tokenre r$ rf c C sF |� dt� ��r
t| |�S |� dt� ��rt| j|�S t�d|� ���)N�:zinvalid token: )�endswithr6 rf �_DEFAULT_TOKENr> r �
ArgumentError)rH r� r( r( r* r� h s
zCreatesToken.tokenN)r� re r$ rf )r� r� r� r� r� r� r( r( r( r* r� b s
r� c @ sP e Zd ZdZdZdZd ZZdZdZ dZ
dZ dd d
�Z dd d �Z
es&e Zd
S d
S )r= zXRoot registry, defers to mappers so that
paths are maintained per-root-mapper.
r( TFrd r r$ �,Union[TokenRegistry, AbstractEntityRegistry]c C sR |t jv rtrt|t�sJ �t| t j| �S z|jW S ty( td|� ���w )Nz/invalid argument for RootRegistry.__getitem__: ) r� r� r �
isinstancere rf �_path_registryrD �
IndexErrorrg r( r( r* �_getitem� s
��zRootRegistry._getitemc C � | S r' r( rM r( r( r* �_truncate_recursive� ri z RootRegistry._truncate_recursiveN)rd r r$ r� )r$ r= )r� r� r� r� r� �
inherit_cacher" r; r� r� r+ r9 r� r� r rh r( r( r( r* r= q s
�r= c @ sD e Zd ZU dZi Zded< dd d
�Zedd d
��Ze ddd��Z
dS )r� zcacheable string tokenzDict[str, PathToken]r� r �
bindparams�List[BindParameter[Any]]r$ �Tuple[Any, ...]c C s
t | �fS r' r� )rH r r� r( r( r* �_gen_cache_key� s
zPathToken._gen_cache_keyrL c C r& r' r( rM r( r( r* rC � r� zPathToken._path_for_compare�strvaluer� c C s* || j v r
| j | S t|� | j |< }|S r' )r� r� )r� r� �resultr( r( r* �intern� s
zPathToken.internN)r r r� r� r$ r� r� )r� r� r$ r� ) r� r� r� r� r� r� r� r� rC r� r� r( r( r( r* r� � s
r� c @ sb e Zd ZU dZdZded< ded< ddd�Zd ZdZdd d
�Z ddd�Z
ddd�Z e s/e Z
dS dS )rf )r� r< r" r; Tre r� r� r< c C s6 t �|�}|| _|| _|j|f | _|j|f | _d S r' )r� r� r� r< r" r; )rH r< r� r( r( r* �__init__� s
zTokenRegistry.__init__Fr$ �Iterator[PathRegistry]c c s� � | j }t|�r
| V d S trt|t�sJ �|js-|j�� D ] }t|j | | j �V q d S |jrPt
d|j �j rP| V t
d|j �j
D ] }t|j | | j �V qAd S | V d S �Nr� )r< r+ r r� rl r� r| �iterate_to_rootrf r� r rd �_is_with_polymorphic�_with_polymorphic_entities�rH r< �mp_ent�entr( r( r* �generate_for_superclasses� s8 ��������
z'TokenRegistry.generate_for_superclasses� Iterator[_PathRepresentation]c c s� � | j }t|�r| jV d S trt|t�sJ �|j�� D ]
}t|j | | j �jV q |j
rOt d|j �j
rO| jV t d|j �jD ]
}t|j | | j �jV q?d S | jV d S r� )r< r+ r; r r� rl r| r� rf r� r� r rd r� r� r� r( r( r* �"_generate_natural_for_superclasses� s6 �������� z0TokenRegistry._generate_natural_for_superclassesrd r c
C s2 z| j | W S ty } zt|� �|�d }~ww r' )r" � TypeErrorr� )rH rd �errr( r( r* r� � s ��zTokenRegistry._getitemN)r< r� r� re )r$ r� )r$ r� �rd r r$ r )r� r� r� r� r� r� r� r� r� r� r� r� r rh r( r( r( r* rf � s
�rf c @ sj e Zd ZU dZdZdZded< ded< ded< dd d �Zddd�Ze ddd��Z
ddd�Z e s3e Z
dS dS )rn ) �propr< r" r; r� rd r| �_wildcard_path_loader_key�_default_path_loader_key� _loader_keyr9 Tr0 r� zOptional[Mapper[Any]]r| z"Optional[_InternalEntityType[Any]]rd r< rl c C s^ t d|d �}|}|jjpt|jj�| _|jr|jr#|j|j }}nB|jrL|jrL|j|jv rL|d � |j�}|j| }|jrI|j|j }d| _n |}n|jre|jre|j|jure|j�
|j�re|j|j }|| _ || _|j |f | _ |j
|f | _
|j| _|jr�tr�t|t�s�J �|j| _|j| _nd | _d | _d|j
| j j f| _| j j| _d| j
f| _d S )Nr/ r� T�loader)r r< r9 r8 r| �inheritsr� �_use_mapper_path�with_polymorphic_mappers�_entity_for_mapperr� r� r" r; �_links_to_entityr� �_is_relationshipr r� r rd �_wildcard_tokenr� r� r� )rH r< r� �insp�natural_parent�subclass_entityr( r( r* r� sZ
� ��
��� �
�
zPropRegistry.__init__r$ c C sX d }t t| jd d� ��D ] \}}|| ju r|}q
|d u r | S | �| jd|d � �S )Nr� r r )� enumerate�reversedr" r� r� )rH �earliestrz r� r( r( r* r� v s
�z PropRegistry._truncate_recursivec C s | j d usJ �| | j S r' )rd rM r( r( r* � entity_path� s
zPropRegistry.entity_path�+Union[int, slice, _InternalEntityType[Any]]�DUnion[AbstractEntityRegistry, _PathElementType, _PathRepresentation]c C s" t |ttf�r | j| S t| |�S r' )r� rO rk r" r� rg r( r( r* r� � s
zPropRegistry._getitemN)r< rl r� r0 )r$ rn �r$ rl )rd r r$ r )r� r� r� r� r� r� r� r� r� r� r r� r rh r( r( r( r* rn s
Z
�rn c @ s� e Zd ZU dZdZdZded< ded< ded< ded < d d
d �Zd d
d�Ze d dd��Z
e d!dd��Z e d"dd��Z d#dd�Z
d$dd �ZesJeZd S d S )%rl )rS r< r� r" rd r; T�!Union[RootRegistry, PropRegistry]r< r/ rS rd r8 r� c C s� || _ || _|j| _|| _|j|f | _|jr>| js |jr>|j�|jd j�r1|j|jf | _d S |j|jd jf | _d S | j| _d S )Nr� ) rS r< r� rd r" r9 r| r� r; �rH r< rd r( r( r* r� � s
� z AbstractEntityRegistry.__init__r$ c C s | j �� | j S r' )r< r� rd rM r( r( r* r� � s z*AbstractEntityRegistry._truncate_recursivec C s
| � d�S )Nr )rP rM r( r( r* � root_entity� rs z"AbstractEntityRegistry.root_entityr# c C r� r' r( rM r( r( r* r � r� z"AbstractEntityRegistry.entity_pathr} c C s | j jS r' )rd r| rM r( r( r* r| � s z AbstractEntityRegistry.mapperc C s dS )NTr( rM r( r( r* �__bool__� ri z AbstractEntityRegistry.__bool__r �:Union[_PathElementType, _PathRepresentation, PathRegistry]c C s< t |ttf�r | j| S |tjv rt| tj| �S t| |�S r' )r� rO rk r" r� r� rf rn rg r( r( r* r� � s
z AbstractEntityRegistry._getitemN�r< r rd r/ r )r$ r/ )r$ r# )r$ r} )r$ r8 )rd r r$ r
)r� r� r� r� r� r. r� r� r� r� r r r| r r� r rh r( r( r( r* rl � s*
.
�rl c @ s e Zd ZdZdS )r� TN)r� r� r� r� r( r( r( r* r� � s r� c @ s e Zd Zd dd�Zd
d d
�Zd S )�_ERDict�registryr� c C s
|| _ d S r' )r )rH r r( r( r* r� � r4 z_ERDict.__init__rS r r$ rn c C s t | j|� | |< }|S r' )rn r )rH rS �itemr( r( r* � __missing__� s z_ERDict.__missing__N)r r� )rS r r$ rn )r� r� r� r� r r( r( r( r* r � s
r c sJ e Zd ZdZdZd� fdd� Zdd
d�Zddd�Zes!eZ � Z
S � Z
S )r� )�_cacheTr< r rd r/ c s t � �||� t| �| _d S r' )�superr� r r r
�r� r( r* r�
s z CachingEntityRegistry.__init__rS r �defaultr$ c C s | j �||�S r' )r �pop)rH rS r r( r( r* r r� zCachingEntityRegistry.popc C s6 t |ttf�r | j| S t |t�rt| |�S | j| S r' )r� rO rk r" r� rf r rg r( r( r* r� s
z CachingEntityRegistry._getitemr )rS r r r r$ r r� ) r� r� r� r� r� r� r r� r rh �
__classcell__r( r( r r* r� s
�r� c C r& r' r( r) r( r( r* �path_is_entity# r� r �TypeGuard[PropRegistry]c C r& r' r( r) r( r( r* �path_is_property' r, r r� )r" r# r$ r% )r" r# r$ r- r� )r" r# r$ r )Or� �
__future__r � functoolsr � itertoolsr �logging�operator�typingr r r r r r
r r r
r r � r r� �_typingr r r �sqlr �
sql.cache_keyr r �
interfacesr r| r �
relationshipsr r r � sql.elementsr � sql.visitorsr � util.typingr r! r+ r. �
attrgetterr1 r� re rj r: ru rv � getLoggerr� rV r3 r6 r� r� r# r� r= r> �InspectionAttrr� rf rn rl r� r r� r r r( r( r( r* �<module> s�
�
&V `