o E6dM@sddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddl m!Z!ddl m"Z"dGd(d)Z#dHd*d+Z$dHd,d-Z%dGd.d/Z&dHd0d1Z' 2dIdJd5d6Z( 2 2dKdLd;d<Z)dMd>d?Z*dNd@dAZ+dOdEdFZ,d2S)P) annotations)Optional)Sequence)map_instance_to_supertype)AssignmentStmt)CallExpr) Expression)FuncDef) LambdaExpr) MemberExpr)NameExpr)RefExpr)StrExpr)TypeInfo)Var)SemanticAnalyzerPluginInterface) is_subtype)AnyType) CallableType)get_proper_type)Instance)NoneType) ProperType) TypeOfAny) UnionType)names)utilapirstmtrnoderleft_hand_explicit_typeOptional[ProperType]infer_from_right_sider returncCst|}|dur dS|tjurt|||||}|S|tjur(t||||}|S|tjur6t||||}|S|tjurDt ||||}|S|tj urQt |||}|S|tj ur_t ||||}|SdSN)rtype_id_for_calleeZMAPPED_infer_type_from_mappedCOLUMN_infer_type_from_decl_columnZ RELATIONSHIP_infer_type_from_relationshipZCOLUMN_PROPERTY%_infer_type_from_decl_column_propertyZSYNONYM_PROPERTY#infer_type_from_left_hand_type_onlyZCOMPOSITE_PROPERTY(_infer_type_from_decl_composite_property)rrr r!r#type_idpython_type_for_typer0pC:\Users\jesus\OneDrive\Desktop\erpjis_fastapi\backend\jisbackend\Lib\site-packages\sqlalchemy/ext/mypy/infer.py#infer_type_from_right_hand_nameexpr(sD        r2c CsZt|jtsJ|jjd}d}t|tr#t|jtr#|j}t|g}t |jd}t |jd}d} |durQ| |durQ|durQd} |durP| t j |g}n|dus\| |dur|durd} t|trj|j}t|trt|jtr|durt|j|g}nyt|trt|jtr|jjdur|durt|jjtrt|jjj} t| trt| j} t| trt| j|g}n?t|d|jd}n4|dur| |dur|durt|d|j|durt|tg}n|durd } t|| |j||durt|||S|dur+| r$t|tsJt|tsJt||||St||||S|S) aInfer the type of mapping from a relationship. E.g.:: @reg.mapped class MyClass: # ... addresses = relationship(Address, uselist=True) order: Mapped["Order"] = relationship("Order") Will resolve in mypy as:: @reg.mapped class MyClass: # ... addresses: Mapped[List[Address]] order: Mapped["Order"] rNZuselistZcollection_classFTz>Expected Python collection type for collection_class parameterzOSending uselist=False and collection_class at the same time does not make sensezCan't infer scalar or collection for ORM mapped expression assigned to attribute '{}' if both 'uselist' and 'collection_class' arguments are absent from the relationship(); please specify a type annotation on the left hand side.) isinstancervaluerargsr r rrrZget_callexpr_kwargZ parse_bool named_typerZNAMED_TYPE_BUILTINS_LISTcalleer typerrZret_typefailrrformatnamer,3_infer_collection_type_from_left_and_inferred_right(_infer_type_from_left_and_inferred_right) rrr r!target_cls_argr/related_object_typeZ uselist_argZcollection_cls_argZtype_is_a_collectionrtZcallable_ret_typemsgr0r0r1r*Ps            r*cCszt|jtsJ|jjd}d}t|tr$t|jtr$|j}t|g}nd}|dur0t|||S|dur;t ||||S|S)z+Infer the type of mapping from a Composite.rN) r3r4rr5r r rrr,r=)rrr r!r>r/r?r0r0r1r-s$   r-cCs&t|jtsJt|}t|||S)zVInfer the type of mapping from a right side expression that returns Mapped. )r3r4rrZtype_for_calleer,)rrr r!r#Zthe_mapped_typer0r0r1r's  r'cCst|jtsJ|jjr+|jjd}t|tr+t|j}|tjur+t|||||dSt|jtrDt|jj}|tj urDt||||St |||S)zInfer the type of mapping from a ColumnProperty. This includes mappings against ``column_property()`` as well as the ``deferred()`` function. r)right_hand_expression) r3r4rr5rr&r7r(r)ZQUERY_EXPRESSIONr,)rrr r!Zfirst_prop_argr.r0r0r1r+6s2       r+NrBOptional[CallExpr]c Cst|tsJd}|durt|jtsdS|j}|jddD]9}t|tr5t|jtr4|j}|j}n%qt|ttfrIt|j t rH|}d}nqt|t frPqt|t frWqJ|dur_dSt|j t rt |j jt jrt||j |}|durt||||St|tgSt|||S)aInfer the type of mapping from a Column. E.g.:: @reg.mapped class MyClass: # ... a = Column(Integer) b = Column("b", String) c: Mapped[int] = Column(Integer) d: bool = Column(Boolean) Will resolve in MyPy as:: @reg.mapped class MyClass: # ... a : Mapped[int] b : Mapped[str] c: Mapped[int] d: Mapped[bool] Nrr0)r3rr4rr5r7r r r r rrr rZ mro_has_idmroZ TYPEENGINE#extract_python_type_from_typeenginer=rrr,) rrr r!rBr7Z column_arg type_argsr/r0r0r1r)gsP&       r)rr/orig_left_hand_typeorig_python_type_for_typec Csh|dur|}|dur |}t||s2|tj|g}d}t|||jt||j t||j ||S)zValidate type when a left hand annotation is present and we also could infer the right hand side:: attrname: SomeType = Column(SomeDBType) NzRLeft hand assignment '{}: {}' not compatible with ORM mapped expression of type {}) rr6rNAMED_TYPE_SQLA_MAPPEDrr9r:r;Z format_typeoptions)rr r!r/rHrIZeffective_typerAr0r0r1r=s(    r=rcCsl|}|}|jrt|jd}t|jd}n|}|}t|ttfs#Jt|ttfs,Jt||||||dS)Nr)rHrI)r5rr3rrr=)rr r!r/rHrIZ left_hand_argZpython_type_argr0r0r1r<s"r<cCs<|durd}t|||j||tjttj gS|S)zDetermine the type based on explicit annotation only. if no annotation were present, note that we need one there to know the type. NzCan't infer type from ORM mapped expression assigned to attribute '{}'; please specify a Python type or Mapped[] on the left hand side.) rr9r:r;r6rrJrrZ special_form)rr r!rAr0r0r1r,s r,rrGSequence[Expression]cCs|jdkr2|r2|d}t|tr+t|jtr+|jjD]}|jdkr)t|jgSqn|tj gS| ds=Jd|| d}|durLt|jtsNJt t|g|j}t |jdS)Nzsqlalchemy.sql.sqltypes.Enumrz enum.Enumz"sqlalchemy.sql.type_api.TypeEnginez+could not extract Python type from node: %s)fullnamer3r r rrErr6rZNAMED_TYPE_BUILTINS_STRZhas_baseZlookup_fully_qualified_or_nonerrr5)rr rGZ first_argZbase_Ztype_engine_symZ type_enginer0r0r1rF.s2     rF) rrrrr rr!r"r#r r$r") rrrrr rr!r"r$r"r%) rrrrr rr!r"rBrCr$r")NN)rrr rr!rr/rrHr"rIr"r$r") rrr rr!rr/rr$r")rrr rr!r"r$r")rrr rrGrLr$r)- __future__rtypingrrZ mypy.maptyperZ mypy.nodesrrrr r r r r rrrZ mypy.pluginrZ mypy.subtypesrZ mypy.typesrrrrrrrrrrr2r*r-r'r+r)r=r<r,rFr0r0r0r1sR                             ( ' 6 h *