관리-도구
편집 파일: decl_class.cpython-38.pyc
U -?�f�C � @ 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 dlmZ d dlm Z d dlm Z d dlm Z d dlm Z d d lm Z d d lm!Z! d d lm"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/d0d1�d2d3�Z*d&d'd4d/d0d5�d6d7�Z+d&d'd8d/d0d5�d9d:�Z,d&d'd0d;�d<d=�Z-d>S )@� )� annotations)�List)�Optional)�Union)�AssignmentStmt)�CallExpr)�ClassDef)� Decorator)� LambdaExpr)�ListExpr)� MemberExpr)�NameExpr)�PlaceholderNode)�RefExpr)�StrExpr)� SymbolNode)�SymbolTableNode)�TempNode)�TypeInfo)�Var)� SemanticAnalyzerPluginInterface)�AnyType)� CallableType)�get_proper_type)�Instance)�NoneType)� ProperType)�Type)� TypeOfAny)� UnboundType)� UnionType� )�apply)�infer)�names)�utilFr r �boolz(Optional[List[util.SQLAlchemyAttribute]])�cls�api� is_mixin_scan�returnc C s t �| |�}|d krd S | j�d�r(d S t �||�}t �|� |d k r\|sXt�| ||� |S g }| jj s�|j � � D ]\}}t | ||||� qrnFt � | jj �D ]6}t|t�r�t| |||� q�t|t�r�t| |||� q�t| |� |s�t�| ||� t �||� |S )N�builtins)r% � info_for_cls�fullname� startswithZget_mapped_attributesZestablish_as_sqlalchemyr" Z re_apply_declarative_assignments�defs�bodyr$ �items�_scan_symbol_table_entryZflatten_typechecking� isinstancer �!_scan_declarative_assignment_stmtr � _scan_declarative_decorator_stmt�_scan_for_mapped_basesZ add_additional_orm_attributesZset_mapped_attributes)r' r( r) �infoZmapped_attributesZsym_name�sym�stmt� r: �N/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/ext/mypy/decl_class.py�,scan_declarative_assignments_and_apply_types0 sT � � � � r< �strr z List[util.SQLAlchemyAttribute]�None)r' r( �name�value� attributesr* c C s� t |j�}t|t�sdS d}t�|j�}d}|tjtjtjtj tj tj hkrh|j rbt |j d �}nd}n�|tj k�r|j s�d}n�t |j d �} t| t�r�| j} t| ttf��r|�| j| �} | dk �rt| jt��rt�| jtj��rtt�|| jg �t� g�}nt�|d�| jj �|� |�rFd} t�|| �|| j�| � t t j �}|dk �r�|jdk �s`t �|�!tj"||jj#|jj$|| j%d�� dS )zaExtract mapping information from a SymbolTableNode that's in the type.names dictionary. NFr T�4Column type should be a TypeEngine subclass not '{}'z�Can't infer type from attribute {} on class {}. please specify a return type from this function that is one of: Mapped[<python type>], relationship[<target class>], Column[<TypeEngine>], MapperProperty[<python type>]�r? �line�column�typr7 )&r �typer3 r r$ �type_id_for_named_node�MAPPED� RELATIONSHIP�COMPOSITE_PROPERTY�MAPPER_PROPERTY�SYNONYM_PROPERTY�COLUMN_PROPERTY�args�COLUMNr r �lookup_qualifiedr? �node�has_base_type_id� TYPEENGINEr r# �#extract_python_type_from_typeenginer r% �fail�formatr- r r � special_form�AssertionError�append�SQLAlchemyAttributerD rE r7 ) r' r( r? r@ rA Z value_type�left_hand_explicit_type�type_id�err�typeengine_argr8 �msgr: r: r; r2 r sz � � ��� ��� ��r2 r )r' r( r9 rA r* c C sv |j D ](}t|tttf�rt�|�tjkr q4qdS | jj � |�}d}t � |j �r�ttj�}t|jj �}|j|_t|gt|��} |jj| _| | jj |< dS t|jjt��r�|jjj} t| t�r�t�| | |�} ndS | tjtj tj tj tj tj hk�r | j!�r t"| j!d �}n�| tj#k�r�| j!�r�| j!d } t| t��r�|�$| j | �} | dk �r�t| jt%��r�t�&| jtj'��r�t(t)�*|| jg �t+� g�}nt �,|d�-| jj.�| � |dk�r�d}t �,||�-|jj �|� ttj�}t|jj �}|j|_t|t��rt"t �/||��}|�0tj1|g�|j_t�2t3|jj4|jj ��}t|g|�} |jj| _|�5t j6|j |j7|j8|| j9d�� | | jj |< dS )a� Extract mapping information from a @declared_attr in a declarative class. E.g.:: @reg.mapped class MyClass: # ... @declared_attr def updated_at(cls) -> Column[DateTime]: return Column(DateTime) Will resolve in mypy as:: @reg.mapped class MyClass: # ... updated_at: Mapped[Optional[datetime.datetime]] Nr rB z�Can't infer type from @declared_attr on function '{}'; please specify a return type from this function that is one of: Mapped[<python type>], relationship[<target class>], Column[<TypeEngine>], MapperProperty[<python type>]rC ):Z decoratorsr3 r r r r$ rH Z DECLARED_ATTRr/ r0 �indexr% Zname_is_dunderr? r r rX �varrR r r rG �funcr Zret_typer Ztype_id_for_unbound_typerI rJ rK rL rM rN rO r rP rQ r rS rT r r# rU r rV rW r- Zunbound_to_instanceZ named_typeZNAMED_TYPE_SQLA_MAPPEDZexpr_to_mapped_constructorr � argumentsrZ r[ rD rE r7 )r' r( r9 rA �decZ dec_indexr\ Zany_Z left_nodeZnew_stmtZ func_typer] r_ r8 r` �rvaluer: r: r; r5 � s� �� ��� � ��� �� � � � � �� r5 r c C s� |j d }t|t�sdS | jj�|j�}|dk s4t�|j}t|t �rHdS ||jksVt�t|t �sdt�|jdkr�|� |j �dkr�t �| j� dS |jdkr�t �| j� nd|j�d�r�dS |jdk�r t|j t�s�t �|d|� n,|j jD ]"}t|ttf�r�t�| |||� q�d}d} |j�s&|jdk�r�t|jt��r�|j} |jjd k�r�|�d | �} | dk �r�| jdk �r�t�| j�tj k�r�t |jj d �} |j}nFt |j�} t| t ��r�t�| j�tj k�r�t | j d �} | }n| } d}t|j t ��r�|dk �r�| } nHt|j t!��r>t|j j"t#��r>t$�%|||| |j j"�} | dk�rBdS ndS | dk �sPt�|�&t j'|j|j(|j)| | jd �� t�*|||| | � dS ) zZExtract mapping information from an assignment statement in a declarative class. r NZ __abstract__TZ __tablename__�__Z_mypy_mapped_attrsz+_mypy_mapped_attrs is expected to be a listZMappedrC )+Zlvaluesr3 r r7 r$ �getr? rY rR r r Z parse_boolrf r% Z set_is_baseZ set_has_tabler. r rV r1 r r" Zapply_mypy_mapped_attrZ is_inferredrG r rQ rH rI r rO r r r Zcalleer r# Z#infer_type_from_right_hand_nameexprrZ r[ rD rE Z apply_type_to_mapped_statement) r' r( r9 rA Zlvaluer8 rR �itemZleft_hand_mapped_typer\ Z mapped_symZ node_typeZpython_type_for_typer: r: r; r4 g s� �� �� � �� � � �� �r4 )r' r( r* c C sN t �| |�}|dkrdS |jdd� D ]"}|j�d�r8q&t|j|dd� q&dS )z�Given a class, iterate through its superclass hierarchy to find all other classes that are considered as ORM-significant. Locates non-mapped mixins and scans them for mapped attributes to be applied to subclasses. Nr! ���r+ T)r) )r% r, �mror- r. r< Zdefn)r' r( r7 Z base_infor: r: r; r6 � s �r6 N)F).� __future__r �typingr r r Z mypy.nodesr r r r r r r r r r r r r r r r Z mypy.pluginr Z mypy.typesr r r r r r r r r r � r" r# r$ r% r< r2 r5 r4 r6 r: r: r: r; �<module> sT �BY