관리-도구
편집 파일: context.cpython-38.pyc
U -?�f=� � @ sh 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 lm Z d d lmZ d dlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z dd lm Z dd lm!Z! d d lm"Z# d d lm$Z$ d d!lm%Z% d d"lm&Z& d d#lmZ d d$l&m'Z' d d%l&m(Z( d d&l&m)Z) d d#l&mZ* 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,l1m2Z2 d d-l1m3Z3 d d.l1m4Z4 d d/l1m5Z5 d d0l1m6Z6 d d1l1m7Z7 d d2l8m9Z9 d d3l:m;Z; d d4l:m<Z< d d5l=m>Z> d d6l=m?Z? d d7l=m@Z@ d d8l=mAZA d d9l=mBZB d d:l=mCZC d d;l=mDZD d d<l=mEZE d d=lFmGZG e �r�dd>lHmIZI dd?lHmJZJ dd@lmKZK ddAlLmMZM ddBlNmOZO ddClPmQZQ ddDlPmRZR d dElSmTZT d dFlUmVZV d dGl,mWZW d dHlXmYZY d dIl8mZZZ d dJl:m[Z[ d dKl=m\Z\ d dLl=m]Z] d dMl=m^Z^ d dNl=m_Z_ d dOl=m`Z` d dPlambZb edQedR�ZcejdZee�f� ZgeCjhZhG dSdT� dT�Zie�fdUdVi�ZjG dWdX� dXe4�ZkG dYdZ� dZek�ZlG d[d\� d\ek�ZmG d]d^� d^�Zne&jj4�od_d`�G dadb� dbem��ZpG dcdd� dde;e6eEe- �Zqe&jj4�od_de�G dfdg� dgele>��Zre&jj4�od_dh�G didj� djemeD��Zsd�dldmdndodp�dqdr�Ztdsdtdu�dvdw�Zudsdtdu�dxdy�Zvd�dzdtd{d|�d}d~�ZwG dd�� d��ZxG d�d�� d�ex�ZyG d�d�� d�ex�ZzG d�d�� d�ex�Z{G d�d�� d�e{�Z|G d�d�� d�e{�Z}G d�d�� d�e}�Z~dS )�� )� annotationsN)�Any)�cast)�Dict)�Iterable)�List)�Optional)�Set)�Tuple)�Type)� TYPE_CHECKING)�TypeVar)�Union� )� attributes)� interfaces)�loading)�_is_aliased_class)�ORMColumnDescription)�ORMColumnsClauseRole)� PathRegistry)�_entity_corresponds_to)�_ORMJoin)�_TraceAdaptRole)� AliasedClass)�Bundle)� ORMAdapter)�ORMStatementAdapter� )�exc)�future)�inspect)�sql)�util)� coercions)� expression)�roles)�visitors)�_TP)�is_dml)�is_insert_update)�is_select_base)�_select_iterables)�CacheableOptions)� CompileState)� Executable)� Generative)�Options)� UpdateBase)�GroupedElement)� TextClause)�CompoundSelectState)� LABEL_STYLE_DISAMBIGUATE_ONLY)�LABEL_STYLE_NONE)� LABEL_STYLE_TABLENAME_PLUS_COL)�Select)�SelectLabelStyle)� SelectState)�TypedReturnsRows)�InternalTraversal)�_InternalEntityType)�OrmExecuteOptionsParameter)�PostLoad)�Mapper)�Query)�_BindArguments)�Session)�Result)�_CoreSingleExecuteParams)�_ColumnsClauseArgument)� SQLCompiler)�_DMLTableElement)� ColumnElement)�_JoinTargetElement)�_LabelConventionCallable)�_SetupJoinsElement)�ExecutableReturnsRows)� SelectBase)� TypeEngine�_T)�boundc @ sd e Zd ZU dZded< ded< ded< G dd � d e�Zdd d d ddddd�dd�Zd d�dd�Zd S )� QueryContext)�top_level_context� compile_state�query�params� load_options�bind_arguments�execution_options�session� autoflush�populate_existing�invoke_all_eagers� version_check� refresh_state�create_eager_joins�propagated_loader_optionsr �runidZpartials�post_load_paths�identity_token� yield_per�loaders_require_buffering�loaders_require_uniquing�intrc z Dict[PathRegistry, PostLoad]rd �ORMCompileStaterU c @ s< e Zd ZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZdS )z!QueryContext.default_load_optionsFTN)�__name__� __module__� __qualname__Z_only_return_tuples�_populate_existing�_version_check�_invoke_all_eagers� _autoflush�_identity_token� _yield_per�_refresh_stateZ_lazy_loaded_fromZ_legacy_uniquing�_sa_top_level_orm_contextZ_is_user_refresh� rv rv �F/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/context.py�default_load_options� s rx Nr. �&Union[Select[Any], FromStatement[Any]]rF rD zQUnion[Type[QueryContext.default_load_options], QueryContext.default_load_options]z$Optional[OrmExecuteOptionsParameter]zOptional[_BindArguments])rU � statementrW r[ rX rZ rY c s� |� _ |p t� _|pt� _|� _|� _|� _d� _d� _|� _ |j � _ |j j }|j } t� fdd�t|| �D ��� _t|j�� _|j� _|j� _|j� _|j� _|j� _ |j � _ |j � _ d S )NFc 3 s$ | ] \}}|j r|�� |�V qd S �N)Zpropagate_to_loadersZ'_adapt_cached_option_to_uncached_option)�.0�optZ uncached_opt��selfrv rw � <genexpr>� s �z(QueryContext.__init__.<locals>.<genexpr>)!rX � _EMPTY_DICTrZ rY rU rV r[ rg rh rW ru rT �select_statement� _with_options�tuple�ziprb �dictr rq r\ rn r] rp r^ ro r_ rt r` rs rf rr re ) r rU rz rW r[ rX rZ rY Zcached_optionsZuncached_optionsrv r~ rw �__init__� s. � zQueryContext.__init__��returnc C s | j p| S r{ )rT r~ rv rv rw �_get_top_level_context� s z#QueryContext._get_top_level_context)NN) rk rl rm � __slots__�__annotations__r1 rx r� r� rv rv rv rw rS f s � 6rS Z _result_disable_adapt_to_contextTc sb e Zd ZdZdd� Zedddd d�� fdd � �Zed d � �Zed d �dd��Zedd� �Z � Z S )�AbstractORMCompileStateFc C s� i | _ |d kr$i | _}|s t�d S |j | _}|r`d|d< |r\|jD ]}|jrF|�| � qFd S |�dd�rpd S |jd }z |d }W n t k r� Y n$X |jD ]}|j r�|jr�|�| � q�d|d< d S )NTZ toplevel_ormFr � selectable) r �global_attributes�AssertionErrorZ_global_attributesr� Z_is_criteria_option�process_compile_state�get�stack�KeyError�_is_compile_state) r rz �compiler�toplevel� process_criteria_for_toplevel�gar} Zstack_0Z toplevel_stmtrv rv rw �_init_global_attributes� s0 z/AbstractORMCompileState._init_global_attributes� Union[Select, FromStatement]�Optional[SQLCompiler]r �rz r� �kwr� c s t � j||f|�S )a� Create a context for a statement given a :class:`.Compiler`. This method is always invoked in the context of SQLCompiler.process(). For a Select object, this would be invoked from SQLCompiler.visit_select(). For the special FromStatement object used by Query to indicate "Query.from_statement()", this is called by FromStatement._compiler_dispatch() that would be called by SQLCompiler.process(). )�super�create_for_statement��clsrz r� r� �� __class__rv rw r� � s z,AbstractORMCompileState.create_for_statementc C s t � �d S r{ ��NotImplementedError)r� r[ rz rW rZ rY � is_pre_eventrv rv rw �orm_pre_session_exec s z,AbstractORMCompileState.orm_pre_session_execrE r� c C s( |j ||p i |d�}| �||||||�S )N)rZ )�execute�orm_setup_cursor_result)r� r[ rz rW rZ rY �conn�resultrv rv rw �orm_execute_statement s ��z-AbstractORMCompileState.orm_execute_statementc C s t � �d S r{ r� �r� r[ rz rW rZ rY r� rv rv rw r� 1 s z/AbstractORMCompileState.orm_setup_cursor_result) rk rl rm �is_dml_returningr� � classmethodr� r� r� r� � __classcell__rv rv r� rw r� � s ' r� c @ s( e Zd ZdZedd� �Zedd� �ZdS )� AutoflushOnlyORMCompileStatez>ORM compile state that is a passthrough, except for autoflush.c C s4 t j�ddh||j�\}}|s,|jr,|�� ||fS )N�_sa_orm_load_optionsr\ )rS rx �from_execution_options�_execution_optionsrq )r� r[ rz rW rZ rY r� rX rv rv rw r� A s ��� z1AutoflushOnlyORMCompileState.orm_pre_session_execc C s |S r{ rv r� rv rv rw r� ^ s z4AutoflushOnlyORMCompileState.orm_setup_cursor_resultN)rk rl rm �__doc__r� r� r� rv rv rv rw r� > s r� c @ s, e Zd ZU G dd� de�Zded< ded< ded< ded< d ed < d ed < d ed< ded< ded< ded< ded< ded< ded< ded< eZd ed < d Zd d � Z e r�e d!d"d#d d$�d%d&��Z d'd(� Z e d)ddd*�d+d,��Ze d-d.� �Ze d/d0� �Ze d1d2� �Zed3d4� �Zd5d6� Zd7d8� Ze d9d:� �Zd;S )<rj c @ s� e Zd Zdejfdejfdejfdejfdejfdejfdejfdejfd ejfd ejfd ejfg Zd Zd Z d Z e Z d Z d ZdZd Zd Zd Zd ZdS )z'ORMCompileState.default_compile_options�_use_legacy_query_style�_for_statement�_bake_ok� _current_path�_enable_single_crit�_enable_eagerloads�_only_load_props�_set_base_alias�_for_refresh_state�_render_for_subquery�_is_starFTN)rk rl rm r= Z dp_boolean�dp_has_cache_keyZ dp_plain_obj�_cache_key_traversalr� r� r� �_path_registryr� r� r� r� r� r� r� r� rv rv rv rw �default_compile_optionsl s. �r� zDict[Any, Any]r r� ry rz r� �List[_QueryEntity]� _entitiesz%Dict[_InternalEntityType, ORMAdapter]�_polymorphic_adaptersz=Union[Type[default_compile_options], default_compile_options]�compile_optionszOptional[_QueryEntity]�_primary_entity�bool�use_legacy_query_stylerL �_label_conventionzList[ColumnElement[Any]]�primary_columns�secondary_columnszSet[ColumnElement[Any]]�dedupe_columnszList[Tuple[Any, ...]]ra r � current_pathFc O s t � �d S r{ r� )r �argr� rv rv rw r� � s zORMCompileState.__init__r� r� r r� c K s d S r{ rv r� rv rv rw r� � s z$ORMCompileState.create_for_statementc C s& | j }||kr"|�|� |�|� d S r{ )r� �add�append)r �objZcol_collectionZdeduperv rv rw � _append_dedupe_col_collection� s z-ORMCompileState._append_dedupe_col_collectionr: )� label_style�legacyr� c C s |rddd�}|S t �|�S d S )Nc S s |r|S t | d�S d S )N�key)�getattr)�col�col_namerv rv rw �name� s z7ORMCompileState._column_naming_convention.<locals>.name)N)r; �_column_naming_convention)r� r� r� r� rv rv rw r� � s z)ORMCompileState._column_naming_conventionc C s t |�S r{ )�_column_descriptions�r� rz rv rv rw �get_column_descriptions� s z'ORMCompileState.get_column_descriptionsc C s t j�ddddddh||j�\}}d|krD|d }|jj�|j|�}|sNt}n |�t�}|j rn|�d|j i�}t |j dd �r�t |j j �dkr�|�d d �d k r�|�d d d ��}||d < z|jd } W n tk r� ds�td��Y nX | r�| j|d< |�s|j�r|�� ||fS )Nr� r] r\ rf re Zsa_top_level_orm_contextr� � �compiled_cacheTz#excess depth for ORM loader options)r� Z_cache_disable_reason�clause�plugin_subjectFz0statement had 'orm' plugin but no plugin_subject�mapper)rS rx r� r� rV Z merge_withrZ �_orm_load_exec_options�unionrs r� �_compile_options�lenr� r� �_propagate_attrsr� r� r� rq ) r� r[ rz rW rZ rY r� rX �ctxr� rv rv rw r� � sb ��� � � ������ z$ORMCompileState.orm_pre_session_execc C sH |j }|jj}|�dtj�} |jjr(|S t||||| ||�} t� || �S )Nr� ) �context�compiledrU r� rS rx r� r� r Z instances) r� r[ rz rW rZ rY r� Zexecution_contextrU rX Z querycontextrv rv rw r� ) s$ �� z'ORMCompileState.orm_setup_cursor_resultc C s dd� | j D �S )z�return all _MapperEntity objects in the lead entities collection. Does **not** include entities that have been replaced by with_entities(), with_only_columns() c S s g | ]}t |t�r|�qS rv �� isinstance� _MapperEntity�r| �entrv rv rw � <listcomp>U s z9ORMCompileState._lead_mapper_entities.<locals>.<listcomp>)r� r~ rv rv rw �_lead_mapper_entitiesM s �z%ORMCompileState._lead_mapper_entitiesc C sD |j s@|jj| jkr@|j�� D ] }| �|ttj||j |d�� q dS )a5 given MapperEntity or ORMColumnEntity, setup polymorphic loading if called for by the Mapper. As of #8168 in 2.0.0rc1, polymorphic adapters, which greatly increase the complexity of the query creation process, are not used at all except in the quasi-legacy cases of with_polymorphic referring to an alias and/or subquery. This would apply to concrete polymorphic loading, and joined inheritance where a subquery is passed to with_polymorphic (which is completely unnecessary in modern use). )� equivalentsr� N) �is_aliased_classr� �persist_selectabler� �iterate_to_root�"_mapper_loads_polymorphically_withr r ZWITH_POLYMORPHIC_ADAPTER�_equivalent_columns)r �ext_infor� �mprv rv rw � _create_with_polymorphic_adapterY s �����z0ORMCompileState._create_with_polymorphic_adapterc C s: |j p |gD ](}|| j|< |�� D ]}|| j|j< q"q d S r{ )�_with_polymorphic_mappersr� r� � local_table)r r� �adapter�m2�mrv rv rw r� v s z2ORMCompileState._mapper_loads_polymorphically_withc C s t d��d S )Nz0this method only works for ORMSelectCompileStater� )r� rV r� rv rv rw �_create_entities_collection} s �z+ORMCompileState._create_entities_collectionN)rk rl rm r- r� r� r� r� �_has_mapper_entitiesr� r r� r� r� r� r� r� r� �propertyr� r� r� r rv rv rv rw rj k sH % Y # rj c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S ) �DMLReturningColFilterz�an adapter used for the DML RETURNING case. Has a subset of the interface used by :class:`.ORMAdapter` and is used for :class:`._QueryEntity` instances to set up their columns as used in RETURNING for a DML statement. )r� �columns� __weakref__c C s: |d k r |j |j k r || _n|| _t�| j� | _| _d S r{ )r r� r# ZWeakPopulateDict�adapt_check_presentr )r � target_mapperZimmediate_dml_mapperrv rv rw r� � s ����z DMLReturningColFilter.__init__c C s. t �|�D ] }| �|�}|d k r | S q d S r{ )�sql_utilZ _find_columnsr )r r� � as_filter�cc�c2rv rv rw �__call__� s z DMLReturningColFilter.__call__c C s. | j }|j�|d �}|d kr d S |jj�|�S r{ )r� Z_columntopropertyr� r �cZcorresponding_column)r r� r� �proprv rv rw r � s z)DMLReturningColFilter.adapt_check_presentN)rk rl rm r� r� r� r r rv rv rv rw r � s r Zorm�orm_from_statementc @ sz e Zd ZU dZdZded< ded< dZded< dZdZdZ dZ e Z e Z ed d d d d �d d��Zdd� Zdd� Zdd� ZdS )� ORMFromStatementCompileStateNF� FromStatement�statement_containerz)Union[SelectBase, TextClause, UpdateBase]�requested_statementzOptional[_DMLTableElement]� dml_tabler� r� r )r r� r� r� c K s� t |t�st�|d k r&|jr&t�d��| �| �}d |_|jj |_ | |_ |_ |j |_}|jrl|j|_d|_g |_i |_|j|_|j r�t |tj�r�|js�|js�|jtkr�|�t �|_ n||_ |� |js�|js�|jnt|j �|_ t j!||j"|jdd� |jj#|_$|j%||ddd� |j&�r8|j&D ]}|j'�r |�(|� �q |j)�rZ|j)D ]\}}||� �qFg |_*g |_+t,� |_-g |_.g |_/d |_0t |j tj1��r�i |_2|jD ]} | �3|� �q�d |_4|_5d|_6|j*D ]} |j7| d|j8d� �q�nt9t:j;|j |j<d�|_=|S )Nz�The ORM FromStatement construct only supports being invoked as the topmost statement, as it is only intended to define how result rows should be returned.T��is_current_entitiesF)r� r� )Zwithin_columns_clauseZadd_to_result_map)Zadapt_on_names)>r� r r� r� �sa_exc� CompileError�__new__r� r� r� r� r r� �elementr r) �tabler r� r� r� r� r% rO Z _is_textual� _label_styler7 Zset_label_styler8 rz r� r� � _QueryEntity�to_compile_state� _raw_columnsr� r� r� r� r� r� �_with_context_optionsr� r� �setr� ra �_fallback_from_clauses�order_byr4 �extra_criteria_entities�setup_compile_stateZ_ordered_columnsZ_textual_ordered_columnsZ_loose_column_name_matching�processZ_add_to_result_mapr r ZADAPT_FROM_STATEMENT�_adapt_on_names�_from_obj_alias) r� r r� r� r rz r} �fnr� �entityr rv rv rw r� � s� � � � �������� � � � � �z1ORMFromStatementCompileState.create_for_statementc C s |S r{ rv �r �cols�current_adapterrv rv rw �_adapt_col_listB s z,ORMFromStatementCompileState._adapt_col_listc C s d S r{ rv r~ rv rv rw �_get_current_adapterE s z1ORMFromStatementCompileState._get_current_adapterc C sV | j j�dd�}t||�}| jjr:t| j�dkr:t� d��| jD ]}|� | |� q@dS )z�used by BulkORMInsert (and Update / Delete?) to set up a handler for RETURNING to return ORM objects and expressions r� Nr �sCan't generate ORM query that includes multiple expressions at the same time as '*'; query for '*' alone if present) rz r� r� r r� r� r� r� r r �!setup_dml_returning_compile_state)r Z dml_mapperr r r. rv rv rw r5 H s � � z>ORMFromStatementCompileState.setup_dml_returning_compile_state)rk rl rm r, r r� r �_has_orm_entities�multi_row_eager_loaders�eager_adding_joins�compound_eager_adapterr� r( � eager_joinsr� r� r2 r3 r5 rv rv rv rw r � s" r c s� e Zd ZU dZdZejZejZ dZ de d< de d< de j fde jfgej Zed e jfg Zd Zd d ddd �d d�Zdd� Zedd� �Zdd� Z� fdd�Zedd� �Zedd� �Z edd � �Z ed d � �Z � Z S ) r a* Core construct that represents a load of ORM objects from various :class:`.ReturnsRows` and other classes including: :class:`.Select`, :class:`.TextClause`, :class:`.TextualSelect`, :class:`.CompoundSelect`, :class`.Insert`, :class:`.Update`, and in theory, :class:`.Delete`. r Nz(Union[ExecutableReturnsRows, TextClause]r r� r+ r# r� Tz%Iterable[_ColumnsClauseArgument[Any]])�entitiesr r+ c sf � fdd�t �|�D �� _|� _|j� _|j� _|j� _|j� _|j� _t |�rV|j nd � _ |� _ d S )Nc s g | ]}t jtj|� d d��qS )T)�apply_propagate_attrsZ post_inspect)r$ �expectr& ZColumnsClauseRoler� r~ rv rw r� � s ��z*FromStatement.__init__.<locals>.<listcomp>) r# Zto_listr# r r) Z is_selectZ is_deleteZ is_insertZ is_updater+ r r+ )r r; r r+ rv r~ rw r� } s � �zFromStatement.__init__c K s2 | j | |f|�}|j }|r"||_|j|jf|�S )z�provide a fixed _compiler_dispatch method. This is roughly similar to using the sqlalchemy.ext.compiler ``@compiles`` extension. )�_compile_state_factoryr� rU r* rz )r r� r� rU r� rv rv rw �_compiler_dispatch� s z FromStatement._compiler_dispatchc C s t tt�| ��j}|| �S )a@ Return a :term:`plugin-enabled` 'column descriptions' structure referring to the columns which are SELECTed by this statement. See the section :ref:`queryguide_inspection` for an overview of this feature. .. seealso:: :ref:`queryguide_inspection` - ORM background )r �ORMSelectCompileStater; Zget_plugin_classr� )r �methrv rv rw �column_descriptions� s �z!FromStatement.column_descriptionsc C s | S r{ rv r~ rv rv rw �_ensure_disambiguated_names� s z)FromStatement._ensure_disambiguated_namesc + s6 t j�dd� | jD ��E d H t� jf |�E d H d S )Nc s s | ] }|j V qd S r{ )� _from_objects�r| r rv rv rw r� � s z-FromStatement.get_children.<locals>.<genexpr>)� itertools�chain� from_iterabler# r� � get_children)r r� r� rv rw rI � s �zFromStatement.get_childrenc C s | j jS r{ )r �_all_selected_columnsr~ rv rv rw rJ � s z#FromStatement._all_selected_columnsc C s t | j�r| jjS d S r{ )r) r �_return_defaultsr~ rv rv rw rK � s z FromStatement._return_defaultsc C s t | j�r| jjS d S r{ )r) r � _returningr~ rv rv rw rL � s zFromStatement._returningc C s t | j�r| jjS d S r{ )r* r �_inliner~ rv rv rw rM � s zFromStatement._inline)T) rk rl rm r� Z__visit_name__r r� r� r� r> �_for_update_argr� r= Zdp_clauseelement_listZdp_clauseelementr/ Z _executable_traverse_internalsZ_traverse_internalsr� r� Zis_from_statementr� r? r rB rC rI rJ rK rL rM r� rv rv r� rw r \ s> ��� � r Zcompound_selectc @ s e Zd ZdS )�CompoundSelectCompileStateN)rk rl rm rv rv rv rw rO � s rO �selectc @ sZ e Zd ZdZeZdZdZdZdZ dZ dZ dZ dZ dZdZedddd d�dd ��Zd d � Zd d � Zedd� �Zedd� �Zedd� �Zedd� �Zedd� �Zdd� Zdd� Zd d � Zd d � Z d d!� Z dDd"d#�Z d$d%� Z d&d'� Z d(d)� Z!d*d+� Z"d,d-� Z#d.d/� Z$d0d1� Z%d2d3� Z&d4d5� Z'd6d7� Z(d8d9� Z)d:d;� Z*e+d<d=� �Z,e+d>d?� �Z-d@dA� Z.dBdC� Z/dS )Er@ rv NFr� r� r r� c s4 | � | �� |dk r |j }nd}|}| j�|j�|_|jrR|�� � _t� � � j_n|� _|jj � _ |jj � _ g � _d� _i � _|j� _|s�� jddd�7 _� j rƈ jjtkrƈ j s�t� _q�t� _n � jj� _|jr� fdd�|jD �� _� �|j� j �� _tj � |j � jdd� |jj � _ d� _ � j!|||dd � |�r�|j"�sL|j�r�|jD ]8}|j"D ]*}|j#�r\|�$� d d � � j| D �� �q\�qR� jj"D ]}|j#�r�|�%� � �q�|j&�r�|j&D ]\}} |� � �q�g � _'g � _(t)� � _*i � _+i � _,g � _-g � _.� �/d d � |j0D ��� _1� �2� t3j4� � j5|f|� � S )z@compiler hook, we arrive here from compiler.visit_select() only.NTF)r� r� c s" i | ]}|t j� |jg d d��qS )Fr )r! r" r# )r| �memoized_entitiesr~ rv rw � <dictcomp>5 s � �z>ORMSelectCompileState.create_for_statement.<locals>.<dictcomp>r rv )r� r� c S s g | ]}t |t�r|�qS rv r� r� rv rv rw r� f s �z>ORMSelectCompileState.create_for_statement.<locals>.<listcomp>c s s | ] }|j V qd S r{ )r� )r| �inforv rv rw r� � s z=ORMSelectCompileState.create_for_statement.<locals>.<genexpr>)6r r� r� Z safe_merger� r� Z_cloner� r# � immutabledictr� � for_statementr� r� r� r� r� r� r �LABEL_STYLE_LEGACY_ORMr8 r� r6 �_memoized_select_entities�_memoized_entitiesr� r� r! r" r# r� r� �eager_order_byr� r� r� Z'process_compile_state_replaced_entitiesr� r$ r� r� r% r� r: r( ra r& �_normalize_froms� _from_obj� from_clauses�_setup_for_generater; r� rz ) r� rz r� r� r� r� rQ r} r- r� rv r~ rw r� � s� � �� � � � �� ���� ��� � z*ORMSelectCompileState.create_for_statementc C s� t d� t d| j� �� | jD ]�}t|t�r||d dkr|t dt�|d �� �� t d| j| � �� t d| j| j� �� q t|t�r |d dkr t d t�|d �� �� t d| j| � �� q d S ) Nz5 --------------------------------------------------- zcurrent path: r �loaderz Loader: r z Zpath_with_polymorphicz With Polymorphic: )�printr� r r� r� r Zcoerce�__dict__)r r� rv rv rw �_dump_option_struct� s z)ORMSelectCompileState._dump_option_structc s� | j }d | _d| _| jjr"| �� |jD ] }|jr(| �|j| j | � q(|jr^| �|j| j � | � � � |j r�|j | _ � r�t � fdd�| j D ��| _ � r�|jdkr�| �|j� �n|j| _|jr�t � fdd�|jD ��| _� r�|jdkr�| �t�|j�� �n |j�pd | _| j�r(| jd j}|�| j�| _|j�rB| �|j� �| _nd| _|j | _ |j �rvt t�dd� |j D ���| _ n8|j d k �r�t t�dd� |j D ���| _!n|j"�s�d | _ |j#| _#| jj$�r�t%| j �d k�r�t&�'d ��| j D ]}|�(| � �q�| j)D ]$}|d }|| f|d d � �� �q�| jj*�r0| �+� | j,�sZ| jj-�rPd �sPt.d ��t&�/d��| j�snt0| j1�| _| jd k�r�d | _| j2�r�| j3�r�| j4�r�| �5� | _n | �6� | _| j7�r�| �8� }|d k �r�| j�9d|i�| _d S )Nrv c 3 s | ]}� |d �V qdS �TNrv �r| �crit�r1 rv rw r� � s �z<ORMSelectCompileState._setup_for_generate.<locals>.<genexpr>�NFc 3 s | ]}� r� |d �n|V qdS rb rv rc re rv rw r� � s �r c s s$ | ] }|d k rt �|�nd V qd S r{ �r Zsurface_selectables�r| �srv rv rw r� � s �c s s$ | ] }|d k rt �|�nd V qd S r{ rg rh rv rv rw r� � s �r{ r r4 Fz,no columns were included in _only_load_propsz4Query contains no columns with which to SELECT from.Z deepentity):r� rz �_join_entitiesr� r� �_set_select_from_aliasrW � _setup_joins�_joinrX r� r3 �_where_criteriar� �_order_by_clausesr2 r' �_having_criteria�_group_by_clausesr# Zflatten_iterator�group_byrY r\ Z_target_adapter�copy_and_processZ _distinct_on� distinct_onZ _distinct�distinctZ _correlate� correlateZ_correlate_except�correlate_exceptZ_auto_correlaterN r� r� r r r) ra r� �_adjust_for_extra_criteriar� r� r� �InvalidRequestError�listr& r7 r8 �_should_nest_selectable�_compound_eager_statement�_simple_statementrU � _mapper_zeroZ _annotate)r rV rQ r r. ZrecZstrategy�ezerorv re rw r] � s� � � ��� � �� � � � �� ��� � ��� �z)ORMSelectCompileState._setup_for_generatec C sF | � | �}g |_d|_i |_|�|j|�|_tj||j |jdd� |S )a Creates a partial ORMSelectCompileState that includes the full collection of _MapperEntity and other _QueryEntity objects. Supports a few remaining use cases that are pre-compilation but still need to gather some of the column / adaption information. NTr ) r r� r� r� r� r r� r! r" r# )r� rV r� r rv rv rw r . s � �z1ORMSelectCompileState._create_entities_collectionc C s |j }t|d �S r{ )rl � _determine_last_joined_entity)r� rz � setup_joinsrv rv rw � determine_last_joined_entityH s z2ORMSelectCompileState.determine_last_joined_entityc c sj |j D ]^}|jrTd|jkrT|jd }|jsB|jsBt|g�E d H qdt|j�E d H qt|g�E d H qd S )NZentity_namespace)r# � is_selectable� _annotationsZ is_mapperr� r, Z_all_column_expressions)r� rz r Zensrv rv rw �all_selected_columnsN s �� z*ORMSelectCompileState.all_selected_columnsc C s | � tj�dd� |jD ���S )Nc s s. | ]&}d |j kr|jn|j d �� gV qdS )� parententityN)r� rD �__clause_element__rE rv rv rw r� ` s ����zAORMSelectCompileState.get_columns_clause_froms.<locals>.<genexpr>)rZ rF rG rH r# r� rv rv rw �get_columns_clause_froms] s ��z.ORMSelectCompileState.get_columns_clause_fromsc C s> t jtj||d�}t|j|�}|jj|j|j |j |j d� |S )N)r< )r� r$ r� r� ) r$ r= r& ZReturnsRowsRoler r# r` �updater� r$ r� r� )r� rz �from_statement�stmtrv rv rw r� n s � �z$ORMSelectCompileState.from_statementc C sT | j }| jjst�t|j�dks$t�| �|jd �}|rP| jddi7 _|| _dS )� used only for legacy Query casesr r r� FN)r� r� r� r� r� r[ � _get_select_from_alias_from_objr, )r rV r rv rv rw rk � s z,ORMSelectCompileState._set_select_from_aliasc C s| |}d|j kr|j d }t|d�r<|js4t�d��qx|jS n<t|jtjj �rt| � � }||jksbt �t t j|j|d�S dS dS )r� r� r� zPA selectable (FromClause) instance is expected when the base alias is being set.�r� N)r� �hasattrr� r � ArgumentError�_adapterr� r� r" ZAliasedReturnsRows� _all_equivsr� r r ZLEGACY_SELECT_FROM_ALIAS)r �from_objrS �equivsrv rv rw r� � s$ ��z5ORMSelectCompileState._get_select_from_alias_from_objc C s | j d jS )z8return the Mapper associated with the first QueryEntity.r )r� r� r~ rv rv rw r~ � s z"ORMSelectCompileState._mapper_zeroc C sD | j D ] }d|jkr|jd S q| jD ]}|jr*|j S q*dS )z�Return the 'entity' (mapper or AliasedClass) associated with the first QueryEntity, or alternatively the 'select from' entity if specified.r� N)r\ r� r� � entity_zero)r r� Zqentrv rv rw � _entity_zero� s z"ORMSelectCompileState._entity_zeroc C s$ | j | jgkr t�d| ��| jjS )Nz4%s() can only be used against a single mapped class.)r� r� r ry r� )r Zmethnamerv rv rw �_only_full_mapper_zero� s ��z,ORMSelectCompileState._only_full_mapper_zeroc C s$ t | j�dkr t�|pd��| �� S )Nr z8This operation requires a Query against a single mapper.)r� r� r ry r� )r Z rationalerv rv rw �_only_entity_zero� s �z'ORMSelectCompileState._only_entity_zeroc C s^ i }| j �� D ]&}dd� |D �D ]}|�|jj� q qdd� | jD �D ]}|�|jj� qF|S )Nc S s g | ]}t |t�r|�qS rv r� r� rv rv rw r� � s �z5ORMSelectCompileState._all_equivs.<locals>.<listcomp>c S s g | ]}t |t�r|�qS rv r� r� rv rv rw r� � s )rX �valuesr� r� r� r� )r r� rQ r� rv rv rw r� � s � � z!ORMSelectCompileState._all_equivsc sH � j r&dd� � j D �}t�� j|�}ng }d }� j� j� fdd�|D � � j� j� j� j� j f� j � j j � j j � j � jd�� j��}|�� }� �� }ttj||d�� _tj|g� j � }� j|_� j d k rވ j jd krވ j |_ |}� j� � D ]}t� |||j �}q�|j � ||� |�r0|j j |f� j�!|��� |j j |f� j"�� |S )Nc S s$ g | ] }t |tjj�r |jn|�qS rv )r� r" �elementsZ_label_referencer )r| �elemrv rv rw r� � s � �zCORMSelectCompileState._compound_eager_statement.<locals>.<listcomp>c s g | ]}|� j kr|�qS rv )r� )r| r r~ rv rw r� � s �� for_update�hints�statement_hintsrv rw r� )#r' r Z expand_column_list_from_order_byr� �_select_statementr\ rn rp r� rN r� �_hints�_statement_hintsrv rw � _select_args�aliasr� r r ZCOMPOUND_EAGER_STATEMENTr9 r rP r� r Zofr: r� Z splice_joinsZstop_on� select_from�non_generativers rY )r Zunwrapped_order_byZorder_by_col_expr�innerr� rz Z from_clauseZ eager_joinrv r~ rw r| � s| � ���� � � �� � ����z/ORMSelectCompileState._compound_eager_statementc C sz | j | j| j t| j�t| j�� � | j| j| j | j f| j | j j | j j| j| jd�| j��}| jrv|j j|f| j�� |S )Nr� )r� r� r� r� r\ r: r� rn rp r� r' rN r� r� r� rv rw r� rY r� )r rz rv rv rw r} 8 s& �� �z'ORMSelectCompileState._simple_statementc C s� t j|||d�}|r||_|r$||_|r:| jt|�7 _|rR|jj|f|�� n|rb|j�|� |rx| jt|�7 _| |_ | |_ ||_ ||_ ||_ ||_|r�||_|r�||_||_|r�||_| r�| |_| r�|jj|f| �� | d k r�|jj|f| �� |S )N)r# r[ r )r9 Z_create_raw_selectrn rp ro r� ru r� rq � _limit_clause�_offset_clause� _fetch_clause�_fetch_clause_options�_independent_ctes�_independent_ctes_opts� _prefixes� _suffixesrN r� r� rv rw )r Z raw_columnsr� Zwhere_criteriaZhaving_criteriar� r' r� r� r� rv rw � limit_clause� offset_clause� fetch_clause�fetch_clause_optionsru rt �prefixes�suffixesrr �independent_ctes�independent_ctes_optsrz rv rv rw r� L sR � ��z'ORMSelectCompileState._select_statementc C sx d|j kr0|j d }| j�|d �}|r0|�|�S t|tj�rB|}nt|d�rT|j}nd S | j�|d �}|rt|�|�S d S )Nr� r ) r� r� r� Z adapt_clauser� r% Z FromClauser� r )r r �searchr� rv rv rw �_adapt_polymorphic_element� s z0ORMSelectCompileState._adapt_polymorphic_elementc s � r� fdd�|D �S |S d S )Nc s g | ]}� |d ��qS )Trv )r| �ore rv rw r� � s z9ORMSelectCompileState._adapt_col_list.<locals>.<listcomp>rv r/ rv re rw r2 � s z%ORMSelectCompileState._adapt_col_listc sJ g � | j r � �d| j jf� | jr2� �d| jf� � s:d S � fdd�}|S )NTFc s � fdd�}t �| i |�S )Nc sF d| j kpd| j k}� D ](\}}|s(|r|| �}|d k r| S qd S )N� _orm_adaptr� )r� )r� Z is_orm_adaptZ always_adaptr �e��adaptersrv rw �replace� s � zRORMSelectCompileState._get_current_adapter.<locals>._adapt_clause.<locals>.replace)r' Zreplacement_traverse)r� r r� r� rv rw � _adapt_clause� s zAORMSelectCompileState._get_current_adapter.<locals>._adapt_clause)r, r� r� r� r� )r r� rv r� rw r3 � s �� z*ORMSelectCompileState._get_current_adapterc C s� |D �]�\}}}}|d }|d }t |�}|d k r:t |�}t|tj�rb|d k rXt�d��|}d }nd|jkrv|jd }|d kr�|js�t|d�s�t� d��d } t|tj��r�t |dd �} |d k�r| r�| }nF|j }z |j }W n4 t k �r } zt� d| �| �W 5 d } ~ X Y nX |j} |j } t|tj��s.| }| || jf| jk�rDq|d k �r�|| k �r�|j�dd �| k �r�t�d |j�d|�|f ��n |d k �r�d } |} nd } } | �|| ||| ||� qd S ) N�isouter�fullzUNo 'on clause' argument may be passed when joining to a relationship path as a targetr� r� z9Expected mapped entity or selectable/table as join targetZ_of_typez0Join target %s does not refer to a mapped entityzMexplicit from clause %s does not match left side of relationship attribute %s)r! r� r �PropComparatorr ry r� r� r� r� r� r r. �AttributeErrorZ _parententityr �QueryableAttributer� �_already_joined_edgesr� �_join_left_to_right) r �args�entities_collection�right�onclause�from_�flagsr� r� Zof_type�err�leftr rv rv rw rm � s� � � ��� � �� ��� �zORMSelectCompileState._joinc C s |dkr,|dkst �| �||||�\}}} n| �||�\}} ||krVt�d||f ��| �||||�\} }}| js~| �| �} nd} |dk r�| j| } | jd|� t | ||||| d�g | j|d d� | _nH| dk r�t || t �s�t �|| j } n|} | jt | | |||| d�g | _dS )z�given raw "left", "right", "onclause" parameters consumed from a particular key within _join(), add a real ORMJoin object to our _from_obj list (or augment an existing one) Nz>Can't construct a join from %s to %s, they are the same entityrv )r� r� Z_extra_criteriar ) r� �"_join_determine_implicit_left_side� _join_place_explicit_left_sider ry � _join_check_and_adapt_right_sider� �_get_extra_criteriar\ r r� r� r� ) r r� r� r� r� r Z outerjoinr� �replace_from_obj_index�use_entity_index�r_infoZextra_criteriaZ left_clauserv rv rw r� @ s� ��� ��� � ��� �� � ��z)ORMSelectCompileState._join_left_to_rightc C sX t |�}d }}| jrrt�| j|j|�}t|�dkrH|d }| j| }n(t|�dkr`t�d��nt�d|f ��n�|�rDi } t|�D ]R\} } | j } | dkr�q�t | �} | |kr�q�t | t �r�| | f| | j< q�d| f| | j< q�t | � � �}t�||j|�}t|�dk�r| ||d \}}n*t|�dk�r2t�d��nt�d|f ��n t�d��|||fS )z�When join conditions don't express the left side explicitly, determine if an existing FROM or entity in this query can serve as the left hand side. Nr r a Can't determine which FROM clause to join from, there are multiple FROMS which can join to this entity. Please use the .select_from() method to establish an explicit left side, as well as providing an explicit ON clause if not present already to help resolve the ambiguity.z�Don't know how to join to %r. Please use the .select_from() method to establish an explicit left side, as well as providing an explicit ON clause if not present already to help resolve the ambiguity.zgNo entities to join from; please use select_from() to establish the left entity/selectable of this join)r! r\ r Z find_left_clause_to_join_fromr� r� r ry � enumerate�entity_zero_or_selectabler� r� rz �keys)r r� r� r� r� r� r� r� �indexesZ potentialZ entity_indexr� r. Zent_infoZ all_clausesrv rv rw r� � sj � � �� �� ���z8ORMSelectCompileState._join_determine_implicit_left_sidec C s� d }}t |�}| jrHt�| j|j�}t|�dkr<t�d��|rH|d }|dkr�|r�t|d�r�t |�D ]$\}}t |t �rf|� |�rf|} q�qf||fS )z�When join conditions express a left side explicitly, determine where in our existing list of FROM clauses we should join towards, or if we need to make a new join, and if so is it from one of our existing entities. Nr zrCan't identify which entity in which to assign the left side of this join. Please use a more specific ON clause.r r� ) r! r\ r Z#find_left_clause_that_matches_givenr� r� r ry r� r� r� r� �corresponds_to) r r� r� r� r� �l_infor� �idxr� rv rv rw r� s0 � � ���z4ORMSelectCompileState._join_place_explicit_left_sidec C s� t |�}t |�}d}t|dd�}|rp|js8t|jtj�rp| jpD|jgD ](} t � |j| �rFt � | |j�rFd} qpqF|r�|j|jkr�t � d|j ��t|dd�|jt|dd� }} } |r�|r�|� |j�s�t � d||f ��t|d�r�| j|f7 _d} |j�r�|�r |j}| j�r0| �� } | dk �r�| |d�}nj|�r�| �|j��s\t � d| j|jjf ��t| tj��r|t�tj| �} d} t|| �}t j d | d d d � |�o�| �o�|}| �s�|�r�t|dd �}d} t j d| dd � | �r|�s�t �t t!j"t |�|j#d�}| �$||� n4|j�sB| �sB|j%�rB| �$|t t!j&||j|j#d�� t|tj'��rh| �� } | �rh| |d�}|�r�| j(|||j)ff7 _(t |�||fS )z�transform the "right" side of the join as well as the onclause according to polymorphic mapping translations, aliasing on the query or on the join, special cases where the right and left side have overlapping tables. Fr� NTz*Can't join table/selectable '%s' to itselfr� zIJoin target %s does not correspond to the right side of join condition %sz(Selectable '%s' is not derived from '%s'z�An alias is being generated automatically against joined entity %s for raw clauseelement, which is deprecated and will be removed in a later release. Use the aliased() construct explicitly, see the linked example.z1.4Zxaj1)�code)Zflatz�An alias is being generated automatically against joined entity %s due to overlapping tables. This is a legacy pattern which may be deprecated in a later release. Use the aliased(<entity>, flat=True) construct explicitly, see the linked example.Zxaj2r� )r� r� )*r! r� �with_polymorphicr� r� r% ZJoinr\ r� r Zselectables_overlapr ry � common_parentr� r� rj �is_clause_elementZ _is_lateralr3 Zis_derived_from� descriptionrO r$ r= r& ZFromClauseRoler r# Zwarn_deprecated�warnr� r r Z DEPRECATED_JOIN_ADAPT_RIGHT_SIDEr� r� Z#_has_aliased_polymorphic_fromclauseZ#WITH_POLYMORPHIC_ADAPTER_RIGHT_JOINZ ClauseElementr� r� )r r� r� r� r r� r� ZoverlapZ right_mapperr� Zright_selectableZright_is_aliasedZ need_adapterr1 Zaliased_entityr rv rv rw r� O s� � � � ���� � �� ��� ���� � �� �� � ����� z6ORMSelectCompileState._join_check_and_adapt_right_sidec C sF | j j| j j| j| j| j j| j j| jp(d | j j| j j | j j | j j d� S )N) r� r� ru rt r� r� rr r� r� r� r� ) r� r� r� ru rt r� r� rr r� r� r� r� r~ rv rv rw r� s �z"ORMSelectCompileState._select_argsc C sF | j }|�d�d k pD|�d�d k pD|�dd�pD|�dd�pD|�dd�S )Nr� r� ru Frt rv rr )r� r� )r �kwargsrv rv rw r{ s � � � �z-ORMSelectCompileState._should_nest_selectablec s< d� j f�jkr4t� �fdd��jd� j f D ��S dS d S )N�additional_entity_criteriac 3 s2 | ]*}|j s|j� kr|���r|�� �V qd S r{ )Zinclude_aliasesr. Z_should_includeZ_resolve_where_criteria)r| Zae�r� r rv rw r� 0 s �z<ORMSelectCompileState._get_extra_criteria.<locals>.<genexpr>rv )r� r� r� )r r� rv r� rw r� + s ���� z)ORMSelectCompileState._get_extra_criteriac C s | j D ]V}|j�dd�}|r|jjdk s8d|jf| jkr|| jkr||jrP|jndf| j|< qt | j� � �}|D ]�\}}|| j kr�qp|jj}| j j r�g }n | �|�}|dk r�||f7 }| �� }|D ]@}|r�|�|�}|r�t�|ddi�}||d�}| j|f7 _q�qpdS )a� Apply extra criteria filtering. For all distinct single-table-inheritance mappers represented in the columns clause of this query, as well as the "select from entity", add criterion to the WHERE clause of the given QueryContext such that only the appropriate subtypes are selected from the total results. Additionally, add WHERE criteria originating from LoaderCriteriaOptions associated with the global context. r� Nr� r� TF)r\ r� r� r� �_single_table_criterionr� r( r� r� r% r� rj r� r� r� r3 �traverser Z_deep_annotatern ) r Z fromclauser� r� r Z single_critr� r1 rd rv rv rw rx ; sB � ���� � z0ORMSelectCompileState._adjust_for_extra_criteria)N)0rk rl rm r� r� rX r, r r6 r7 r8 r9 rv rw rn rp r� r� ra r] r r� r� r� r� rk r� r~ r� r� r� r� r| r} r� r� r2 r3 rm r� r� r� r� r r� r{ r� rx rv rv rv rw r@ � sj ' [M0_cjB > r@ Fz#Union[Query, Select, FromStatement]z Optional[ORMSelectCompileState]r� zList[ORMColumnDescription])�query_or_select_stmtrU r� r� c C s8 |d krt j| |d�}|}dd� dd� |jD �D �}|S )N)r� c S sJ g | ]B\}}|j |jt|d d�|j|jdk r>|js>t|dd�ndd��qS )r� FNr. )r� �type�aliased�exprr. )� _label_namer� r� r� r� r� )r| r� Zinsp_entrv rv rw r� s � ���z(_column_descriptions.<locals>.<listcomp>c S s g | ]}||j f�qS rv �r� )r| Z_entrv rv rw r� � s )r@ r r� )r� rU r� r� �drv rv rw r� u s � ��r� z Union[Query[Any], Select[Any]]z"Optional[_InternalEntityType[Any]])�query_or_augmented_selectr� c C sJ | }|j r |j}|d k r |S |jrBd|jd jkrB|jd jd S t|�S )Nr� r )rl �_last_joined_entityr[ r� �_entity_from_pre_ent_zero)r� r r� rv rv rw � _legacy_filter_by_entity_zero� s r� c C sX | }|j sd S |j d }d|jkr,|jd S t|t�r<|jS d|jkrP|jd S |S d S )Nr r� �bundle)r# r� r� r r. )r� r r� rv rv rw r� � s r� z Tuple[_SetupJoinsElement, ...]z=Optional[Union[_InternalEntityType[Any], _JoinTargetElement]])r� r� r� c C s2 | sd S | d \}}}}t |tj�r*|jS |S d S )N���)r� r r� r. )r� r� �targetr� r� r� rv rv rw r� � s �r� c @ s� e Zd ZU dZdZded< dZdZdZded< ded < d ed < d ed < ddd�dd�Z dddd�dd�Z dd� Z e dd� �Z dS ) r! z:represent an entity column returned within a Query result.rv r� �supports_single_entityFz Optional[str]r� z!Union[Type[Any], TypeEngine[Any]]r� z.Union[_InternalEntityType, ColumnElement[Any]]r� z Optional[_InternalEntityType]r� rj �None)rU r� c C s t � �d S r{ r� �r rU rv rv rw r) � s z _QueryEntity.setup_compile_stater �rU r r� c C s t � �d S r{ r� �r rU r rv rv rw r5 � s z._QueryEntity.setup_dml_returning_compile_statec C s t � �d S r{ r� )r r� r� rv rv rw � row_processor� s z_QueryEntity.row_processorc C s� t |�D ]�\}}|jr8|jr2| �||j||� qn|j}|jr�|jrtd|jkr^t||||� q�t � ||j |||� q�|j� dd�r�t ||||� q�|jr�t � ||j |||� q�t � ||g|||� q|jrt |||� q|S )Nr� r� F)r� Z_is_lambda_elementZ _is_sequencer" Z _resolvedr� r� r� r� � _ColumnEntity� _for_columnsZ_select_iterabler� � _BundleEntityZ_is_clause_listZ is_bundle)r� rU r; r� r r� r. rv rv rw r" � sh � �����z _QueryEntity.to_compile_stateN)rk rl rm r� r� r� �_non_hashable_value�_null_column_type�use_id_for_hashr) r5 r� r� r" rv rv rv rw r! � s r! c @ s� e Zd ZU dZdZded< ded< ded< ded < d ed < d ed < dd� ZdZdZdZ e dd� �Z e dd� �Z dd� Z dd� Zdd� Zdd d d �d d �Zd!d"� Zd#S )$r� z mapper/class/AliasedClass entity) r� r� r� r� �path�_extra_entitiesr� r� r� �_polymorphic_discriminatorr> r� z Mapper[Any]r� r� r� r� r r �strr� c C s� |� | � |r*|jd kr | |_d|_d|_|jd }|j | }| _|j}|| _|j | _ }| jf| _ |j rv|j | _ n |jj| _ |j | _ |j| _|j| _|j| _|j| _|jr�|�|| j� d S )NTr� )r� r� r r6 r� � _post_inspectr� r. r� r� r r� r� r� �class_rk r� r r� Zwith_polymorphic_mappersr� Zpolymorphic_onr � _should_select_with_poly_adapterr� )r rU r. r� r r� r� rv rv rw r� = s4 �z_MapperEntity.__init__Tc C s | j jS r{ )r� r r~ rv rv rw r� g s z_MapperEntity.typec C s | j S r{ r� r~ rv rv rw r� k s z'_MapperEntity.entity_zero_or_selectablec C s t | j|�S r{ )r r� �r r. rv rv rw r� o s z _MapperEntity.corresponds_toc C sR d }| j s"|jr*|j�| jd �}n| jj}|rH|jrB|�|j�}qN|}n|j}|S r{ )r� r� r� r� r� r� r, �wrap)r rU r �retrv rv rw �_get_entity_clausesr s �z!_MapperEntity._get_entity_clausesc C s� |j }| �|�}|jr(|r(|�|j�}n |s2|j}|j| krL|jj}|j}nd }}tj | | j ||| j |||| j d� }|| j | jfS )N)�only_load_propsr` �polymorphic_discriminator)rU r r9 r r� r� r� r` r Z_instance_processorr� r r r� r )r r� r� rU r r r` Z _instancerv rv rw r� � s, � z_MapperEntity.row_processorrj r r� r� c C s. t j|| j| | j||j| j|jj| jd� d S )N�r� r r ) r �_setup_entity_queryr� r r� r� r� r� r r� rv rv rw r5 � s �z/_MapperEntity.setup_dml_returning_compile_statec C s� | � |�}| jj}|d k s*d| jf|jkrJ| j}||jr>|jnd f|j|< tj || j| | j ||j | j |j j| jd� |j�| j� d S )Nr� r )r r� r� r� r� r� r� r( r r r r� r� r� r� r r& r� r� )r rU r �single_table_critr� rv rv rw r) � s0 ���� � z!_MapperEntity.setup_compile_stateN)rk rl rm r� r� r� r� r� r r r r� r� r� r r� r5 r) rv rv rv rw r� & s* % r� c @ s� e Zd ZU dZdZded< ded< ded< d ed < d ed < ded < d$dd�Zedd� �Zedd� �Z dd� Z edd� �Z dd� Z d d d d �d d!�Z d"d#� ZdS )%r� rv )r� r� r� r� r� r� r� r� r r� z Type[Any]r� r r� r� r� r� TNc C s� d|_ |jd }|r"|j�| � n |�| � t|tjtjf�rH|� � }n|}| | _ | _ t |�| _ |j | _g | _|r�|jD ]X}d|jkr�t||||| d� qzt|t�r�t||||| d� qztj||g|d || d� qz| j j| _d S )NTr� �� parent_bundle)r6 r� r� r� r� r r� r r� r� r� r� r� r� r� Zexprsr� r �_ORMColumnEntityr� Z single_entityr� )r rU r� r� r Zsetup_entitiesr r� rv rv rw r� � sT � � �� z_BundleEntity.__init__c C s | j }|d k r|jS d S d S r{ )r� r� )r r rv rv rw r� s z_BundleEntity.mapperc C s& | j D ]}|j}|d k r| S qd S r{ )r� r� �r r� r rv rv rw r� ( s z_BundleEntity.entity_zeroc C s dS rf rv r rv rv rw r� 1 s z _BundleEntity.corresponds_toc C s& | j D ]}|j}|d k r| S qd S r{ )r� r� r rv rv rw r� 6 s z'_BundleEntity.entity_zero_or_selectablec C s | j D ]}|�|� qd S r{ )r� r) )r rU r� rv rv rw r) ? s z!_BundleEntity.setup_compile_staterj r r� r� c C s | � |�S r{ �r) r� rv rv rw r5 C s z/_BundleEntity.setup_dml_returning_compile_statec s@ t � �fdd�| jD �� \}}}| j�� j||�}|| j| jfS )Nc s g | ]}|� � ���qS rv )r� r� �r� r� rv rw r� L s z/_BundleEntity.row_processor.<locals>.<listcomp>)r� r� r� Zcreate_row_processorrV r� r )r r� r� Zprocs�labels�extra�procrv r rw r� J s � z_BundleEntity.row_processor)TN)rk rl rm r r� r� r� r r� r� r� r� r) r5 r� rv rv rv rw r� � s* � ; r� c @ sJ e Zd ZdZed dd��Zedd� �Zedd� �Zed d � �Z d d � Z dS )r� )� _fetch_column�_row_processor�raw_column_index�translate_raw_columnNc C s� |D ]z}|j }d|kr |d } n t�|d�} | rjd|j krRt|||| |||d� q~t|||| |||d� qt||||||d� qd S )Nr� re r )r� r Z extract_first_column_annotation�_IdentityTokenEntityr �_RawColumnEntity) r� rU r r� r r r �columnr �_entityrv rv rw r� \ sH � � � �z_ColumnEntity._for_columnsc C s | j jS r{ )r" r� r~ rv rv rw r� � s z_ColumnEntity.typec C s | j jj S r{ )r" r� Zhashabler~ rv rv rw r � s z!_ColumnEntity._non_hashable_valuec C s | j jjS r{ )r" r� Z_isnullr~ rv rv rw r � s z _ColumnEntity._null_column_typec C s� |j }| jd k r@| j\}}}| jr6||jj| j f7 }|||fS | jd k rR| j}n&| j}|jrj|jj | }|j rx|� � }|j r�|j j | }|� |�}|| j| jf}|| _| jr�| j|jj| j f }|| j|fS |S d S r{ )rU r r rV r# r r r" r, r r� � _deannotater9 �_getterr� r ) r r� r� rU �getterZ label_nameZextra_entitiesr" r rv rv rw r� � s4 � � z_ColumnEntity.row_processor)N) rk rl rm r� r� r� r r� r r r� rv rv rv rw r� T s � 1 r� c @ sH e Zd ZdZdZdZdZddd�Zdd� Zdd d d �d d �Z dd� Z dS )r! NF)r� r" r� r� r c C s� || _ || _|d k | _|jr.| jddi7 _|r8|jr@d | _n|rN|j| _n |�|�| _|rl|j � | � n |� | � || _ | j j r�| j j d nd | _ | j | j f| _d | _| _d S )Nr� Tr )r� r r r� r� Z_is_text_clauser� � _proxy_keyr� r� r� r" rD r� r r r )r rU r" r� r r r rv rv rw r� � s$ �z_RawColumnEntity.__init__c C s dS rf rv r rv rv rw r� � s z _RawColumnEntity.corresponds_torj r r� r� c C s | � |�S r{ r r� rv rv rw r5 s z2_RawColumnEntity.setup_dml_returning_compile_statec C s\ |� � }|r&|| jd�}|d kr,d S n| j}|jr:|�� }|j�|� |j�|� || _d S rf ) r3 r" r� r$ r� r� r� r� r )r rU r1 r" rv rv rw r) s z$_RawColumnEntity.setup_compile_state)N) rk rl rm r� r� r� r� r� r� r5 r) rv rv rv rw r! � s � $r! c @ sD e Zd ZdZdZdZddd�Zdd� Zd d d d �d d�Zdd� Z dS )r z Column/expression based entity.F)r� r� r" r� r� r� r Nc C s |j }|} |�dd �} |�d| �} | r<t| j| �| _d| _n|| _|d k | _|| _|r~|rl| rb| n|j| _q�|j || d�| _nd | _| j | | _ | _ } | j | _ } |r�|j�| � n |�| � d|_|| _d | _| _| j| jf| _| jr�|�| | j� d S )N� proxy_key� proxy_ownerF)r� T)r� r� r� r. r� r r r' r� r� r r� r� r� r� r� r6 r" r r r r r� r� )r rU r" r� r� r r r r r# Zorm_keyr) r r� rv rv rw r� * sB � �z_ORMColumnEntity.__init__c C s. t |�r|| jkS t | j� o(|�| j�S d S r{ )r r� r� r rv rv rw r� m s � �z _ORMColumnEntity.corresponds_torj r r� r� c C s8 | j | _|| j d�}|d k r4|j�|� |j�|� d S rf )r" r r� r� r� r� )r rU r r" rv rv rw r5 v s z2_ORMColumnEntity.setup_dml_returning_compile_statec C s� |� � }|r8|| jd�}|d kr>|js*t�| j| _d S n| j}| j}| jj}|d k sdd| jf|jkr~||j rr|j nd f|j |< |j r�|j s�|�� }t| jj��|jj�r�|j�|j� |j�|� |j�|� || _d S )NFr� )r3 r" r� r� r r� r� r� r� r� r� r( r� Z_expression_labelr$ r% rD � intersectionr� r& r� r� r� r� )r rU r1 r" r r rv rv rw r) � s8 ���� � z$_ORMColumnEntity.setup_compile_state)N) rk rl rm r� r� r� r� r� r5 r) rv rv rv rw r s � C r c @ s e Zd ZdZdd� Zdd� ZdS )r Fc C s d S r{ rv r� rv rv rw r) � s z(_IdentityTokenEntity.setup_compile_statec s � fdd�}|| j | jfS )Nc s � j jS r{ )rX rr )�row�r� rv rw r& � s z2_IdentityTokenEntity.row_processor.<locals>.getter)r� r )r r� r� r& rv r, rw r� � s z"_IdentityTokenEntity.row_processorN)rk rl rm r r) r� rv rv rv rw r � s r )NF)N)� __future__r rF �typingr r r r r r r r r r r r � r r r �baser r r Z path_registryr r# r r r r r r r r r r r! r" r$ r% r& r r' Z sql._typingr( r) r* r+ Zsql.baser, r- r. r/ r0 r1 Zsql.dmlr2 Z sql.elementsr3 r4 Zsql.selectabler5 r6 r7 r8 r9 r: r; r< Z sql.visitorsr= Z_typingr> r? r@ r� rA rV rB r[ rC rD ZenginerE Zengine.interfacesrF rG Z sql.compilerrH rI rJ rK rL rM rN rO Z sql.type_apirP rQ �rootr� rT r� rV rS r� r� r� rj r Z plugin_forr r rO r@ r� r� r� r� r! r� r� r� r! r r rv rv rv rw �<module> s g�l- + -x � & � �_ - yN