U
|
¸ý°d÷M ã @ s: d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dl m
|
Z
|
d dl m Z d dl m Z d d l m Z d d
|
l mZ d d l mZ d d l mZ d d l mZ d dl mZ d dl mZ d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d dlm Z d dlm Z d dlm!Z! dd l"m#Z# dd l"m$Z$ d d d d!d"d!d#d$d%Z%d d d d!d!d&d'd(Z&d d d d!d!d&d)d*Z'd d d d!d"d!d#d+d,Z(d d d d!d!d&d-d.Z)dDd d d d!d0d!d1d2d3Z*dEd d d4d4d!d!d!d5d6d7Z+d d d8d8d!d9d:d;Z,d d d!d!d<d=d>Z-d d?d@d4dAdBdCZ.d/S )Fé )Ú annotations)ÚOptional)ÚSequence)Úmap_instance_to_supertype)Ú format_type)Ú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)Úutilr r r zOptional[ProperType]r )ÚapiÚstmtÚnodeÚleft_hand_explicit_typeÚinfer_from_right_sideÚreturnc C sº t |¡}|d krd S |t jkr2t| ||||}n|t jkrLt| |||}nj|t jkrft| |||}nP|t jkrt | |||}n6|t j
|
krt | ||}n |t j kr²t | |||}nd S |S )N)r Útype_id_for_calleeZMAPPEDÚ_infer_type_from_mappedÚCOLUMNÚ _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)r r r! r" r# Útype_idÚpython_type_for_type© r/ úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/ext/mypy/infer.pyÚ#infer_type_from_right_hand_nameexpr) s^
|
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿr1 )r r r! r" r$ c C s t |jtst|jjd }d}t |trFt |jtrF|j}t|g }t
|
|jd¡}t
|
|jd¡}d} |dk r¤| |¡dkr¤|dkr¤d} |dk r | t j|g¡}nt|dks¼| |¡dkr¦|dk r¦d} t |trÚ|j}t |trt |jtr|dk r¤t|j|g}nt |trt |jtr|jjdk r|dk r¤t |jjtr¤t|jjj}
|
t |
|
tr¤t|
|
j} t | tr¤t| j|g}nt | d|j¡ d}nr|dk rô| |¡dkrô|dk rÚt | d|j¡ |dk rt|t g}n$|dkrd } t | | |j¡|¡ |dkr.t| ||S |dk r|| rlt |tsNtt |ts^tt| |||S t | |||S n|S dS )
|
a¦ Infer 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"]
|
|
r NZuselistZcollection_classFTz>Expected 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 ÚAssertionErrorÚargsr r! r r r Zget_callexpr_kwargZ
|
parse_boolÚ
|
named_typer ZNAMED_TYPE_BUILTINS_LISTÚcalleer
|
Útyper r Zret_typeÚfailr r ÚformatÚnamer+ Ú3_infer_collection_type_from_left_and_inferred_rightÚ(_infer_type_from_left_and_inferred_right) r r r! r" Útarget_cls_argr. Úrelated_object_typeZ uselist_argZcollection_cls_argZtype_is_a_collectionÚrtZcallable_ret_typeÚmsgr/ r/ r0 r) Q sÒ ÿ
|
ÿÿ þý ÿÿ ÿþ
|
ÿ
|
ÿÿ
|
þ
|
ý
|
|
þü
|
ü
|
ÿ
|
ÿ
|
ÿ
|
ÿür) c C s~ t |jtst|jjd }d}t |trHt |jtrH|j}t|g }nd}|dkr`t | ||S |dk rvt
|
| |||S |S dS )z+Infer the type of mapping from a Composite.r N) r2 r3 r r4 r5 r r! r r r+ r= )r r r! r" r> r. r? r/ r/ r0 r, ÷ s0 ÿ ÿ ÿr, c C s& t |jtstt |¡}t| ||S )zVInfer the type of mapping from a right side expression
|
that returns Mapped.
|
|
|
)r2 r3 r r4 r Ztype_for_calleer+ )r r r! r" r# Zthe_mapped_typer/ r/ r0 r& s
|
ÿr& c C s t |jtst|jjrV|jjd }t |trVt |j¡}|tjkrVt | ||||dS t |jtrt |jj¡}|tj
|
krt | |||S t | ||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) r2 r3 r r4 r5 r r% r7 r' r( ZQUERY_EXPRESSIONr+ )r r r! r" Zfirst_prop_argr- r/ r/ r0 r* 7 s6
|
|
û
|
ü ÿr* NzOptional[CallExpr])r r r! r" rB r$ c C s4 t |tstd}|dkr0t |jts*dS |j}|jdd
D ]}t |trjt |jtr¾|j}|j} qÀq>t |tt frt |j
|
t r>|}d} qÀq¾q>q>t |t fr¦q>q>t |t fr¶q>q>ds>tq>|dkrÌdS t |j
|
t r$t |j
|
jtj¡r$t| |j
|
|}|dk rt| |||S t|t gS n t| ||S dS )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/ F)r2 r r4 r3 r r5 r7 r r r r! r r r r Z
|
mro_has_idÚmroZ
|
TYPEENGINEÚ#extract_python_type_from_typeenginer= r r r+ ) r r r! r" rB r7 Z
|
column_argÚ type_argsr. r/ r/ r0 r( h s^ &
|
|
ÿ ÿ
|
ÿ ÿr( r )r r! r" r. Úorig_left_hand_typeÚorig_python_type_for_typer$ c C s\ |dkr |}|dkr|}t ||sX| 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 {}) r r6 r ÚNAMED_TYPE_SQLA_MAPPEDr r9 r: r; r )r r! r" r. rG rH Zeffective_typerA r/ r/ r0 r= Ì s*
|
ÿÿýù
|
r= r )r r! r" r. r$ c C sl |}|}|j r,t|j d }t|j d }n|}|}t|ttfsFtt|ttfsXtt| |||||dS )Nr )rG rH )r5 r r2 r r r4 r= )r r! r" r. rG rH Z left_hand_argZpython_type_argr/ r/ r0 r< ö s" úr< )r r! r" r$ c C s@ |dkr8d}t | | |j¡|¡ | tjttj g¡S |S dS )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[<python type>] on the left hand side.)
|
r r9 r: r; r6 r rI r r Z special_form)r r! r" rA r/ r/ r0 r+ s ÿ
|
ÿr+ r zSequence[Expression])r r! rF r$ c C s¼ |j dkrd|rd|d }t|trVt|jtrV|jjD ] }|j dkr4t|jg S q4n| tj g ¡S |
|
d¡szt d| | d¡}|d k rt|jtst t t|g |j}t|jd S )Nz sqlalchemy.sql.sqltypes.Enumr z enum.Enumz"sqlalchemy.sql.type_api.TypeEnginez+could not extract Python type from node: %séÿÿÿÿ)Úfullnamer2 r r! r rD r r6 r ZNAMED_TYPE_BUILTINS_STRZhas_baser4 Z lookup_fully_qualified_or_noner r r5 )r r! rF Z first_argZbase_Ztype_engine_symZ type_enginer/ r/ r0 rE 0 s2 ÿ
|
ÿÿ
|
ÿþrE )N)NN)/Ú
|
__future__r Útypingr r Z mypy.maptyper Z mypy.messagesr Z
|
mypy.nodesr r r r
|
r r r r r r r Z mypy.pluginr Z mypy.subtypesr Z
|
mypy.typesr r r r r r r r Ú r r r1 r) r, r& r* r( r= r< r+ rE r/ r/ r/ r0 Ú<module> sR ( ' 6 ûi ú *
|
