o ãRc˜Hã@s°ddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddl m!Z!ddl m"Z"dededed eed!ed"eef d#d$„Z#dededed eed"eef d%d&„Z$dededed eed"eef d'd(„Z%dededed eed"eef d)d*„Z& +d;dededed eed,ee d"eef d-d.„Z' + +dExpected Python collection type for collection_class parameterzOSending uselist=False and collection_class at the same time does not make sensezäCan'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.)Ú isinstanceÚrvaluerÚargsr r rrrZget_callexpr_kwargZ parse_boolÚ named_typerZNAMED_TYPE_BUILTINS_LISTÚcalleer ÚtyperrZret_typeÚfailrrÚformatÚnamer*Ú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_collectionÚrtZcallable_ret_typeÚmsgr.r.r/r(MsÆ  ÿ ÿÿ þýÿ€ÿ ÿþ  ÿÿ€ÿ þ ý   þ€üüÿ€ÿ ÿ ÿür(cCszt|jtƒsJ‚|jjd}d}t|tƒr$t|jtƒr$|j}t|gƒ}nd}|dur0t|||ƒS|dur;t ||||ƒS|S)z3Infer the type of mapping from a CompositeProperty.rN) r1r2rr3r r rrr*r;)rrr r!r<r-r=r.r.r/r+ós$  ÿ ÿÿr+cCs”t|jtƒsJ‚|jjr+|jjd}t|tƒr+t |j¡}|tjur+t|||||dSt|jtƒrDt |jj¡}|tj urDt||||ƒSt |||ƒS)zœInfer the type of mapping from a ColumnProperty. This includes mappings against ``column_property()`` as well as the ``deferred()`` function. r)Úright_hand_expression) r1r2rr3rr%r5r&r'ZQUERY_EXPRESSIONr*)rrr r!Zfirst_prop_argr,r.r.r/r)s2     û  üÿr)Nr@c Cst|tƒsJ‚d}|durt|jtƒsdS|j}|jdd…D]9}t|tƒr5t|jtƒr4|j}|j}n%qt|ttfƒrIt|j t ƒrH|}d}nqt|t fƒrPqt|t fƒrWqJ‚|dur_dSt|j t ƒr‡t  |j jt j¡r‡t||j |ƒ}|dur€t||||ƒSt|tƒgƒSt|||ƒS)aãInfer 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] Nrér.)r1rr2rr3r5r r r r rrr rZ mro_has_idÚmroZ TYPEENGINEÚ#extract_python_type_from_typeenginer;rrr*) rrr r!r@r5Z column_argÚ type_argsr-r.r.r/r'DsP&   ü    ÿÿÿÿr'r-Úorig_left_hand_typeÚorig_python_type_for_typec Cs\|dur|}|dur |}t||ƒs,| tj|g¡}d}t || |jt|ƒt|ƒ¡|¡|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 {}) rr4rÚNAMED_TYPE_SQLA_MAPPEDrr7r8r9r)rr r!r-rErFZeffective_typer?r.r.r/r;¨s( ÿÿýù r;cCsl|}|}|jrt|jdƒ}t|jdƒ}n|}|}t|ttfƒs#J‚t|ttfƒs,J‚t||||||dS)Nr)rErF)r3rr1rrr;)rr r!r-rErFZ left_hand_argZpython_type_argr.r.r/r:Òs"úr:cCs<|durd}t || |j¡|¡| tjttj ƒg¡S|S)z‘Determine the type based on explicit annotation only. if no annotation were present, note that we need one there to know the type. Nz”Can't infer type from ORM mapped expression assigned to attribute '{}'; please specify a Python type or Mapped[] on the left hand side.) rr7r8r9r4rrGrrZ special_form)rr r!r?r.r.r/r*ïs ÿÿr*rDcCs¼|jdkr2|r2|d}t|tƒr+t|jtƒr+|jjD]}|jdkr)t|jgƒSqn| tj g¡S|  d¡s=Jd|ƒ‚|  d¡}|durLt|jtƒsNJ‚t t|gƒ|jƒ}t |jdƒS)Nzsqlalchemy.sql.sqltypes.Enumrz enum.Enumz"sqlalchemy.sql.type_api.TypeEnginez+could not extract Python type from node: %séÿÿÿÿ)Úfullnamer1r r rrBrr4rZNAMED_TYPE_BUILTINS_STRZhas_baseZlookup_fully_qualified_or_nonerrr3)rr rDZ first_argZbase_Ztype_engine_symZ type_enginer.r.r/rC s2 ÿ  ÿÿ ÿÿ ÿþrCr$)NN),ÚtypingrrZ mypy.maptyperZ mypy.messagesrZ mypy.nodesrrrr r r r r rrrZ mypy.pluginrZ mypy.subtypesrZ mypy.typesrrrrrrrrÚrrr0r(r+r)r'r;r:r*rCr.r.r.r/Ús                            ÿþýüû ú&ÿþýü û'ÿþýü û ÿþýü û6ûÿþýüû úiúÿþýüûú ù*ÿþýü ûÿþý üÿþýü