U
|
¸ý°dêE ã @ 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 dlmZ d dlm Z d dlm Z d dlm Z d dlm Z d d lm Z d d lm!Z! d d lm"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/d0d1d2d3Z*d&d'd4d/d0d5d6d7Z+d&d'd8d/d0d5d9d:Z,d&d'd0d;d<d=Z-d>S )@é )Ú annotations)ÚList)ÚOptional)ÚUnion)ÚAssignmentStmt)ÚCallExpr)ÚClassDef)Ú Decorator)Ú
|
LambdaExpr)ÚListExpr)Ú
|
MemberExpr)ÚNameExpr)ÚPlaceholderNode)ÚRefExpr)ÚStrExpr)Ú
|
SymbolNode)ÚSymbolTableNode)ÚTempNode)ÚTypeInfo)ÚVar)Ú SemanticAnalyzerPluginInterface)ÚAnyType)Ú CallableType)Úget_proper_type)ÚInstance)ÚNoneType)Ú
|
ProperType)ÚType)Ú TypeOfAny)Ú UnboundType)Ú UnionTypeé )Úapply)Úinfer)Únames)ÚutilFr r Úboolz(Optional[List[util.SQLAlchemyAttribute]])ÚclsÚapiÚ is_mixin_scanÚreturnc C s t | |¡}|d krd S | j d¡r(d S t ||¡}t |¡ |d k r\|sXt | ||¡ |S g }| jj s|j
|
¡ D ]\}}t | |||| qrnFt | jj ¡D ]6}t|trºt| ||| qt|trt| ||| qt| | |sðt | ||¡ t ||¡ |S )NÚbuiltins)r% Ú info_for_clsÚfullnameÚ
|
startswithZget_mapped_attributesZestablish_as_sqlalchemyr" Z re_apply_declarative_assignmentsÚdefsÚbodyr$ ÚitemsÚ_scan_symbol_table_entryZflatten_typecheckingÚ
|
isinstancer Ú!_scan_declarative_assignment_stmtr Ú _scan_declarative_decorator_stmtÚ_scan_for_mapped_basesZ add_additional_orm_attributesZset_mapped_attributes)r' r( r) ÚinfoZmapped_attributesZsym_nameÚsymÚstmt© r: úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/ext/mypy/decl_class.pyÚ,scan_declarative_assignments_and_apply_types0 sT
|
þ
|
ÿ
|
ÿ
|
ÿ
|
r< Ústrr z List[util.SQLAlchemyAttribute]ÚNone)r' r( ÚnameÚvalueÚ
|
attributesr* c C s t |j}t|tsdS d}t |j¡}d}|tjtjtjtj tj
|
tj hkrh|j rbt |j d }nd}n²|tj kr|j sd}nt |j d } t| tr| j} t| ttfr| | j| ¡}
|
|
|
dk rt|
|
jtrt |
|
jtj¡rtt ||
|
jg ¡t g}nt |d |
|
jj ¡|¡ |rFd} t || || j¡| ¡ t t j }|dk r|jdk s`t | !tj"||jj#|jj$|| j%d¡ dS )zaExtract mapping information from a SymbolTableNode that's in the
|
type.names dictionary.
|
|
NFr Tú4Column type should be a TypeEngine subclass not '{}'zØCan't infer type from attribute {} on class {}. please specify a return type from this function that is one of: Mapped[<python type>], relationship[<target class>], Column[<TypeEngine>], MapperProperty[<python type>]©r? ÚlineÚcolumnÚtypr7 )&r Útyper3 r r$ Útype_id_for_named_nodeÚMAPPEDÚ RELATIONSHIPÚCOMPOSITE_PROPERTYÚMAPPER_PROPERTYÚSYNONYM_PROPERTYÚCOLUMN_PROPERTYÚargsÚCOLUMNr r Úlookup_qualifiedr? ÚnodeÚhas_base_type_idÚ
|
TYPEENGINEr r# Ú#extract_python_type_from_typeenginer r% ÚfailÚformatr- r r Ú special_formÚAssertionErrorÚappendÚSQLAlchemyAttributerD rE r7 ) r' r( r? r@ rA Z
|
value_typeÚleft_hand_explicit_typeÚtype_idÚerrÚtypeengine_argr8 Úmsgr: r: r; r2 s sz
|
|
ú ÿ
|
ÿüÿ ÿüÿ
|
|
ûÿr2 r )r' r( r9 rA r* c C sv |j D ](}t|tttfrt |¡tjkr q4qdS | jj
|
|¡}d}t |j ¡rttj}t|jj }|j|_t|gt|} |jj| _| | jj |< dS t|jjtr¢|jjj}
|
t|
|
trÎt |
|
| |¡} ndS | tjtj tj tj tj tj hkr |
|
j!r t"|
|
j!d }n| tj#kr¢|
|
j!r¢|
|
j!d } t| tr¢| $| j | ¡} | dk r¢t| jt%r¢t &| jtj'¡rt(t) *|| jg ¡t+ g}nt ,|d -| jj.¡|
|
¡ |dkrÒd}t ,|| -|jj ¡|¡ ttj}t|jj }|j|_t|trt"t /||¡}| 0tj1|g¡|j_t 2t3|jj4|jj ¡}t|g|} |jj| _| 5t j6|j |j7|j8|| j9d¡ | | jj |< dS )a° Extract mapping information from a @declared_attr in a declarative
|
class.
|
|
E.g.::
|
|
@reg.mapped
|
class MyClass:
|
# ...
|
|
@declared_attr
|
def updated_at(cls) -> Column[DateTime]:
|
return Column(DateTime)
|
|
Will resolve in mypy as::
|
|
@reg.mapped
|
class MyClass:
|
# ...
|
|
updated_at: Mapped[Optional[datetime.datetime]]
|
|
Nr rB zàCan't infer type from @declared_attr on function '{}'; please specify a return type from this function that is one of: Mapped[<python type>], relationship[<target class>], Column[<TypeEngine>], MapperProperty[<python type>]rC ):Z
|
decoratorsr3 r r r r$ rH Z DECLARED_ATTRr/ r0 Úindexr% Zname_is_dunderr? r r rX ÚvarrR r r rG Úfuncr Zret_typer Ztype_id_for_unbound_typerI rJ rK rL rM rN rO r rP rQ r rS rT r r# rU r rV rW r- Zunbound_to_instanceZ
|
named_typeZNAMED_TYPE_SQLA_MAPPEDZexpr_to_mapped_constructorr
|
Ú argumentsrZ r[ rD rE r7 )r' r( r9 rA ÚdecZ dec_indexr\ Zany_Z left_nodeZnew_stmtZ func_typer] r_ r8 r` Úrvaluer: r: r; r5 Í s®
|
ÿþ
|
|
|
|
úÿÿ
|
ö
|
ÿüÿ ÿü
|
ÿ
|
|
ÿ ÿ ÿ
|
ûÿ r5 r c C s |j d }t|tsdS | jj |j¡}|dk s4t|j}t|t rHdS ||jksVtt|t
|
sdt|jdkr| |j ¡dkrt | j¡ dS |jdkr¦t | j¡ nd|j d¡r¶dS |jdkr
|
t|j tsÞt |d|¡ n,|j jD ]"}t|ttfræt | |||¡ qæd}d} |js&|jdkrt|jtrÜ|j} |jjd krÜ| d | ¡}
|
|
|
dk rÜ|
|
jdk rÜt |
|
j¡tj krÜt |jj d } |j}nFt |j} t| t rÔt | j¡tj krÔt | j d } | }n| } d}t|j t rú|dk rú| } nHt|j t!r>t|j j"t#r>t$ %|||| |j j"¡} | dkrBdS ndS | dk sPt| &t j'|j|j(|j)| | jd
|
¡ t *|||| | ¡ dS ) zZExtract mapping information from an assignment statement in a
|
declarative class.
|
|
r NZ __abstract__TZ __tablename__Ú__Z_mypy_mapped_attrsz+_mypy_mapped_attrs is expected to be a listZMappedrC )+Zlvaluesr3 r r7 r$ Úgetr? rY rR r r Z
|
parse_boolrf r% Z set_is_baseZ set_has_tabler. r rV r1 r r" Zapply_mypy_mapped_attrZ is_inferredrG r rQ rH rI r rO r r r Zcalleer r# Z#infer_type_from_right_hand_nameexprrZ r[ rD rE Z apply_type_to_mapped_statement) r' r( r9 rA Zlvaluer8 rR ÚitemZleft_hand_mapped_typer\ Z
|
mapped_symZ node_typeZpython_type_for_typer: r: r; r4 i sª
|
|
|
|
|
|
ÿþ
|
ÿý
|
ÿ
|
ÿþ ÿ ÿ
|
ûÿ
|
ûr4 )r' r( r* c C sN t | |¡}|dkrdS |jdd
D ]"}|j d¡r8q&t|j|dd q&dS )zèGiven a class, iterate through its superclass hierarchy to find
|
all other classes that are considered as ORM-significant.
|
|
Locates non-mapped mixins and scans them for mapped attributes to be
|
applied to subclasses.
|
|
Nr! éÿÿÿÿr+ T)r) )r% r, Úmror- r. r< Zdefn)r' r( r7 Z base_infor: r: r; r6 ì s ÿr6 N)F).Ú
|
__future__r Útypingr r r Z
|
mypy.nodesr r r r r
|
r r r r r r r r r r r Z mypy.pluginr Z
|
mypy.typesr r r r r r r r r r Ú r" r# r$ r% r< r2 r5 r4 r6 r: r: r: r; Ú<module> sT ýCZ
|
