관리-도구
편집 파일: path_registry.cpython-38.pyc
U -?�f@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 e�r�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/ d#d$d%�d&d'�Z0d#d(d%�d)d*�Z1ne�2d'�Z0e�2d*�Z1ee Z3e4Z5ee5d+d,f Z6ee6d-f Z7ed+ Z8eed,e4f Z9e�:e;�Z<d.d#d%�d/d0�Z=d1Z>d2e?d3< d4Z@G d5d#� d#e �ZAG d6d7� d7eA�ZBG d8d9� d9eB�ZCeC� eA_DG d:d;� d;ejEe e4�ZFG d<d=� d=eA�ZGG d>d?� d?eA�ZHG d@dA� dAeB�ZIG dBdC� dCeI�ZJG dDdE� dEe e e f �ZKG dFdG� dGeI�ZLe�r d#d(d%�dHdI�ZMd#dJd%�dKdL�ZNne�2d*�ZMe�2dM�ZNdS )Nz@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� PathRegistryzTypeGuard[RootRegistry]��path�returnc C s d S �N� �r$ r' r' �L/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/path_registry.py�is_root0 � r* z!TypeGuard[AbstractEntityRegistry]c C s d S 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 s r1 �*r �_WILDCARD_TOKENZ _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< ddd�dd�Zddd�dd�Zedd�dd��Zddd�d d �Zd ddd d �d!d"�Zd ddd d �d#d$�Zdgd dd&dd �d'd(�Zdd�d)d*�Zdd�d+d,�Zed-d.d/�d0d1��Zedd2d/�d3d1��Zed4dd/�d5d1��Zedd6d/�d7d1��Zed8d9d/�d:d1��Zd;d<d/�d=d1�Zedd�d>d?��Zd@d�dAdB�Z dCddD�dEdF�Z d dddG�dHdI�Z dd�dJdK�Z e ddLdM�dNdO��Z!e dLddM�dPdQ��Z"dLd�dRdS�Z#e dLd dM�dTdU��Z$ee dCdVdD�dWdX���Z%ee dYdZdD�d[dX���Z%e dd6dD�d\dX��Z%e dd d]�d^d_��Z&d d d�d`da�Z'dbd�dcdd�Z(dbd�dedf�Z)d%S )hr" 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_traversalr )�otherr% c C sD z|d k o| j |jkW S tk r> t�dt|� � Y dS X d S )N�1Comparison of PathRegistry to %r is not supportedF�r$ �_path_for_compare�AttributeErrorr �warn�type��selfr< r' r' r) �__eq__� s ��zPathRegistry.__eq__c C sD z|d kp| j |jkW S tk r> t�dt|� � Y dS X d S )Nr= Tr> rC r' r' r) �__ne__� s ��zPathRegistry.__ne__� Optional[_PathRepresentation]�r% c C s | j S r& r( �rD r' r' r) r? � s z PathRegistry._path_for_compare�intr- )�indexr% c C s | j | S r& r( )rD rK r' r' r) � odd_element� s zPathRegistry.odd_elementzDict[Any, Any]�None)� attributes�key�valuer% c C s" t �d|| |� |||| jf< d S )Nz set '%s' on path '%s' to '%s')�log�debugr7 �rD rN rO rP r' r' r) �set� s zPathRegistry.setc C s& t �d|| |� |�|| jf|� d S )Nz$setdefault '%s' on path '%s' to '%s')rQ rR � setdefaultr7 rS r' r' r) rU � s zPathRegistry.setdefaultNz Optional[Any]c C s" || j f}||kr|| S |S d S r& )r7 rS r' r' r) �get� s zPathRegistry.getc C s t | j�S r& ��lenr$ rI r' r' r) �__len__� s zPathRegistry.__len__c C s t | �S r& )�idrI r' r' r) �__hash__� s zPathRegistry.__hash__� _StrPathToken� TokenRegistry��entityr% c C s d S r& r' �rD r_ r' r' r) � __getitem__� s zPathRegistry.__getitem__�_PathElementTypec C s d S r& r' r` r' r' r) ra � s �slicec C s d S r& r' r` r' r' r) ra � s �AbstractEntityRegistryc C s d S r& r' r` r' r' r) ra � s r. � PropRegistryc C s d S r& r' r` r' r' r) ra � s zTUnion[_StrPathToken, int, slice, _InternalEntityType[Any], StrategizedProperty[Any]]zaUnion[TokenRegistry, _PathElementType, _PathRepresentation, PropRegistry, AbstractEntityRegistry]c C s t � �d S r& )�NotImplementedErrorr` r' r' r) ra � s c C s t | j�S r& rW rI r' r' r) �length� s zPathRegistry.lengthzOIterator[Tuple[_InternalEntityType[Any], Union[str, StrategizedProperty[Any]]]]c c sF t t| j�}t t|�}tdt|�d�D ]}|| ||d fV q&d S )Nr r r )r �_OddPathRepresentationr$ �_EvenPathRepresentation�rangerX )rD Zodd_pathZ even_path�ir' r' r) �pairs� s zPathRegistry.pairs� Mapper[Any])�mapperr% c sF t t| j�� � fdd�tdt� �d�D �D ]}|j�|�r* dS q*dS )Nc s g | ] }� | �qS r' r' ��.0rk ��_m_pathr' r) � <listcomp>� s z0PathRegistry.contains_mapper.<locals>.<listcomp>r r TF)r rh r$ rj rX rn �isa)rD rn Z path_mapperr' rq r) �contains_mapper� s " z PathRegistry.contains_mapper)rN rO r% c C s || j f|kS r& r( )rD rN rO r' r' r) �contains� s zPathRegistry.containsc C s t | �� ffS r& )r1 � serializerI r' r' r) � __reduce__� s zPathRegistry.__reduce__r/ r# 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)rp �mr' r' r) � <genexpr> s �z/PathRegistry._serialize_path.<locals>.<genexpr>c s g | ] }� | �qS r' r' ro rq r' r) rs s z0PathRegistry._serialize_path.<locals>.<listcomp>r r c s s$ | ] }t |�r|jnt|�V qd S r& )r rO r| )rp �pr' r' r) r~ s �c s g | ] }� | �qS r' r' ro )�_p_pathr' r) rs s r r& )r rh ri �list�zip�tuplerj rX )�clsr$ r' )rr r� r) �_serialize_path� s � ����z PathRegistry._serialize_pathc s^ ddd�dd��dddd�dd�� t t� �fdd �|D �� �}|rZ|d d krZ|d d � }|S ) Nr )�mclsr% c S s" | t jkrtj| 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_token)r� rO r% c S sH |d kr d S |t jkr t j| S tj| dd�}|d k s:t�|j| S d S r� )r� r� r� r� �AssertionError�attrs)r� rO �mpr' r' r) �_deserialize_key_token s z>PathRegistry._deserialize_path.<locals>._deserialize_key_tokenc s" g | ]\}}�|�� ||�f�qS r' r' )rp r� rO �r� r� r' r) rs % s ��z2PathRegistry._deserialize_path.<locals>.<listcomp>���r )r� r �r� r$ r r' r� r) �_deserialize_path s ��� z PathRegistry._deserialize_pathc C s | j }| �|�S r& )r$ r� )rD r$ r' r' r) rw 2 s zPathRegistry.serializec C s |d k s t �| �|�}| �|�S r& )r� r� �coercer� r' r' r) r0 6 s zPathRegistry.deserialize�CachingEntityRegistryc C s d S r& r' �r� rn r' r' r) � per_mapper< s zPathRegistry.per_mapper�AliasedInsp[Any]�SlotsEntityRegistryc C s d S r& r' r� r' r' r) r� @ s c C s" |j rt| j|�S t| j|�S d S r& )ry r� r: r� r� r' r' r) r� D s )�rawr% c C s dddd�dd�}t ||| j�S )Nr" rb ��prev�next_r% c S s | | S r& r' �r� r� r' r' r) �_redO s z!PathRegistry.coerce.<locals>._red)r r: )r� r� r� r' r' r) r� M s zPathRegistry.coercec C s dddd�dd�}t ||j| �S )Nr" rb r� c S s | | S r& r' r� r' r' r) r� V s z"PathRegistry.__add__.<locals>._red)r r$ )rD 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| )rp �elemr' r' r) r~ \ s z'PathRegistry.__str__.<locals>.<genexpr>�])�joinr$ rI r' r' r) �__str__[ s zPathRegistry.__str__c C s | j j� d| j�d�S )N�(�))� __class__�__name__r$ rI r' r' r) �__repr__^ s zPathRegistry.__repr__)N)*r� � __module__� __qualname__�__doc__� __slots__�is_tokenr* � has_entity� is_propertyr, �__annotations__r ZExtendedInternalTraversalZdp_has_cache_key_listr; rE rF �propertyr? rL rT rU rV rY r[ r ra rg rl ru rv rx � classmethodr� r� rw r0 r� r� r� r� r� r' r' r' r) r" X s| � � #c @ s2 e Zd ZU dZded< ded< ddd�dd �Zd S ) � CreatesTokenr' r4 rz r* r\ r] )�tokenr% c C sJ |� dt� ��rt| |�S |� dt� ��r6t| j|�S t�d|� ���d S )N�:zinvalid token: )�endswithr3 r] �_DEFAULT_TOKENr: r � ArgumentError)rD r� r' r' r) r� h s zCreatesToken.tokenN)r� r� r� r� r� r� r' r' r' r) r� b s r� c @ sV e Zd ZdZdZdZd ZZdZdZ dZ dZ ddd�dd �Z d d �d d �Z esRe Zd S )r9 zXRoot registry, defers to mappers so that paths are maintained per-root-mapper. r' TFr z,Union[TokenRegistry, AbstractEntityRegistry]r^ c C s\ |t jkr,tr t|t�s t�t| t j| �S z|jW S tk rV t d|� ���Y nX d S )Nz/invalid argument for RootRegistry.__getitem__: ) r� r� r � isinstancer\ r� r] Z_path_registryr@ � IndexErrorr` r' r' r) �_getitem� s �zRootRegistry._getitemrH c C s | S r& r' rI r' r' r) �_truncate_recursive� s z RootRegistry._truncate_recursiveN)r� r� r� r� r� � inherit_cacher$ r7 r� rz r* r5 r� r� r ra r' r' r' r) r9 q s r9 c @ sV e Zd ZU dZi Zded< dddd�dd �Zed d �d d ��Ze dd d�dd��Z dS )r� zcacheable string tokenzDict[str, PathToken]r� r zList[BindParameter[Any]]zTuple[Any, ...])r � bindparamsr% c C s t | �fS r& r� )rD r r� r' r' r) �_gen_cache_key� s zPathToken._gen_cache_keyrG rH c C s d S r& r' rI r' r' r) r? � s zPathToken._path_for_comparer| )�strvaluer% c C s. || j kr| j | S t|� | j |< }|S d 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� � s r� c @ sr e Zd ZU dZdZded< ded< ddd�dd �Zd ZdZd d �d d�Z dd �dd�Z ddd�dd�Z e sne Z dS )r] )r� r8 r$ r7 Tr\ r� r� r8 )r8 r� c C s6 t �|�}|| _|| _|j|f | _|j|f | _d S r& )r� r� r� r8 r$ r7 )rD r8 r� r' r' r) �__init__� s zTokenRegistry.__init__FzIterator[PathRegistry]rH c c s� | j }t|�r| V d S tr*t|t�s*t�|jsV|j�� D ]}t |j | | j �V q:nJ|jr�t d|j �j r�| V t d|j �jD ]}t |j | | j �V q~n| V d S �Nr� )r8 r* r r� rd r� rz rn �iterate_to_rootr] r� r r_ �_is_with_polymorphic�_with_polymorphic_entities�rD r8 Zmp_entZentr' r' r) �generate_for_superclasses� s. ��� � z'TokenRegistry.generate_for_superclassesz Iterator[_PathRepresentation]c c s� | j }t|�r| jV d S tr,t|t�s,t�|j�� D ]}t |j | | j �jV q6|j r�t d|j �jr�| jV t d|j �jD ]}t |j | | j �jV q|n| jV d S r� )r8 r* r7 r r� rd r� rn r� r] r� rz r r_ r� r� r� r' r' r) �"_generate_natural_for_superclasses� s. ��� � �z0TokenRegistry._generate_natural_for_superclassesr r^ c C s@ z | j | W S tk r: } zt|� �|�W 5 d }~X Y nX d S r& )r$ � TypeErrorr� )rD r_ �errr' r' r) r� � s zTokenRegistry._getitemN)r� r� r� r� r� r� r� r� r� r� r� r� r ra r' r' r' r) r] � s r] c @ sz e Zd ZU dZdZdZded< ded< ded< d dd �d d �Zd d �dd�Ze d d �dd��Z ddd�dd�Z e sve Z dS )re ) �propr8 r$ r7 r� r_ rn �_wildcard_path_loader_key�_default_path_loader_key� _loader_keyr5 Tr. r� zOptional[Mapper[Any]]rn z"Optional[_InternalEntityType[Any]]r_ rd )r8 r� c C sd t d|d �}|}|jjp$t|jj�| _|jr4|jrF|j|j }}n�|jr�|jr�|j|jkr�|d � |j�}|j| }|jr�|j|j }d| _q�|}n2|jr�|jr�|j|jk r�|j� |j�r�|j|j }|| _ || _|j |f | _ |j |f | _ |j| _|j�r*t�rt|t��st�|j| _|j| _n d | _d | _d|j | j j f| _| j j| _d| j f| _d S )Nr- r� T�loader)r r8 r5 r4 rn Zinheritsrz Z_use_mapper_pathZwith_polymorphic_mappersZ_entity_for_mapperrt r� r$ r7 Z_links_to_entityr� Z_is_relationshipr r� r r� r_ Z_wildcard_tokenr� r� r� )rD r8 r� ZinspZnatural_parentZsubclass_entityr' r' r) r� s^ � �� � ���� � � zPropRegistry.__init__rH c C s\ d }t t| jd d� ��D ]\}}|| jkr|}q|d kr>| S | �| jd|d � �S d S )Nr� r r )� enumerate�reversedr$ r� r� )rD Zearliestrk r� r' r' r) r� v s z PropRegistry._truncate_recursivec C s | j d k st�| | j S r& )r_ r� rI r' r' r) � entity_path� s zPropRegistry.entity_pathz+Union[int, slice, _InternalEntityType[Any]]zDUnion[AbstractEntityRegistry, _PathElementType, _PathRepresentation]r^ c C s& t |ttf�r| j| S t| |�S d S r& )r� rJ rc r$ r� r` r' r' r) r� � s zPropRegistry._getitemN)r� r� r� r� r� r� r� r� r� r� r� r� r ra r' r' r' r) re s Z re c @ s� e Zd ZU dZdZdZded< ded< ded< ded < ddd �d d �Zd d �dd�Ze dd �dd��Z e dd �dd��Z e dd �dd��Z dd �dd�Z ddd �d d �Zes�eZd S ) rd )rO r8 rz r$ r_ r7 T�!Union[RootRegistry, PropRegistry]r8 r- rO r_ r4 rz �r8 r_ c C s� || _ || _|j| _|| _|j|f | _|jrx| js:|jrx|j�|jd j�r`|j|jf | _q�|j|jd jf | _n| j| _d S )Nr� ) rO r8 rz r_ r$ r5 rn rt r7 �rD r8 r_ r' r' r) r� � s � z AbstractEntityRegistry.__init__rH c C s | j �� | j S r& )r8 r� r_ rI r' r' r) r� � s z*AbstractEntityRegistry._truncate_recursivec C s | � d�S )Nr )rL rI r' r' r) � root_entity� s z"AbstractEntityRegistry.root_entityr" c C s | S r& r' rI r' r' r) r� � s z"AbstractEntityRegistry.entity_pathrm c C s | j jS r& )r_ rn rI r' r' r) rn � s z AbstractEntityRegistry.mapperc C s dS )NTr' rI r' r' r) �__bool__� s z AbstractEntityRegistry.__bool__r z:Union[_PathElementType, _PathRepresentation, PathRegistry]r^ c C s@ t |ttf�r| j| S |tjkr2t| tj| �S t| |�S d S r& )r� rJ rc r$ r� r� r] re r` r' r' r) r� � s z AbstractEntityRegistry._getitemN)r� r� r� r� r� r, r� r� r� r� r� r� rn r� r� r ra r' r' r' r) rd � s& . rd c @ s e Zd ZdZdS )r� TN)r� r� r� r� r' r' r' r) r� � s r� c @ s* e Zd Zdd�dd�Zddd�dd �Zd S ) �_ERDictr� ��registryc C s || _ d S r& r� )rD r� r' r' r) r� � s z_ERDict.__init__r re )rO r% c C s t | j|� | |< }|S r& )re r� )rD rO �itemr' r' r) � __missing__� s z_ERDict.__missing__N)r� r� r� r� r� r' r' r' r) r� � s r� c sV e Zd ZdZdZddd�� fdd� Zdddd �d d �Zddd �d d�ZesNeZ � Z S )r� )�_cacheTr� r- r� c s t � �||� t| �| _d S r& )�superr� r� r� r� �r� r' r) r� s z CachingEntityRegistry.__init__r )rO �defaultr% c C s | j �||�S r& )r� �pop)rD rO r� r' r' r) r� s zCachingEntityRegistry.popr^ c C s: t |ttf�r| j| S t |t�r,t| |�S | j| S d S r& )r� rJ rc r$ r� r] r� r` r' r' r) r� s z CachingEntityRegistry._getitem) r� r� r� r� r� r� r� r� r ra � __classcell__r' r' r� r) r� s r� c C s d S r& r' r( r' r' r) �path_is_entity# s r� zTypeGuard[PropRegistry]c C s d S r& r' r( r' r' r) �path_is_property' r+ r� r� )Or� � __future__r � functoolsr � itertoolsr �logging�operator�typingr r r r r r r r r r r � r r� Z_typingr r r �sqlr Z sql.cache_keyr r Z interfacesr rn r Z relationshipsr r r Z sql.elementsr Z sql.visitorsr Z util.typingr r! r* r, � attrgetterr/ r| r\ rb r6 rh ri � getLoggerr� rQ r1 r3 r� r� r" r� r9 r: ZInspectionAttrr� r] re rd r� r� r� r� r� r' r' r' r) �<module> s� � &V `