File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/ext/mypy/__pycache__/decl_class.cpython-310.pyc
o
���g�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( %dCdDd.d/�Z)dEd7d8�Z*dFd;d<�Z+dGd>d?�Z,dHd@dA�Z-dBS )I� )� 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)�utilF�clsr �apir �
is_mixin_scan�bool�return�(Optional[List[util.SQLAlchemyAttribute]]c C s t �| |�}|d u r d S | j�d�rd S t �||�}t �|� |d ur.|s,t�| ||� |S g }| jj sG|j
� � D ] \}}t | ||||� q9n#t �
| jj �D ]}t|t�r]t| |||� qNt|t�rit| |||� qNt| |� |sxt�| ||� t �||� |S )N�builtins)r% � info_for_cls�fullname�
startswith�get_mapped_attributes�establish_as_sqlalchemyr" � re_apply_declarative_assignments�defs�bodyr$ �items�_scan_symbol_table_entry�flatten_typechecking�
isinstancer �!_scan_declarative_assignment_stmtr � _scan_declarative_decorator_stmt�_scan_for_mapped_bases� add_additional_orm_attributes�set_mapped_attributes)r& r' r( �info�mapped_attributes�sym_name�sym�stmt� rC �I/usr/local/lib/python3.10/dist-packages/sqlalchemy/ext/mypy/decl_class.py�,scan_declarative_assignments_and_apply_types0 sD
�
��
�
��
rE �name�str�valuer �
attributes� List[util.SQLAlchemyAttribute]�Nonec C s| t |j�}t|t�s dS d}t�|j�}d}|tjtjtjtj tj
tj hv r4|j r1t |j d �}nWd}nT|tj
u r�|j s?d}nIt |j d �} t| t�rN| j} t| ttf�r�|�| j| �}
|
dur�t|
jt�r�t�|
jtj�r|tt�||
jg �t� g�}n t�|d�|
jj �|� |r�d} t�|| �|| j�| � t t j �}|dur�|jdus�J �|� tj!||jj"|jj#|| j$d�� dS 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>]�rF �line�column�typr> )%r �typer8 r r$ �type_id_for_named_node�MAPPED� RELATIONSHIP�COMPOSITE_PROPERTY�MAPPER_PROPERTY�SYNONYM_PROPERTY�COLUMN_PROPERTY�args�COLUMNr r �lookup_qualifiedrF �node�has_base_type_id�
TYPEENGINEr r# �#extract_python_type_from_typeenginer r% �fail�formatr. r r � special_form�append�SQLAlchemyAttributerN rO r> ) r& r' rF rH rI �
value_type�left_hand_explicit_type�type_id�err�typeengine_argrA �msgrC rC rD r6 r sv
�
�
���
��
���r6 rB r c C s` |j D ]}t|tttf�rt�|�tju r nqdS | jj �
|�}d}t � |j
�rMttj�}t|jj
�}|j|_t|gt|��} |jj| _| | jj |< dS t|jjt�r�|jjj}
t|
t�rft�|
| |�} ndS | tjtj tj tj tj tj hv r�|
j!r�t"|
j!d �}nE| tj#u r�|
j!r�|
j!d } t| t�r�|�$| j
| �}
|
dur�t|
jt%�r�t�&|
jtj'�r�t(t)�*||
jg �t+� g�}n t �,|d�-|
jj.�|
� |du r�d}t �,||�-|jj
�|� ttj�}t|jj
�}|j|_t|t�r�t"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 rL 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>]rM ):�
decoratorsr8 r
r r r$ rR �
DECLARED_ATTRr3 r4 �indexr% �name_is_dunderrF r r rb �varr\ r r rQ �funcr �ret_typer �type_id_for_unbound_typerS rT rU rV rW rX rY r rZ r[ r r] r^ r r# r_ r r` ra r. �unbound_to_instance�
named_type�NAMED_TYPE_SQLA_MAPPED�expr_to_mapped_constructorr
� argumentsrc rd rN rO r> )r& r' rB rI �dec� dec_indexrf �any_� left_node�new_stmt� func_typerg ri rA rj �rvaluerC rC rD r: � s�
��
�� �
���
��
�� �
�� r: r c
C sp |j d }t|t�s dS | jj�|j�}|dusJ �|j}t|t�r$dS ||ju s+J �t|t �s2J �|jdkrG|�
|j �du rEt �
| j� dS |jdkrSt �| j� n1|j�d�r[dS |jdkr�t|j t�snt �|d|� n|j jD ]}t|ttf�r�t�| |||� qrd}d} |js�|jdu r�t|jt�r�|j} |jjd kr�|�d | �}
|
dur�|
jdur�t�|
j�tju r�t |jj d �} |j}n!t |j�} t| t �r�t�| j�tju r�t | j d �} | }n| } d}t|j t �r�|dur�| } n$t|j t ��rt|j j!t"��rt#�$|||| |j j!�} | du �rdS ndS | du�s J �|�%t j&|j|j'|j(| | jd
�� t�)|||| | � dS ) zZExtract mapping information from an assignment statement in a
declarative class.
r N� __abstract__T�
__tablename__�__�_mypy_mapped_attrsz+_mypy_mapped_attrs is expected to be a list�MappedrM )*�lvaluesr8 r
r> r$ �getrF r\ r r �
parse_boolr~ r% � set_is_base�
set_has_tabler/ r r` r5 r r" �apply_mypy_mapped_attr� is_inferredrQ r r[ rR rS r rY r r r �calleer r# �#infer_type_from_right_hand_nameexprrc rd rN rO � apply_type_to_mapped_statement)
r& r' rB rI �lvaluerA r\ �item�left_hand_mapped_typerf �
mapped_sym� node_type�python_type_for_typerC rC rD r9 g s�
�
�
��
���
���
�r9 c C sN t �| |�}|du r dS |jdd� D ]}|j�d�r qt|j|dd� qdS )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/ rE �defn)r& r' r> � base_inforC rC rD r; � s ��r; N)F)r& r r' r r( r) r* r+ ) r& r r' r rF rG rH r rI rJ r* rK )
r& r r' r rB r rI rJ r* rK )
r& r r' r rB r rI rJ r* rK )r& r r' r r* rK ).�
__future__r �typingr r r �
mypy.nodesr r r r r
r r r
r r r r r r r r � mypy.pluginr �
mypy.typesr r r r r r r r r r � r" r# r$ r% rE r6 r: r9 r; rC rC rC rD �<module> sV �
B
Y