관리-도구
편집 파일: plugin.cpython-38.pyc
U -?�f& � @ sR d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd l m Z dd l mZ dd l mZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddl m Z ddl m!Z! ddl mZ dd l"m#Z# dd l"m$Z$ dd l"m%Z% z e&d � W n e'k �r� Y n X e'd ��G d!d"� d"e �Z(d#d$d%�d&d'�Z)d(d)d*�d+d,�Z*d-d)d*�d.d/�Z+d-d)d*�d0d1�Z,d-d)d*�d2d3�Z-d-d)d*�d4d5�Z.d-d)d*�d6d7�Z/d-d)d*�d8d9�Z0d:d;d*�d<d=�Z1d>d)d*�d?d@�Z2dAdBd)dC�dDdE�Z3dFS )Gz" Mypy plugin for SQLAlchemy ORM. � )� annotations)�Callable)�List)�Optional)�Tuple)�Type)�Union)�nodes)� calculate_mro)�MroError)�Block)�ClassDef)�GDEF)�MypyFile)�NameExpr)� SymbolTable)�SymbolTableNode)�TypeInfo)�AttributeContext)�ClassDefContext)�DynamicClassDefContext)�Plugin)� SemanticAnalyzerPluginInterface)�get_proper_type)�Instance� )� decl_class)�names)�utilzsqlalchemy-stubsz�The SQLAlchemy mypy plugin in SQLAlchemy 2.0 does not work with sqlalchemy-stubs or sqlalchemy2-stubs installed, as well as with any other third party SQLAlchemy stubs. Please uninstall all SQLAlchemy stubs packages.c @ s| e Zd Zddd�dd�Zddd�dd�Zddd�d d �Zddd�d d �Zddd�d d�Zddd�dd�Zddd�dd�Z dS )�SQLAlchemyPlugin�strz2Optional[Callable[[DynamicClassDefContext], None]])�fullname�returnc C s t �|�t jkrtS d S �N)r �type_id_for_fullnameZDECLARATIVE_BASE�_dynamic_class_hook��selfr! � r( �J/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/ext/mypy/plugin.py�get_dynamic_class_hook< s z'SQLAlchemyPlugin.get_dynamic_class_hookz+Optional[Callable[[ClassDefContext], None]]c C s t S r# )�_fill_in_decoratorsr& r( r( r) � get_customize_class_mro_hookC s z-SQLAlchemyPlugin.get_customize_class_mro_hookc C s\ | � |�}|d k rX|jd k rXt�|j�}|tjkr6tS |tjtjfkrJtS |tj krXt S d S r# ) �lookup_fully_qualified�noder �type_id_for_named_nodeZMAPPED_DECORATOR�_cls_decorator_hookZAS_DECLARATIVEZAS_DECLARATIVE_BASE�_base_cls_decorator_hookZDECLARATIVE_MIXIN�_declarative_mixin_hook)r' r! �symZtype_idr( r( r) �get_class_decorator_hookH s � z)SQLAlchemyPlugin.get_class_decorator_hookc C s t �|�t jkrtS d S r# )r r$ ZDECLARATIVE_META�_metaclass_cls_hookr& r( r( r) �get_metaclass_hook[ s z#SQLAlchemyPlugin.get_metaclass_hookc C s. | � |�}|r*t|jt�r*t�|j�r*tS d S r# )r- � isinstancer. r r Zhas_declarative_base�_base_cls_hook)r' r! r3 r( r( r) �get_base_class_hooke s � � �z$SQLAlchemyPlugin.get_base_class_hookz,Optional[Callable[[AttributeContext], Type]]c C s |� d�rtS d S )Nz-sqlalchemy.orm.attributes.QueryableAttribute.)� startswith�_queryable_getattr_hookr& r( r( r) �get_attribute_hooks s �z#SQLAlchemyPlugin.get_attribute_hookr zList[Tuple[int, str, int]])�filer" c C s dddgS )N)� �sqlalchemy.orm���)r> zsqlalchemy.orm.attributesr@ )r> zsqlalchemy.orm.decl_apir@ r( )r' r= r( r( r) �get_additional_deps} s �z$SQLAlchemyPlugin.get_additional_depsN) �__name__� __module__� __qualname__r* r, r4 r6 r9 r< rA r( r( r( r) r ; s r r z TypingType[SQLAlchemyPlugin])�versionr" c C s t S r# )r )rE r( r( r) �plugin� s rF r �None)�ctxr" c C s4 t | � t| jtg ��}| j�| j�|_tt� || jj �}||_ t | j|� t j | jdtfd�}|dk r�t|jt�r�t �|j� tj|jj| jdd� t|jg �g|_n| j�tj�}|g|_z t|� W nD t k �r t � | jd| j� | j�tj�}|g|_d|_ Y nX | j� | jt t!|�� t �|� dS )zZGenerate a declarative Base class when the declarative_base() function is encountered.�cls)Z expr_typesNT�Z is_mixin_scanz.Not able to calculate MRO for declarative base)"� _add_globalsr �namer �apiZqualified_namer! r r Z cur_mod_id�info�_set_declarative_metaclassr Zget_callexpr_kwarg�callr r7 r. � set_is_baser �,scan_declarative_assignments_and_apply_typesZdefnr �basesZ named_typer ZNAMED_TYPE_BUILTINS_OBJECTr r �failZfallback_to_anyZadd_symbol_table_noder r )rH rI rN Zcls_arg�objr( r( r) r% � s< � � r% r c C s� | j jD ]�}t|tj�r:t|jtj�r:|jjdkr:|j}n t|tj�r|jdkr|}nqt|jt �r| j j |jj|dd�}nq|r|j rt |j�}t|t�r�|jj� d|j� �|_qt�| j d|j |j � qd S )NZas_declarative_baseZmappedT)Zsuppress_errors�.z�Class decorator called %s(), but we can't tell if it's from an ORM registry. Please annotate the registry assignment, e.g. my_registry: registry = registry())rI Z decoratorsr7 r ZCallExprZcallee� MemberExprrL �exprr rM Zlookup_qualifiedr. r �typer r! r rT )rH � decorator�targetr3 Zsym_typer( r( r) r+ � s@ � � � �� � ��r+ c C s� t | � t| jtj�st�| jj}t|tj�r<t|jtj �s@t�t |jj �}t|t �rht �|j �t jkslt�t�| j| j� d S r# )rK r7 �reasonr rW �AssertionErrorrX ZRefExprr. ZVarr rY r r r/ ZREGISTRYr rR rI rM )rH rX Z node_typer( r( r) r0 � s ��r0 c C s> t | � | j}t| j|� t�| jj� tj|| jdd� d S �NTrJ ) rK rI rO rM r rQ rN r rR )rH rI r( r( r) r1 � s �r1 c C s. t | � t�| jj� tj| j| jdd� d S r^ )rK r rQ rI rN r rR rM �rH r( r( r) r2 s �r2 c C s t �| jj� d S r# )r rQ rI rN r_ r( r( r) r5 s r5 c C s t | � t�| j| j� d S r# )rK r rR rI rM r_ r( r( r) r8 s r8 r r c C s | j S r# )Zdefault_attr_typer_ r( r( r) r; s r; z.Union[ClassDefContext, DynamicClassDefContext]c C s t �| ddd� dS )z`Add __sa_DeclarativeMeta and __sa_Mapped symbol to the global space for all class defs r? ZMappedZ __sa_MappedN)r Z add_globalr_ r( r( r) rK s rK r r )rM � target_clsr" c C s@ |j }| �d�}|d k r$t|jt�s(t�t|jg � |_|_d S )Nz'sqlalchemy.orm.decl_api.DeclarativeMeta) rN Z lookup_fully_qualified_or_noner7 r. r r] r Zdeclared_metaclassZmetaclass_type)rM r` rN r3 r( r( r) rO ' s �rO N)4�__doc__� __future__r �typingr r r r r Z TypingTyper Zmypyr Zmypy.mror r Z mypy.nodesr r r r r r r r Z mypy.pluginr r r r r Z mypy.typesr r � r r r � __import__� ImportErrorr rF r% r+ r0 r1 r2 r5 r8 r; rK rO r( r( r( r) �<module> s` � M'6