관리-도구
편집 파일: infer.cpython-38.pyc
U -?�f�K � @ s. d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d d lm Z d d lm Z d d lmZ d d lmZ d d lmZ 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 ddl 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'dCd d d d d/d d0�d1d2�Z(dDd d d3d3d d d d4�d5d6�Z)d d d7d7d d8�d9d:�Z*d d d d d;�d<d=�Z+d d>d?d3d@�dAdB�Z,d.S )E� )� 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)�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 jkr�t | |||�}n6|t j kr�t | ||�}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. �I/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/ext/mypy/infer.py�#infer_type_from_right_hand_nameexpr( s^ � � � � � �r0 )r r r r! r# c C s� t |jt�st�|jjd }d}t |t�rFt |jt�rF|j}t|g �}t � |jd�}t � |jd�}d} |dk r�| � |�dkr�|dkr�d} |dk r�| � t j|g�}�nt|dks�| � |�dk�r�|dk �r�d} t |t�r�|j}t |t��rt |jt��r|dk �r�t|j|g�}n�t |t��r�t |jt��r�|jjdk �r�|dk �r�t |jjt��r�t|jjj�} t | t��r�t| j�} t | t��r�t| j|g�}nt �| d|j� d}nr|dk �r�| � |�dk�r�|dk �r�t �| d|j� |dk �rt|t� g�}n$|dk�rd } t �| | �|j�|� |dk�r.t| ||�S |dk �r|| �rlt |t��sNt�t |t��s^t�t| |||�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. r/ r( P s� � �� �� �� �� � �� � � �� � � � � ��r( c C s~ t |jt�st�|jjd }d}t |t�rHt |jt�rH|j}t|g �}nd}|dkr`t | ||�S |dk rvt | |||�S |S dS )z+Infer the type of mapping from a Composite.r N) r1 r2 r r3 r4 r r r r r* r< )r r r r! r= r- r> r. r. r/ r+ � s0 � � �r+ c C s& t |jt�st�t�|�}t| ||�S )zVInfer the type of mapping from a right side expression that returns Mapped. )r1 r2 r r3 r Ztype_for_calleer* )r r r r! r" Zthe_mapped_typer. r. r/ r% s �r% c C s� t |jt�st�|jjrV|jjd }t |t�rVt�|j�}|tjkrVt | ||||d�S t |jt�r�t�|jj�}|tj kr�t | |||�S t | ||�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) r1 r2 r r3 r4 r r$ r6 r& r' ZQUERY_EXPRESSIONr* )r r r r! Zfirst_prop_argr, r. r. r/ r) 6 s6 � � �r) NzOptional[CallExpr])r r r r! rA r# c C s4 t |t�st�d}|dkr0t |jt�s*dS |j}|jdd� D ]�}t |t�rjt |jt�r�|j}|j} q�q>t |tt f�r�t |j t �r>|}d} q�q�q>q>t |t f�r�q>q>t |t f�r�q>q>ds>t�q>|dkr�dS t |j t ��r$t�|j jtj��r$t| |j |�}|dk �rt| |||�S t|t� g�S 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)r1 r r3 r2 r r4 r6 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! rA r6 Z column_arg� type_argsr- r. r. r/ r' g s^ & � � � �r' r )r r r! r- �orig_left_hand_type�orig_python_type_for_typer# c C sh |dkr |}|dkr|}t ||�sd| �tj|g�}d}t�| |�|jt�|| j �t�|| j ��|� |S )z�Validate 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 r5 r �NAMED_TYPE_SQLA_MAPPEDr r8 r9 r: Z format_type�options)r r r! r- rF rG Zeffective_typer@ r. r. r/ r< � s* �� �� r< r )r r r! r- r# c C sl |}|}|j r,t|j d �}t|j d �}n|}|}t|ttf�sFt�t|ttf�sXt�t| |||||d�S )Nr )rF rG )r4 r r1 r r r3 r< )r r r! r- rF rG Z left_hand_argZpython_type_argr. r. r/ r; � s" �r; )r r r! r# c C s@ |dkr8d}t �| |�|j�|� | �tjttj �g�S |S dS )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[<python type>] on the left hand side.) r r8 r9 r: r5 r rH r r Z special_form)r r r! r@ r. r. r/ r* s � �r* r zSequence[Expression])r r rE r# c C s� |j dkrd|rd|d }t|t�rVt|jt�rV|jjD ] }|j dkr4t|jg � S q4n| �tj g �S |� d�szt d| ��| � d�}|d k r�t|jt�s�t �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���)�fullnamer1 r r r rC r r5 r ZNAMED_TYPE_BUILTINS_STRZhas_baser3 Z lookup_fully_qualified_or_noner r r4 )r r rE Z first_argZbase_Ztype_engine_symZ type_enginer. r. r/ rD . s2 � �� ��rD )N)NN)-� __future__r �typingr r Z mypy.maptyper 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 r0 r( r+ r% r) r' r< r; r* rD r. r. r. r/ �<module> sP ( ' 6 �h � *