
    e)                    l   U d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl	m
Z
 d d	l	mZ d d
l	mZ d dl	mZ d dl	mZ d dl	mZ d dl	mZ d dl	mZ d dl	mZ d dl	mZ d dl	mZ d dl	mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ ddlmZ  ej        d          Z de!d<    ej        d          Z"de!d<    ej        d          Z#de!d<    ej        d          Z$de!d<    ej        d           Z%de!d!<    ej        d"          Z&de!d"<    ej        d#          Z'de!d#<    ej        d$          Z(de!d$<    ej        d%          Z)de!d%<    ej        d&          Z*de!d&<    ej        d'          Z+de!d'<    ej        d(          Z,de!d(<    ej        d)          Z-de!d)<    ej        d*          Z.de!d*<    ej        d+          Z/de!d+<    ej        d,          Z0de!d,<    ej        d-          Z1de!d-<   d.Z2d/Z3d0Z4d1Z5i d2e d3d4hfd5e"h d6fd7e"h d6fd8e#d9d:hfd;e$h d<fd=e$h d<fd>e*h d?fd@e*h d?fdAe+h dBfdCe+h dBfdDe-dEdFhfdGe%dHhfdIe&e5hfdJe'h dKfdLe(h dMfdNe)dOdPhfdQe.h dRfe/dSdThfe,dUdVhfe0dWdXhfe1dYdZhfd[Z6d\e!d]<   d|dcZ7d}dfZ8d~dnZ9ddqZ:ddtZ;ddwZ<ddzZ=d{S )    )annotations)Dict)List)Optional)Set)Tuple)Union)ARG_POS)CallExpr)ClassDef)	Decorator)
Expression)FuncDef)
MemberExpr)NameExpr)OverloadedFuncDef)
SymbolNode)	TypeAlias)TypeInfo)SemanticAnalyzerPluginInterface)CallableType)get_proper_type)Instance)UnboundType   )utilCOLUMNintRELATIONSHIPREGISTRYCOLUMN_PROPERTY	TYPEENGNE
TYPEENGINEMAPPEDDECLARATIVE_BASEDECLARATIVE_METAMAPPED_DECORATORSYNONYM_PROPERTYCOMPOSITE_PROPERTYDECLARED_ATTRMAPPER_PROPERTYAS_DECLARATIVEAS_DECLARATIVE_BASEDECLARATIVE_MIXINQUERY_EXPRESSIONzbuiltins.objectzbuiltins.strzbuiltins.listzsqlalchemy.orm.base.MappedColumnzsqlalchemy.sql.schema.Columnzsqlalchemy.sql.ColumnRelationship>   sqlalchemy.orm.Relationship#sqlalchemy.orm.RelationshipProperty)sqlalchemy.orm.relationships.Relationship1sqlalchemy.orm.relationships.RelationshipPropertyRelationshipPropertyregistryz sqlalchemy.orm.decl_api.registryzsqlalchemy.orm.registryColumnProperty>   sqlalchemy.orm.ColumnProperty"sqlalchemy.orm.MappedSQLExpression(sqlalchemy.orm.properties.ColumnProperty-sqlalchemy.orm.properties.MappedSQLExpressionMappedSQLExpressionSynonym>   sqlalchemy.orm.Synonymsqlalchemy.orm.SynonymProperty'sqlalchemy.orm.descriptor_props.Synonym/sqlalchemy.orm.descriptor_props.SynonymPropertySynonymProperty	Composite>   sqlalchemy.orm.Composite sqlalchemy.orm.CompositeProperty)sqlalchemy.orm.descriptor_props.Composite1sqlalchemy.orm.descriptor_props.CompositePropertyCompositePropertyMapperPropertyz(sqlalchemy.orm.interfaces.MapperPropertyzsqlalchemy.orm.MapperProperty
TypeEnginez"sqlalchemy.sql.type_api.TypeEngineMappeddeclarative_base>   sqlalchemy.orm.declarative_base(sqlalchemy.orm.decl_api.declarative_base+sqlalchemy.ext.declarative.declarative_baseDeclarativeMeta>   sqlalchemy.orm.DeclarativeMeta'sqlalchemy.orm.decl_api.DeclarativeMeta*sqlalchemy.ext.declarative.DeclarativeMetamappedz'sqlalchemy.orm.decl_api.registry.mappedzsqlalchemy.orm.registry.mappedas_declarative>   sqlalchemy.orm.as_declarative&sqlalchemy.orm.decl_api.as_declarative)sqlalchemy.ext.declarative.as_declarativez4sqlalchemy.orm.decl_api.registry.as_declarative_basez+sqlalchemy.orm.registry.as_declarative_basez%sqlalchemy.orm.decl_api.declared_attrzsqlalchemy.orm.declared_attrz)sqlalchemy.orm.decl_api.declarative_mixinz sqlalchemy.orm.declarative_mixinzsqlalchemy.orm.query_expressionz1sqlalchemy.orm._orm_constructors.query_expression)as_declarative_basedeclared_attrdeclarative_mixinquery_expressionzDict[str, Tuple[int, Set[str]]]_lookupinfor   type_idreturnboolc                    | j         D ]-}t                              |j        d          \  }}||k    r n.dS |dS |j        |v S NNNF)mror^   getnamefullname)r_   r`   mrcheck_type_id	fullnamess        9lib/python3.11/site-packages/sqlalchemy/ext/mypy/names.pyhas_base_type_idrn      sa    h  #*;;rw#E#E yG##E $ uu;)##    rf   List[TypeInfo]c                    | D ]-}t                               |j        d          \  }}||k    r n.dS |dS |j        |v S rd   r^   rg   rh   ri   )rf   r`   rj   rk   rl   s        rm   
mro_has_idrs      s_      #*;;rw#E#E yG##E $ uu;)##ro   type_r   clsr   apir   Optional[int]c                V   |                     | j        |           }|t          |j        t                    rCt          |j        j                  }t          |t                    rt          |j	                  S n.t          |j        t                    rt          |j                  S d S N)lookup_qualifiedrh   
isinstancenoder   r   targetr   type_id_for_named_nodetyper   )rt   ru   rv   symtarget_types        rm   type_id_for_unbound_typer      s     

