U
|
¸ý°d[$ ã @ s d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d lm
|
Z
|
d d
|
lm Z d d lm Z d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dl 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'eeef d(Z.G d)d* d*Z/d+d,d-d.d/Z0d0d+d1d2d3d4d5Z1d0d+d6d7d8d9Z2d0d+d6d7d:d;Z3d0d2d<d=d>Z4d0d2d<d?d@Z5d0d,d<dAdBZ6d0d,d<dCdDZ7d0d2d<dEdFZ8d0d,d<dGdHZ9d0dIdJdKdLdMZ:d0dNd2dOdPdQZ;dId+dRd2dSdTdUZ<dVd+d+d+d2dWdXdYZ=e dZd[d\d+d2d]d^d_d`Z>e d\d+dadbd^dcd`Z>dd[d\d+ddd6d^ded`Z>dfdgdhdidjZ?dkdldmdndoZ@dIdpdpdqdrdsZAdtdIdudvdwdxZBdS )yé )Ú annotationsN)ÚAny)ÚIterable)ÚIterator)ÚList)ÚOptional)Úoverload)ÚTuple)ÚType)ÚTypeVar)ÚUnion)ÚCallExpr)ÚClassDef)ÚCLASSDEF_NO_INFO)ÚContext)Ú
|
Expression)ÚFuncDef)ÚIfStmt)ÚJsonDict)Ú
|
MemberExpr)ÚNameExpr)Ú Statement)ÚSymbolTableNode)Ú TypeAlias)ÚTypeInfo)ÚClassDefContext)ÚDynamicClassDefContext)Ú SemanticAnalyzerPluginInterface)Údeserialize_and_fixup_type)Úmap_type_from_supertype)Ú CallableType)Úget_proper_type)ÚInstance)ÚNoneType)Ú TypeVarType)Ú UnboundType)Ú UnionTypeÚ _TArgType)Úboundc @ sZ e Zd ZdddddddddZd d
|
d d Zddd ddZedd dd dddZdS )ÚSQLAlchemyAttributeÚstrÚintzOptional[Type]r ÚNone)ÚnameÚlineÚcolumnÚtypÚinfoÚreturnc C s" || _ || _|| _|| _|| _d S ©N)r- r. r/ Útyper1 )Úselfr- r. r/ r0 r1 © r6 úOd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/ext/mypy/util.pyÚ__init__7 s
|
z SQLAlchemyAttribute.__init__r )r2 c C s$ | j s
|
t| j| j| j| j ¡ dS )N)r- r. r/ r4 )r4 ÚAssertionErrorr- r. r/ Ú serialize)r5 r6 r6 r7 r: E s
|
üz SQLAlchemyAttribute.serialize)Úsub_typer2 c C s& t | jtsdS t| j|| j| _dS )zxExpands type vars in the context of a subtype when an attribute is
|
inherited from a generic super type.
|
N)Ú
|
isinstancer4 r$ r r1 )r5 r; r6 r6 r7 Úexpand_typevar_from_subtypeN s z/SQLAlchemyAttribute.expand_typevar_from_subtyper )r1 ÚdataÚapir2 c C s, | ¡ }t| d¡|}| f ||d|S )Nr4 )r0 r1 )Úcopyr Úpop)Úclsr1 r> r? r0 r6 r6 r7 Ú deserializeW s z SQLAlchemyAttribute.deserializeN)Ú__name__Ú
|
__module__Ú __qualname__r8 r: r= Ú classmethodrC r6 r6 r6 r7 r) 6 s
|
r) r* Úbool)r- r2 c C s t t d| ¡S )Nz ^__.+?__$)rH ÚreÚmatch)r- r6 r6 r7 Úname_is_dunderc s rK r r r, )r1 Úkeyr> r2 c C s || j di ¡|< d S ©NZ
|
sqlalchemy©ÚmetadataÚ
|
setdefault)r1 rL r> r6 r6 r7 Ú_set_info_metadatag s rQ z Optional[Any])r1 rL r2 c C s | j di ¡ |d ¡S rM )rO Úget)r1 rL r6 r6 r7 Ú_get_info_metadatak s rS c C s0 | j r,| j D ] }t||}|d k r | S q d S r3 )ÚmrorS )r1 rL ÚbaserO r6 r6 r7 Ú_get_info_mro_metadatao s
|
|
|
rV )r1 r2 c C s | j di ¡ d S rM rN ©r1 r6 r6 r7 Úestablish_as_sqlalchemyx s rX c C s t | dd d S ©NÚis_baseT©rQ rW r6 r6 r7 Ú set_is_base| s r\ c C s t | d}|dkS rY ©rS ©r1 rZ r6 r6 r7 Ú get_is_base s
|
r_ c C s t | d}|dkS rY )rV r^ r6 r6 r7 Úhas_declarative_base
s
|
r` c C s t | dd d S ©NZ has_tableTr[ rW r6 r6 r7 Ú set_has_table s rb c C s t | d}|dkS ra r] r^ r6 r6 r7 Ú get_has_table s
|
rc r z#Optional[List[SQLAlchemyAttribute]])r1 r? r2 c C sJ t | d}|d krd S g }|D ]&}t | ||¡}| | ¡ | |¡ q |S )NÚmapped_attributes)rS r) rC r= Úappend)r1 r? rd Ú
|
attributesr> Úattrr6 r6 r7 Úget_mapped_attributes s ÿ
|
rh zList[SQLAlchemyAttribute])r1 rf r2 c C s t | ddd |D d S )Nrd c S s g | ] }| ¡ qS r6 )r: )Ú.0Ú attributer6 r6 r7 Ú
|
<listcomp>¬ s z)set_mapped_attributes.<locals>.<listcomp>r[ )r1 rf r6 r6 r7 Úset_mapped_attributes¦ s
|
ýrl r )r? ÚmsgÚctxr2 c C s d| }| ||¡S )Nz[SQLAlchemy Mypy plugin] %s)Úfail)r? rm rn r6 r6 r7 ro ° s ro z.Union[ClassDefContext, DynamicClassDefContext])rn ÚmoduleÚ symbol_nameÚasnamer2 c C s8 | j j| j j j}||kr4| j j| j| }|||< d S r3 )r? ÚmodulesZ
|
cur_mod_idÚnames)rn rp rq rr Úmodule_globalsZ
|
lookup_symr6 r6 r7 Ú
|
add_globalµ s ÿrv .)Ú
|
expr_typesr z#Optional[Union[CallExpr, NameExpr]])Úcallexprr- rw r2 c C s d S r3 r6 ©rx r- rw r6 r6 r7 Úget_callexpr_kwargÅ s rz z!Tuple[TypingType[_TArgType], ...]zOptional[_TArgType]c C s d S r3 r6 ry r6 r6 r7 rz Ì s z%Optional[Tuple[TypingType[Any], ...]]c C sT z| j |¡}W n tk
|
r& Y d S X | j| }t||d k rB|nttfrP|S d S r3 )Ú arg_namesÚindexÚ
|
ValueErrorÚargsr< r r )rx r- rw Zarg_idxÚkwargr6 r6 r7 rz Ö s
|
ÿzIterable[Statement]zIterator[Statement])Ústmtsr2 c c sR | D ]H}t |trFt |jd trF|jd jdkrF|jd jE d H q|V qd S )Nr ztyping.TYPE_CHECKING)r< r Úexprr ÚfullnameÚbody)r Zstmtr6 r6 r7 Úflatten_typecheckingê s ÿþýr r z#Optional[Union[Instance, TypeInfo]])Úcalleer2 c C s t | ttfrt | jtrP| jjrLt | jjtrLt| jjj}t |t rL|S d S t | jt
|
rxt| jj }t |t r|S nt | jt r| jS d S r3 ) r< r r Únoder r4 r r! Úret_typer" r Útargetr )r
r Z target_typer6 r6 r7 Útype_for_calleeö s
|
|
r r
|
)r? r0 r2 c s t |ts|S |jdkr>t t fdd|jD t g S |j|¡}|dk rt |trt |j t
|
r|j }t | fdd|jD S |S dS )zÔTake the UnboundType that we seem to get as the ret_type from a FuncDef
|
and convert it into an Instance/TypeInfo kind of structure that seems
|
to work as the left-hand type of an AssignmentStatement.
|
|
r c s g | ]}t |qS r6 )Úunbound_to_instance)ri Ztyp_arg©r? r6 r7 rk s z'unbound_to_instance.<locals>.<listcomp>Nc s$ g | ] }t |tr t |n|qS r6 )r< r% r )ri Úargr r6 r7 rk . s þÿ ) r< r% r- r r& r~ r# Úlookup_qualifiedr r r r" )r? r0 r Z
|
bound_typer6 r r7 r s4
|
|
ÿÿþÿþ
|
ý
|
üþ
|
r r zOptional[TypeInfo])rB r? r2 c C sD | j tkr>| | j| ¡}|d kr$d S |r4t|jts8t|jS | j S r3 )r1 r r r- r< r r r9 )rB r? Úsymr6 r6 r7 Ú info_for_cls9 s
|
r )CÚ
|
__future__r rI Útypingr r r r r r r r
|
Z
|
TypingTyper r Z
|
mypy.nodesr r r r r r r r r r r r r r Z mypy.pluginr r r Zmypy.plugins.commonr Z mypy.typeopsr Z
|
mypy.typesr r! r" r# r$ r% r& r' r) rK rQ rS rV rX r\ r_ r` rb rc rh rl ro rv rz r r r r r6 r6 r6 r7 Ú<module> s -
|
ÿ ü 0
|
