a
    [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Z6e d3d4hfe"e6fe"e6fe"e6fe#d5d6hfe$h d7fe$h d7fe*h d8fe*h d8fe+h d9fe+h d9fe-d:d;hfe%d<hfe&e5hfe'h d=fe(h d>fe)d?d@hfe.h dAfe/dBdChfe,dDdEhfe0dFdGhfe1dHdIhfdJZ7dKe!dL< dMddNdOdPdQZ8dRddNdSdTdUZ9dVdWdXdYdZd[d\Z:d]dYd^d_d`Z;dadYdbdcddZ<dedYdfdgdhZ=d]didjdkdlZ>dmS )n    )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_PROPERTYZ	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>   z#sqlalchemy.orm.RelationshipPropertyz1sqlalchemy.orm.relationships.RelationshipPropertyzsqlalchemy.orm.Relationshipz)sqlalchemy.orm.relationships.Relationshipz2sqlalchemy.orm.relationships._RelationshipDeclaredzsqlalchemy.sql.schema.Columnzsqlalchemy.sql.Columnz sqlalchemy.orm.decl_api.registryzsqlalchemy.orm.registry>   z"sqlalchemy.orm.MappedSQLExpressionz(sqlalchemy.orm.properties.ColumnPropertyz-sqlalchemy.orm.properties.MappedSQLExpressionzsqlalchemy.orm.ColumnProperty>   zsqlalchemy.orm.SynonymPropertyz'sqlalchemy.orm.descriptor_props.Synonymz/sqlalchemy.orm.descriptor_props.SynonymPropertyzsqlalchemy.orm.Synonym>   z)sqlalchemy.orm.descriptor_props.Compositez1sqlalchemy.orm.descriptor_props.CompositePropertyzsqlalchemy.orm.Compositez sqlalchemy.orm.CompositePropertyz(sqlalchemy.orm.interfaces.MapperPropertyzsqlalchemy.orm.MapperPropertyz"sqlalchemy.sql.type_api.TypeEngine>   z+sqlalchemy.ext.declarative.declarative_basezsqlalchemy.orm.declarative_basez(sqlalchemy.orm.decl_api.declarative_base>   z*sqlalchemy.ext.declarative.DeclarativeMetaz'sqlalchemy.orm.decl_api.DeclarativeMetazsqlalchemy.orm.DeclarativeMetaz'sqlalchemy.orm.decl_api.registry.mappedzsqlalchemy.orm.registry.mapped>   z)sqlalchemy.ext.declarative.as_declarativezsqlalchemy.orm.as_declarativez&sqlalchemy.orm.decl_api.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)ColumnZRelationshipZRelationshipPropertyZ_RelationshipDeclaredregistryZColumnPropertyZMappedSQLExpressionZSynonymZSynonymPropertyZ	CompositeZCompositePropertyZMapperPropertyZ
TypeEngineZMappedZdeclarative_baseZDeclarativeMetaZmappedZas_declarativeZas_declarative_baseZdeclared_attrZdeclarative_mixinZquery_expressionzDict[str, Tuple[int, Set[str]]]_lookupr   bool)infotype_idreturnc                 C  sD   | j D ]"}t|jd\}}||kr q.qdS |d u r:dS |j|v S NNNF)mror0   getnamefullname)r2   r3   mrcheck_type_id	fullnames r>   g/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/sqlalchemy/ext/mypy/names.pyhas_base_type_id   s    
r@   zList[TypeInfo])r7   r3   r4   c                 C  sB   | D ]"}t |jd\}}||kr q,qdS |d u r8dS |j|v S r5   r0   r8   r9   r:   )r7   r3   r;   r<   r=   r>   r>   r?   
mro_has_id   s    rB   r   r   r   zOptional[int])type_clsapir4   c                 C  s^   | | j| }|d urZt|jtrDt|jj}t|trZt|j	S nt|jt
rZt|jS d S N)Zlookup_qualifiedr9   
isinstancenoder   r   targetr   type_id_for_named_nodetyper   )rC   rD   rE   symtarget_typer>   r>   r?   type_id_for_unbound_type   s    

rN   r   )calleer4   c                 C  sf  t | ttfrbt | jtrnt | jjtrn| jjjrjt | jjjtrjt	| jjjj
}t |trjt|jjS d S t | jtr| jjr| jjjrt | jjjtrt	| jjjj
}t |trt|jjS d S t | jtr| jjrt | jjtrt	| jjj
}t |trt|jjS d S t | jtrLt	| jj}t |trbt|jjS nt | jtrbt| S d S rF   )rG   r   r   rH   r   funcr   rK   r   r   ret_typer   type_id_for_fullnamer:   r   implr   rI   r   rJ   )rO   rQ   rM   r>   r>   r?   type_id_for_callee   sF    


rT   z'Union[NameExpr, MemberExpr, SymbolNode])rH   r4   c                 C  s<   t | jd\}}|d u s"|d u r&d S | j|v r4|S d S d S )Nr6   rA   )rH   r3   r=   r>   r>   r?   rJ   +  s    
rJ   str)r:   r4   c                 C  sJ   |  d}|d }t|d\}}|d u s2|d u r6d S | |v rB|S d S d S )N.r6   )splitr0   r8   )r:   tokensZ	immediater3   r=   r>   r>   r?   rR   8  s    
rR   r
   )exprr4   c                 C  s,   t d}t|_t|d}t|| gtgdgS )NZ__sa_MappedZ_empty_constructorZarg1)r   NAMED_TYPE_SQLA_MAPPEDr:   r   r
   r	   )rZ   Zcolumn_descriptorZmember_exprr>   r>   r?   expr_to_mapped_constructorF  s    
r\   N)?
__future__r   typingr   r   r   r   r   r   Z
mypy.nodesr	   r
   r   r   r   r   r   r   r   r   r   r   Zmypy.pluginr   Z
mypy.typesr   r   r   r    r   symbolr   __annotations__r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   ZNAMED_TYPE_BUILTINS_OBJECTZNAMED_TYPE_BUILTINS_STRZNAMED_TYPE_BUILTINS_LISTr[   Z_RelFullNamesr0   r@   rB   rN   rT   rJ   rR   r\   r>   r>   r>   r?   <module>   s   






			  ,