관리-도구
편집 파일: apply.cpython-38.pyc
U -?�f6) � @ s� d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d d lm Z d d lm Z d d lm Z d d lmZ d d lmZ d dlmZ 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 d dlm Z d d lm!Z! d d l"m#Z# d d l"m$Z$ d d l%m&Z& d d!l%m'Z' d"d#d$d%d&d'�d(d)�Z(d"d#d%d&d*�d+d,�Z)d#d-d.d/d/d&d0�d1d2�Z*d"d#d%d&d*�d3d4�Z+d#d"d5d5d&d6�d7d8�Z,d9S ):� )� annotations)�List)�Optional)�Union)� ARG_NAMED_OPT)�Argument)�AssignmentStmt)�CallExpr)�ClassDef)�MDEF)� MemberExpr)�NameExpr)�RefExpr)�StrExpr)�SymbolTableNode)�TempNode)�TypeInfo)�Var)� SemanticAnalyzerPluginInterface)�add_method_to_class)�AnyType)�get_proper_type)�Instance)�NoneTyp)� ProperType)� TypeOfAny)� UnboundType)� UnionType� )�infer)�util)�expr_to_mapped_constructor)�NAMED_TYPE_SQLA_MAPPEDr r zUnion[NameExpr, StrExpr]z List[util.SQLAlchemyAttribute]�None)�cls�api�item� attributes�returnc C s� t |t�r|j}nt |t�r$|j}nd S | jjD ]2}t |t�r0t |jd t�r0|jd j|kr0 q|q0t � |d|� �| � d S |j d kr�t � |d|� d S t |j �}t |t ttf�s�t�|�t j||j|j|| jd�� t|||jd |d � d S )Nr z Can't find mapped attribute zBStatement linked from _mypy_mapped_attrs has no typing information)�name�line�column�typ�info)� isinstancer r) r �value�defs�bodyr �lvaluesr Zfail�typer r r r �AssertionError�appendZSQLAlchemyAttributer* r+ r- � apply_type_to_mapped_statement)r$ r% r& r' r) �stmt�left_hand_explicit_type� r9 �I/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/ext/mypy/apply.py�apply_mypy_mapped_attr- sV ��� � ��� �r; )r$ r% r' r( c C s� dd� |D �}d}| j jD �]�}t|t�rt|jd t�r|jd j|krt|jd jt�r|jd j}||jd j j }t |j �}t|t ��rxt|t ��sxt|j t ��rxt|j jt��rxt|j jjt��rx|j jjjdk �rx|j jjjjtk�rx|j jjdk�rxt|j jd t ��rxt|j jd jt��rxt�|||||j jd j�} | dk �rxt| t ��sx| }|||jd j _ d}t|j t��r�|j j jtkr|dk �s�t�|�t|g�|_ q|�r�t�| j |� dS )zlFor multiple class passes, re-apply our left-hand side types as mypy seems to reset them in place. c S s i | ] }|j |�qS r9 )r) ��.0�attrr9 r9 r: � <dictcomp>k s z4re_apply_declarative_assignments.<locals>.<dictcomp>Fr NZ_empty_constructorT) r0 r1 r. r r2 r r) �noder r3 r r �rvaluer Zcalleer �expr�fullnamer" �argsr r Z#infer_type_from_right_hand_nameexprr r4 � named_typer Zset_mapped_attributesr- ) r$ r% r' Zmapped_attr_lookupZupdate_cls_metadatar7 � left_node�python_type_for_typeZleft_node_proper_typeZnew_python_type_for_typer9 r9 r: � re_apply_declarative_assignmentsb s� ���� � �� � ��� �� � � � �� �� � � �rH r r zOptional[ProperType])r% r7 �lvaluer8 rG r( c C s� |j }t|t�st�|dk r4d|_| �t|g�|_n*d|_| �t|dkrTtt j �gn|g�|_t |j �|_ |jdk r�|dk r�||_dS )a� Apply the Mapped[<type>] annotation and right hand object to a declarative assignment statement. This converts a Python declarative class statement such as:: class User(Base): # ... attrname = Column(Integer) To one that describes the final Python behavior to Mypy:: class User(Base): # ... attrname : Mapped[Optional[int]] = <meaningless temp node> NF) r@ r. r r4 Zis_inferred_defrE r" r3 r r � special_formr! rA )r% r7 rI r8 rG rF r9 r9 r: r6 � s$ ��� r6 c C s& t �| |�}|dkrdS t �|�}d|jkr�|s�dd� |D �}|jdd� D ]@}d|jkr\qLt �||�}|dkrrqL|D ]}|�|j|j � qvqLg } |� � D ]:\} } | dkr�t t j �} | �tt| | �| t| �td�� q�t|| d| t� � d |jk�rt �|��rt|| d d � |�s"t|| d d � dS ) zCApply __init__, __table__ and other attributes to the mapped class.N�__init__c S s i | ]}|j |j�qS r9 )r) r3 r< r9 r9 r: r? s z1add_additional_orm_attributes.<locals>.<dictcomp>r ���Z sqlalchemy)�variableZtype_annotationZ initializer�kindZ __table__zsqlalchemy.sql.schema.Tablez sqlalchemy.orm.mapper.MapperZ __mapper__)r Z info_for_clsZ get_is_base�names�mro�metadataZget_mapped_attributes� setdefaultr) r3 �itemsr r rJ r5 r r r r r r Z get_has_table� _apply_placeholder_attr_to_class) r$ r% r' r- Zis_baseZmapped_attr_names�baseZbase_cls_attributesr> � argumentsr) r, r9 r9 r: � add_additional_orm_attributes� sR �� � �rW �str)r% r$ �qualified_name�attrnamer( c C sr | � |�}|r,t|jt�s t�t|jg �}n ttj�}t |�}|j d | |_ |j |_ ||_ tt|�|j j|< d S )N�.)Z lookup_fully_qualified_or_noner. r@ r r4 r r r rJ r rC Z _fullnamer- r3 r r rO )r% r$ rY rZ �sym�type_�varr9 r9 r: rT 0 s rT N)-� __future__r �typingr r r Z mypy.nodesr r r r r r r r r r r r r r Z mypy.pluginr Zmypy.plugins.commonr Z mypy.typesr r r r r r r r � r r rO r! r" r; rH r6 rW rT r9 r9 r9 r: �<module> sH 5UD5