관리-도구
편집 파일: util.cpython-38.pyc
U -?�f^; � @ s8 d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d dlm Z d d lm Z d d lmZ d d lmZ d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlZddlmZ ddlm Z ddl m Z ddl m Z ddl m Z ddl 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"m+Z+ dd%l"m,Z, dd&l"m-Z- dd'l"m.Z. dd(l"m/Z/ dd)l"m0Z0 dd*l"m1Z1 dd+l"m2Z2 dd,l"m3Z3 dd-l"m4Z4 dd.l5m6Z6 dd/l5m7Z7 dd0l5m8Z8 dd1l5m9Z9 dd2l5m:Z: dd3l;m<Z< d4d5lm=Z= d4dlm Z> d4d6lm?Z? d4d7lm@Z@ d4d8lmAZA d4d9lBmCZC d4d:l@mDZD d4d;l@mEZE d4d<l@mFZF d4d=l@mGZG d4d8l@mAZH d4d>l@mIZI d4d?lJmKZK d4d@lLmMZM d4dAlNmOZO d4dBlPmQZQ d4dClPmRZR d4dDlSmTZT d4dElSmUZU d4dFlVmWZW d4dGlXmYZY d4dHlZm[Z\ d4dIlZm]Z^ d4dJlZm_Z` d4dKlZmaZa d4dLlZmbZb d4dMlZmcZc d4dNlZmdZd ej�rddOl meZe ddPl mfZf ddQl mgZg ddRl mhZh ddSlimjZj ddTlimkZk ddUllmmZm ddVl;mnZn ddWlompZp ddXlqmrZr d4dYlsmtZt d4dZlsmuZu d4d[lJmvZv d4d\lJmwZw d4d]lJmxZx d4d^lJmyZy d4d_lJmzZz d4d`lJm{Z{ d4dalLm|Z| d4dblNm}Z} d4dclSm~Z~ d4ddlVmZ d4delVm�Z� d4dflVm�Z� d4dgl�m�Z� d4dhlZm�Z� d4dilZm�Z� edjedk�Z�e�dl�Z�ej�ej�eA��e,e4e)dm��dn�Z�G dodp� dpec�Z�e e�e�e\��Z[G dqdr� drec�Z�e e�e�e^��Z]G dsdt� dtec�Z�e e�e�e`��Z_G dudv� dve e� �Z�dwdx� Z�d�d{d|�Z�d�dddd}�d~dd�d�d�d�d��d�d��Z�G d�d�� d�ej��Z�G d�d�� d�eHj��Z�G d�d�� d�eHj��Z�G d�d�� d�e?j�d� e8e �Z�e?j�G d�d�� d�e9e e:eQe*eYe?j�d� e e � �Z�G d�d�� d��Z�G d�d�� d�e6�Z�e?��e��d�d�� � e?��e��d�d�d��d�d���Z�e�ee �Z�e?��e��d�d�d��d�d���Z�e?j�G d�d�� d�e8e� eMeRe?j�d� e*��Z�d�d�d�d�d��d�d��Z�d�d�d��d�d��Z�d�d�d��d�d��Z�G d�d�� d�eEj��Z�d�d�d�d�d�d��d�d��Z�d�d�d��d�d��Z�d�d�d��d�d��Z�d�d�d�d��d�d��Z�d�d�d�d��d�d��Z�d�d�d�d��d�dÄZ�d�d�d�dƜd�dȄZ�d�d�d�d�d˜d�d̈́Z�G d�dτ d�e��Z�d�d�d�dќd�dӄZ�d�d�d�d�d�d�d�d�d�d�d�dל d�dلZ�d�d�dڜd�d܄Z�dS )�� )� annotationsN)� AbstractSet)�Any)�Callable)�cast)�Dict)� FrozenSet)�Generic)�Iterable)�Iterator)�List)�Match)�Optional)�Sequence)�Tuple)�Type)� TYPE_CHECKING)�TypeVar)�Union� )� attributes)�exc)�_O)�insp_is_aliased_class)�insp_is_mapper)�prop_is_relationship)�_class_to_mapper)�_MappedAnnotationBase)� _never_set)� _none_set)� attribute_str)� class_mapper)� DynamicMapped)�InspectionAttr)� instance_str)�Mapped)� object_mapper)� object_state)�opt_manager_of_class)� ORMDescriptor)�state_attribute_str)�state_class_str)� state_str)�WriteOnlyMapped)�CriteriaOption)�MapperProperty)�ORMColumnsClauseRole)�ORMEntityColumnsClauseRole)�ORMFromClauseRole)� PathRegistry� )�event)� inspection)�sql)�util�� result_tuple)� coercions)� expression)�lambdas)�roles)�visitors)� is_selectable)�SupportsCloneAnnotations)�ColumnCollection)� HasCacheKey)�MemoizedHasCacheKey)� ColumnElement)�KeyedColumnElement)� FromClause)� MemoizedSlots)�de_stringify_annotation)�de_stringify_union_elements)�eval_name_only)�is_origin_of_cls)�Literal)�Protocol)�typing_get_origin)� _EntityType)�_IdentityKeyType)�_InternalEntityType)� _ORMCOLEXPR)� _MapperEntity)�ORMCompileState)�Mapper)�AbstractEntityRegistry)�Query)�RelationshipProperty)�Row)� RowMapping)�_CE)�_ColumnExpressionArgument)�_EquivalentColumnMap)�_FromClauseArgument)�_OnClauseArgument)�_PropagateAttrsType)�_SA)�ReadOnlyColumnCollection)� BindParameter)�_ColumnsClauseElement)�Select)� Selectable)�anon_map)�_AnnotationScanType)�ArgsTypeProcotol�_T)�bound)�delete� delete-orphan�all�merge�expunge� save-update�refresh-expire�none)r% r- r" )Zlocals_c @ s, e Zd Zddd�ddddddd �d d �ZdS ) �_DeStringifyAnnotationNF)�str_cleanup_fn�include_generic� Type[Any]ri �str�#Optional[Callable[[str, str], str]]�bool)�cls� annotation�originating_modulerv rw �returnc C s d S �N� )�selfr| r} r~ rv rw r� r� �C/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/util.py�__call__� s z _DeStringifyAnnotation.__call__��__name__� __module__� __qualname__r� r� r� r� r� ru � s �ru c @ s( e Zd Zdd�dddddd�dd �ZdS ) �_DeStringifyUnionElementsN�rv rx rj ry rz )r| r} r~ rv r c C s d S r� r� )r� r| r} r~ rv r� r� r� r� � s z"_DeStringifyUnionElements.__call__r� r� r� r� r� r� � s �r� c @ s e Zd Zdddd�dd�ZdS )� _EvalNameOnlyry r )�name� module_namer c C s d S r� r� )r� r� r� r� r� r� r� � � z_EvalNameOnly.__call__Nr� r� r� r� r� r� � s r� c s� e Zd ZU dZe�dddg�ZeZdddddgZdZ d e d < d e d < d e d < d e d< d e d< d e d < dd d�� fdd� Z dd� Z e dd� �Z� ZS )�CascadeOptionszHKeeps track of the options sent to :paramref:`.relationship.cascade`ro rt rn rq rs rp )� save_updaterm �refresh_expirerp rq � delete_orphanr{ r� rm r� r� z#Optional[Union[Iterable[str], str]])� value_listr c s� t |t�s|d kr | �|�S t|�}|�| j�rZt�dd�dd� t |�| j��D �� ��d|krn|� | j � d|kr~|� � |� d� t� �| |�}d|k|_d|k|_d |k|_d |k|_d |k|_d |k|_|jr�|js�t�d � |S )Nz Invalid cascade option(s): %s�, c S s g | ] }t |��qS r� )�repr��.0�xr� r� r� � <listcomp>� s �z*CascadeOptions.__new__.<locals>.<listcomp>ro rt rr rm rs rp rq rn z5The 'delete-orphan' cascade option requires 'delete'.)� isinstancery � from_string�set� difference�_allowed_cascades�sa_exc� ArgumentError�join�sorted�update�_add_w_all_cascades�clear�discard�super�__new__r� rm r� rp rq r� r8 �warn)r| r� �valuesr� �� __class__r� r� r� � s< ����� zCascadeOptions.__new__c C s dd� dd� t| �D �� S )NzCascadeOptions(%r)�,c S s g | ]}|�qS r� r� r� r� r� r� r� s z+CascadeOptions.__repr__.<locals>.<listcomp>)r� r� �r� r� r� r� �__repr__ s zCascadeOptions.__repr__c C s" dd� t �d|pd�D �}| |�S )Nc S s g | ] }|r|�qS r� r� )r� �cr� r� r� r� s z.CascadeOptions.from_string.<locals>.<listcomp>z\s*,\s*� )�re�split)r| �argr� r� r� r� r� s zCascadeOptions.from_string)r� r� r� �__doc__� all_cascadesr� r� r� Z_viewonly_cascades� __slots__�__annotations__r� r� � classmethodr� � __classcell__r� r� r� r� r� � s" � %r� c s� �s�fdd�� |r^� ���fdd�}� ���fdd�}� ���fdd�}� ���fd d �}n6� ���fd d�}� ���fd d�}� ���fd d�}t j| d|ddd� t j| d|dd� t j| d|ddd� |r�t j| d|ddd� dS )zNRuns a validation method on an attribute value to be set or appended. c s | j � j}|j|k S r� )�manager�impl)�state� initiatorr� )�keyr� r� �detect_is_backref s z,_validator_events.<locals>.detect_is_backrefc s4 |j tjk r,�s� | |�s,�| �� �|d�S |S d S �NF��opr ZOP_BULK_REPLACE�obj�r� �valuer� �r� �include_backrefsr� � validatorr� r� �append s ��z!_validator_events.<locals>.appendc s8 �s�| |�s4| � � � �� �fdd�|D �|d d �< d S )Nc s g | ]}��� |d ��qS )Fr� �r� r� �r� r� r� r� r� r� ' s �7_validator_events.<locals>.bulk_set.<locals>.<listcomp>�r� �r� r� r� r� r� r� �bulk_set$ s �z#_validator_events.<locals>.bulk_setc s( �s� | |�s �| � � �|d�S |S d S r� r� �r� r� Zoldvaluer� r� r� r� �set_+ s z _validator_events.<locals>.set_c s$ �s� | |�s �| � � �|d� d S )NTr� r� r� r� r� �remove1 s z!_validator_events.<locals>.removec s2 |j tjk r*�s� | |�s*�| �� �|�S |S d S r� r� r� r� r� r� r� 7 s ��c s8 �s�| |�s4| � � � �� �fdd�|D �|d d �< d S )Nc s g | ]}��� |��qS r� r� r� r� r� r� r� B s r� r� r� r� r� r� r� ? s c s& �s� | |�s �| � � �|�S |S d S r� r� r� r� r� r� r� D s r� T)�raw�retvalZ bulk_replace)r� r� r� N)r5 �listen) �descr� r� Zinclude_removesr� r� r� r� r� r� r� r� �_validator_events s r� �p_unionTc s6 t �� }i �i �| D ]|}| | �tjtj�dd���| |< i }�jD ]B}|j|krdt� d||f ��|� |j� |||j< |j �|j< qD|��< q� ��fdd��g }| � � D ]v\} �|dk r�|� tj��fdd�|D �t�t�| ���|�g � ���� q�|� tj��fd d�|D �� ���� q�tj|� �|�S ) a0 Create a ``UNION`` statement used by a polymorphic mapper. See :ref:`concrete_inheritance` for an example of how this is used. :param table_map: mapping of polymorphic identities to :class:`_schema.Table` objects. :param typecolname: string name of a "discriminator" column, which will be derived from the query, producing the polymorphic identity for each row. If ``None``, no polymorphic discriminator is generated. :param aliasname: name of the :func:`~sqlalchemy.sql.expression.alias()` construct generated. :param cast_nulls: if True, non-existent columns, which are represented as labeled NULLs, will be passed into CAST. This is a legacy behavior that is problematic on some backends such as Oracle - in which case it can be set to False. T)Z allow_selectz�Polymorphic union can't use '%s' as the discriminator column due to mapped column %r; please apply the 'typecolname' argument; this is available on ConcreteBase as '_concrete_discriminator_name'c sh z�| | W S t k rb � r@t�t�� �| ��| � Y S t�t�� �| ��| � Y S Y nX d S r� )�KeyErrorr7 r �null�labelZ type_coerce)r� �table)� cast_nulls� colnamemaps�typesr� r� �col� s z polymorphic_union.<locals>.colNc s g | ]}� |���qS r� r� �r� r� �r� r� r� r� r� � s z%polymorphic_union.<locals>.<listcomp>c s g | ]}� |���qS r� r� r� r� r� r� r� � s )r8 Z OrderedSetr; �expectr> ZStrictFromClauseRoler� r� r� ZInvalidRequestError�add�type�itemsr� r7 �selectZliteral_column�sql_utilZ_quote_ddl_exprr� � select_fromZ union_all�alias) Z table_mapZ typecolnameZ aliasnamer� Zcolnamesr� �mr� �result�type_r� )r� r� r� r� r� r� �polymorphic_unionQ sb � �� ����� �� ���r� )�instance�row�identity_tokenzOptional[Type[_T]]zUnion[Any, Tuple[Any, ...]]z Optional[_T]z%Optional[Union[Row[Any], RowMapping]]� Optional[Any]z_IdentityKeyType[_T])�class_�identr� r� r� r c C sz | dk rRt | �}|dkrB|dkr*t�d��|jtt�|��|d�S |j||d�S n$|dk rlt|�}|� |�S t�d��dS )aQ Generate "identity key" tuples, as are used as keys in the :attr:`.Session.identity_map` dictionary. This function has several call styles: * ``identity_key(class, ident, identity_token=token)`` This form receives a mapped class and a primary key scalar or tuple as an argument. E.g.:: >>> identity_key(MyClass, (1, 2)) (<class '__main__.MyClass'>, (1, 2), None) :param class: mapped class (must be a positional argument) :param ident: primary key, may be a scalar or tuple argument. :param identity_token: optional identity token .. versionadded:: 1.2 added identity_token * ``identity_key(instance=instance)`` This form will produce the identity key for a given instance. The instance need not be persistent, only that its primary key attributes are populated (else the key will contain ``None`` for those missing values). E.g.:: >>> instance = MyClass(1, 2) >>> identity_key(instance=instance) (<class '__main__.MyClass'>, (1, 2), None) In this form, the given instance is ultimately run though :meth:`_orm.Mapper.identity_key_from_instance`, which will have the effect of performing a database check for the corresponding row if the object is expired. :param instance: object instance (must be given as a keyword arg) * ``identity_key(class, row=row, identity_token=token)`` This form is similar to the class/tuple form, except is passed a database result row as a :class:`.Row` or :class:`.RowMapping` object. E.g.:: >>> row = engine.execute(\ text("select * from table where a=1 and b=2")\ ).first() >>> identity_key(MyClass, row=row) (<class '__main__.MyClass'>, (1, 2), None) :param class: mapped class (must be a positional argument) :param row: :class:`.Row` row returned by a :class:`_engine.CursorResult` (must be given as a keyword arg) :param identity_token: optional identity token .. versionadded:: 1.2 added identity_token Nzident or row is required)r� z class or instance is required) r! r� r� Z identity_key_from_primary_key�tupler8 Zto_listZidentity_key_from_rowr& Zidentity_key_from_instance)r� r� r� r� r� �mapperr� r� r� � identity_key� s" G � � r� c @ sh e Zd ZdZe�� Ze�� Ze�� Ze�� Z e�� Z e�� Z e�� Z e�� Z e�� Ze�� Ze�� ZdS )�_TraceAdaptRolea� Enumeration of all the use cases for ORMAdapter. ORMAdapter remains one of the most complicated aspects of the ORM, as it is used for in-place adaption of column expressions to be applied to a SELECT, replacing :class:`.Table` and other objects that are mapped to classes with aliases of those tables in the case of joined eager loading, or in the case of polymorphic loading as used with concrete mappings or other custom "with polymorphic" parameters, with whole user-defined subqueries. The enumerations provide an overview of all the use cases used by ORMAdapter, a layer of formality as to the introduction of new ORMAdapter use cases (of which none are anticipated), as well as a means to trace the origins of a particular ORMAdapter within runtime debugging. SQLAlchemy 2.0 has greatly scaled back ORM features which relied heavily on open-ended statement adaption, including the ``Query.with_polymorphic()`` method and the ``Query.select_from_entity()`` methods, favoring user-explicit aliasing schemes using the ``aliased()`` and ``with_polymorphic()`` standalone constructs; these still use adaption, however the adaption is applied in a narrower scope. N)r� r� r� r� �enum�auto� ALIASED_INSPZ JOINEDLOAD_USER_DEFINED_ALIASZ JOINEDLOAD_PATH_WITH_POLYMORPHICZJOINEDLOAD_MEMOIZED_ADAPTERZMAPPER_POLYMORPHIC_ADAPTERZWITH_POLYMORPHIC_ADAPTERZ#WITH_POLYMORPHIC_ADAPTER_RIGHT_JOINZ DEPRECATED_JOIN_ADAPT_RIGHT_SIDEZADAPT_FROM_STATEMENTZCOMPOUND_EAGER_STATEMENTZLEGACY_SELECT_FROM_ALIASr� r� r� r� r� � s r� c sH e Zd ZdZdZddddddd�ddd d d d d d d �� fd d�Z� ZS )�ORMStatementAdapterz.ColumnAdapter which includes a role attribute.)�roleNFT�� equivalents�adapt_required�allow_label_resolve�anonymize_labels�adapt_on_names�adapt_from_selectablesr� rg � Optional[_EquivalentColumnMap]r{ �!Optional[AbstractSet[FromClause]])r� � selectabler r r r r r c s$ || _ t� j|||||||d� d S )Nr� )r� r� �__init__) r� r� r r r r r r r r� r� r� r > s �z ORMStatementAdapter.__init__)r� r� r� r� r� r r� r� r� r� r� r� 9 s �r� c sj e Zd ZU dZdZded< ded< ddd ddd ddd �d d d dddddddd� � fdd�Zdd� Z� ZS )� ORMAdapterzaColumnAdapter subclass which excludes adaptation of entities from non-matching mappers. )r� r� �is_aliased_class� aliased_inspr{ r �Optional[AliasedInsp[Any]]r NFT)r r r r r �limit_on_entityr r r� �_InternalEntityType[Any]r zOptional[Selectable]r ) r� �entityr r r r r r r r c sh || _ |j| _|d kr |j}t|�r2d| _|| _n d| _d | _t� j||||||rX| jnd | | d� d S )NTF)r r r Z include_fnr r ) r� r� r r r r r� r � _include_fn) r� r� r r r r r r r r r r� r� r� r a s& �zORMAdapter.__init__c C s, |j �dd �}| p*|�| j�p*| j�|�S )N� parentmapper)� _annotations�get�isar� )r� �elemr r� r� r� r � s zORMAdapter._include_fn) r� r� r� r� r� r� r r r� r� r� r� r� r V s �($r c @ s� e Zd ZU dZded< d"dddd d d d d d d d � dd�Zeddd�dd��Zddd�dd�Zddddd�dd �Z dd �d d �Z dd �d d!�Z dS )#� AliasedClassa� Represents an "aliased" form of a mapped class for usage with Query. The ORM equivalent of a :func:`~sqlalchemy.sql.expression.alias` construct, this object mimics the mapped class using a ``__getattr__`` scheme and maintains a reference to a real :class:`~sqlalchemy.sql.expression.Alias` object. A primary purpose of :class:`.AliasedClass` is to serve as an alternate within a SQL statement generated by the ORM, such that an existing mapped entity can be used in multiple contexts. A simple example:: # find all pairs of users with the same name user_alias = aliased(User) session.query(User, user_alias).\ join((user_alias, User.id > user_alias.id)).\ filter(User.name == user_alias.name) :class:`.AliasedClass` is also capable of mapping an existing mapped class to an entirely new selectable, provided this selectable is column- compatible with the existing mapped selectable, and it can also be configured in a mapping as the target of a :func:`_orm.relationship`. See the links below for examples. The :class:`.AliasedClass` object is constructed typically using the :func:`_orm.aliased` function. It also is produced with additional configuration when using the :func:`_orm.with_polymorphic` function. The resulting object is an instance of :class:`.AliasedClass`. This object implements an attribute scheme which produces the same attribute and method interface as the original mapped class, allowing :class:`.AliasedClass` to be compatible with any attribute technique which works on the original class, including hybrid attributes (see :ref:`hybrids_toplevel`). The :class:`.AliasedClass` can be inspected for its underlying :class:`_orm.Mapper`, aliased selectable, and other information using :func:`_sa.inspect`:: from sqlalchemy import inspect my_alias = aliased(MyClass) insp = inspect(my_alias) The resulting inspection object is an instance of :class:`.AliasedInsp`. .. seealso:: :func:`.aliased` :func:`.with_polymorphic` :ref:`relationship_aliased_class` :ref:`relationship_to_window_function` ry r� NFz_EntityType[_O]�Optional[FromClause]� Optional[str]r{ � Optional[Sequence[Mapper[Any]]]� Optional[ColumnElement[Any]]r ) �mapped_class_or_acr� r� �flatr �with_polymorphic_mappers� with_polymorphic_discriminator� base_alias�use_mapper_path�represents_outer_joinc C s� t dt�|��} | j} d} |d krN| jr<| jjr<| j�� }qX| jj ||d�}n | jrXd} |d k sdt �t | | |||rv|n| j |d k r�|n| j || || | � | _d| jj� d�| _d S )N�_InternalEntityType[_O]F�r� r T�aliased(�))r r6 �inspectr� r r Z _is_subqueryr� Z _with_polymorphic_selectable�_anonymous_fromclause�AssertionError� AliasedInspr �polymorphic_on� _aliased_inspr� r� )r� r r� r� r r r r r r! r" Zinspr� � nest_adaptersr� r� r� r � sF � �� ���zAliasedClass.__init__�AliasedInsp[_O]�AliasedClass[_O])r r c C s\ | � | �}d|jjj� d�|_||_|jrX|jD ]&}||k r0t�|�}t ||jj|� q0|S )Nr% r& ) r� r� r� r� r, �_is_with_polymorphic�_with_polymorphic_entitiesr � _reconstitute_from_aliased_insp�setattr)r| r r� Zsub_aliased_insp�entr� r� r� r2 s �z,AliasedClass._reconstitute_from_aliased_inspr )r� r c C s� z| j d }W n tk r( t� �Y nX |j}t||�}t|d�r\t|d�r\t�|j| �S t|d�rr|� d | �}t|d�r�|� |�}t | ||� |S )Nr, r� �__self__�__get__�adapt_to_entity) �__dict__r� �AttributeError�_target�getattr�hasattrr� � MethodType�__func__r6 r7 r3 )r� r� r, �target�attrr� r� r� � __getattr__ s zAliasedClass.__getattr__r )r� � mapped_classr r c C sr t ||�}t|d�r,t|d�r,t�|j| �S t|d�rB|�d | �}t|d�rnt�| �|_|� |�}t | ||� |S )Nr� r5 r6 r7 ) r; r< r� r= r> r6 �weakref�ref� _weak_entityr7 r3 )r� r� rB r r@ r� r� r� �_get_from_serialized3 s z!AliasedClass._get_from_serialized�r c C s dt | �| jjjf S )Nz<AliasedClass at 0x%x; %s>)�idr, r: r� r� r� r� r� r� J s �zAliasedClass.__repr__c C s t | j�S r� )ry r, r� r� r� r� �__str__P s zAliasedClass.__str__) NNFFNNNFF) r� r� r� r� r� r r� r2 rA rF r� rI r� r� r� r� r � s$ : �": r r. c @ s� e Zd ZU dZdZdejjfdejjfdejjfdejj fdejj fdej j fd ej j fgZ d ed < d ed< d ed< ded< ded< ded< ded< ddd ddddddddd � d d �Zed\d!d"dddd#d$�d%d&��Zed]d'd(d)ddddddddd*� d+d,��Zedd-�d.d/��Zd0Zd d-�d1d2�Zedd-�d3d4��Zed5d-�d6d7��Zed8d-�d9d:��Zd;d-�d<d=�Zd;d>d?�d@dA�ZdBdBdC�dDdE�Z d^dFddFdG�dHdI�Z e �r�d_dJddJdK�dLdM�Z ne Z dNdO� Z dPdQ� Z!dRdS� Z"dTdU� Z#dVdW� Z$dXdY� Z%dZd[� Z&d S )`r* a� Provide an inspection interface for an :class:`.AliasedClass` object. The :class:`.AliasedInsp` object is returned given an :class:`.AliasedClass` using the :func:`_sa.inspect` function:: from sqlalchemy import inspect from sqlalchemy.orm import aliased my_alias = aliased(MyMappedClass) insp = inspect(my_alias) Attributes on :class:`.AliasedInsp` include: * ``entity`` - the :class:`.AliasedClass` represented. * ``mapper`` - the :class:`_orm.Mapper` mapping the underlying class. * ``selectable`` - the :class:`_expression.Alias` construct which ultimately represents an aliased :class:`_schema.Table` or :class:`_expression.Select` construct. * ``name`` - the name of the alias. Also is used as the attribute name when returned in a result tuple from :class:`_query.Query`. * ``with_polymorphic_mappers`` - collection of :class:`_orm.Mapper` objects indicating all those mappers expressed in the select construct for the :class:`.AliasedClass`. * ``polymorphic_on`` - an alternate column or SQL expression which will be used as the "discriminator" for a polymorphic load. .. seealso:: :ref:`inspection_toplevel` )� __weakref__rE r� r r� �_adapt_on_namesr r+ �_use_mapper_path� _base_aliasr" �persist_selectable� local_tabler0 r1 �_adapterr: �__clause_element__�_memoized_valuesZ_all_column_expressions�_nest_adaptersr� rK rL r: r r r+ z Mapper[_O]r� rG r rP zSequence[Mapper[Any]]zSequence[AliasedInsp[Any]]r1 z weakref.ref[AliasedClass[_O]]rE z!Union[Type[_O], AliasedClass[_O]]r/ r# r r r r r{ ) r � inspectedr r� r r+ rM rL r r" r- c s0 |j } |j} t�|�| _| | _| | _ | _| _|| _|| _ t�|pF| �| _ || _ | | _ | | _ |r�d| _|| _g | _| jD ]@}|| k rzt|j|| � |d�}t| j |jj|� | j�|j� qznd| _| g| _ttj| || j� d� fdd�| jD �dd�| _| �r t |t ��st �|j� | j�| _� | _ | | _!d S )NT)r r r! Fc s h | ]}� s|j �qS r� )r )r� r� �r r� r� � <setcomp>� s �z'AliasedInsp.__init__.<locals>.<setcomp>)r r r r r r )"r r� rC rD rE r rN rO r� r+ rM rL r" rS r0 r r1 r r� r3 r� r� r, r r� r� Z_equivalent_columnsrP r� r* r) �wraprK r: )r� r rT r r� r r+ rM rL r r" r- r r� Zpolyr4 r� rU r� r � s\ � � ��zAliasedInsp.__init__NFz"Union[_EntityType[_O], FromClause]r z#Union[AliasedClass[_O], FromClause])�elementr� r� r r r c C sT t |t�r>|rt�d��|r*|j||d�S tjtj||d�S nt |||||d�S d S )Nz+adapt_on_names only applies to ORM elementsr$ �r )r� r r� r ) r� rG r� r� r� r; r� r> ZAnonymizedFromClauseRoler )r| rX r� r� r r r� r� r� �_alias_factory s& � ��zAliasedInsp._alias_factoryzUnion[Type[_O], Mapper[_O]]z/Union[Literal['*'], Iterable[_EntityType[Any]]]z)Union[Literal[(False, None)], FromClause]) �base�classesr r r+ �aliased� innerjoinr r� rL r c C sl t |�} |dkr |r t�d��| j|||d�\} }|s:|rR|d k sFt�|j|d�}t||| | ||| | d�S )Nr� z[the 'flat' and 'selectable' arguments cannot be passed simultaneously to with_polymorphic()�r^ rY )r� r r r r! r" )r r� r� �_with_polymorphic_argsr) r( r ) r| r[ r\ r r r+ r] r^ r r� rL �primary_mapper�mappersr� r� r� �_with_polymorphic_factory# s. � � �z%AliasedInsp._with_polymorphic_factoryrG c C s* | � � }|d kr&t�| �}t�|�| _ |S r� )rE r r2 rC rD )r� r4 r� r� r� r K s zAliasedInsp.entityTc C s" | j �| j| | d���d| d��S )N)r � parententity�entity_namespace�orm�Zcompile_state_plugin�plugin_subject)r � _annotater� �_set_propagate_attrsr� r� r� r� �#_memoized_method___clause_element__\ s ���z/AliasedInsp._memoized_method___clause_element__c C s | j S r� )r r� r� r� r� re g s z AliasedInsp.entity_namespace�Type[_O]c C s | j jS )zUReturn the mapped class ultimately represented by this :class:`.AliasedInsp`.)r� r� r� r� r� r� r� k s zAliasedInsp.class_rW c C s | j r| jjS t�| �S d S r� )rL r� �_path_registryr3 Z per_mapperr� r� r� r� rm q s zAliasedInsp._path_registryzDict[str, Any]c C s4 | j | j| j| j| j| j| j| �� | j| j | j d� S )N) r r� r� r� r r r r r! r" r- ) r r� r r� rK r r+ rM rL r" rS r� r� r� r� � __getstate__x s �zAliasedInsp.__getstate__�None)r� r c C sN | � |d |d |d |d |d |d |d |d |d |d |d � d S ) Nr r� r� r� r r r r! r r" r- )r )r� r� r� r� r� � __setstate__� s �zAliasedInsp.__setstate__r. )�otherr c C s� |j }| j |kst�t�dd� | jD ��}dd� |jD �}||krF|S |�|�}|j|d |j d�\}}|jdd�}t ||||j |j |jd�j S ) Nc s s | ] }|j V qd S r� �r� �r� �mpr� r� r� � <genexpr>� s z*AliasedInsp._merge_with.<locals>.<genexpr>c S s h | ] }|j �qS r� rr rs r� r� r� rV � s z*AliasedInsp._merge_with.<locals>.<setcomp>r_ TrY )r r r! r" ) r� r) r8 Zto_setr �unionr` r" r( r r+ rL r, )r� rq ra Z our_classesZ new_classesr\ rb r r� r� r� � _merge_with� s. � � �zAliasedInsp._merge_withrS )�exprr� r c C sD t |t�st�| | jd�}|r&||d< | j�|��|��d| d��S )N)rd r � proxy_keyrf rg )r� rE r) r� rP �traverseri rj )r� rx r� �dr� r� r� �_adapt_element� s � ���zAliasedInsp._adapt_elementr\ )r� r� r c C s d S r� r� )r� r� r� r� r� r� �_orm_adapt_element� s z AliasedInsp._orm_adapt_elementc C sX | j }||kr0|| jkr | S t| j|jj�jS n$|�| j�r@| S dsTtd|| f ��d S )NFz"mapper %s doesn't correspond to %s) r r� r; r r� r� r, r r) )r� r� Z self_polyr� r� r� �_entity_for_mapper� s � z AliasedInsp._entity_for_mapperc s. � j j\}}� j�|�� fdd�|�� D �fS )Nc s i | ]\}}� j �|�|�qS r� )rP rz )r� r� �paramr� r� r� � <dictcomp>� s � z:AliasedInsp._memoized_attr__get_clause.<locals>.<dictcomp>)r� Z _get_clauserP rz r� )r� �onclauseZ replacemapr� r� r� �_memoized_attr__get_clause� s ��z&AliasedInsp._memoized_attr__get_clausec C s i S r� r� r� r� r� r� � _memoized_attr__memoized_values� s z+AliasedInsp._memoized_attr__memoized_valuesc sD � j r � j�dd� � jD ��}n � j�� }� fdd�|D �}t|�S )Nc S s g | ] }|j �qS r� )r� )r� r4 r� r� r� r� � s zFAliasedInsp._memoized_attr__all_column_expressions.<locals>.<listcomp>c s g | ]\}}|� � |�f�qS r� )r| )r� r� r� r� r� r� r� � s )r0 r� Z_columns_plus_keysr1 rB )r� Zcols_plus_keysr� r� r� �&_memoized_attr__all_column_expressions� s � �z2AliasedInsp._memoized_attr__all_column_expressionsc O s0 || j kr| j | S |||� | j |< }|S d S r� )rR )r� r� Z callable_�args�kwr� r� r� r� �_memo� s zAliasedInsp._memoc C s< | j r"dd�dd� | j D �� }nd}dt| �| jj|f S )Nz(%s)r� c s s | ]}|j jV qd S r� )r� r� rs r� r� r� ru s z'AliasedInsp.__repr__.<locals>.<genexpr>r� z<AliasedInsp at 0x%x; %s%s>)r r� rH r� r� )r� Z with_polyr� r� r� r� s � �zAliasedInsp.__repr__c s> � j r,d� jjd�� fdd�� jD ��f S d� jjf S d S )Nzwith_polymorphic(%s, [%s])r� c 3 s | ]}|� j k r|jjV qd S r� )r� r� r� rs r� r� r� ru s �z&AliasedInsp.__str__.<locals>.<genexpr>z aliased(%s))r0 r: r� r� r r� r� r� r� rI s �� zAliasedInsp.__str__)NNFF)FFNFFFNF)N)N)'r� r� r� r� r� r? �ExtendedInternalTraversalZ dp_string� dp_booleanZdp_inspectable�dp_clauseelement�InternalTraversalZdp_has_cache_key_listZ_cache_key_traversalr� r r� rZ rc �propertyr r rk re r� rm rn rp rw r| r r} r~ r� r� r� r� r� rI r� r� r� r� r* T s| & � � "L � �&' � � r* c @ s. e Zd ZdZdZdd� Ze�d�dd� �ZdS ) �_WrapUserEntitya A wrapper used within the loader_criteria lambda caller so that we can bypass declared_attr descriptors on unmapped mixins, which normally emit a warning for such use. might also be useful for other per-lambda instrumentations should the need arise. ��subjectc C s || _ d S r� r� )r� r� r� r� r� r - s z_WrapUserEntity.__init__zsqlalchemy.orm.decl_apic C sP t jjj}t�| d�}||jkrBt|j| |j�rB|j| � |�S t ||�S d S )Nr� ) r8 Z preloadedrf �decl_api�object�__getattribute__r8 r� Z declared_attr�fgetr; )r� r� r� r� r� r� r� r� 0 s �z _WrapUserEntity.__getattribute__N) r� r� r� r� r� r r8 Zpreload_moduler� r� r� r� r� r� ! s r� c @ s e Zd ZU dZdZdejjfdejjfdej j fdej j fdej j fgZ de d< d e d< d e d< d e d < d e d< d e d< d e d< d3ddd d d d d�dd�Zedd� �Zdd� Zdd�d d �Zd d d �d d!�Zd"d#d$�d%d&�Zd d'd(d)�d*d+�Zd d(d �d,d-�Zd.d(d/�d0d1�Zd2S )4�LoaderCriteriaOptiona Add additional WHERE criteria to the load for all occurrences of a particular entity. :class:`_orm.LoaderCriteriaOption` is invoked using the :func:`_orm.with_loader_criteria` function; see that function for details. .. versionadded:: 1.4 )� root_entityr �deferred_where_criteria�where_criteria�_where_crit_orig�include_aliases�propagate_to_loadersr� r r� r� r� zOptional[Type[Any]]�"Optional[_InternalEntityType[Any]]z9Union[ColumnElement[bool], lambdas.DeferredLambdaElement]r{ r� r r� FTz_EntityType[Any]zXUnion[_ColumnExpressionArgument[bool], Callable[[Any], _ColumnExpressionArgument[bool]]])�entity_or_baser� � loader_onlyr� r� �track_closure_variablesc C s� t dt�|d��}|d kr.t d|�| _d | _n d | _|| _|| _t|�r�| jd k rZ| j}n|d k sft�|j}d| _t j |t j t |�ft j|d�d�| _nd| _t�t j |�| _|| _|| _d S )Nr Frx T)r� )Z lambda_args�opts)r r6 r' r� r r� �callabler) r� r= ZDeferredLambdaElementr> ZWhereHavingRoler� Z LambdaOptionsr� r; r� r� r� ) r� r� r� r� r� r� r� r Z wrap_entityr� r� r� r d s> � �� �z LoaderCriteriaOption.__init__c C s t ||||d�S )N)r� r� )r� )r| r r� r� r� r� r� r� � _unreduce� s �z LoaderCriteriaOption._unreducec C s( t j| jr| jjn| j| j| j| jffS r� )r� r� r r� r� r� r� r� r� r� r� r� � __reduce__� s ��z LoaderCriteriaOption.__reduce__zIterator[Mapper[Any]]rG c c sz | j r| j jjE d H n^| js"t�t| j�� �}|rv|�d�}tdt j |dd��}|rf|jjE d H q0|� |�� � q0d S )Nr r F)�raiseerr) r r� Zself_and_descendantsr� r) �list�__subclasses__�popr r6 r' �extend)r� �stack�subclassr4 r� r� r� � _all_mappers� s �z!LoaderCriteriaOption._all_mappersrU )� compile_stater c C s |j j�dd �| krdS dS )N�for_loader_criteriaFT)Zselect_statementr r �r� r� r� r� r� �_should_include� s ���z$LoaderCriteriaOption._should_includer �ColumnElement[bool])�ext_infor c C sF | j r td| j�|j��}n| j}t|t�s0t�tj |d| iddd�S )Nr� r� T)Zdetect_subquery_colsZind_cols_on_fromclause) r� r r� Z_resolve_with_argsr r� rE r) r� �_deep_annotate)r� r� �critr� r� r� �_resolve_where_criteria� s ��z,LoaderCriteriaOption._resolve_where_criteriazIterable[_MapperEntity]ro )r� �mapper_entitiesr c C s | � |� d S r� )�process_compile_state)r� r� r� r� r� r� �'process_compile_state_replaced_entities� s z<LoaderCriteriaOption.process_compile_state_replaced_entitiesc C s | � |j� dS )z7Apply a modification to a given :class:`.CompileState`.N)�get_global_criteriaZglobal_attributesr� r� r� r� r� � s z*LoaderCriteriaOption.process_compile_statezDict[Any, Any])r r c C s, | � � D ] }|�d|fg �}|�| � qd S )NZadditional_entity_criteria)r� � setdefaultr� )r� r rt Z load_criteriar� r� r� r� � s �z(LoaderCriteriaOption.get_global_criteriaN)FFTT)r� r� r� r� r� r? r� Z dp_plain_objZdp_has_cache_keyr� r� r� Z_traverse_internalsr� r r� r� r� r� r� r� r� r� r� r� r� r� r� r� = s: � �1 r� c C s | j S r� )r, )r? r� r� r� �<lambda>� r� r� rl zOptional[Mapper[_O]])r� r c C sH z&t | �}|d ks|js W d S |j}W n tjk r> Y d S X |S d S r� )r( Z is_mappedr� r ZNO_STATE)r� Z class_managerr� r� r� r� � _inspect_mc� s r� c C s t dt| ��}t|�S )Nrl )r rO r� )r� �originr� r� r� �_inspect_generic_alias s r� c @ s� e Zd ZU dZdZdZdZdZdZe � � Z de d< e j Zde d< dd d d �d d �Zdddd�dd�Zedd�dd��Zedd�dd��Zedd�d d ��Zde d < de d < d d!� Zd"d#� Zed$d%� �Zd&d'� Zd(d)d*d+d,�d-d.�Zd/S )0�Bundlea+ A grouping of SQL expressions that are returned by a :class:`.Query` under one namespace. The :class:`.Bundle` essentially allows nesting of the tuple-based results returned by a column-oriented :class:`_query.Query` object. It also is extensible via simple subclassing, where the primary capability to override is that of how the set of expressions should be returned, allowing post-processing as well as custom return types, without involving ORM identity-mapped classes. .. seealso:: :ref:`bundles` FTra �_propagate_attrszList[_ColumnsClauseElement]�exprsry z _ColumnExpressionArgument[Any]r )r� r� r� c s` | � _ � _� fdd�|D �}|� _tdd� dd� |D �D ���� � _� _|�d� j�� _dS )aQ Construct a new :class:`.Bundle`. e.g.:: bn = Bundle("mybundle", MyClass.x, MyClass.y) for row in session.query(bn).filter( bn.c.x == 5).filter(bn.c.y == 4): print(row.mybundle.x, row.mybundle.y) :param name: name of the bundle. :param \*exprs: columns or SQL expressions comprising the bundle. :param single_entity=False: if True, rows for this :class:`.Bundle` can be returned as a "single entity" outside of any enclosing tuple in the same manner as a mapped entity. c s g | ]}t jtj|� d ��qS ))Zapply_propagate_attrs)r; r� r> �ColumnsClauseRole�r� rx r� r� r� r� N s � �z#Bundle.__init__.<locals>.<listcomp>c s s | ]}t |d |j�|fV qdS )r� N)r; �_label)r� r� r� r� r� ru V s �z"Bundle.__init__.<locals>.<genexpr>c S s g | ]}|j �d |��qS ��bundle�r r �r� �er� r� r� r� X s � single_entityN) r� r� r� rB Z as_readonlyr� �columnsr� r� )r� r� r� r� Z coerced_exprsr� r� r� r 9 s � �zBundle.__init__rh zList[BindParameter[Any]]zTuple[Any, ...])rh � bindparamsr c s* | j | j| jft� �fdd�| jD �� S )Nc s g | ]}|� � ���qS r� )�_gen_cache_keyr� �rh r� r� r� r� ` s z)Bundle._gen_cache_key.<locals>.<listcomp>)r� r� r� r� r� )r� rh r� r� r� r� r� \ s �zBundle._gen_cache_keyzOptional[Mapper[Any]]rG c C s | j d j�dd �}|S )Nr r �r� r r )r� rt r� r� r� r� c s �z Bundle.mapperr� c C s | j d j�dd �}|S )Nr rd r� )r� Zier� r� r� r j s � �z Bundle.entityz6ReadOnlyColumnCollection[str, KeyedColumnElement[Any]]c C s | j S r� )r� r� r� r� r� re q s zBundle.entity_namespacer� r� c K s | j �| j �}|j�| j� |S r� )r� r� r8 r� )r� r� �clonedr� r� r� �_clone� s z Bundle._clonec C s^ | | d�}|� | j� | jd j�d| j�}tjdd� | jD �tj dd��� |�� d|d ��S ) N)r� re r rh c S s g | ]}|j �d |��qS r� r� r� r� r� r� r� � s z-Bundle.__clause_element__.<locals>.<listcomp>F)Z_literal_as_text_role�grouprf rg ) r� r r� r� r r r< Z ClauseListr> r� ri rj )r� r rh r� r� r� rQ � s$ ���� ���zBundle.__clause_element__c C s | � � jS r� )rQ �clausesr� r� r� r� r� � s zBundle.clausesc C s | � � }||_|S )z<Provide a copy of this :class:`.Bundle` passing a new label.)r� r� )r� r� r� r� r� r� r� � s z Bundle.labelz Select[Any]z#Sequence[Callable[[Row[Any]], Any]]z Sequence[str]zCallable[[Row[Any]], Any])�query�procs�labelsr c s. t |dd� |D ��� ddd�� �fdd� }|S )ag Produce the "row processing" function for this :class:`.Bundle`. May be overridden by subclasses to provide custom behaviors when results are fetched. The method is passed the statement object and a set of "row processor" functions at query execution time; these processor functions when given a result row will return the individual attribute value, which can then be adapted into any kind of return data structure. The example below illustrates replacing the usual :class:`.Row` return structure with a straight Python dictionary:: from sqlalchemy.orm import Bundle class DictBundle(Bundle): def create_row_processor(self, query, procs, labels): 'Override create_row_processor to return values as dictionaries' def proc(row): return dict( zip(labels, (proc(row) for proc in procs)) ) return proc A result from the above :class:`_orm.Bundle` will return dictionary values:: bn = DictBundle('mybundle', MyClass.data1, MyClass.data2) for row in session.execute(select(bn)).where(bn.c.data1 == 'd1'): print(row.mybundle['data1'], row.mybundle['data2']) c S s g | ]}d �qS )r� r� )r� �lr� r� r� r� � s z/Bundle.create_row_processor.<locals>.<listcomp>zRow[Any]r )r� r c s �� fdd��D ��S )Nc s g | ] }|� ��qS r� r� )r� �proc�r� r� r� r� � s z=Bundle.create_row_processor.<locals>.proc.<locals>.<listcomp>r� r� �Z keyed_tupler� r� r� r� � s z)Bundle.create_row_processor.<locals>.procr9 )r� r� r� r� r� r� r� r� �create_row_processor� s 'zBundle.create_row_processorN)r� r� r� r� r� Zis_clause_elementZ is_mapperr Z is_bundler8 � immutabledictr� r� � EMPTY_SETZ proxy_setr r� r� r� r re r� rQ r� r� r� r� r� r� r� r� s2 # r� z Bundle[_T]rb )rX �excluder c C s t �| ddi|�S )z�Deep copy the given ClauseElement, annotating each element with the "_orm_adapt" flag. Elements within the exclude collection will be cloned but not annotated. � _orm_adaptT)r� r� )rX r� r� r� r� � _orm_annotate� s r� )rX r c C s t j| dd�S )z�Remove annotations that link a column to a particular mapping. Note this doesn't affect "remote" and "foreign" annotations passed by the :func:`_orm.foreign` and :func:`_orm.remote` annotators. )r� rd )r� �r� Z_deep_deannotate�rX r� r� r� �_orm_deannotate� s �r� c C s t �| �S r� r� r� r� r� r� �_orm_full_deannotate s r� c @ sp e Zd ZdZejjZdZddddddd d d d �d d �Zdd� Z dddddd d�dd�Z ddddd d�dd�Z dS )�_ORMJoinz/Extend Join to support ORM constructs as input.TNFr� r_ zOptional[_OnClauseArgument]r{ r� z Tuple[ColumnElement[bool], ...])�left�rightr� �isouter�full� _left_memo� _right_memo�_extra_criteriac C s* t dt�|��} t dt�|��} | j} || _|| _t|tj�rpt rTt|j t j �sTt �|j �� } |j} ||j7 }n t|t�r�|} | jj} nd } d } | j}| �r&t�| |�r�| }nt|t�s�t �|}| j|| d| d|d�\}}}}}}|d k �r |�rt�|||�}|}nt�||||�}|}n|}|| _t| ��r@|j� dd �}n t | ��sTt | ��rZ| }nd }|d k �rz| j� d|i�| _t |��o�| }t!j"�#| |||||� | j$d k �s�t �|�r�| j$tj%|� M _$| �s&t&| dd ��r&| j'j(�r&t d| �} | j'j)}|d k �r&t | ��r| j*�+|�}| j$|@ | _$d S )Nz+Union[FromClause, _InternalEntityType[Any]]T)Zsource_selectableZdest_selectableZsource_polymorphicZof_type_entityZalias_secondaryZextra_criteriard r� r ),r r6 r' r r� r� r� r �QueryableAttributer Z comparatorrY Z Comparatorr) Z_source_selectabler� r� r/ �parentr� Zclause_is_presentrG Z _create_joinsr7 r� Z_target_adapterr@ r r r r rv r{ r< �Joinr r� �and_r; r� ZsingleZ_single_table_criterionrP rz )r� r� r� r� r� r� r� r� r� Z left_infoZ right_infoZadapt_toZ on_selectable�propZleft_selectableZ adapt_fromZpjZsj�source�destZ secondaryZtarget_adapterrd Zaugment_onclauseZ single_critr� r� r� r s� �� � �� � �� �� z_ORMJoin.__init__c C s^ |}t |tj�r|j}q| j|ks&t�t| j|j| j| j| j dd�}t||j|j|j|j d�S )zlSplice a join into the center. Given join(a, b) and join(b, c), return join(a, b).join(c) N)r� r� r� )r� r� ) r� r7 r� r� r� r) r� r� r� r� r� )r� rq �leftmostr� r� r� r� �_splice_into_center� s& � �z _ORMJoin._splice_into_center)r� r� r� r� r c C s t | ||||d�S )N)r� r� �r� )r� r� r� r� r� r� r� r� r� � s z _ORMJoin.join)r� r� r� r c C s t | ||d|d�S )NT)r� r� r� )r� r� r� r� r� r� r� � outerjoin� s z_ORMJoin.outerjoin)NFFNNr� )NFF)NF) r� r� r� r� r< r� Z__visit_name__Z inherit_cacher r� r� r� r� r� r� r� r� s$ � y � �r� r� z"attributes.QueryableAttribute[Any]zOptional[_EntityType[Any]]r� )r� r� � from_entityr c C sl t |t�rt�d��nHt |tj�rZ|jr.|j}|j}|dksDt|�sTt�d|� ���|}n|}|j | |d�S )a Create filtering criterion that relates this query's primary entity to the given related instance, using established :func:`_orm.relationship()` configuration. E.g.:: stmt = select(Address).where(with_parent(some_user, User.addresses)) The SQL rendered is the same as that rendered when a lazy loader would fire off from the given parent on that attribute, meaning that the appropriate state is taken from the parent object in Python without the need to render joins to the parent table in the rendered statement. The given property may also make use of :meth:`_orm.PropComparator.of_type` to indicate the left side of the criteria:: a1 = aliased(Address) a2 = aliased(Address) stmt = select(a1, a2).where( with_parent(u1, User.addresses.of_type(a2)) ) The above use is equivalent to using the :func:`_orm.with_parent.from_entity` argument:: a1 = aliased(Address) a2 = aliased(Address) stmt = select(a1, a2).where( with_parent(u1, User.addresses, from_entity=a2) ) :param instance: An instance which has some :func:`_orm.relationship`. :param property: Class-bound attribute, which indicates what relationship from the instance should be used to reconcile the parent/child relationship. :param from_entity: Entity in which to consider as the left side. This defaults to the "zero" entity of the :class:`_query.Query` itself. .. versionadded:: 1.2 z@with_parent() accepts class-bound mapped attributes, not stringsNz6Expected relationship property for with_parent(), got )r� ) r� ry r� r� r r� Z_of_typer� r Z _with_parent)r� r� r� Zmapper_propertyZprop_tr� r� r� � with_parent� s" 9 � ��r� r{ )�object_r c C s t �| �}|jS )z�Return True if the given object has a database identity. This typically corresponds to the object being in either the persistent or detached state. .. seealso:: :func:`.was_deleted` )r �instance_state� has_identity�r� r� r� r� r� r� s r� c C s t �| �}|jS )z�Return True if the given object was deleted within a session flush. This is regardless of whether or not the object is persistent or detached. .. seealso:: :attr:`.InstanceState.was_deleted` )r r� � was_deletedr� r� r� r� r� s r� r )�givenr r c C s^ t |�r(t | �r$|�� | �� kr$dS dS t | �rH| jr@|| jkS || kS t| �sTt�|�| �S )z�determine if 'given' corresponds to 'entity', in terms of an entity passed to Query that would match the same entity being referred to elsewhere in the query. TF)r rM rL r r r) Z common_parent�r� r r� r� r� �_entity_corresponds_to' s r� c C sR t | �r*t |�o(|j o(| |kp(|| jkS t |�s>| �|j�S |joL| |jkS dS )a determine if 'given' corresponds to 'entity', in terms of a path of loader options where a mapped attribute is taken to be a member of a parent entity. e.g.:: someoption(A).someoption(A.b) # -> fn(A, A) -> True someoption(A).someoption(C.d) # -> fn(A, C) -> False a1 = aliased(A) someoption(a1).someoption(A.b) # -> fn(a1, A) -> False someoption(a1).someoption(a1.b) # -> fn(a1, a1) -> True wp = with_polymorphic(A, [A1, A2]) someoption(wp).someoption(A1.foo) # -> fn(wp, A1) -> False someoption(wp).someoption(wp.A1.foo) # -> fn(wp, wp.A1) -> True N)r rL r1 r r� r r� r� r� r� �$_entity_corresponds_to_use_path_impl> s �� �r z Mapper[Any])r� r� r c C sD | j r || jkp| j�|�S | jr6|| jkp4| �|�S | �|�S dS )zedetermine if 'given' "is a" mapper, in terms of the given would load rows of type 'mapper'. N)r r r� r )r� r� r� r� r� � _entity_isac s �r z Query[Any]r )�iterable_query�itemr c C s� dd� }t |t�r�t�|�\}}}t |t�rFt |t�rF|| dkrFg S t |t�rX|dk sjt |t�rp|dk rp|� | �||�}|dk r�t|�dd|j� S t|�S n(|dkr�|� nt| ||d � �d S dS )zdcalculate __getitem__ in terms of an iterable query object that also has a slice() method. c S s t d��d S )Nz@negative indexes are not accepted by SQL index / slice operators)� IndexErrorr� r� r� r� �_no_negative_indexesx s �z&_getitem.<locals>._no_negative_indexesr N���r )r� �slicer8 Z decode_slice�intr� �step)r r r �start�stopr �resr� r� r� �_getitemr s. �� ��� r ri rx )�raw_annotationr| �originating_clsr c C s8 zt || |j�}W n tk r( Y dS X t|t�S d S r� )rI r� � NameErrorrL r )r r| r � annotatedr� r� r� �_is_mapped_annotation� s �r c @ s e Zd ZdS )� _CleanupErrorN)r� r� r� r� r� r� r� r � s r ry )r} r~ r c sn t �d| �}|s| S zt|�d�|�}W n@ tk rh } z"td| � d|�d�� d��|�W 5 d }~X Y nX |tjkrzd}n6z t|t �r�|j }n| W S W n t k r� | Y S X g }|}|� ||kr�|n|�d�� |�d�}t �d|�}|d kr�|� |� �qq�t �d|d ��sjt �d |d ��sjd � d � � fd d�|d �d�D ��|d < d� |�dt|�d } | S )Nz^(.+?)\[(.+)\]$r zFor annotation "z%", could not resolve container type "z[". Please ensure this type is imported at the module level outside of TYPE_CHECKING blocks�ClassVarr4 z ^["'].*["']$r z.*\[.*\]z"' r� c 3 s | ]}d |� � �� d �V qdS )�"N)�strip)r� r �Z stripcharsr� r� ru s z1_cleanup_mapped_str_annotation.<locals>.<genexpr>r� �[�])r� �matchrK r� r r �typingr � issubclassr r� � TypeErrorr� r� r� �len) r} r~ �mmr� �neZ real_symbolr� �inner�g2r� r r� � _cleanup_mapped_str_annotation� sJ �� �� � r# z Optional[_AnnotationScanType]r� z1Optional[Tuple[Union[type, str], Optional[type]]]) r r| r~ r� �attr_cls�required�is_dataclass_field� expect_mappedr� r c C s� | dkr2|r.t �d|j� d|� d|j� d���dS zt|| |td�} W nz tk r| } zt �d| � d��| �W 5 d} ~ X Y nF tk r� } z(|r�d | kr�t �d| � d��| �| } W 5 d} ~ X Y nX |r�| dfS t| d �r�t| t ��sR|�rJ|s�dS t | d d�} | t j k�rdS t | t��r*t| t��r*dS t jd |j� d|� d �d d��n| dfS t| j�dk�rlt �d��| jd | jfS dS )z�given an annotation, figure out if it's ``Mapped[something]`` and if so, return the ``something`` part. Includes error raise scenarios and other options. Nz4Python typing annotation is required for attribute "�.z " when primary argument(s) for "z#" construct are None or not presentr� z Could not interpret annotation zt. Check that it uses names that are correctly imported at the module level. See chained stack trace for more hints.zMapped[� __origin__zType annotation for "a " can't be correctly interpreted for Annotated Declarative Table form. ORM annotations should normally make use of the ``Mapped[]`` generic type, or other ORM-compatible generic type, as a container for the actual type, which indicates the intent that the attribute is mapped. Class variables that are not intended to be mapped by the ORM should use ClassVar[]. To allow Annotated Declarative to disregard legacy annotations which don't use Mapped[] to pass, set "__allow_unmapped__ = True" on the class or a superclass this class.Zzlpr)�coder z)Expected sub-type for Mapped[] annotationr )r� r� r� rI r# r r r< rL r r; r r r� r� r r) r �__args__r) ) r r| r~ r� r$ r% r& r' r� r Zcer r� r� r� r� �_extract_mapped_subtype sl �� �� �� � � ��r, )r� r c C s$ t | d�r| �� }nd }t�| |�S )N�_mapper_property_name)r< r- r8 Zclsname_as_plain_name)r� r� r� r� r� � _mapper_property_as_plain_namem s r. )r� T)NN)N)N)TT)�� __future__r r� � functoolsr� r� r r r r r r r r r r r r r r r r r r r rC r� r r Z_typingr r r r r[ r r r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- Z interfacesr. r/ r0 r1 r2 Z path_registryr3 r5 r� r6 r7 r8 Z engine.resultr: r; r< r= r> r� r? Z sql._typingr@ Zsql.annotationrA Zsql.baserB Z sql.cache_keyrC rD Z sql.elementsrE rF Zsql.selectablerG Zutil.langhelpersrH Z util.typingrI Z_de_stringify_annotationrJ Z _de_stringify_union_elementsrK Z_eval_name_onlyrL rM rN rO rP rQ rR rS �contextrT rU r� rV rW r� rX Z relationshipsrY ZenginerZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg Z sql.visitorsrh ri rj rk � frozensetr� �partialr� Z_de_stringify_partialru r� r� ry r� r� r� r� �Enumr� Z ColumnAdapterr� r Z Inspectabler Z_self_inspectsr* r� r� Z _inspectsr� r� � GenericAliasr� r� r� r� r� r� r� r� r� r� r� r r r r � Exceptionr r# r, r. r� r� r� r� �<module> s� ���� � �IC � S �� Z< 5 � J � O 4 � _ 3 �O%'` �"c