o
    g(                     @  s  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dZ de!d< edZ"de!d< edZ#de!d< edZ$de!d< ed Z%de!d!< ed"Z&de!d"< ed#Z'de!d#< ed$Z(de!d$< ed%Z)de!d%< ed&Z*de!d&< ed'Z+de!d'< ed(Z,de!d(< ed)Z-de!d)< ed*Z.de!d*< ed+Z/de!d+< ed,Z0de!d,< ed-Z1de!d-< d.Z2d/Z3d0Z4d1Z5h d2Z6i d3e d4d5hfd6e"e6fd7e"e6fd8e"e6fd9e#d:d;hfd<e$h d=fd>e$h d=fd?e*h d@fdAe*h d@fdBe+h dCfdDe+h dCfdEe-dFdGhfdHe%dIhfdJe&e5hfdKe'h dLfdMe(h dNfdOe)dPdQhfe.h dRfe/dSdThfe,dUdVhfe0dWdXhfe1dYdZhfd[Z7d\e!d]< ddcddZ8ddgdhZ9ddpdqZ:ddtduZ;ddxdyZ<dd|d}Z=dddZ>dS )    )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>   sqlalchemy.orm.Relationship#sqlalchemy.orm.RelationshipProperty)sqlalchemy.orm.relationships.Relationship1sqlalchemy.orm.relationships.RelationshipProperty2sqlalchemy.orm.relationships._RelationshipDeclaredColumnzsqlalchemy.sql.schema.Columnzsqlalchemy.sql.ColumnRelationshipRelationshipProperty_RelationshipDeclared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>   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as_declarative_basedeclared_attrdeclarative_mixinquery_expressionzDict[str, Tuple[int, Set[str]]]_lookupinfor   type_idreturnboolc                 C  sD   | j D ]}t|jd\}}||kr nqdS |d u rdS |j|v S NNNF)mror_   getnamefullname)r`   ra   mrcheck_type_id	fullnames rm   S/var/www/html/api-tag/env/lib/python3.10/site-packages/sqlalchemy/ext/mypy/names.pyhas_base_type_id   s   

ro   rf   List[TypeInfo]c                 C  sB   | D ]}t |jd\}}||kr nqdS |d u rdS |j|v S rd   r_   rg   rh   ri   )rf   ra   rj   rk   rl   rm   rm   rn   
mro_has_id   s   
rr   type_r   clsr   apir   Optional[int]c                 C  s`   | | j| }|d ur.t|jtr#t|jj}t|tr!t|j	S d S t|jt
r.t|jS d S N)lookup_qualifiedrh   
isinstancenoder   r   targetr   type_id_for_named_nodetyper   )rs   rt   ru   symtarget_typerm   rm   rn   type_id_for_unbound_type   s   


r   calleer   c                 C  sX  t | ttfrt | jtr6t | jjtr6| jjjr4t | jjjtr4t	| jjjj
}t |tr4t|jjS d S t | jtrb| jjr`| jjjr`t | jjjtr`t	| jjjj
}t |tr`t|jjS d S t | jtr| jjrt | jjtrt	| jjj
}t |trt|jjS d S t | jtrt	| jj}t |trt|jjS d S t | jtrt| S d S rw   )ry   r   r   rz   r   funcr   r}   r   r   ret_typer   type_id_for_fullnameri   r   implr   r{   r   r|   )r   r   r   rm   rm   rn   type_id_for_callee   sH   




r   rz   'Union[NameExpr, MemberExpr, SymbolNode]c                 C  s8   t | jd\}}|d u s|d u rd S | j|v r|S d S )Nre   rq   )rz   ra   rl   rm   rm   rn   r|   +  s   
r|   ri   strc                 C  sF   |  d}|d }t|d\}}|d u s|d u rd S | |v r!|S d S )N.re   )splitr_   rg   )ri   tokens	immediatera   rl   rm   rm   rn   r   8  s   
r   exprr
   c                 C  s,   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_exprrm   rm   rn   expr_to_mapped_constructorF  s   
r   N)r`   r   ra   r   rb   rc   )rf   rp   ra   r   rb   rc   )rs   r   rt   r   ru   r   rb   rv   )r   r   rb   rv   )rz   r   rb   rv   )ri   r   rb   rv   )r   r   rb   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   _RelFullNamesr_   ro   rr   r   r   r|   r   r   rm   rm   rm   rn   <module>   s>  	
$-6?H
O
PQYai	  




,
