U
|
¸ý°dI* ã @ sð d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z 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 dlm Z d dlm Z d dlm Z d dlm Z d dlm Z d d lm!Z! d d l"m#Z# d d l"m$Z$ d d l%m&Z& d d!l%m'Z' d"d#d$d%d&d'd(d)Z(d"d#d%d&d*d+d,Z)d#d-d.d/d/d&d0d1d2Z*d"d#d%d&d*d3d4Z+d#d"d5d5d&d6d7d8Z,d9S ):é )Ú annotations)ÚList)ÚOptional)ÚUnion)Ú ARG_NAMED_OPT)ÚArgument)ÚAssignmentStmt)ÚCallExpr)ÚClassDef)ÚMDEF)Ú
|
MemberExpr)ÚNameExpr)ÚRefExpr)ÚStrExpr)ÚSymbolTableNode)ÚTempNode)ÚTypeInfo)ÚVar)Ú SemanticAnalyzerPluginInterface)Úadd_method_to_class)ÚAnyType)Úget_proper_type)ÚInstance)ÚNoneTyp)Ú
|
ProperType)Ú TypeOfAny)Ú UnboundType)Ú UnionTypeé )Úinfer)Úutil)Úexpr_to_mapped_constructor)ÚNAMED_TYPE_SQLA_MAPPEDr
|
r zUnion[NameExpr, StrExpr]z List[util.SQLAlchemyAttribute]ÚNone)ÚclsÚapiÚitemÚ
|
attributesÚreturnc C sð t |tr|j}nt |tr$|j}nd S | jjD ]2}t |tr0t |jd tr0|jd j|kr0 q|q0t
|
|d| | ¡ d S |j d krt
|
|d|¡ d S t |j }t |t ttfs¶t| t j||j|j|| jd¡ t|||jd |d d S )Nr z Can't find mapped attribute zBStatement linked from _mypy_mapped_attrs has no typing information)ÚnameÚlineÚcolumnÚtypÚinfo)Ú
|
isinstancer r) r ÚvalueÚdefsÚbodyr Úlvaluesr ZfailÚtyper r r r ÚAssertionErrorÚappendZSQLAlchemyAttributer* r+ r- Ú apply_type_to_mapped_statement)r$ r% r& r' r) ÚstmtÚleft_hand_explicit_type© r9 úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/ext/mypy/apply.pyÚapply_mypy_mapped_attr- sV
|
|
ÿþý
|
ü
|
ÿûÿ
|
ÿr; )r$ r% r' r( c
|
C sÌ dd |D }d}| j jD ]}t|trt|jd tr|jd j|krt|jd jtr|jd j}||jd j j }t
|
|j }t|t rxt|t sxt|j t rxt|j jtrxt|j jjtrx|j jjjdk rx|j jjjjtkrx|j jjdkrxt|j jd t rxt|j jd jtrxt |||||j jd j¡} | dk rxt| t sx| }|||jd j _ d}t|j tr|j j jtkr|dk s¢t| t|g¡|_ q|rÈt | j |¡ dS )zlFor multiple class passes, re-apply our left-hand side types as mypy
|
seems to reset them in place.
|
|
c S s i | ] }|j |qS r9 )r) ©Ú.0Úattrr9 r9 r: Ú
|
<dictcomp>k s z4re_apply_declarative_assignments.<locals>.<dictcomp>Fr NZ_empty_constructorT) r0 r1 r. r r2 r r) Únoder r3 r r Úrvaluer Zcalleer ÚexprÚfullnamer" Úargsr r Z#infer_type_from_right_hand_nameexprr r4 Ú
|
named_typer Zset_mapped_attributesr- )
|
r$ r% r' Zmapped_attr_lookupZupdate_cls_metadatar7 Ú left_nodeÚpython_type_for_typeZleft_node_proper_typeZnew_python_type_for_typer9 r9 r: Ú re_apply_declarative_assignmentsb s ÿþýü
|
ÿ
|
ÿþ
|
ü ûúù ÿø
|
ö õ ô ûÿ
|
ÿþ
|
ÿ
|
ÿ þ ÿrH r r zOptional[ProperType])r% r7 Úlvaluer8 rG r( c C s |j }t|tst|dk r4d|_| t|g¡|_n*d|_| t|dkrTtt j
|
gn|g¡|_t |j |_ |jdk r|dk r||_dS )aµ Apply the Mapped[<type>] annotation and right hand object to a
|
declarative assignment statement.
|
|
This converts a Python declarative class statement such as::
|
|
class User(Base):
|
# ...
|
|
attrname = Column(Integer)
|
|
To one that describes the final Python behavior to Mypy::
|
|
class User(Base):
|
# ...
|
|
attrname : Mapped[Optional[int]] = <meaningless temp node>
|
|
NF) r@ r. r r4 Zis_inferred_defrE r" r3 r r Ú special_formr! rA )r% r7 rI r8 rG rF r9 r9 r: r6 ¹ s$ ÿÿü r6 c C s& t | |¡}|dkrdS t |¡}d|jkrè|sèdd |D }|jdd
D ]@}d|jkr\qLt ||¡}|dkrrqL|D ]}| |j|j ¡ qvqLg } |
|
¡ D ]:\}
|
} | dkr´t t j } | tt|
|
| | t| td¡ qt|| d| t d |jkrt |¡rt|| d
|
d |s"t|| d d dS ) zCApply __init__, __table__ and other attributes to the mapped class.NÚ__init__c S s i | ]}|j |jqS r9 )r) r3 r< r9 r9 r: r?
|
s z1add_additional_orm_attributes.<locals>.<dictcomp>r éÿÿÿÿZ
|
sqlalchemy)ÚvariableZtype_annotationZ initializerÚkindZ __table__zsqlalchemy.sql.schema.Tablez sqlalchemy.orm.mapper.MapperZ
|
__mapper__)r Z info_for_clsZ get_is_baseÚnamesÚmroÚmetadataZget_mapped_attributesÚ
|
setdefaultr) r3 Úitemsr r rJ r5 r r r r r r Z get_has_tableÚ _apply_placeholder_attr_to_class) r$ r% r' r- Zis_baseZmapped_attr_namesÚbaseZbase_cls_attributesr> Ú argumentsr) r, r9 r9 r: Ú add_additional_orm_attributesû sR
|
|
|
üÿ ÿ ÿrW Ústr)r% r$ Úqualified_nameÚattrnamer( c C sr | |¡}|r,t|jts tt|jg }n
|
ttj}t |}|j
|
d | |_ |j |_ ||_ tt||j j|< d S )NÚ.)Z lookup_fully_qualified_or_noner. r@ r r4 r r r rJ r rC Z _fullnamer- r3 r r rO )r% r$ rY rZ ÚsymÚtype_Úvarr9 r9 r: rT 0 s
|
|
rT N)-Ú
|
__future__r Útypingr r r Z
|
mypy.nodesr r r r r
|
r r r r r r r r r Z mypy.pluginr Zmypy.plugins.commonr Z
|
mypy.typesr r r r r r r r Ú r r rO r! r" r; rH r6 rW rT r9 r9 r9 r: Ú<module> sH 5WB5
|
