관리-도구
편집 파일: decl_base.cpython-38.pyc
U -?�f�C � @ s d Z ddlmZ ddlZddlZddlZddlmZ ddlmZ ddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddl m Z ddl m!Z! dd lm"Z" dd lm#Z# dd l$m%Z% dd l$m&Z& dd l'm(Z( dd!l'm)Z) dd"l*m+Z+ dd#l*m,Z, dd$l*m-Z- dd%l*m.Z. dd&l*m/Z/ dd'l*m0Z0 dd(l1m2Z2 dd)l3m4Z4 dd*l3m5Z5 dd+l6m7Z7 dd,l6m8Z8 dd-l6m9Z9 dd.l6m:Z: d/d0lm;Z; d/dlmZ d/d1lm6Z6 d/d2l<m=Z= d/d3l>m?Z? d/d4l@mAZA d/d5l@mBZB d/d6l6mCZC d/d7lDmEZE d/d8lDmFZF d/d9lDmGZG d/d:lDmHZH d/d;lDmIZI d/d<lDmJZJ e�r6dd=l mKZK dd>l mLZL dd?l$mMZM dd@lNmOZO ddAl mPZP d/dBlQmRZR d/dCl@mSZS d/dDlTmUZU edEedF�ZVe eWef ZXeeedGf e eWef f ZYG dHdI� dIeHe �ZZG dJdK� dKeZe eH�Z[G dLdM� dMeI�Z\dNdOdP�dQdR�Z]dNdSdP�dTdU�Z^dNdVdP�dWdX�Z_d�dNdZdSd[d\�d]d^�Z`d_d`dP�dadb�ZadcdNdddedf�dgdh�Zbdcd_didjdkdl�dmdn�Zce6�ddo�dpdSdq�drds��ZedpdZd_dSdt�dudv�ZfG dwdx� dx�ZgG dydz� dzeg�ZhG d{d|� d|e�ZiG d}d~� d~eg�Zje6�ddo�ddZdpd��d�d���ZkG d�d�� d�ej�ZldNdZd�d�d��d�d��ZmdNdZd�d��d�d��Zndpdpd�d��d�d��Zod�eo_pdZd�d�d��d�d��ZqdS )�z(Internal implementation for declarative.� )� annotationsN)�Any)�Callable)�cast)�Dict)�Iterable)�List)�Mapping)� NamedTuple)�NoReturn)�Optional)�Sequence)�Tuple)�Type)� TYPE_CHECKING)�TypeVar)�Union� )� attributes)� clsregistry)�exc)�instrumentation)� mapperlib)�_O)�attr_is_internal_proxy)�InstrumentedAttribute)�QueryableAttribute)�_is_mapped_class)�InspectionAttr)�CompositeProperty)�SynonymProperty)�_AttributeOptions)�_DCAttributeOptions)�_IntrospectsAnnotations)�_MappedAttribute)� _MapsColumns)�MapperProperty)�Mapper)�ColumnProperty)� MappedColumn)�_extract_mapped_subtype)�_is_mapped_annotation)� class_mapper)�de_stringify_annotation� )�event)�util)� expression)�_NoArg)�Column)�Table)� topological)�_AnnotationScanType)� is_fwd_ref)� is_literal)�Protocol)� TypedDict)�typing_get_args)� _ClassDict)� _RegistryType)�Mapped)� declared_attr)� ClassManager)� NamedColumn)�MetaData)� FromClause�_T)�bound.c @ s: e Zd ZU dZded< ded< ded< dd d �d d �Zd S )�MappedClassProtocolz�A protocol representing a SQLAlchemy mapped class. The protocol is generic on the type of class, use ``MappedClassProtocol[Any]`` to allow any mapped class. �str�__name__� Mapper[_O]� __mapper__rC � __table__r r )�kw�returnc K s d S �N� )�selfrL rO rO �H/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/decl_base.py�__call__e � z MappedClassProtocol.__call__N)rH � __module__� __qualname__�__doc__�__annotations__rR rO rO rO rQ rF Z s rF c @ sV e Zd ZU dZded< ded< ded< ded < d ed < d d �dd�Zd d �dd�ZdS )�_DeclMappedClassProtocolz:Internal more detailed version of ``MappedClassProtocol``.rB �metadatarG � __tablename__� _MapperKwArgs�__mapper_args__�Optional[_TableArgsType]�__table_args__� Optional[_DataclassArguments]�_sa_apply_dc_transforms�None�rM c C s d S rN rO �rP rO rO rQ �__declare_first__q rS z*_DeclMappedClassProtocol.__declare_first__c C s d S rN rO rc rO rO rQ �__declare_last__s rS z)_DeclMappedClassProtocol.__declare_last__N)rH rT rU rV rW rd re rO rO rO rQ rX h s rX c @ sN e Zd ZU ded< ded< ded< ded< ded< ded< ded< d ed < d S ) �_DataclassArgumentszUnion[_NoArg, bool]�init�repr�eq�order� unsafe_hash� match_args�kw_onlyz'Union[_NoArg, Callable[..., Type[Any]]]�dataclass_callableN�rH rT rU rW rO rO rO rQ rf v s rf � Type[Any]z3Optional[Union[_DeferredMapperConfig, Mapper[Any]]]��clsrM c C s0 t �| �rt �| �S t| �r(t| dd�S d S d S )NF)� configure)�_DeferredMapperConfig�has_cls�config_for_clsr r, �rr rO rO rQ �_declared_mapping_info� s rx �boolc C s. t �| �rt| ddd� S t| �r&dS dS dS )a� return True if this class will be used as a superclass to set in 'inherits'. This includes deferred mapper configs that aren't mapped yet, however does not include classes with _sa_decl_prepare_nocascade (e.g. ``AbstractConcreteBase``); these concrete-only classes are not set up as "inherits" until after mappers are configured using mapper._set_concrete_base() �_sa_decl_prepare_nocascadeT��strictFN)rt ru �_get_immediate_cls_attrr rw rO rO rQ �_is_supercls_for_inherits� s �r~ �Optional[Type[Any]]c C s\ | t kr d S | j�dd�rB| jD ] }t|�}|d k r | S q d S t| �}|rT|jS | S d S )N� __abstract__F)�object�__dict__�get� __bases__�"_resolve_for_abstract_or_classical�_dive_for_cls_manager�class_)rr �base_�supZ clsmanagerrO rO rQ r� � s r� FrG z Optional[Any])rr �attrnamer| rM c C s� |dks t �t| t�sdS || jkr.t| |�S | jdd� D ]D}t|�dk }||jkr<|| ksr|rn|| jkr<n |s<t||� S q<dS )a� return an attribute of the class that is either present directly on the class, e.g. not on a superclass, or is from a superclass but this superclass is a non-mapped mixin, that is, not a descendant of the declarative base and is also not classically mapped. This is used to detect attributes that indicate something about a mapped class independently from any mapped classes that it may inherit from. r� Nr T)�AssertionError� issubclassr� r� �getattr�__mro__r� r� )rr r� r| �baseZ_is_classical_inheritsrO rO rQ r} � s( ��� � �r} �Type[_O]zOptional[ClassManager[_O]]c C s& | j D ]}t�|�}|r| S qd S rN )r� r �opt_manager_of_class)rr r� �managerrO rO rQ r� � s � r� r= r< �Optional[_MapperConfig])�registryrr �dict_rM c C s t �| ||d i �S rN )� _MapperConfig� setup_mapping)r� rr r� rO rO rQ �_as_declarative� s r� �Optional[FromClause]r[ rI )r� rr �table� mapper_kwrM c C s t | |||� td|�jS )NzMappedClassProtocol[_O])�_ImperativeMapperConfigr rJ )r� rr r� r� rO rO rQ �_mapper� s r� zsqlalchemy.orm.decl_apir )�objrM c C s t jjj}t| |t jf�S rN )r0 � preloaded� orm_decl_api�_declared_attr_common� isinstanceZ classproperty)r� r� rO rO rQ �_is_declarative_props s r� )r� �namerr rM c C s2 t | �r*t| dd�r&t�d||f � dS dS d S )N� _cascadingFz~@declared_attr.cascading is not supported on the %s attribute on class %s. This attribute invokes for subclasses in any case.T)r� r� r0 �warn)r� r� rr rO rO rQ � _check_declared_props_nocascade s ��r� c @ s� e Zd ZU dZded< ded< ded< ded < ed d d d ddd�dd��Zd ddd�dd�Zdddd�dd�Zd#ddd �d d �Z dd d �d d!�Z d"S )$r� )rr � classname� properties�declared_attr_reg� __weakref__rp rr rG r� z_util.OrderedDict[str, Union[Sequence[NamedColumn[Any]], NamedColumn[Any], MapperProperty[Any]]]r� z Dict[declared_attr[Any], Any]r� r= r� r< r� r[ r� )r� �cls_r� r� r� rM c C s| t �| �}|r*|j|kr*t�d| �d���|j�dd�r<d S t|ddd�pRt|d�}|rht |||||�S t |||||�S d S ) N�Class z, already has been instrumented declarativelyr� Frz Tr{ Z_sa_decl_prepare) r r� r� r �InvalidRequestErrorr� r� r} �hasattrrt �_ClassScanMapperConfig)rr r� r� r� r� r� r� Z defer_maprO rO rQ r� * s: � �� � �z_MapperConfig.setup_mapping)r� r� r� c C sx t �|td�| _|j| _t �� | _i | _|� dd�sNt j | jd|| |j d� n&t �| j�}|rd|jstt�d| j ��d S )Nr� � non_primaryF)�finalizer� �declarative_scanZ init_methodztClass %s has no primary mapper configured. Configure a primary mapper first before setting up a non primary Mapper.)r0 Zassert_arg_type�typerr rH r� � OrderedDictr� r� r� r Zregister_class� constructorr r� � is_mappedr r� )rP r� r� r� r� rO rO rQ �__init__I s& � ��z_MapperConfig.__init__rD )r� �valuerM c C s t �| j�}|�||� |S rN )r �manager_of_classrr Zinstall_member)rP r� r� r� rO rO rQ �set_cls_attributee s z _MapperConfig.set_cls_attribute.� Mapper[Any]�r� rM c C s t � �d S rN )�NotImplementedError�rP r� rO rO rQ �mapj s z_MapperConfig.mapra c C s | � |� d S rN )r� r� rO rO rQ �_early_mappingm s z _MapperConfig._early_mappingN).) rH rT rU � __slots__rW � classmethodr� r� r� r� r� rO rO rO rQ r� s r� c sR e Zd ZdZddddd�� fdd� Zejfdd d �d d �Zdd d �dd�Z� Z S )r� )� local_table�inheritsr= r� r� r[ )r� r� r� r� c sh t � �|||� | �d|�| _tj�: |�dd�sFt�| j | j |j � | � |� | � |� W 5 Q R X d S )NrK r� F)�superr� r� r� r �_CONFIGURE_MUTEXr� r � add_classr� rr �_class_registry�_setup_inheritancer� )rP r� r� r� r� �� __class__rO rQ r� t s � z _ImperativeMapperConfig.__init__r� r� c C s t }| �d|| j| jf|��S )NrJ )r' r� rr r� �rP r� Z mapper_clsrO rO rQ r� � s �z_ImperativeMapperConfig.mapra c C s� | j }|�dd �}|d kr�g }|jD ]0}t|�}|d kr:q$t|�r$||kr$|�|� q$|r�t|�dkrxt�d||f ��|d }nt |t �r�|j }|| _ d S )Nr� r �&Class %s has multiple mapped bases: %rr ) rr r� r� r� r~ �append�lenr r� r� r' r� r� )rP r� rr r� �inherits_searchr� �crO rO rQ r� � s* �� z*_ImperativeMapperConfig._setup_inheritance) rH rT rU r� r� r0 � EMPTY_DICTr� r� � __classcell__rO rO r� rQ r� q s r� c @ sF e Zd ZU ded< ded< ded< ded< d ed < d ed < d ed< dS )�_CollectedAnnotationr6 �raw_annotationzOptional[Type[Mapped[Any]]]�mapped_containerzUnion[Type[Any], str]�extracted_mapped_annotationry � is_dataclassr � attr_valuerG �originating_modulerp �originating_classNro rO rO rO rQ r� � s r� c s2 e Zd ZU dZdZded< ded< ded< d ed < d ed < d ed < ded< ded< ded< ded< ded< ded< ded< d ed < d ed < d ed < d ed!< d"ed#< d ed$< dd%dd d&d'�� fd(d)� Zd*d+�d,d-�Zd%d.d/�d0d1�Zd2d3d4d5hZ e � d6�Z d7d8d/�d9d:�Z d*d+�d;d<�Zd*d+�d=d>�Zed%d?d@�dAdB��ZedCd%d?d*dD�dEdF��ZedCd*dG�dHdI��ZdJdKd7dLdMdNdO�dPdQ�Zd7dJdMd*dR�dSdT�Zd8d.d d7dUdV�dWdX�Zd*d+�dYdZ�Zd*d+�d[d\�Zdpd d*d^�d_d`�Zdadbdc�ddde�Zd&d*df�dgdh�Zd&d*df�didj�Z d&d*df�dkdl�Z e j fd&dmdf�dndo�Z � Z!S )qr� )r� � clsdict_view�collected_attributes�collected_annotationsr� �persist_selectable�declared_columns�column_ordering� column_copies� table_args� tablename� mapper_args�mapper_args_fn�table_fnr� �single�allow_dataclass_fields�dataclass_setup_arguments� is_dataclass_prior_to_mapping�allow_unmapped_annotationsFr= r� r< r� z Dict[str, _CollectedAnnotation]r� �Dict[str, Any]r� r� r� r� z util.OrderedSet[Column[Any]]r� zDict[Column[Any], int]r� zRDict[Union[MappedColumn[Any], Column[Any]], Union[MappedColumn[Any], Column[Any]]]r� z Optional[str]r� �Mapping[str, Any]r� r] r� z&Optional[Callable[[], Dict[str, Any]]]r� r r� ry r� r� r� r_ r� r� r� r[ )r� r� r� r� r� c s@ |rt �|�nt j| _t� �|||� || _d | _i | _i | _ t � � | _ i | _ i | _ d| _t| jdd � | _}t| jdd�p�t| j�| _t�|� | _}t|d�}|r�|r�|r�t�d��t|o�|�| _| �� | � � | � � t j �R t �!| j"| j|j#� | �$|� | �%� | �&� | �'|� | �(|� | �)|� W 5 Q R X d S )NFr` Z__allow_unmapped__� __sa_dataclass_metadata_key__z�SQLAlchemy mapped dataclasses can't consume mapping information from dataclass.Field() objects if the immediate class is not already a dataclass.)*r0 Z immutabledictr� r� r� r� r� r� r� r� Z OrderedSetr� r� r� r� r� rr r� ry r� � dataclassesr� r� r} r r� r� �_setup_declared_events�_scan_attributes� _setup_dataclasses_transformsr r� r r� r� r� �_setup_inheriting_mapper� _extract_mappable_attributes�_extract_declared_columns� _setup_table�_setup_inheriting_columnsr� ) rP r� r� r� r� r� ZdcaZcldZsdkr� rO rQ r� � s^ � � ��� � � z _ClassScanMapperConfig.__init__ra rb c sX t � jd�r*t�td�dd�� fdd� �}t � jd�rTt�td�dd�� fd d � �}d S ) Nre �after_configuredra rb c s t d� j��� d S �N� _DeclMappedClassProtocol[Any])r rr re rO rc rO rQ r� I s �zG_ClassScanMapperConfig._setup_declared_events.<locals>.after_configuredrd �before_configuredc s t d� j��� d S r� )r rr rd rO rc rO rQ r� Q s �zH_ClassScanMapperConfig._setup_declared_events.<locals>.before_configured)r} rr r/ Z listens_forr' )rP r� r� rO rc rQ r� F s z-_ClassScanMapperConfig._setup_declared_eventszCallable[[str, Any], bool]rq c s� | j rt�d��nd��s2dddd��fdd� }nR�fd d �t���D ���fd d �t���D ��t� � dddd�� ���fd d� }|S ) z�Produce a function that checks if a class has overridden an attribute, taking SQLAlchemy-enabled dataclass fields into account. r� NrG r ry )�keyr� rM c s t � | |�|k S rN )r� )r� r� rw rO rQ �attribute_is_overriddenh s zR_ClassScanMapperConfig._cls_attr_override_checker.<locals>.attribute_is_overriddenc s$ i | ] }� |j kr|j|j � �qS rO �rY r� ��.0�f��sa_dataclass_metadata_keyrO rQ � <dictcomp>l s � zE_ClassScanMapperConfig._cls_attr_override_checker.<locals>.<dictcomp>c s$ i | ] }� |j kr|j|j � �qS rO r r r rO rQ r q s � c s� t |�r|j}��| � �}t |�r(|j}||kr4dS |� k r@dS ��| � �}t�| |�}||krddS |� k rzt|t�rzdS ||kr�dS |� k r�dS dS )NFT)r� �fgetr� r� r� r )r� r� �retZ all_field)�absent�all_datacls_fieldsrr �local_datacls_fieldsrO rQ r� y s. �)r� r} r0 �dataclass_fields�local_dataclass_fieldsr� )rP rr r� rO )r r rr r r rQ �_cls_attr_override_checkerW s" � � � (z1_ClassScanMapperConfig._cls_attr_override_checkerrK r\ rZ r^ z ^(?:_sa_|__)rp z2Callable[[], Iterable[Tuple[str, Any, Any, bool]]]c s� t �|��t|��| j� | j�� �fdd�t �t��t���D ��| jrTt|d��nd��stdd����fdd� }n4d d � t � |�D ����dd�������fd d� }|S ) z�produce a function to iterate the "attributes" of a class which we want to consider for mapping, adjusting for SQLAlchemy fields embedded in dataclass fields. c s" g | ]}�� |�r|� kr|�qS rO )�match)r �n)�_include_dunders�_match_exclude_dundersrO rQ � <listcomp>� s �z=_ClassScanMapperConfig._cls_attr_resolver.<locals>.<listcomp>r� Nz$Iterable[Tuple[str, Any, Any, bool]]rb c s � �fdd��D �S )Nc 3 s& | ] }|�� |�� � |�d fV qdS )FN�r� )r r� )�cls_annotations�cls_varsrO rQ � <genexpr>� s ��z`_ClassScanMapperConfig._cls_attr_resolver.<locals>.local_attributes_for_class.<locals>.<genexpr>rO rO )r r �namesrO rQ �local_attributes_for_class� s �zM_ClassScanMapperConfig._cls_attr_resolver.<locals>.local_attributes_for_classc S s i | ] }|j |�qS rO )r� )r �fieldrO rO rQ r � s z=_ClassScanMapperConfig._cls_attr_resolver.<locals>.<dictcomp>c 3 sd �D ]Z} �� | d �}|rD�|jkrD|jt|j��� � |j�dfV q| �� | �� � | �dfV qd S )NTF)r� rY r� �_as_dc_declaredattr)r� r )r r r � fixed_sa_dataclass_metadata_keyr r rO rQ r � s � ���) r0 �get_annotations�varsr r Zmerge_lists_w_ordering�listr� r} r )rP rr r rO )r r r r r r r r rQ �_cls_attr_resolver� s0 �� �� z)_ClassScanMapperConfig._cls_attr_resolverc s� | j }td|�� | j}| j}| j}| j}d }d }}d } d } d|k} | �| j �} g } |jD ]R}||k olt|�}| � |�}|s�||k r�| � || | |�}ni }| � ||||f� qZ| D �]�\}}}}|� |� |� D �]�\}}}}||k�r�|dk�r(t |||�}|�s�|�r|�r�dd�� fdd� }|}n�|dk�rXt |||�}| �s�|�rP|�r�� j} n�|dk�r�t |||�}|�r�| �s�d d�� fd d � }|} nh|d k�r�t |||�}|�s�|�r�|�r�� j}t|tttd �f��s�t�d ��||k �r�d}n d�s�t�q�|�r,t|�r�|js�t�d|j|||f � q�q�||k �r�t|t t f��rLqАq�t|t ��rft� d���q�t|��rt|d k �s~t�|j �r�||k�r�t�d||f � |�!||� ||< ||< }t"|||� nl|�r�t#||d �}t|t$��s�|�%� }n t#||�}t|t$��r(t&|��r(t|j't ��s(|j(}| ||< ||< t|t t f��r\|j)d k�r\|j*|_)| �+||�,� |d|� n:t-|||��r�| s�||ks�| |d �s�t�q�n| �.|||� q�|�r ||k�s�|| |k �r | ||��r�t�t|��r�|�%� }|||< | �+|||d|� q�| �+|||d |�}|d k �o0|j/d k }|d k �rB|j0n|} |d k�rv| �sv|�rv| d k �rj| nt � ||< q�||kr�|||< q�q�|�r�| �s�d }|| _1| | _2|| _3| | _4d S )Nr� rK r\ r� rb c s t � j�S rN )�dictr\ rO �� cls_as_DeclrO rQ �_mapper_args_fnA s z@_ClassScanMapperConfig._scan_attributes.<locals>._mapper_args_fnrZ rC c s � j S rN )rK rO r" rO rQ � _table_fnW s z:_ClassScanMapperConfig._scan_attributes.<locals>._table_fnr^ z3__table_args__ value must be a tuple, dict, or NoneTFz�Regular (i.e. not __special__) attribute '%s.%s' uses @declared_attr, but owning class %s is mapped - not applying to subclass %s.z�Mapper properties (i.e. deferred,column_property(), relationship(), etc.) must be declared as @declared_attr callables on declarative mixin classes. For dataclass field() objects, use a lambda:zXAttribute '%s' on class %s cannot be processed due to @declared_attr.cascading; skipping)5rr r r� r� r� r r r� r~ r �_produce_column_copiesr� �updater� rZ r^ r� �tupler! r� r � ArgumentErrorr� r� Z_quietr0 r� rH r3 r) r& r� r� �__get__�setattrr� r r r Zoriginal_propertyZ descriptor�docrV �_collect_annotationZ_collect_return_annotationr+ �_warn_for_decl_attributesr� r� r� r� r� r� ) rP rr r� r� r� r r� r� Zinherited_table_argsr� r� � fixed_tabler� �basesr� Z class_mappedr Zlocally_collected_columnsr� r� � annotation�is_dataclass_fieldZ check_declr$ r% r �collected_annotationr� Z generated_objrO r" rQ r� � s� ���� � ��� � � � �� �� � �� � �� ��� ������ � � � �� ��� z'_ClassScanMapperConfig._scan_attributesc sp �j }|sd S d�jjkr.t�d�j� d���t�t��dddd���fdd � � t� �j�}|d k sht ��fd d �� �fd d ��j � � D �D �}�r܈� � D ]>\}}t jd�j� dd�dd � |D ��� d|� d�ddd� q�i }i }|D ]P}t|�dk�r|\} } n,t|�dk�r&|\} } } | || < n d�s0t �| || < q�|� � D ]\} } t�j| | � �qB��|�j|� d S )N�__dataclass_fields__r� z� is already a dataclass; ensure that base classes / decorator styles of establishing dataclasses are not being mixed. This can happen if a class that inherits from 'MappedAsDataclass', even indirectly, is been mapped with '@registry.mapped_as_dataclass'rG rp ry )r� r� rM c s& |� j k r"d|jkr"�| �| � dS )Nr4 T)rr r� r� )r� r� )rP �warn_for_non_dc_attrsrO rQ �_allow_dataclass_field0 s ��zT_ClassScanMapperConfig._setup_dataclasses_transforms.<locals>._allow_dataclass_fieldc s. g | ]&\}}}t �|||� j�|tj���qS rO )r! Z!_get_arguments_for_make_dataclassr� r� r2 �NO_ARG)r r� �annor� rc rO rQ r > s ��zH_ClassScanMapperConfig._setup_dataclasses_transforms.<locals>.<listcomp>c 3 sV | ]N\}\}}}}}}}� ||�r|�j ks<t�j | t�s||rF|n||fV qd S rN )r� r� r ) r r� Zraw_annor� Z mapped_annoZis_dcr� r� r� )r6 rP rO rQ r E s$ �� �zG_ClassScanMapperConfig._setup_dataclasses_transforms.<locals>.<genexpr>zWhen transforming z to a dataclass, attribute(s) �, c s s | ]}t |�V qd S rN �rh )r r� rO rO rQ r k s z originates from superclass a , which is not a dataclass. This usage is deprecated and will raise an error in SQLAlchemy 2.1. When declaring SQLAlchemy Declarative Dataclasses, ensure that all mixin classes and other superclasses which include attributes are also a subclass of MappedAsDataclass.z2.0Zdcmx��coder. � F)r� rr r� r r� � collections� defaultdictr r r� r� r� �itemsr0 Zwarn_deprecated�joinr� r+ � _apply_dataclasses_to_any_class)rP r� r� � field_listr� Z non_dc_attrsr �defaults�itemr� �tp�spec�k�vrO )r6 rP r5 rQ r� sV � ��'�( � �z4_ClassScanMapperConfig._setup_dataclasses_transformsz!Mapping[str, _AnnotationScanType])�klassrM c C sl t �|�}i }|�� D ]P\}}t|||�r^t|||j|td �dddd�}|rf|\}}|||< q|||< q|S )NF��requiredr2 � expect_mapped)r0 r r@ r+ r* rT r� ) rr rJ r Znew_annor� r1 � extracted�inner�_rO rO rQ �(_update_annotations_for_non_mapped_class� s& � z?_ClassScanMapperConfig._update_annotations_for_non_mapped_classrf )r� rJ �use_annotationsrM c C s� | � |� |d }|tjkr"tj}|r@t|dd �}td|�|_nd }zjz ||fdd� |�� D �� W nF t t fk r� } z$t j d|j �d|�d�d d �|�W 5 d }~X Y nX W 5 |r�|d kr�|`n||_X d S ) Nrn rW r� c S s( i | ] \}}|t jk r|d kr||�qS )rn )r2 r7 )r rH rI rO rO rQ r � s � zJ_ClassScanMapperConfig._apply_dataclasses_to_any_class.<locals>.<dictcomp>zAPython dataclasses error encountered when creating dataclass for z: zN. Please refer to Python dataclasses documentation for additional information.Zdcter; )�_assert_dc_argumentsr2 r7 r� � dataclassr� r rW r@ � TypeError� ValueErrorr r� rH )rr r� rJ rR rn Zrestored�exrO rO rQ rB � s6 ����� z6_ClassScanMapperConfig._apply_dataclasses_to_any_class)� argumentsrM c C sT ddddddddh}t |��|�}|rPd �d d � t|�D ��}t�d |� d ���d S )Nrg rh rj ri rk rm rl rn r9 c s s | ] }|�V qd S rN rO )r �argrO rO rQ r � s z>_ClassScanMapperConfig._assert_dc_arguments.<locals>.<genexpr>zDataclass argument(s) z are not accepted)�set� differencerA �sortedr r) )rr rX �allowedZdisallowed_args�msgrO rO rQ rS � s � �z+_ClassScanMapperConfig._assert_dc_argumentsrG r6 zOptional[bool]r z Optional[_CollectedAnnotation])r� r� r� rM r� rM c C s || j kr| j | S |d kr d S | j}| j}|d kr`t|tj�}| o\| o\|d kp\t|t�}nd}d}t|| j|j |t |�d||o�| d�} | d kr�d S | \} } |d kr�t | �s�t | �D ]@} t| t �s�t| dd�r�t| j| |j dd�} t| t�r�| �� }q�t|| | |||j |� | j |< } | S )NFrK T)Z check_generic)Zinclude_generic)r� r� r� r� r� �Fieldr$ r* rr rT r� r8 r; rG r7 r- r# Zfound_in_pep593_annotatedr� )rP r� r� r� rM r� r� Zallow_unmappedr2 rN r� r� �elem�carO rO rQ r- � sl �� � �� � z*_ClassScanMapperConfig._collect_annotation)rr r� r� rM c C s( t |tj�r$t�d|� d|� d�� d S )N� Attribute '� ' on class z� appears to be a non-schema SQLAlchemy expression object; this won't be part of the declarative mapping. To map arbitrary expressions, use ``column_property()`` or a similar function such as ``deferred()``, ``query_expression()`` etc. )r� r1 � ColumnElementr0 r� )rP rr r� r� rO rO rQ r. : s �z0_ClassScanMapperConfig._warn_for_decl_attributesz0Dict[str, Union[Column[Any], MappedColumn[Any]]])�attributes_for_classr� r/ r� rM c C sd | j }| j}i }| j}|� D �]@\} } } } |s�| d kr�t| ||�r�|| | �rNq | �| | |d| �} | d k rn| jn| } | d kr�t� } | || < t|| | � q t| t tf�r || | �r�q | �| | |d| �} | d k r�| jn| } | |kr d|k�r t | dd �p�| |d j ks | j �r:| j D ]*}|j d k �r|j jd k�rt�d���q| �� || < }||| < t|| |� q |S )NTrK r� z�Columns with foreign keys to non-table-bound columns must be declared as @declared_attr callables on declarative mixin classes. For dataclass field() objects, use a lambda:.)rr r� r� r+ r- r� r) r+ r� r3 r� r� Z foreign_keys� _table_columnr� r r� Z_copy)rP re r� r/ r� rr r� Z locally_collected_attributesr� r� r� r1 r� r3 ZfkZcopy_rO rO rQ r& G s| �� � ��� ������ � �� z-_ClassScanMapperConfig._produce_column_copiesc s� | j }| j}| j}| j}t|ddd�}| j}|p4| j}t| j�}t |�D �]�} | |krXqH|| � t � �r�� j r~t � d| j � t|| �� n4t� t�r�� j|k r�� j| kr�t� j�� t|| � � t� t�r�t� �dkr�t� d ttf�r�t � d| � qH�n�|�rt� tj��rqH�n�t� ttf��sR|� | � | � || � � |sHt|| � � qH�n�| dk�rrt � d | � d ����nvt� t��r�t | | j!�"� � �� �nRt� t#��r>| j$�"| d �\} } } } }}}| d k �s�| d k�s�|�r4z � �%| | j&||| | | | | � W n8 t'k �r0 } zt �(d | � d ��|�W 5 d }~X Y nX n |�s>t)�t� t��r� j*�r�|�s�t� t+��rrddddg}n dddg}� fdd�|D �}t �(d| � d|� dd�,t-dd� |D ���� d���t� t+t.f��s�|� | � t|| � � qH� || < qHd S )Nrz Tr{ z�Use of @declared_attr.cascading only applies to Declarative 'mixin' and 'abstract' classes. Currently, this flag is ignored on mapped class %sr r z�Ignoring declarative-like tuple value of attribute '%s': possibly a copy-and-paste error with a comma accidentally placed at the end of the line?)rY zAttribute name 'z-' is reserved when using the Declarative API.)NNNFNNNz7Could not resolve all types within mapped annotation: "zU". Ensure all types are written correctly and are imported within the module in use.rg �default_factoryrh �defaultc s( h | ] }t � jd |� ��tjk r|�qS )Z dataclasses_)r� Z_attribute_optionsr2 r7 �r �a�r� rO rQ � <setcomp>@ s ���zF_ClassScanMapperConfig._extract_mappable_attributes.<locals>.<setcomp>rb rc z# includes dataclasses argument(s): r9 c s s | ]}t |�V qd S rN r: ri rO rO rQ r J s zF_ClassScanMapperConfig._extract_mappable_attributes.<locals>.<genexpr>zF but class does not specify SQLAlchemy native dataclass configuration.)/rr r� r� r r} r� r� ry r� r r� r� r0 r� r� r� r r� r� r r+ r( r� r3 r$ r� r_ r" �popr. r r� �_undefer_column_namer� r� r# r� r� r� � NameErrorr) r� Z_has_dataclass_argumentsr& rA r\ r% )rP rr r� � our_stuffr Z late_mappedr� Z expect_annotations_wo_mappedZlook_for_dataclass_thingsrH r1 r� r� r� r� r� r� �ne�argnames�argsrO rk rQ r� � s� �� �� ��� � ���� � � � �� ��� � �� �� � � *� z3_ClassScanMapperConfig._extract_mappable_attributesc C s2 | j }| j}| j}t�t�}t|�� �D ]�\}}t|t �r�|j }|rN|||< n||= |j D ]N\}} t|t �s|||j �|� |�|� | ||< |d krZ||jkrZ|||< qZq(t|t�r(|j d k s�t�||j �|� |�|� ||jkr(||= q(|�� D ]4\} } t| �dkr�t�d| j| d�t| ��f � q�d S )Nr zzOn class %r, Column object %r named directly multiple times, only one will be used: %s. Consider using orm.synonym insteadr9 )r� r� r� r> r? rZ r r@ r� r% �mapper_property_to_assign�columns_to_assignr r� �addr� r3 r� r� r0 r� r� rA r\ ) rP rp r� r� Zname_to_prop_keyr� r� Z mp_to_assign�colZ sort_orderr� �keysrO rO rQ r� f s< ��z0_ClassScanMapperConfig._extract_declared_columnsN)r� rM c s� | j }td|�}| j}| j}| j}| j}| j� t�|�}| j d k�r>d|k�r>|d k�r>t |d�rztt t t �|j��} nt } |d k �r�d} i } |r�t|t�r�|} n4t|t�r�t|d t�r�|dd� |d } } n|} |�d�} | r�| | d< |�d �} | �rd | d < t|� fd d �d�}| �d| || �|�f|| �| ��}nZ|d k�rh| j �rb| �d| � � �}n|j}|�r�|D ]$}|j�|��srt�d|j ���qr|| _ d S )N�MappedClassProtocol[Any]rK � __table_cls__rO ���r Z__autoload_with__� autoload_withZ __autoload__T�autoloadc s � � | d�S )Nr r )r� �r� rO rQ �<lambda>� rS z5_ClassScanMapperConfig._setup_table.<locals>.<lambda>�r� z8Can't add additional column %r when specifying __table__) rr r r� r� r� r� r� r r� r� r� r r4 r0 �unbound_method_to_callablerz r� r! r( r� r\ r� �_metadata_for_clsrK r� Zcontains_columnr r) r� r� )rP r� rr r# r� r� r� r� r� Z table_clsrs Ztable_kwr| r} Zsorted_columnsr� rO r~ rQ r� � s� ��� � ����� ���z#_ClassScanMapperConfig._setup_tablezClassManager[Any]rB )r� rM c C s& t | jdd �}|d k r|S |jjS d S )NrY )r� rr r� rY )rP r� �metarO rO rQ r� � s z(_ClassScanMapperConfig._metadata_for_clsr� c C s� | j }|�dd �}|d kr�g }|jD ]0}t|�}|d kr:q$t|�r$||kr$|�|� q$|r�t|�dkrxt�d||f ��|d }nt |t �r�|j }|| _ | j }d|kr�| jd kr�d| _d S )Nr� r r� r rK T)rr r� r� r� r~ r� r� r r� r� r' r� r� r� r� r� )rP r� rr r� r� r� r� r� rO rO rQ r� � s0 �� z/_ClassScanMapperConfig._setup_inheriting_mapperc C s` | j }| j}| j}| j}|d krF| jd krFt|d�sFt�d| ���n| j�r\t| j�}|d k sdt �|j }|j }|d k�r\|r�t� d��|r�t |t �s�t� d| j� d| j� d���|D ]�} |d k s�t �| j|jk�r|j| j | kr�q�t� d| � d|j� d |j| j � d ���| j�r t� d ��t�r6t |t ��s6t �|�| � |d k r�||k r�|�| � q�d S ) NZ __no_table__zwClass %r does not have a __table__ or __tablename__ specified and does not inherit from an existing table-mapped class.z?Can't place __table_args__ on an inherited class with no table.z9Can't declare columns on single-table-inherited subclass z ; superclass z is not mapped to a TablezColumn 'rc z! conflicts with existing column 'zs'. If using Declarative, consider using the use_existing_column parameter of mapped_column() to resolve conflicts.zDCan't place primary key columns on an inherited class with no table.)r� rr r� r� r� r} r r� rx r� r� r) r� r4 r� r� rH � primary_keyr � append_columnZ_refresh_for_new_column) rP r� r� rr r� r� Zinherited_mapper_or_config�inherited_tableZ inherited_persist_selectablerw rO rO rQ r� sh ����� � �� �� ���z0_ClassScanMapperConfig._setup_inheriting_columnsc s� �j }�jr��� }ni }|r(|�|� d|krFt|�}|�|d � dD ]&}||krJ|| }�j�||�||< qJd|kr��fdd�t�|d �D �|d< d|kr�|d }t|t �r�|j }|�j k r�t � d|d ���j r�j |d< �j �r�|�dd ��s�t�j d �� � j}d |k�rX� fd d �|jD ��� j�p6d � |d < }|�dd� �jD �� t|�� �D ]N\}} t| tj��s~�qd|� jk�rd� j| } t| t��rd| g| j ||< �qd|� � } || d< | �_ d S )Nr� )Zversion_id_colZpolymorphic_onr� c s g | ]}� j �||��qS rO )r� r� )r rI rc rO rQ r v s �zD_ClassScanMapperConfig._prepare_mapper_arguments.<locals>.<listcomp>r� z:mapper inherits argument given for non-inheriting class %sZconcreteF�exclude_propertiesc s h | ]}|� j kr|j�qS rO )Z_columntopropertyr� �r r� )�inherited_mapperrO rQ rl � s �zC_ClassScanMapperConfig._prepare_mapper_arguments.<locals>.<setcomp>rO c S s g | ] }|j �qS rO r� r� rO rO rQ r � s ) r� r� r' r! r� r� r0 Zto_listr� r' r� r� r r� r, r� r� �unionr� �difference_updater� r r@ r1 rd Z_propsr( �columns�copyr� ) rP r� r� r� rH rI Z inherits_argr� r� rw �pZresult_mapper_argsrO )r� rP rQ �_prepare_mapper_arguments_ sf � �� � � � z0_ClassScanMapperConfig._prepare_mapper_argumentsr� c C sL | � |� t| jd�r,tdt�| jj��}nt}| �d|| j| j f| j ��S )N�__mapper_cls__zType[Mapper[Any]]rJ ) r� r� rr r r0 r� r� r' r� r� r� r� rO rO rQ r� � s ���z_ClassScanMapperConfig.map)N)"rH rT rU r� � is_deferredrW r� r� r r �re�compiler r r� r� r� rQ rB rS r- r. r& r� r� r� r� r� r� r� r0 r� r� r� rO rO r� rQ r� � sj MM� D 0n4L Z F;N!HQr� r� )�field_metadatar rM c C s6 t jj}| | }t|�r.t||j�s.|�|�S |S d S rN )r0 r� r� �callabler� r? )r� r �decl_apir� rO rO rQ r � s r c s� e Zd ZU ded< dZe�� Zded< ddd�d d �Ze d d �d d��Z e j d dd�dd��Z e ddd�dd��Z e d dd�dd��Ze d dd�dd��Ze d d d�dd��Ze d#d dd d �d d ��Zejfdd d�� fd!d"� Z� ZS )$rt zweakref.ref[Type[Any]]�_clsTz?util.OrderedDict[weakref.ref[Type[Any]], _DeferredMapperConfig]�_configsr[ ra r� c C s d S rN rO r� rO rO rQ r� � s z$_DeferredMapperConfig._early_mappingrp rb c C s | � � S rN )r� rc rO rO rQ rr � s z_DeferredMapperConfig.cls)r� rM c C s t �|| j�| _| | j| j< d S rN )�weakref�ref�_remove_config_clsr� r� )rP r� rO rO rQ rr � s )r� rM c C s | j �|d � d S rN )r� rm )rr r� rO rO rQ r� � s z(_DeferredMapperConfig._remove_config_clsry c C s t |t�ot�|�| jkS rN )r� r� r� r� r� �rr r� rO rO rQ ru � s z _DeferredMapperConfig.has_clsr c C s2 t |d�r|�� tj|dt�|�� d�d��d S )N�_sa_raise_deferred_configr� zG has a deferred mapping on it. It is not yet usable as a mapped class.)r^ )r� r� �orm_excZUnmappedClassErrorZ_safe_cls_namer� rO rO rQ �raise_unmapped_for_cls� s �z,_DeferredMapperConfig.raise_unmapped_for_clsc C s | j t�|� S rN )r� r� r� r� rO rO rQ rv � s z$_DeferredMapperConfig.config_for_clszList[_DeferredMapperConfig])�base_cls�sortrM c sr �fdd�dd� | j �� D �D �}|s*|S dd� |D �� g }� D ] �|�� �fdd��jD �� q@tt�||��S )Nc s& g | ] \}}|d k rt |� �r|�qS rN )r� )r �mr� )r� rO rQ r s �z:_DeferredMapperConfig.classes_for_base.<locals>.<listcomp>c S s g | ]}||j f�qS rO rw �r r� rO rO rQ r s c S s i | ] }|j |�qS rO rw r� rO rO rQ r s z:_DeferredMapperConfig.classes_for_base.<locals>.<dictcomp>c 3 s& | ] }|� kr� | � � fV qd S rN rO )r r� )� all_m_by_cls�m_clsrO rQ r s �z9_DeferredMapperConfig.classes_for_base.<locals>.<genexpr>)r� �values�extendr� r r5 r� )rr r� r� �classes_for_baseZtuplesrO )r� r� r� rQ r� s �� z&_DeferredMapperConfig.classes_for_baser� c s | j �| jd � t� �|�S rN )r� rm r� r� r� r� r� rO rQ r� s z_DeferredMapperConfig.map)T)rH rT rU rW r� r0 r� r� r� �propertyrr �setterr� r� ru r� rv r� r� r� r� rO rO r� rQ rt � s* � �rt zMapperProperty[Any]ra )rr r� r� rM c C s6 d| j k�r$td| �}ddd�dd�}t|t�r\t||� ||�j|dd� |j�||� n�t|t�r�|j }|j D ]6\}}t||� ||�j|dd� |sr|j�||� qr|r�|j�||� ndt|t �r�|j�||� nJt|t ��r |j |k�r t|j �}|j�||� nt�| ||� |j�� nt�| ||� d S ) z�add an attribute to an existing declarative class. This runs through the logic to determine MapperProperty, adds it to the Mapper, adds a column to the mapped Table, etc. rJ ry r4 )�mcrM c S s* t | jt�r| jS t�d| j�d���d S )Nz+Cannot add a new attribute to mapped class z) because it's not mapped against a table.)r� rK r4 r r� rH )r� rO rO rQ �_table_or_raise+ s �z'_add_attribute.<locals>._table_or_raiseT)Zreplace_existingN)r� r r� r3 rn r� rJ Z add_propertyr% rt ru r& r r� r r� � __setattr__�_expire_memoizations)rr r� r� Z mapped_clsr� �mprw rP rO rO rQ �_add_attribute s< � � r� )rr r� rM c C sv d| j krf|| j krftd| �jjsf| j | }t|ttttf�rHt d��qrt � | |� td| �j� � n t � | |� d S )NrJ ry z<Can't un-map individual mapped attributes on a mapped class.) r� r rJ Z_dispose_calledr� r3 r% r& r r� r� � __delattr__r� )rr r� r� rO rO rQ �_del_attributeR s. �� �� �� � r� )rP �kwargsrM c K sB t | �}|D ]0}t||�s,td||jf ��t| ||| � q dS )a= A simple constructor that allows initialization from kwargs. Sets attributes on the constructed instance using the names and values in ``kwargs``. Only keys that are present as attributes of the instance's class are allowed. These could be, for example, any mapped columns or relationships. z(%r is an invalid keyword argument for %sN)r� r� rU rH r+ )rP r� r� rH rO rO rQ �_declarative_constructorj s �r� r� z Column[Any])r� �columnrM c C s$ |j d kr| |_ |jd kr | |_d S rN )r� r� )r� r� rO rO rQ rn � s rn )F)rrV � __future__r r> r� r� �typingr r r r r r r r r r r r r r r r r� � r r r r� r r Z_typingr r r r r� r r Zdescriptor_propsr r Z interfacesr! r" r# r$ r% r& Zmapperr' r� r( r) r0 r* r+ r, r- r/ �sqlr1 Zsql.baser2 Z sql.schemar3 r4 r5 Z util.typingr6 r7 r8 r9 r: r; r< r= r> r� r? r@ Z sql.elementsrA rB Zsql.selectablerC rD rG r[ Z_TableArgsTyperF rX rf rx r~ r� r} r� r� r� Zpreload_moduler� r� r� r� r� r� r rt r� r� r� rH rn rO rO rO rQ �<module> s� �' Z> M4