U
|
¸ý°d[H ã @ s¦ U d Z ddlmZ ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm
|
Z
|
dd
|
lm Z dd lm Z dd lm Z dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlmZ ddlmZ ddlmZ ddl m Z ddl m Z dd lm Z dd lm!Z! dd lm Z dd l"m#Z# dd l$m%Z% erdd!l&m'Z' dd"l"m(Z( dd#l"m)Z) ed$ed%Z*e e+ee,d&f f Z-e. Z/d'e0d(< d)d*d+d,d-d.d/Z1d)d0d+d,d-d1d2Z2d)d+d3d4d5d6d7Z3G d8d& d&Z4G d9d: d:e4Z5G d;d< d<e4Z6G d=d> d>Z7G d?d@ d@Z8e! 9e8¡dAdB G dCdD dDZ:d)dEd@dFdGdHZ;G dIdJ dJZ<da=dKe0dL< d0dMdNdOdPdQZ>dS )Rz±Routines to handle the string class registry used by declarative.
|
|
This system allows specification of classes and expressions used in
|
:func:`_orm.relationship` using strings.
|
|
é )Ú annotationsN)ÚAny)ÚCallable)Úcast)ÚDict)Ú Generator)ÚIterable)ÚList)ÚMapping)ÚMutableMapping)ÚNoReturn)ÚOptional)ÚSet)ÚTuple)ÚType)Ú TYPE_CHECKING)ÚTypeVar)ÚUnioné )Ú
|
attributes)Ú
|
interfaces)ÚSynonymProperty)ÚColumnProperty)Ú class_mapperé )Úexc)Ú
|
inspection)Úutil)Ú_get_table_key)ÚCallableReference)ÚRelationshipProperty)ÚMetaData)ÚTableÚ_T)ÚboundÚClsRegistryTokenzSet[ClsRegistryToken]Ú _registriesÚstrzType[_T]Ú_ClsRegistryTypeÚNone)Ú classnameÚclsÚdecl_class_registryÚreturnc
|
C s | |kr6|| }t |ts>t|td|g }|| < n||| < ztt|d }W n& tk
|
rv tdd |d< }Y nX |j d¡}|r | d¡}| |¡}|D ]}| |¡}q¢z| | |¡ W q t
|
k
|
r } z&t |tsöt d| d¡|n W 5 d}~X Y qX qdS )z^Add a class to the _decl_class_registry associated with the
|
given declarative class.
|
|
ú Type[Any]Ú_sa_module_registryNÚ.r úname "ú-" matches both a class name and a module name) Ú
|
isinstanceÚ_MultipleClassMarkerr Ú _ModuleMarkerÚKeyErrorÚ
|
__module__ÚsplitÚpopÚ
|
get_moduleÚ add_classÚAttributeErrorr ÚInvalidRequestError) r* r+ r, ÚexistingÚ root_moduleÚtokensÚtokenÚmoduleÚae© rD úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/orm/clsregistry.pyr; @ sF
|
ÿ ÿþÿþ
|
|
|
|
|
ÿýr; r. c C sÈ | |kr,|| }t |tr&| |¡ n|| = ztt|d }W n tk
|
rT Y d S X |j d¡}|rÄ| d¡}| |¡}|D ]}| |¡}q~z|
|
| |¡ W qb t k
|
rÀ t |tsºn Y qbX qbd S )Nr/ r0 r ) r3 r4 Ú remove_itemr r5 r6 r7 r8 r9 r: Ú remove_classr< )r* r+ r, r> r? r@ rA rB rD rD rE rG v s0
|
ÿ
|
|
|
rG zCallable[[Any], bool]Úbool)Úkeyr, Útestr- c C sP | |kr dS || }t |trB|jD ]}||r$ dS q$tdn
|
|| S dS )a¶ test if a key is empty of a certain object.
|
|
used for unit tests against the registry to see if garbage collection
|
is working.
|
|
"test" is a callable that will be passed an object should return True
|
if the given object is the one we were looking for.
|
|
We can't pass the actual object itself b.c. this is for testing garbage
|
collection; the caller will have to have removed references to the
|
object itself.
|
|
TFzunknown codepathN)r3 r4 ÚcontentsÚNotImplementedError)rI r, rJ ÚthingZ sub_thingrD rD rE Ú _key_is_empty s
|
|
|
rN c @ s e Zd ZdZdZdS )r% zAan object that can be in the registry._class_registry as a value.rD N)Ú__name__r7 Ú __qualname__Ú__doc__Ú __slots__rD rD rD rE r% · s c @ s e Zd ZU dZdZded< ded< d"dd d
|
d d Zd ddddZddddZddd dddZ ddd d d Z
|
d dd d d!Z dS )#r4 zRrefers to multiple classes of the same name
|
within _decl_class_registry.
|
|
)Ú on_removerK Ú __weakref__zSet[weakref.ref[Type[Any]]]rK z/CallableReference[Optional[Callable[[], None]]]rS NzIterable[Type[Any]]z Optional[Callable[[], None]])ÚclassesrS c s( | _ fdd|D _t ¡ d S )Nc s h | ]}t | j¡qS rD )ÚweakrefÚrefÚ _remove_item)Ú.0Úitem©ÚselfrD rE Ú <setcomp>Î s z0_MultipleClassMarker.__init__.<locals>.<setcomp>)rS rK r& Úadd)r\ rU rS rD r[ rE Ú__init__È s
|
|
ÿz _MultipleClassMarker.__init__r. r) )r+ r- c C s | t |¡¡ d S ©N)rX rV rW ©r\ r+ rD rD rE rF Ó s z _MultipleClassMarker.remove_itemz*Generator[Optional[Type[Any]], None, None]©r- c C s dd | j D S )Nc s s | ] }| V qd S r` rD ©rY rW rD rD rE Ú <genexpr>× s z0_MultipleClassMarker.__iter__.<locals>.<genexpr>©rK r[ rD rD rE Ú__iter__Ö s z _MultipleClassMarker.__iter__ú List[str]r' )ÚpathrI r- c C sV t | jdkr*t dd ||g ¡ ¡n(t| jd }| }|d krNt||S d S )Nr zxMultiple classes found for path "%s" in the registry of this declarative base. Please use a fully module-qualified path.r0 r )ÚlenrK r r= ÚjoinÚlistÚ NameError)r\ rh rI rW r+ rD rD rE Ú attempt_getÙ s ýÿz _MultipleClassMarker.attempt_getzweakref.ref[Type[Any]])rW r- c C s. | j |¡ | j s*t | ¡ | jr*| ¡ d S r` )rK Údiscardr& rS )r\ rW rD rD rE rX è s
|
|
z!_MultipleClassMarker._remove_item)rZ r- c C sT dd dd | j D D }|j|kr:t d|j|jf ¡ | j t || j¡¡ d S )Nc S s h | ]}|d k r|j qS r` )r7 )rY r+ rD rD rE r] ó s þz0_MultipleClassMarker.add_item.<locals>.<setcomp>c S s g | ]
|
}| qS rD rD rc rD rD rE Ú
|
<listcomp>õ s z1_MultipleClassMarker.add_item.<locals>.<listcomp>zThis declarative base already contains a class with the same class name and module name as %s.%s, and will be replaced in the string-lookup table.) rK r7 r ÚwarnrO r^ rV rW rX )r\ rZ ÚmodulesrD rD rE Úadd_itemï s þ
|
|
ýÿz _MultipleClassMarker.add_item)N) rO r7 rP rQ rR Ú__annotations__r_ rF rf rm rX rr rD rD rD rE r4 ½ s
|
ý r4 c @ sº e Zd ZU dZdZded< ded< ded< d ed
|
< d dd d dZd ddddZd ddddZd ddddZ d dddd Z
|
d d dd d Z d d dd d!d"Z d d dd d#d$Z d%S )&r5 z>Refers to a module name within
|
_decl_class_registry.
|
|
)ÚparentÚnamerK Úmod_nsrh rT zOptional[_ModuleMarker]rt z5Dict[str, Union[_ModuleMarker, _MultipleClassMarker]]rK Ú_ModNSrv rg rh r' )ru rt c C sJ || _ || _i | _t| | _| j r6| j j| jg | _ng | _t | ¡ d S r` )rt ru rK rw rv rh r& r^ )r\ ru rt rD rD rE r_ s
|
z_ModuleMarker.__init__rH )ru r- c C s
|
|| j kS r` re ©r\ ru rD rD rE Ú __contains__ s z_ModuleMarker.__contains__r% c C s
|
| j | S r` re rx rD rD rE Ú __getitem__ s z_ModuleMarker.__getitem__r) c C s: | j |d ¡ | j s6| jd k r6| j | j¡ t | ¡ d S r` )rK r9 rt rX ru r& rn rx rD rD rE rX s z_ModuleMarker._remove_itemúUnion[_ModNS, Type[Any]]©rI r- c C s | j |¡S r` )rv Ú __getattr__©r\ rI rD rD rE Ú resolve_attr& s z_ModuleMarker.resolve_attrc C s4 || j kr t|| }|| j |< ntt| j | }|S r` )rK r5 r )r\ ru ÚmarkerrD rD rE r: ) s
|
|
|
z_ModuleMarker.get_moduler. )ru r+ r- c
|
s j krpttj }z| |¡ W q tk
|
rl } z&t|tsZt d d¡|n W 5 d }~X Y qX n"t|g fddd }j < d S )Nr1 r2 c s
|
¡S r` )rX rD ©ru r\ rD rE Ú<lambda>@ ó z)_ModuleMarker.add_class.<locals>.<lambda>)rS )rK r r4 rr r< r3 r r= )r\ ru r+ r> rC rD r rE r; 1 s
|
|
|
ÿý ÿz_ModuleMarker.add_classc C s( || j kr$tt| j | }| |¡ d S r` )rK r r4 rF )r\ ru r+ r> rD rD rE rG C s
|
z_ModuleMarker.remove_classN)rO r7 rP rQ rR rs r_ ry rz rX r r: r; rG rD rD rD rE r5 s
|
r5 c @ s8 e Zd ZU dZded< ddddZddd d
|
d Zd S ) rw )Z__parentr5 Ú_ModNS__parent)rt c C s
|
|| _ d S r` )r )r\ rt rD rD rE r_ N s z_ModNS.__init__r' r{ r| c C st z| j j| }W n tk
|
r$ Y n8X |d k r\t|tr>|jS t|tsLt| | j j |¡S t
|
d| j j |f d S )Nz<Module %r has no mapped classes registered under the name %r) r rK r6 r3 r5 rv r4 ÚAssertionErrorrm rh rl ru )r\ rI ÚvaluerD rD rE r} Q s
|
|
ÿÿz_ModNS.__getattr__N©rO r7 rP rR rs r_ r} rD rD rD rE rw I s
|
rw c @ s8 e Zd ZU dZded< ddddZdddd d
|
Zd S ) Ú _GetColumns©r+ r. r+ c C s
|
|| _ d S r` r ra rD rD rE r_ h s z_GetColumns.__init__r' r r| c C s t | jdd}|r||jkr.td| j|f |j| }|jtjjkrt|t j
|
sVt |j }t|t rn|j}nt|tst d| ¡t| j|S )NF)Ú configurez/Class %r does not have a mapped column named %rzaProperty %r is not an instance of ColumnProperty (i.e. does not correspond directly to a Column).)r r+ Zall_orm_descriptorsr< Zextension_typer Z NotExtensionZ NOT_EXTENSIONr3 r ZQueryableAttributer
Úpropertyr ru r r r= Úgetattr)r\ rI ÚmpÚdescÚproprD rD rE r} k s*
|
ÿÿ
|
|
|
þÿz_GetColumns.__getattr__Nr rD rD rD rE r c s
|
r c C s t | j¡S r` )r Úinspectr+ )ÚtargetrD rD rE r r r c @ sB e Zd ZU dZded< ded< dddddZddd d
|
d Zd S ) Ú _GetTable©rI Úmetadatar' rI r! r c C s || _ || _d S r` r )r\ rI r rD rD rE r_ s z_GetTable.__init__r" r| c C s | j jt|| j S r` )r Útablesr rI r~ rD rD rE r} s z_GetTable.__getattr__Nr rD rD rD rE r s
|
|
r r )rI r r- c C s t |tr| g | ¡}t|S r` )r3 r4 rm r )rI r rD rD rE Ú_determine_container s
|
r c @ s e Zd ZU dZded< ded< ded< ded < d
|
ed < d ed < d"ddddd
|
dddZdddddZddddddZdd d d Zdd d d Z d!S )#Ú_class_resolver)r+ r ÚargÚfallbackÚ_dictÚ
|
_resolversÚ favor_tablesr. r+ úRelationshipProperty[Any]r úMapping[str, Any]r r' r rH r z Tuple[Callable[[str], Any], ...]r F)r+ r r r r c C s6 || _ || _|| _|| _t | j¡| _d| _|| _ d S )NrD )
|
r+ r r r r Z PopulateDictÚ _access_clsr r r )r\ r+ r r r r rD rD rE r_ ® s z_class_resolver.__init__r r| c
|
C s | j }t |¡}|j}|d k s"t|j}|j}| jrd||jkrH|j| S ||j krdt
|
|t |d|S ||krzt ||| S | js°||jkr|j| S ||j kr°t
|
|t |d|S d|krâ|t t|d krât t|d }| |¡S | jr| jD ] }||} | d k rð| S qð| j| S )Nr r/ )r+ r Zmanager_of_classÚregistryr
Z_class_registryr r r Z_schemasr r r r r5 r r r )
|
r\ rI r+ ÚmanagerZ decl_baser, r r Zresolvr rD rD rE r ¾ sB
|
|
|
|
|
|
|
ÿ ÿ
|
|
|
z_class_resolver._access_clsÚ Exceptionr )ru Úerrr- c C s t d|¡}|r^| d¡ d¡}t d| jj d| jd| jj d| d¡ d | d
|
¡|n t d | jj| j|| j
|
f ¡|d S ) Nz (.+)\[(.+)\]r ú'zWhen initializing mapper z, expression "relationship(z
)" seems to be using a generic class as the argument to relationship(); please state the generic argument using an annotation, e.g. "z : Mapped[r z['z']] = relationship()"zÅWhen initializing mapper %s, expression %r failed to locate a name (%r). If this is a class name, consider adding this relationship() to the %r class after both dependent classes have been defined.) ÚreÚmatchÚgroupÚstripr r= r rt r rI r+ )r\ ru r£ Z generic_matchZclsargrD rD rE Ú_raise_for_nameå s 2ÿø
|
üÿúz _class_resolver._raise_for_namez Union[Table, Type[Any], _ModNS]rb c
|
C sÒ | j }| j}d }z0| d¡D ] }|d kr2|| }q t||}q W n` tk
|
rn } z| ||¡ W 5 d }~X Y n` tk
|
r } z| |jd |¡ W 5 d }~X Y n.X t|t r²|j
|
S t rÊt|t t tfsÊt|S d S )Nr0 r )r r r8 r r6 r© rl Úargsr3 r r+ r Útyper" rw r
)r\ ru ÚdÚrvalrA r£ ÚnrD rD rE Ú _resolve_nameü s"
|
$
|
z _class_resolver._resolve_namec
|
C sh z.t | jt | j}t|tr&|jW S |W S W n4 tk
|
rb } z| |j d |¡ W 5 d }~X Y nX d S )Nr )
|
Úevalr Úglobalsr r3 r r+ rl r© rª )r\ Úxr® rD rD rE Ú__call__ s
|
|
z_class_resolver.__call__N)F)
|
rO r7 rP rR rs r_ r r© r¯ r³ rD rD rD rE r s
|
|
ú'r r Ú_fallback_dictr zmTuple[Callable[[str], Callable[[], Union[Type[Any], Table, _ModNS]]], Callable[[str, bool], _class_resolver]])r+ r r- c sx t d kr@dd l}ddlm} ddlm} t |j¡ ||d¡a dddd d
|
fd d }dd d fdd }||fS )Nr r )Úforeign)Úremote)rµ r¶ Fr' rH r )r r r- c s t t| |dS )N)r )r r´ )r r ©r+ r rD rE Ú resolve_arg3 s ÿz _resolver.<locals>.resolve_argz-Callable[[], Union[Type[Any], Table, _ModNS]])r r- c s t t| jS r` )r r´ r¯ )r r· rD rE Ú resolve_name8 s z _resolver.<locals>.resolve_name)F) r´ Ú
|
sqlalchemyÚ rµ r¶ r Z immutabledictÚ__dict__Úunion)r+ r rº rµ r¶ r¸ r¹ rD r· rE Ú _resolver! s ÿr¾ )?rQ Ú
|
__future__r r¥ Útypingr r r r r r r r
|
r r r r r r r r r rV r» r r Zdescriptor_propsr Z
|
propertiesr r r r r Z
|
sql.schemar Z util.typingr Z relationshipsr r! r" r# r' r« r( Úsetr& rs r; rG rN r% r4 r5 rw r Z _inspectsr r r r´ r¾ rD rD rD rE Ú<module> sl 6! EG ÿ
|
