a
    8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.Relationshipz1sqlalchemy.orm.relationships.RelationshipPropertyz2sqlalchemy.orm.relationships._RelationshipDeclaredz#sqlalchemy.orm.RelationshipPropertyz)sqlalchemy.orm.relationships.Relationshipz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.ColumnPropertyz-sqlalchemy.orm.properties.MappedSQLExpression>   zsqlalchemy.orm.SynonymPropertyz'sqlalchemy.orm.descriptor_props.Synonymz/sqlalchemy.orm.descriptor_props.SynonymPropertyzsqlalchemy.orm.Synonym>   z sqlalchemy.orm.CompositePropertyz)sqlalchemy.orm.descriptor_props.Compositezsqlalchemy.orm.Compositez1sqlalchemy.orm.descriptor_props.CompositePropertyz(sqlalchemy.orm.interfaces.MapperPropertyzsqlalchemy.orm.MapperPropertyz"sqlalchemy.sql.type_api.TypeEngine>   z+sqlalchemy.ext.declarative.declarative_basez(sqlalchemy.orm.decl_api.declarative_basezsqlalchemy.orm.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.orm.decl_api.as_declarativezsqlalchemy.orm.as_declarativez)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)Z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)mror/   getnamefullname)r1   r2   mrcheck_type_id	fullnames r=   v/var/www/vosh/data/www/fastworkle.ru/webshop-tgbot-v.1.0/venv/lib/python3.9/site-packages/sqlalchemy/ext/mypy/names.pyhas_base_type_id   s    
r?   zList[TypeInfo])r6   r2   r3   c                 C  sB   | D ]"}t |jd\}}||kr q,qdS |d u r8dS |j|v S r4   r/   r7   r8   r9   )r6   r2   r:   r;   r<   r=   r=   r>   
mro_has_id   s    rA   r   r   r   zOptional[int])type_clsapir3   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_qualifiedr8   
isinstancenoder   r   targetr   type_id_for_named_nodetyper   )rB   rC   rD   symtarget_typer=   r=   r>   type_id_for_unbound_type   s    

rM   r   )calleer3   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 rE   )rF   r   r   rG   r   funcr   rJ   r   r   ret_typer   type_id_for_fullnamer9   r   implr   rH   r   rI   )rN   rP   rL   r=   r=   r>   type_id_for_callee   sF    


rS   z'Union[NameExpr, MemberExpr, SymbolNode])rG   r3   c                 C  s<   t | jd\}}|d u s"|d u r&d S | j|v r4|S d S d S )Nr5   r@   )rG   r2   r<   r=   r=   r>   rI   +  s    
rI   str)r9   r3   c                 C  sJ   |  d}|d }t|d\}}|d u s2|d u r6d S | |v rB|S d S d S )N.r5   )splitr/   r7   )r9   tokensZ	immediater2   r<   r=   r=   r>   rQ   8  s    
rQ   r
   )exprr3   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_MAPPEDr9   r   r
   r	   )rY   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_LISTrZ   Z_RelFullNamesr/   r?   rA   rM   rS   rI   rQ   r[   r=   r=   r=   r>   <module>   s   






			  ,