uz5
1
1C
ch	** 	4)#(/::K+x00 @-k.>???@(++ 	4)#(3334ro   calleer   c                Z   t          | t          t          f          rt          | j        t                    rt          | j        j        t                    r| j        j        j        rut          | j        j        j        t                    rQt          | j        j        j        j
                  }t          |t                    rt          |j        j                  S d S t          | j        t                    r| j        j        r| j        j        j        rut          | j        j        j        t                    rQt          | j        j        j        j
                  }t          |t                    rt          |j        j                  S d S t          | j        t                    ry| j        j        rkt          | j        j        t                    rLt          | j        j        j
                  }t          |t                    rt          |j        j                  S d S t          | j        t                     rHt          | j        j                  }t          |t                    rt          |j        j                  S n)t          | j        t$                    rt'          |           S d S ry   )r{   r   r   r|   r   funcr   r   r   r   ret_typer   type_id_for_fullnameri   r   implr   r}   r   r~   )r   r   r   s      rm   type_id_for_calleer     sH   &:x011 '2fk9-- &	2*Kg3
 3
 &	2 {$ H %|* * H +6;+;+@+IJJh11 H/0FGGG4%677 	2 HK$)H v{/4lCCH
 +6;+;+@+IJJh11 H/0FGGG4W-- 	2{ HJv{/?$N$N H*6;+;+DEEh11 H/0FGGG4Y// 	2)&+*<==K+x00 G+K,<,EFFFGX.. 	2)&1114ro   r|   'Union[NameExpr, MemberExpr, SymbolNode]c                n    t                               | j        d          \  }}||d S | j        |v r|S d S )Nre   rr   )r|   r`   rl   s      rm   r~   r~   2  sD     !TY==GY)+t	)	#	#tro   ri   strc                    |                      d          }|d         }t                              |d          \  }}||d S | |v r|S d S )N.re   )splitr^   rg   )ri   tokens	immediater`   rl   s        rm   r   r   ?  sX    ^^C  Fr
I Y==GY)+t	Y		tro   exprr   c                    t          d          }t          |_        t          |d          }t	          || gt
          gdg          S )N__sa_Mapped_empty_constructorarg1)r   NAMED_TYPE_SQLA_MAPPEDri   r   r   r
   )r   column_descriptormember_exprs      rm   expr_to_mapped_constructorr   M  sN     //!7.0DEEK					  ro   N)r_   r   r`   r   ra   rb   )rf   rp   r`   r   ra   rb   )rt   r   ru   r   rv   r   ra   rw   )r   r   ra   rw   )r|   r   ra   rw   )ri   r   ra   rw   )r   r   ra   r   )>
__future__r   typingr   r   r   r   r   r	   
mypy.nodesr
   r   r   r   r   r   r   r   r   r   r   r   mypy.pluginr   
mypy.typesr   r   r   r    r   symbolr   __annotations__r   r    r!   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   NAMED_TYPE_BUILTINS_OBJECTNAMED_TYPE_BUILTINS_STRNAMED_TYPE_BUILTINS_LISTr   r^   rn   rs   r   r   r~   r   r    ro   rm   <module>r      s8   # " " " " " "                                                                   ! ! ! ! ! !       ! ! ! ! ! !       ( ( ( ( ( ( ! ! ! ! ! !                   7 7 7 7 7 7 # # # # # # & & & & & &       " " " " " "      dk(## # # # #DK// / / / /J'' ' ' ' '"t{#455 5 5 5 5$+k**
 * * * *dk(## # # # ##$677  7 7 7 7#$677  7 7 7 7#$677  7 7 7 7#$677  7 7 7 7%$+&:;;  ; ; ; ; T[11 1 1 1 1"t{#455 5 5 5 5!dk"233 3 3 3 3&4;'<==  = = = =$%899  9 9 9 9#$677  7 7 7 7 / ( * 5 [,*#	
[, 	
 	
 	
[," 	
 	
 	
#[,4 .%	
5[,B 	
 	
 	
C[,T 	
 	
 	
U[,f 	
 	
 	
g[,x 	
 	
 	
y[,J 	
 	
 	
K[,\ 	
 	
 	
][,n 6+	
o[,| : DEF}[,~ v./0[,@ 	
 	
 	
A[,P 	
 	
 	
Q[,` 5,	
a[,n 	
 	
 	
o[,@ 	B9	
 	3*	
 	7.	
 	-?	
i[, [, [, [ [ [ [|$ $ $ $$ $ $ $   ) ) ) )X
 
 
 
   	 	 	 	 	 	ro   