File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/ext/mypy/__pycache__/util.cpython-310.pyc
o
���g�& � @ s6 d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ 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 dlm Z 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!l)m*Z* d d"l+m,Z, d d#l-m.Z. d d$l-m/Z/ d d%l-m0Z0 d d&l-m1Z1 d d
l-m Z d d'l-m2Z2 d d(l-m3Z3 d d)l-m4Z4 e5d*d+� ej6�7d,�D ��Z8e8d-kZ9e
d.eee f d/�Z:G d0d1� d1�Z;d�d6d7�Z<d�d>d?�Z=d�dAdB�Z>d�dCdD�Z?d�dEdF�Z@d�dGdH�ZAd�dIdJ�ZBd�dKdL�ZCd�dMdN�ZDd�dOdP�ZEd�dTdU�ZFd�dZd[�Zd�d^d_�ZGd�dcdd�ZHd�didj�ZIe dkdl�d�dqdr��ZJe d�dudr��ZJddl�d�dwdr�ZJd�d{d|�ZKd�d�d��ZLd�d�d��ZMd�d�d��ZNd�d�d��ZOdS )�� )� annotationsN)�Any)�Iterable)�Iterator)�List)�Optional)�overload)�Tuple)�Type)�TypeVar)�Union)�version)� format_type)�CallExpr)�ClassDef)�CLASSDEF_NO_INFO)�Context)�
Expression)�FuncDef)�IfStmt)�JsonDict)�
MemberExpr)�NameExpr)� Statement)�SymbolTableNode)� TypeAlias)�TypeInfo)�Options)�ClassDefContext)�DynamicClassDefContext)� SemanticAnalyzerPluginInterface)�deserialize_and_fixup_type)�map_type_from_supertype)� CallableType)�get_proper_type)�Instance)�NoneType)� TypeVarType)� UnboundType)� UnionTypec C s g | ] }t �d |�rt|��qS )z^\d+$)�re�match�int)�.0�x� r/ �C/usr/local/lib/python3.10/dist-packages/sqlalchemy/ext/mypy/util.py�
<listcomp>6 s r1 �.)� � � _TArgType)�boundc @ s8 e Zd Zdd d
�Zddd�Zd dd�Zed dd��ZdS ) �SQLAlchemyAttribute�name�str�liner, �column�typ�Optional[Type]�infor �return�Nonec C s" || _ || _|| _|| _|| _d S �N)r8 r: r; �typer> )�selfr8 r: r; r< r> r/ r/ r0 �__init__? s
z SQLAlchemyAttribute.__init__r c C s$ | j sJ �| j| j| jt| j �d�S )N)r8 r: r; rB )rB r8 r: r; �serialize_type)rC r/ r/ r0 � serializeM s
�z SQLAlchemyAttribute.serialize�sub_typec C s& t | jt�sdS t| j|| j�| _dS )zxExpands type vars in the context of a subtype when an attribute is
inherited from a generic super type.
N)�
isinstancerB r' r" r> )rC rG r/ r/ r0 �expand_typevar_from_subtypeV s z/SQLAlchemyAttribute.expand_typevar_from_subtype�data�apir c C s, |� � }t|�d�|�}| d||d�|��S )NrB )r< r> r/ )�copyr! �pop)�clsr> rJ rK r< r/ r/ r0 � deserialize_ s z SQLAlchemyAttribute.deserializeN) r8 r9 r: r, r; r, r< r= r> r r? r@ )r? r )rG r r? r@ )r> r rJ r rK r r? r7 )�__name__�
__module__� __qualname__rD rF rI � classmethodrO r/ r/ r/ r0 r7 > s
r7 r8 r9 r? �boolc C s t t�d| ��S )Nz ^__.+?__$)rT r* r+ )r8 r/ r/ r0 �name_is_dunderk � rU r> r �keyrJ r r@ c C s || j �di �|< d S �N�
sqlalchemy��metadata�
setdefault)r> rW rJ r/ r/ r0 �_set_info_metadatao � r] �
Optional[Any]c C s | j �di ��|d �S rX )r[ �get)r> rW r/ r/ r0 �_get_info_metadatas r^ ra c C s0 | j r| j D ]}t||�}|d ur| S qd S rA )�mrora )r> rW �baser[ r/ r/ r0 �_get_info_mro_metadataw s
�rd c C s | j �di � d S rX rZ �r> r/ r/ r0 �establish_as_sqlalchemy� s rf c C � t | dd� d S �N�is_baseT�r] re r/ r/ r0 � set_is_base� rV rk c C � t | d�}|du S rh �ra �r> ri r/ r/ r0 � get_is_base� �
ro c C rl rh )rd rn r/ r/ r0 �has_declarative_base� rp rq c C rg �N� has_tableTrj re r/ r/ r0 �
set_has_table� rV rt c C rl rr rm rn r/ r/ r0 �
get_has_table� rp ru rK r �#Optional[List[SQLAlchemyAttribute]]c C sJ t | d�}|d u r d S g }|D ]}t�| ||�}|�| � |�|� q|S )N�mapped_attributes)ra r7 rO rI �append)r> rK rw �
attributesrJ �attrr/ r/ r0 �get_mapped_attributes� s �
r{ �typ_r
�optionsr c C s t rt| |�S t| �S rA )�mypy_14�_mypy_format_type)r| r} r/ r/ r0 r � s
r ry �List[SQLAlchemyAttribute]c C s t | ddd� |D �� d S )Nrw c S s g | ]}|� � �qS r/ )rF )r- � attributer/ r/ r0 r1 � s z)set_mapped_attributes.<locals>.<listcomp>rj )r> ry r/ r/ r0 �set_mapped_attributes� s
�r� �msg�ctxr c C s d| }| � ||�S )Nz[SQLAlchemy Mypy plugin] %s)�fail)rK r� r� r/ r/ r0 r� � s r� �.Union[ClassDefContext, DynamicClassDefContext]�module� symbol_name�asnamec C s< | j j| j j j}||vr | j j| j| }|||< d S d S rA )rK �modules�
cur_mod_id�names)r� r� r� r� �module_globals�
lookup_symr/ r/ r0 �
add_global� s � �r� .)�
expr_types�callexprr r� �#Optional[Union[CallExpr, NameExpr]]c C � d S rA r/ �r� r8 r� r/ r/ r0 �get_callexpr_kwarg� s r� �!Tuple[TypingType[_TArgType], ...]�Optional[_TArgType]c C r� rA r/ r� r/ r/ r0 r� � s �%Optional[Tuple[TypingType[Any], ...]]c C sR z| j �|�}W n
ty Y d S w | j| }t||d ur |nttf�r'|S d S rA )� arg_names�index�
ValueError�argsrH r r )r� r8 r� �arg_idx�kwargr/ r/ r0 r� � s �
��stmts�Iterable[Statement]�Iterator[Statement]c c sT � | D ]$}t |t�r$t |jd t�r$|jd jdkr$|jd jE d H q|V qd S )Nr ztyping.TYPE_CHECKING)rH r �exprr �fullname�body)r� �stmtr/ r/ r0 �flatten_typechecking� s ����r� �calleer �#Optional[Union[Instance, TypeInfo]]c C s� t | ttf�rFt | jt�r(| jjr&t | jjt�r&t| jjj�}t |t �r&|S d S t | jt
�r=t| jj �}t |t �r;|S d S t | jt �rF| jS d S rA )
rH r r �noder rB r# r$ �ret_typer% r �targetr )r� r� � target_typer/ r/ r0 �type_for_callee s
�r� r< c s� t |t�s|S |jdkr t� t� fdd�|jD �t� g ��S � �|j|�}|durEt |t�rEt |j t
�rE|j }t |� fdd�|jD ��S |S )z�Take the UnboundType that we seem to get as the ret_type from a FuncDef
and convert it into an Instance/TypeInfo kind of structure that seems
to work as the left-hand type of an AssignmentStatement.
r c s g | ]}t � |��qS r/ )�unbound_to_instance)r- �typ_arg�rK r/ r0 r1 + s z'unbound_to_instance.<locals>.<listcomp>Nc s$ g | ]}t |t�rt� |�n|�qS r/ )rH r( r� )r- �argr� r/ r0 r1 ; s � ��) rH r( r8 r� r) r� r& �lookup_qualifiedr r� r r% )rK r< r� �
bound_typer/ r� r0 r� s2
����
�
�� r� rN r �Optional[TypeInfo]c C sD | j tu r |�| j| �}|d u rd S |rt|jt�s J �|jS | j S rA )r> r r� r8 rH r� r )rN rK �symr/ r/ r0 � info_for_clsH s
r� �Union[str, JsonDict]c C s` z| � � W S ty Y nw t| d�r#tdd� | jD ��| _| � � S t| d�r,| �� } | � � S )Nr� c s s& � | ]}t |d �r
|�� n|V qdS )�resolve_string_annotationN)�hasattrr� )r- �ar/ r/ r0 � <genexpr>[ s � �
�
�z!serialize_type.<locals>.<genexpr>r� )rF � Exceptionr� �tupler� r� )r< r/ r/ r0 rE U s
�
�
�rE )r8 r9 r? rT )r> r rW r9 rJ r r? r@ )r> r rW r9 r? r_ )r> r r? r@ )r> r r? rT )r> r rK r r? rv )r| r
r} r r? r9 )r> r ry r� r? r@ )rK r r� r9 r� r r? r@ )
r� r� r� r9 r� r9 r� r9 r? r@ )r� r r8 r9 r� r@ r? r� )r� r r8 r9 r� r� r? r� )r� r r8 r9 r� r� r? r_ )r� r� r? r� )r� r r? r� )rK r r< r
r? r
)rN r rK r r? r� )r< r
r? r� )P�
__future__r r* �typingr r r r r r r r
�
TypingTyper r �mypyr
�
mypy.messagesr r �
mypy.nodesr r r r r r r r r r r r r r � mypy.optionsr � mypy.pluginr r r �mypy.plugins.commonr! � mypy.typeopsr" �
mypy.typesr# r$ r% r& r' r( r) r� � __version__�split�_versr~ r5 r7 rU r] ra rd rf rk ro rq rt ru r{ r� r� r� r� r� r� r� r� rE r/ r/ r/ r0 �<module> s� �
-
� �
2