o =a@ @sJddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd lm Z dd lm Z dd lm Z dd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!dd l"m#Z#dd!l"m$Z$dd"l"m%Z%dd#l"m&Z& $d8d%ed&ed'e'd(eee&j(fd)d*Z)d%ed&ed+e*d,ed-ee&j(d(d.f d/d0Z+d%ed&ed1ed-ee&j(d(d.f d2d3Z,d%ed&ed1ed-ee&j(d(d.f d4d5Z-d%ed&ed(d.fd6d7Z.d.S)9)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)utilFclsapi is_mixin_scanreturncCst||}|dur dS|jdrdSt||}|dur)|s't||||Sg}|jjsB|j D] \}}t |||||q4n#t |jjD]}t |trXt||||qIt |trdt||||qIt|||sst|||t|||S)Nbuiltins)r$ info_for_clsfullname startswithget_mapped_attributesr! re_apply_declarative_assignmentsdefsbodyr#items_scan_symbol_table_entryflatten_typechecking isinstancer!_scan_declarative_assignment_stmtr _scan_declarative_decorator_stmt_scan_for_mapped_basesadd_additional_orm_attributesset_mapped_attributes)r%r&r'infomapped_attributessym_namesymstmtr?T/var/www/html/gps/gps/lib/python3.10/site-packages/sqlalchemy/ext/mypy/decl_class.py,scan_declarative_assignments_and_apply_types.sB        rAnamevalue attributesNc Cs|t|j}t|ts dSd}t|j}d}|tjtjtjtj tj tj hvr4|j r1t|j d}nWd}nT|tj ur|j s?d}nIt|j d} t| trN| j} t| ttfr|| j| } | durt| jtrt| jtjr|tt|| jgtg}n t|d| jj||rd} t|| ||j|ttj}|dur|jdusJ| tj!||jj"|jj#||j$ddSdS)zaExtract mapping information from a SymbolTableNode that's in the type.names dictionary. NFrT4Column type should be a TypeEngine subclass not '{}'zCan't infer type from attribute {} on class {}. please specify a return type from this function that is one of: Mapped[], relationship[], Column[], MapperProperty[]rBlinecolumntypr:)%rtyper4rr#type_id_for_named_nodeMAPPED RELATIONSHIPCOMPOSITE_PROPERTYMAPPER_PROPERTYSYNONYM_PROPERTYCOLUMN_PROPERTYargsCOLUMNrrlookup_qualifiedrBnodehas_base_type_id TYPEENGINErr"#extract_python_type_from_typeenginerr$failformatr+rr special_formappendSQLAlchemyAttributerGrHr:) r%r&rBrCrD value_typeleft_hand_explicit_typetype_iderrtypeengine_argr=msgr?r?r@r2nsv       r2r>c Cs |jD]}t|tttfrt|tjurnqdS|jj |}d}t|j j t r|j j j}t|tr Column[DateTime]: return Column(DateTime) Will resolve in mypy as:: @reg.mapped class MyClass: # ... updated_at: Mapped[Optional[datetime.datetime]] NrrEzCan't infer type from @declared_attr on function '{}'; please specify a return type from this function that is one of: Mapped[], relationship[], Column[], MapperProperty[] __sa_MappedrF)7 decoratorsr4r r rr#rK DECLARED_ATTRr/r0indexfuncrJrret_typertype_id_for_unbound_typerLrMrNrOrPrQrRrrSrTrBrUrrVrWrr"rXrr$rYrZr+varrrr[unbound_to_instance named_typeexpr_to_mapped_constructorr argumentsrr\r]rGrHr:)r%r&r>rDdec dec_indexr_ func_typer`rbr=rc left_nodervaluenew_stmtr?r?r@r6s                 r6c Csp|jd}t|ts dS|jj|j}|dusJ|j}t|tr$dS||jus+Jt|t s2J|jdkrG| |j durEt |jdS|jdkrSt |jn1|jdr[dS|jdkrt|j tsnt |d|n|j jD]}t|ttfrt||||qrd}d} |js|jdurt|jtr|j} |jjd kr|d |} | dur| jdurt| jtjurt|jjd} |j}n!t|j} t| trt| jtjurt| jd} | }n| } d}t|j tr|dur| } n$t|j t rt|j j!t"rt#$|||| |j j!} | durdSndS| dusJ|%t j&|j|j'|j(| |jd t)|||| | dS) zZExtract mapping information from an assignment statement in a declarative class. rN __abstract__T __tablename_____mypy_mapped_attrsz+_mypy_mapped_attrs is expected to be a listMappedrF)*lvaluesr4r r:r#getrBrUr r parse_boolrtr$ set_is_base set_has_tabler,r rYr1rr!apply_mypy_mapped_attr is_inferredrJrrTrKrLrrRrrrcalleerr"#infer_type_from_right_hand_nameexprr\r]rGrHapply_type_to_mapped_statement) r%r&r>rDlvaluer=rUitemleft_hand_mapped_typer_ mapped_sym node_typepython_type_for_typer?r?r@r5Ws                  r5cCsNt||}|dur dS|jddD]}|jdrqt|j|ddqdS)zGiven 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,rAdefn)r%r&r: base_infor?r?r@r7s  r7)F)/typingrrr mypy.nodesrrrrr r r r r rrrrrrr mypy.pluginr mypy.typesrrrrrrrrrrr!r"r#r$boolr]rAstrr2r6r5r7r?r?r?r@s                                    @ Z