U
|
¸ý°dÄb ã @ 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/ d#d$d%d&d'Z0d#d(d%d)d*Z1ne 2d'¡Z0e 2d*¡Z1ee
|
Z3ee4d+d,f Z5ee5d-f Z6ed+ Z7eed,e4f Z8e 9e:¡Z;d.d#d%d/d0Z<d1Z=d2e>d3< d4Z?G d5d# d#e Z@G d6d7 d7e@ZAG d8d9 d9eAZBeB e@_CG d:d; d;ejDe e4ZEG d<d= d=e@ZFG d>d? d?e@ZGG d@dA dAeAZHG dBdC dCeHZIG dDdE dEe e
|
e
|
f ZJG dFdG dGeHZKerüd#d(d%dHdIZLd#dJd%dKdLZMne 2d*¡ZLe 2dM¡ZMdS )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)ÚMapperProperty)ÚMapper)ÚRelationshipProperty)Ú AliasedInsp)Ú_CacheKeyTraversalType)Ú BindParameter)Úanon_map)Ú _LiteralStar)Ú TypeGuardÚ PathRegistryzTypeGuard[RootRegistry]©ÚpathÚreturnc C s d S ©N© ©r$ r' r' úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/orm/path_registry.pyÚis_root0 s r* z!TypeGuard[AbstractEntityRegistry]c C s d S r& r' r( r' r' r) Ú is_entity3 s r+ ú_InternalEntityType[Any]úMapperProperty[Any].Ú_SerializedPathc C s
|
t | ¡S r& )r" Ú deserializer( r' r' r) Ú_unreduce_pathP s r0 Ú*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< dddddZdddddZeddddZdddddd d Zdddddd d Zdcddd!ddd"d#Zd$dd%d&Zd$dd'd(Zed)d*d+d,d-Zed$d.d+d/d-Zed0dd+d1d-Zed2d3d+d4d-Zed5d6d+d7d-Zd8d9d+d:d-Zed$dd;d<Zd=dd>d?Zd@ddAdBdCZ ddddDdEdFZ dddGdHZ e ddIdJdKdLZ e dIddJdMdNZ!dIddOdPZ"e dId dJdQdRZ#ee d@dSdAdTdUZ$ee dVdWdAdXdUZ$e d2d3dAdYdUZ$e dd dZd[d\Z%d d dd]d^Z&d)dd_d`Z'd)ddadbZ(d S )dr" 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= rB r' r' r) Ú__ne__ s ÿÿzPathRegistry.__ne__ú Optional[_PathRepresentation]©r% c C s | j S r& r( ©rC r' r' r) r> s z PathRegistry._path_for_comparezDict[Any, Any]ÚNone)Ú
|
attributesÚkeyÚvaluer% c C s" t d|| |¡ |||| jf< d S )Nz set '%s' on path '%s' to '%s')ÚlogÚdebugr6 ©rC rJ rK rL r' r' r) Úset s zPathRegistry.setc C s& t d|| |¡ | || jf|¡ d S )Nz$setdefault '%s' on path '%s' to '%s')rM rN Ú
|
setdefaultr6 rO r' r' r) rQ ¡ s zPathRegistry.setdefaultNz Optional[Any]c C s" || j f}||kr|| S |S d S r& )r6 rO r' r' r) Úget§ s
|
zPathRegistry.getÚintc C s
|
t | jS r& ©Úlenr$ rH r' r' r) Ú__len__° s zPathRegistry.__len__c C s t | S r& )ÚidrH r' r' r) Ú__hash__³ s zPathRegistry.__hash__ÚstrÚ TokenRegistry©Úentityr% c C s d S r& r' ©rC r\ r' r' r) Ú __getitem__¶ s zPathRegistry.__getitem__Ú_PathElementTypec C s d S r& r' r] r' r' r) r^ º s Úslicec C s d S r& r' r] r' r' r) r^ ¾ s r, ÚAbstractEntityRegistryc C s d S r& r' r] r' r' r) r^ Â s r- Ú PropRegistryc C s d S r& r' r] r' r' r) r^ È s zEUnion[str, int, slice, _InternalEntityType[Any], MapperProperty[Any]]zaUnion[TokenRegistry, _PathElementType, _PathRepresentation, PropRegistry, AbstractEntityRegistry]c C s
|
t d S r& )ÚNotImplementedErrorr] r' r' r) r^ Ì s c C s
|
t | jS r& rT rH r' r' r) ÚlengthÛ s zPathRegistry.lengthzJIterator[Tuple[_InternalEntityType[Any], Union[str, MapperProperty[Any]]]]c c sF t t| j}t t|}tdt|dD ]}|| ||d fV q&d S )Nr r r )r Ú_OddPathRepresentationr$ Ú_EvenPathRepresentationÚrangerU )rC Zodd_pathZ even_pathÚir' r' r) Úpairsß s
|
zPathRegistry.pairsú Mapper[Any])Úmapperr% c sF t t| j fddtdt dD D ]}|j |¡r* dS q*dS )Nc s g | ] } | qS r' r' ©Ú.0rh ©Ú_m_pathr' r) Ú
|
<listcomp>ë s z0PathRegistry.contains_mapper.<locals>.<listcomp>r r TF)r re r$ rg rU rk Úisa)rC rk Z path_mapperr' rn r) Úcontains_mapperé s
|
" z PathRegistry.contains_mapper)rJ rK r% c C s || j f|kS r& r( )rC rJ rK r' r' r) Úcontainsñ s zPathRegistry.containsc C s t | ¡ ffS r& )r0 Ú serializerH r' r' r) Ú
|
__reduce__ô s zPathRegistry.__reduce__r. r# c
|
sv t t| t t|tttdd fddtdt dD D tdd fddtd tdD D d
|
S ) Nc s s( | ] }|j s|jr|jnt|V qd S r& )Ú is_mapperÚis_aliased_classÚclass_rY )rm Úmr' r' r) Ú <genexpr>þ s ÿz/PathRegistry._serialize_path.<locals>.<genexpr>c s g | ] } | qS r' r' rl rn r' r) rp s z0PathRegistry._serialize_path.<locals>.<listcomp>r r c s s$ | ] }t |r|jnt|V qd S r& )r rK rY )rm Úpr' r' r) rz s ÿc s g | ] } | qS r' r' rl )Ú_p_pathr' r) rp s r r& )r re rf ÚlistÚzipÚtuplerg rU )Úclsr$ r' )ro r| r) Ú_serialize_path÷ s
|
|
þ þüûÿz PathRegistry._serialize_pathc s^ ddddddddddd t t fdd |D }|rZ|d
|
d krZ|d d
|
}|S ) Nr )Úmclsr% c S s" | t jkrtj| ddS 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 rK 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 rK Úmpr' r' r) Ú_deserialize_key_token s
|
|
z>PathRegistry._deserialize_path.<locals>._deserialize_key_tokenc s" g | ]\}}| ||fqS r' r' )rm r rK ©r r r' r) rp ! 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 )rC r$ r' r' r) rt . s zPathRegistry.serializec C s |d k s t | |¡}| |¡S r& )r r Úcoercer r' r' r) r/ 2 s
|
zPathRegistry.deserializeÚCachingEntityRegistryc C s d S r& r' ©r rk r' r' r) Ú
|
per_mapper8 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& )rv r r9 r r r' r' r) r B s )Úrawr% c C s dddddd}t ||| jS )Nr" r_ ©ÚprevÚnext_r% c S s | | S r& r' ©r r r' r' r) Ú_redM s z!PathRegistry.coerce.<locals>._red)r r9 )r r r r' r' r) r K s zPathRegistry.coercec C s dddddd}t ||j| S )Nr" r_ r c S s | | S r& r' r r' r' r) r T s z"PathRegistry.__add__.<locals>._red)r r$ )rC r; r r' r' r) Ú__add__S s zPathRegistry.__add__c C s dd dd | jD ¡ dS )Nz ORM Path[z -> c s s | ]}t |V qd S r& ©rY )rm Úelemr' r' r) rz Z s z'PathRegistry.__str__.<locals>.<genexpr>ú])Újoinr$ rH r' r' r) Ú__str__Y s zPathRegistry.__str__c C s | j j d| jdS )Nú(ú))Ú __class__Ú__name__r$ rH 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: rD rE Úpropertyr> rP rQ rR rV rX r r^ rd ri rr rs ru Ú classmethodr r rt r/ r r r r£ r¨ r' r' r' r) r" X sz
|
|
ÿ
|
|
ÿ
|
#c @ s2 e Zd ZU dZded< ded< ddddd Zd
|
S ) Ú CreatesTokenr' r3 rw r* rY rZ )Útokenr% c C sJ | dt ¡rt| |S | dt ¡r6t| j|S t d| ¡d S )Nú:zinvalid token: )Úendswithr2 rZ Ú_DEFAULT_TOKENr9 r Ú ArgumentError)rC r´ r' r' r) r´ f s
|
|
zCreatesToken.tokenN)r§ r© rª r¬ r° r´ r' r' r' r) r³ ` s
|
r³ c @ sV e Zd ZdZdZdZd ZZdZdZ dZ
|
dZ ddddd Z d d
|
d d Z esRe Zd S )r8 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|ts tt| t j| S z|jW S tk
|
rV t d| Y nX d S )Nz/invalid argument for RootRegistry.__getitem__: )
|
r
r r Ú
|
isinstancerY r rZ Z_path_registryr? Ú
|
IndexErrorr] r' r' r) Ú_getitem s
|
ÿzRootRegistry._getitemrG c C s | S r& r' rH r' r' r) Ú_truncate_recursive s z RootRegistry._truncate_recursiveN)r§ r© rª r« r¬ Ú inherit_cacher$ r6 r® rw r* r4 r» r¼ r r^ r' r' r' r) r8 o s r8 c @ sV e Zd ZU dZi Zded< dddddd Zed
|
d d d Ze dd dddZ
|
dS )r
zcacheable string tokenzDict[str, PathToken]r r zList[BindParameter[Any]]zTuple[Any, ...])r Ú
|
bindparamsr% c C s
|
t | fS r& r )rC r r¾ r' r' r) Ú_gen_cache_key s zPathToken._gen_cache_keyrF rG c C s d S r& r' rH r' r' r) r> ¢ s zPathToken._path_for_comparerY )Ú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 @ sd e Zd ZU dZdZded< ded< ddddd Zd
|
ZdZd d d dZ dddddZ
|
e s`e
|
Z dS )rZ )r´ r7 r$ r6 TrY r´ r³ r7 )r7 r´ c C s6 t |¡}|| _|| _|j|f | _|j|f | _d S r& )r
r r´ r7 r$ r6 )rC r7 r´ r' r' r) Ú__init__· s
|
|
zTokenRegistry.__init__FzIterator[PathRegistry]rG c c s¤ | j }t|r| V d S tr*t|ts*t|jsV|j ¡ D ]}t |j | | j
|
V q:nJ|jrt d|j j r| V t d|j jD ]}t |j | | j
|
V q~n| V d S )Nr )r7 r* r r¹ ra r rw rk Ziterate_to_rootrZ r´ r r\ Z_is_with_polymorphicZ_with_polymorphic_entities)rC r7 Zmp_entÚentr' r' r) Úgenerate_for_superclassesà s. ÿþþ ÿ
|
z'TokenRegistry.generate_for_superclassesr r[ c
|
C s@ z | j | W S tk
|
r: } zt| |W 5 d }~X Y nX d S r& )r$ Ú TypeErrorrº )rC r\ Úerrr' r' r) r» Ý s zTokenRegistry._getitemN) r§ r© rª r¬ r½ r° rà r® r rÅ r» r r^ r' r' r' r) rZ ¯ s
|
rZ c @ sz e Zd ZU dZdZdZded< ded< ded< d dd
|
d d Zd d ddZe d d ddZ
|
dddddZ e sve Z dS )rb ) Úpropr7 r$ r6 r® r\ rk Ú_wildcard_path_loader_keyÚ_default_path_loader_keyÚ _loader_keyr4 Tr- rÈ zOptional[Mapper[Any]]rk z"Optional[_InternalEntityType[Any]]r\ ra )r7 rÈ c C sd t d|d }|}|jjr"d| _nd| _|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 | _ |jr*trt|tst|j| _|j| _n d | _d | _d|j
|
| j j f| _| j j| _d| j f| _d S )Nr, r TFÚloader)r r7 r4 rw Z_use_mapper_pathZwith_polymorphic_mappersZ_entity_for_mapperrk rq rÈ r$ r6 Z_links_to_entityr® Z_is_relationshipr r¹ r r r\ Z_wildcard_tokenrÉ rÊ rË )rC r7 rÈ ZinspZnatural_parentZsubclass_entityr' r' r) rà ü s^ ÿþ
|
ý
|
ÿþÿý û
|
þ
|
zPropRegistry.__init__rG 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 )rC Zearliestrh r´ r' r' r) r¼ V s
|
z PropRegistry._truncate_recursivec C s | j d k st| | j S r& )r\ r rH r' r' r) Ú entity_patha s zPropRegistry.entity_pathz+Union[int, slice, _InternalEntityType[Any]]zDUnion[AbstractEntityRegistry, _PathElementType, _PathRepresentation]r[ c C s& t |ttfr| j| S t| |S d S r& )r¹ rS r` r$ r r] r' r' r) r» f s
|
zPropRegistry._getitemN)r§ r© rª r¬ r½ r¯ r° rà r¼ r± rÏ r» r r^ r' r' r' r) rb ç s
|
Z rb c @ s´ e Zd ZU dZdZdZded< ded< ded< ded < ddd
|
d d Zd d ddZe dd ddZ
|
e dd ddZ e dd ddZ dd ddZ ddd d d Zes°eZd S ) ra )rK r7 rw r$ r\ r6 Tú!Union[RootRegistry, PropRegistry]r7 r, rK r\ r3 rw ©r7 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 ) rK r7 rw r\ r$ r4 rk rq r6 ©rC r7 r\ r' r' r) rà s
|
ÿ
|
z AbstractEntityRegistry.__init__rG c C s | j ¡ | j S r& )r7 r¼ r\ rH r' r' r) r¼ ¯ s z*AbstractEntityRegistry._truncate_recursivec C s t d| jd S )Nr, r )r r$ rH r' r' r) Ú root_entity² s z"AbstractEntityRegistry.root_entityr" c C s | S r& r' rH r' r' r) rÏ ¶ s z"AbstractEntityRegistry.entity_pathrj c C s | j jS r& )r\ rk rH r' r' r) rk º s z AbstractEntityRegistry.mapperc C s dS )NTr' rH r' r' r) Ú__bool__¾ s z AbstractEntityRegistry.__bool__r z:Union[_PathElementType, _PathRepresentation, PathRegistry]r[ c C s@ t |ttfr| j| S |tjkr2t| tj| S t| |S d S r& )r¹ rS r` r$ r
r rZ rb r] r' r' r) r» Á s
|
|
|
z AbstractEntityRegistry._getitemN)r§ r© rª r¬ r® r+ r° rà r¼ r± rÓ rÏ rk rÔ r» r r^ r' r' r' r) ra r s&
|
+
|
ra c @ s e Zd ZdZdS )r TN)r§ r© rª r½ r' r' r' r) r Ï s r c @ s* e Zd ZddddZddddd Zd
|
S ) Ú_ERDictr ©Úregistryc C s
|
|| _ d S r& rÖ )rC r× r' r' r) rÃ Ö s z_ERDict.__init__r rb )rK r% c C s t | j| | |< }|S r& )rb r× )rC rK Ú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 dZesNeZ 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 )rK Údefaultr% c C s | j ||¡S r& )rÚ Úpop)rC rK rÝ r' r' r) rÞ ï s zCachingEntityRegistry.popr[ c C s: t |ttfr| j| S t |tr,t| |S | j| S d S r& )r¹ rS r` r$ r
rZ rÚ r] r' r' r) r» ò s
|
|
|
|
z CachingEntityRegistry._getitem) r§ r© rª r¬ r½ rà rÞ r» r r^ Ú __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 s rá r¯ )Nr« Ú
|
__future__r Ú functoolsr Ú itertoolsr ÚloggingÚoperatorÚtypingr r r r r r
|
r r r r r Ú r r Ú_typingr r r Zsqlr Z sql.cache_keyr r Z
|
interfacesr rk r Z relationshipsr r r Z sql.elementsr Z sql.visitorsr Z util.typingr r! r* r+ Ú
|
attrgetterr. rY r_ r5 re rf Ú getLoggerr§ rM r0 r2 r° r· r" r³ r8 r9 ZInspectionAttrr
rZ rb ra r rÕ r rà rá r' r' r' r) Ú<module> s
|
|
ÿ
|
|
|
&8 ]
|
|
|
