?????????? ????????? - ??????????????? - /home/agenciai/.local/lib/python3.9/site-packages/sqlalchemy/orm/__pycache__/context.cpython-39.pyc
???????
a ��Jh�� � @ s� 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^�ZnG d_d`� d`en�ZoG dadb� dben�Zpe&jj4�qdcdd�G dedf� dfem��ZrG dgdh� dhe;e6eEe- �Zse&jj4�qdcdi�G djdk� dkele>��Zte&jj4�qdcdl�G dmdn� dnemeD��Zud�dpdqdrdsdt�dudv�Zvdwdxdy�dzd{�Zwdwdxdy�d|d}�Zxd�d~dxdd��d�d��ZyG d�d�� d��ZzG d�d�� d�ez�Z{G d�d�� d�ez�Z|G d�d�� d�ez�Z}G d�d�� d�e}�Z~G d�d�� d�e}�ZG 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 @ sf e Zd ZU dZded< ded< ded< G dd � d e�Zdd d d d ddddd�dd�Zd d�dd�Zd S )� QueryContext)�top_level_context� compile_state�query�user_passed_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�intrd z Dict[PathRegistry, PostLoad]re �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� rw rw �K/home/agenciai/.local/lib/python3.9/site-packages/sqlalchemy/orm/context.py�default_load_options� s ry Nr. �2Union[Select[Any], FromStatement[Any], UpdateBase]rF rD zQUnion[Type[QueryContext.default_load_options], QueryContext.default_load_options]z$Optional[OrmExecuteOptionsParameter]zOptional[_BindArguments])rU � statementrW rX r\ rY r[ rZ 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��selfrw rx � <genexpr>� s �z(QueryContext.__init__.<locals>.<genexpr>)"rY � _EMPTY_DICTr[ rZ rU rV rW r\ rh ri rX rv rT �select_statement� _with_options�tuple�ziprc �dictr rr r] ro r^ rq r_ rp r` ru ra rt rg rs rf ) r� rU r{ rW rX r\ rY r[ rZ Zcached_optionsZuncached_optionsrw r rx �__init__� s0 � zQueryContext.__init__��returnc C s | j p| S r| )rT r rw rw rx �_get_top_level_context� s z#QueryContext._get_top_level_context)NN) rl rm rn � __slots__�__annotations__r1 ry r� r� rw rw rw rx rS f s � BrS Z _result_disable_adapt_to_contextTc sb e Zd ZdZdd� Zeddddd�� fd d � �Zed d � �Zed d�dd��Zedd� �Z � Z S )�AbstractORMCompileStateFc C s� i | _ |d u r i | _|s J �d S |j | _}|r\d|d< |rX|jD ]}|jrB|�| � qBd S |�dd�rld S |jd }z |d }W n ty� Y n$0 |jD ]}|j r�|jr�|�| � q�d|d< d S )NTZ toplevel_ormFr � selectable) r �global_attributesZ_global_attributesr� Z_is_criteria_option�process_compile_state�get�stack�KeyError�_is_compile_state) r� r{ �compiler�toplevel� process_criteria_for_toplevel�gar~ Zstack_0Z toplevel_stmtrw rw rx �_init_global_attributes� s0 z/AbstractORMCompileState._init_global_attributesr/ rH r r. �r{ r� �kwr� c s t � j||fi |��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��clsr{ r� r� �� __class__rw rx r� s z,AbstractORMCompileState.create_for_statementc C s t � �d S r| ��NotImplementedError)r� r\ r{ rX r[ rZ � is_pre_eventrw rw rx �orm_pre_session_exec s z,AbstractORMCompileState.orm_pre_session_execrE r� c C s( |j ||p i |d�}| �||||||�S )N)r[ )�execute�orm_setup_cursor_result)r� r\ r{ rX r[ rZ �conn�resultrw rw rx �orm_execute_statement( s ��z-AbstractORMCompileState.orm_execute_statementc C s t � �d S r| r� �r� r\ r{ rX r[ rZ r� rw rw rx r� > s z/AbstractORMCompileState.orm_setup_cursor_result) rl rm rn �is_dml_returningr� � classmethodr� r� r� r� � __classcell__rw rw r� rx 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 ry �from_execution_options�_execution_optionsrr )r� r\ r{ rX r[ rZ r� rY rw rw rx r� N s ��� z1AutoflushOnlyORMCompileState.orm_pre_session_execc C s |S r| rw r� rw rw rx r� k s z4AutoflushOnlyORMCompileState.orm_setup_cursor_resultN)rl rm rn �__doc__r� r� r� rw rw rw rx r� K s r� c @ sB 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 d!d"d#d d$�d%d&��Z e d'dd(d#d d)�d*d+��Z d,d-� Z e d.ddd/�d0d1��Ze d2d3� �Ze d4d5� �Ze d6d7� �Zed8d9� �Zd:d;� Zd<d=� Ze d>d?� �Zd@S )Ark 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)rl rm rn 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� rw rw rw rx �default_compile_optionsy s. �r� zDict[Any, Any]r r� rz r{ 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, ...]]rb r � current_pathFc O s t � �d S r| r� )r� �argr� rw rw rx r� � s zORMCompileState.__init__r/ rH r r� c K s" | j td|�f|j |d�|��S )N� Union[Select, FromStatement])r� r� )�_create_orm_contextr r� r� rw rw rx r� � s ���z$ORMCompileState.create_for_statementr� �Optional[SQLCompiler]�r{ r� r� r� r� c K s t � �d S r| r� )r� r{ r� r� r� rw rw rx r� � s z#ORMCompileState._create_orm_contextc C s& | j }||vr"|�|� |�|� d S r| )r� �add�append)r� �objZcol_collectionZdeduperw rw rx � _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_namerw rw rx �name� s z7ORMCompileState._column_naming_convention.<locals>.name)N)r; �_column_naming_convention)r� r� r� r� rw rw rx r� � s z)ORMCompileState._column_naming_conventionc C s t |�S r| )�_column_descriptions�r� r{ rw rw rx �get_column_descriptions� s z'ORMCompileState.get_column_descriptionsc C s t j�dh d�||j�\}}d|v r>|d }|jj�|j|�}|sHt}n |�t�}|j rh|�d|j i�}t |j dd �r�t |j j �dkr�|�dd�d ur�|�d d d ��}||d < z|jd } W n ty� d s�J d��Y n0 | r�| j|d< |s�|jr�|�� ||fS )Nr� > �sa_top_level_orm_contextr] r^ rg rf r� rg r� � �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 ry r� r� rV Z merge_withr[ �_orm_load_exec_options�unionrt r� �_compile_options�lenr� r� �_propagate_attrsr� r� rr ) r� r\ r{ rX r[ rZ r� rY �ctxr� rw rw rx r� � sV ��� � ������ z$ORMCompileState.orm_pre_session_execc C sJ |j }|jj}|�dtj�} |jjr(|S t|||||| ||�} t� || �S )Nr� ) �contextZcompiledrU r� rS ry r� r� r Z instances) r� r\ r{ rX r[ rZ r� Zexecution_contextrU rY Z querycontextrw rw rx r� E 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 rw �� isinstance� _MapperEntity�r} �entrw rw rx � <listcomp>r s z9ORMCompileState._lead_mapper_entities.<locals>.<listcomp>)r� r rw rw rx �_lead_mapper_entitiesj s �z%ORMCompileState._lead_mapper_entitiesc C sD |j s@|jj| jvr@|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� �mprw rw rx � _create_with_polymorphic_adapterv 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�mrw rw rx r� � s z2ORMCompileState._mapper_loads_polymorphically_withc C s t d��d S )Nz0this method only works for ORMSelectCompileStater� )r� rV r� rw rw rx �_create_entities_collection� s �z+ORMCompileState._create_entities_collectionN)rl rm rn r- r� r� r� r� �_has_mapper_entitiesr� r� r� r� r� r� r� r� r� �propertyr� r r� r rw rw rw rx rk x sJ % Y $ rk c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S ) �_DMLReturningColFilterz�a base for an adapter used for the DML RETURNING cases 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 ur |j |j ur || _n|| _t�| j� | _| _d S r| )r r� r# ZWeakPopulateDict�adapt_check_presentr )r� � target_mapperZimmediate_dml_mapperrw rw rx r� � s ����z _DMLReturningColFilter.__init__c C s. t �|�D ] }| �|�}|d ur | S q d S r| )�sql_utilZ _find_columnsr )r� r� � as_filter�cc�c2rw rw rx �__call__� s z _DMLReturningColFilter.__call__c C s t � �d S r| r� )r� r� rw rw rx r � s z*_DMLReturningColFilter.adapt_check_presentN)rl rm rn r� r� r� r r rw rw rw rx r � s r c @ s e Zd ZdZdd� ZdS )� _DMLBulkInsertReturningColFiltera[ an adapter used for the DML RETURNING case specifically for ORM bulk insert (or any hypothetical DML that is splitting out a class hierarchy among multiple DML statements....ORM bulk insert is the only example right now) its main job is to limit the columns in a RETURNING to only a specific mapped table in a hierarchy. c C s. | j }|j�|d �}|d u r d S |jj�|�S r| )r� �_columntopropertyr� r �c�corresponding_column�r� r� r� �proprw rw rx r � s z4_DMLBulkInsertReturningColFilter.adapt_check_presentN�rl rm rn r� r rw rw rw rx r � s r c @ s e Zd ZdZdd� ZdS )�"_DMLUpdateDeleteReturningColFiltera_ an adapter used for the DML RETURNING case specifically for ORM enabled UPDATE/DELETE its main job is to limit the columns in a RETURNING to include only direct persisted columns from the immediate selectable, not expressions like column_property(), or to also allow columns from other mappers for the UPDATE..FROM use case. c C s. | j }|j�|d �}|d ur*|jj�|�S |S r| )r� r r� r� r r r rw rw rx r � s z6_DMLUpdateDeleteReturningColFilter.adapt_check_presentNr rw rw rw rx r � s r Zorm�orm_from_statementc @ s| e Zd ZU dZdZded< ded< dZded< dZdZdZ dZ e Z e Z ed d d d d d �dd��Zdd� Zdd� Zdd� ZdS )� ORMFromStatementCompileStateNF� FromStatement�statement_containerz)Union[SelectBase, TextClause, UpdateBase]�requested_statementzOptional[_DMLTableElement]� dml_tabler� r� r� r r� c K s |}t |t�sJ �|d ur*|jr*t�d��| �| �}d |_|jj|_ | |_ |_ |j |_ }|jrp|j|_d|_g |_i |_|j|_|j r�t |tj�r�|js�|js�|jtu r�|�t�|_ n||_ |� |js�|js�|jnt|j �|_ t j ||j!|jdd� |jj"|_#|j$||ddd� |j%�r<|j%D ]}|j&�r"|�'|� �q"|j(�r^|j(D ]\}} ||� �qJg |_)g |_*t+� |_,g |_-g |_.d |_/t |j tj0��r�i |_1|jD ]} | �2|� �q�d |_3|_4d|_5|j)D ]} |j6| d|j7d� �q�nt8t9j:|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� �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 r{ r� r� � _QueryEntity�to_compile_state� _raw_columnsr� r� r� r� r� r� �_with_context_optionsr� r� �setr� rb �_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� r r� r~ �fnr� �entityr rw rw rx r� s� � � � �������� � � � � �z0ORMFromStatementCompileState._create_orm_contextc C s |S r| rw �r� �cols�current_adapterrw rw rx �_adapt_col_list� s z,ORMFromStatementCompileState._adapt_col_listc C s d S r| rw r rw rw rx �_get_current_adapter� s z1ORMFromStatementCompileState._get_current_adapterc C s� | j j�dd�}| j jr$t||�}n | j js4| j jr@t||�}nd}| jj rdt | j �dkrdt � d��| j D ]}|�| |� qjdS )zused by BulkORMInsert, 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)r{ r� r� � is_insertr � is_update� is_deleter r� r� r� r� r# r$ �!setup_dml_returning_compile_state)r� Z dml_mapperr r r6 rw rw rx r@ � s$ ���� z>ORMFromStatementCompileState.setup_dml_returning_compile_state)rl rm rn r4 r r� r �_has_orm_entities�multi_row_eager_loaders�eager_adding_joins�compound_eager_adapterr� r0 � eager_joinsr� r� r: r; r@ rw rw rw rx 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� r3 r+ r� Tz%Iterable[_ColumnsClauseArgument[Any]])�entitiesr& r3 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 rw rx r� � s ��z*FromStatement.__init__.<locals>.<listcomp>) r# Zto_listr+ r& r) Z is_selectr? r= r> r+ r( r3 )r� rF r& r3 rw r rx r� � s � �zFromStatement.__init__c K s: | j | |fi |��}|j }|r&||_|j|jfi |��S )z�provide a fixed _compiler_dispatch method. This is roughly similar to using the sqlalchemy.ext.compiler ``@compiles`` extension. )�_compile_state_factoryr� rU r2 r{ )r� r� r� rU r� rw rw rx �_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� �methrw rw rx �column_descriptions� s �z!FromStatement.column_descriptionsc C s | S r| rw r rw rw rx �_ensure_disambiguated_names s z)FromStatement._ensure_disambiguated_namesc + s: t j�dd� | jD ��E d H t� jf i |��E d H d S )Nc s s | ] }|j V qd S r| )� _from_objects�r} r& rw rw rx r� s z-FromStatement.get_children.<locals>.<genexpr>)� itertools�chain� from_iterabler+ r� � get_children)r� r� r� rw rx rT s �zFromStatement.get_childrenc C s | j jS r| )r& �_all_selected_columnsr rw rw rx rU s z#FromStatement._all_selected_columnsc C s t | j�r| jjS d S r| )r) r& �_return_defaultsr rw rw rx rV s z FromStatement._return_defaultsc C s t | j�r| jjS d S r| )r) r& � _returningr rw rw rx rW s zFromStatement._returningc C s t | j�r| jjS d S r| )r* r& �_inliner rw rw rx rX ! s zFromStatement._inline)T) rl rm rn r� Z__visit_name__r r� r� r� rI �_for_update_argr� r= Zdp_clauseelement_listZdp_clauseelementr/ Z _executable_traverse_internalsZ_traverse_internalsr� r� Zis_from_statementr� rJ r rM rN rT rU rV rW rX r� rw rw r� rx r � s> ��� � r Zcompound_selectc @ s e Zd ZdS )�CompoundSelectCompileStateN)rl rm rn rw rw rw rx rZ & s rZ �selectc @ s\ e Zd ZdZeZdZdZdZdZ dZ dZ dZ dZ dZdZeddddd d�d d ��Zd d � Zd d� Zedd� �Zedd� �Zedd� �Zedd� �Zedd� �Zdd� Zdd � Zd d � Zd d � Z d!d"� Z dEd#d$�Z d%d&� Z d'd(� Z d)d*� Z!d+d,� Z"d-d.� Z#d/d0� Z$d1d2� Z%d3d4� Z&d5d6� Z'd7d8� Z(d9d:� Z)d;d<� Z*e+d=d>� �Z,e+d?d@� �Z-dAdB� Z.dCdD� Z/dS )FrK rw NFr� r� r� r r� c s" | � | �� |}| j�|j�|_|jr<|�� � _t�� � j_n|� _|jj � _ |jj � _ g � _ d � _i � _|j� _|s�� jddd�7 _� j r�� jjtu r�� j s�t� _q�t� _n � jj� _|jrև fdd�|jD �� _� �|j� j �� _tj� |j � j dd� |jj � _ d� _ � j |||dd� |�r�|j!�s6|j�r�|jD ]8}|j!D ]*}|j"�rF|�#� d d � � j| D �� �qF�q<� jj!D ]}|j"�r~|�$� � �q~|j%�r�|j%D ]\}} |� � �q�g � _&g � _'t(� � _)i � _*i � _+g � _,g � _-� �.d d � |j/D ��� _0� �1� t2j3� � j4|fi |�� � S ) NFT)r� r� c s" i | ]}|t j� |jg d d��qS )Fr! )r) r* r+ )r} �memoized_entitiesr rw rx � <dictcomp>� s ��z=ORMSelectCompileState._create_orm_context.<locals>.<dictcomp>r! rw )r� r� c S s g | ]}t |t�r|�qS rw r� r� rw rw rx r� � s �z=ORMSelectCompileState._create_orm_context.<locals>.<listcomp>c s s | ] }|j V qd S r| )r� )r} �inforw rw rx r� � s z<ORMSelectCompileState._create_orm_context.<locals>.<genexpr>)5r% 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� rE r0 rb r. �_normalize_froms� _from_obj� from_clauses�_setup_for_generater; r� r{ ) r� r{ r� r� r� r� r\ r~ r5 r� rw r rx r� @ s� � �� � � ��� ���� ��� � z)ORMSelectCompileState._create_orm_contextc 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� rw rw rx �_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�|jdvr�| �|j� �n|j| _|jr�t � fdd�|jD ��| _� r�|jdvr�| �t�|j�� �n |j�pd | _| j�r(| jd j}|�| j�| _|j�rB| �|j� �| _nd| _|j | _ |j �rvt t�dd� |j D ���| _ n8|j d u�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 }|| g|d d � �R � �q�| jj*�r2| �+� | j,�s\| jj-�rRd �sRJ d ��t&�.d��| j�spt/| j0�| _| jd u �r�d | _| j1�r�| j2�r�| j3�r�| �4� | _n | �5� | _| j6�r�| �7� }|d u�r�| j�8d|i�| _d S )Nrw c 3 s | ]}� |d �V qdS �TNrw �r} �crit�r9 rw rx r� s �z<ORMSelectCompileState._setup_for_generate.<locals>.<genexpr>�NFc 3 s | ]}� r� |d �n|V qdS rm rw rn rp rw rx r� s �r c s s$ | ] }|d urt �|�nd V qd S r| �r Zsurface_selectables�r} �srw rw rx r� 7 s �c s s$ | ] }|d urt �|�nd V qd S r| rr rs rw rw rx r� > s �r| r r< Fz,no columns were included in _only_load_propsz4Query contains no columns with which to SELECT from.Z deepentity)9r� r{ �_join_entitiesr� r� �_set_select_from_aliasrb � _setup_joins�_joinrc r� r; �_where_criteriar� �_order_by_clausesr: r/ �_having_criteria�_group_by_clausesr# Zflatten_iterator�group_byrd rg Z_target_adapter�copy_and_processZ _distinct_on� distinct_onZ _distinct�distinctZ _correlate� correlateZ_correlate_except�correlate_exceptZ_auto_correlaterY r� r� r# r$ r1 rb r� �_adjust_for_extra_criteriar� r� �InvalidRequestError�listr. rB rC �_should_nest_selectable�_compound_eager_statement�_simple_statementr` � _mapper_zeroZ _annotate)r� rV r\ r r6 ZrecZstrategy�ezerorw rp rx rh � 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� rw rw rx r | s � �z1ORMSelectCompileState._create_entities_collectionc C s |j }t|d �S r| )rw � _determine_last_joined_entity)r� r{ � setup_joinsrw rw rx � determine_last_joined_entity� s z2ORMSelectCompileState.determine_last_joined_entityc c sj |j D ]^}|jrTd|jv rT|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� r{ r& Zensrw rw rx �all_selected_columns� s �� z*ORMSelectCompileState.all_selected_columnsc C s | � tj�dd� |jD ���S )Nc s s. | ]&}d |j vr|jn|j d �� gV qdS )� parententityN)r� rO �__clause_element__rP rw rw rx r� � s ����zAORMSelectCompileState.get_columns_clause_froms.<locals>.<genexpr>)re rQ rR rS r+ r� rw rw rx �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)rG )r� r, r� r� ) r$ rH r& ZReturnsRowsRoler r+ rk �updater� r, r� r� )r� r{ �from_statement�stmtrw rw rx r� � s � �z$ORMSelectCompileState.from_statementc C sT | j }| jjsJ �t|j�dks$J �| �|jd �}|rP| jddi7 _|| _dS )� used only for legacy Query casesr r r� FN)r� r� r� r� rf � _get_select_from_alias_from_objr4 )r� rV r rw rw rx rv � s z,ORMSelectCompileState._set_select_from_aliasc C s| |}d|j v r|j d }t|d�r<|js4t�d��qx|jS n<t|jtjj �rt| � � }||ju sbJ �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 ZLEGACY_SELECT_FROM_ALIAS)r� �from_objr^ �equivsrw rw rx 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 rw rw rx r� � s z"ORMSelectCompileState._mapper_zeroc C sD | j D ] }d|jv r|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)rg r� r� � entity_zero)r� r� Zqentrw rw rx � _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# r� r� )r� Zmethnamerw rw rx �_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# r� r� )r� Z rationalerw rw rx �_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 rw r� r� rw rw rx r� s �z5ORMSelectCompileState._all_equivs.<locals>.<listcomp>c S s g | ]}t |t�r|�qS rw r� r� rw rw rx r� % s )rc �valuesr� r� r� r� )r� r� r\ r� rw rw rx r� s � � z!ORMSelectCompileState._all_equivsc sL � 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 urވ j jd u rވ j |_ |}� j� � D ]}t� |||j �}q�|j � ||� |�r2|j j |g� j�!|��R � |j j |g� j"�R � |S )Nc S s$ g | ] }t |tjj�r |jn|�qS rw )r� r" �elementsZ_label_referencer& )r} �elemrw rw rx r� 7 s � �zCORMSelectCompileState._compound_eager_statement.<locals>.<listcomp>c s g | ]}|� j vr|�qS rw )r� )r} r r rw rx r� K � �� for_update�hints�statement_hintsr� r� r� )#r/ r Z expand_column_list_from_order_byr� �_select_statementrg ry r{ r� rY r� �_hints�_statement_hintsr� r� � _select_args�aliasr� r r ZCOMPOUND_EAGER_STATEMENTrD r r[ r� r( ZofrE r� Z splice_joinsZstop_on� select_from�non_generativer~ rd )r� Zunwrapped_order_byZorder_by_col_expr�innerr� r{ Z from_clauseZ eager_joinrw r rx r� + sr � ���� �� �� �����z/ORMSelectCompileState._compound_eager_statementc C s| | j | j| j t| j�t| j�� � | j| j| j | j f| j | j j | j j| j| jd�| j��}| jrx|j j|g| j�R � |S )Nr� )r� r� r� r� rg rE r� ry r{ r� r/ rY r� r� r� r� r� r� rd r� )r� r{ rw rw rx r� � s& �� �z'ORMSelectCompileState._simple_statementc C s t j|||d�}|r||_|r$||_|r:| jt|�7 _|rT|jj|g|�R � n|rd|j�|� |rz| jt|�7 _| |_ | |_ ||_ ||_ ||_ ||_|r�||_|r�||_||_|r�||_| r�| |_| r�|jj|g| �R � | d u�r|jj|g| �R � |S )N)r+ rf r( )r9 Z_create_raw_selectry r{ rz r� r� r� r| � _limit_clause�_offset_clause� _fetch_clause�_fetch_clause_options�_independent_ctes�_independent_ctes_opts� _prefixes� _suffixesrY r� r� r� r� )r� Z raw_columnsr� Zwhere_criteriaZhaving_criteriar� r/ r� r� r� r� r� � limit_clause� offset_clause� fetch_clause�fetch_clause_optionsr� r �prefixes�suffixesr} �independent_ctes�independent_ctes_optsr{ rw rw rx r� � sR � ��z'ORMSelectCompileState._select_statementc C sx d|j v r0|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� rw rw rx �_adapt_polymorphic_element� s z0ORMSelectCompileState._adapt_polymorphic_elementc s � r� fdd�|D �S |S d S )Nc s g | ]}� |d ��qS )Trw )r} �orp rw rx r� � r� z9ORMSelectCompileState._adapt_col_list.<locals>.<listcomp>rw r7 rw rp rx r: � 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 v pd| j v }� D ](\}}|s(|r|| �}|d ur| S qd S )N� _orm_adaptr� )r� )r� Z is_orm_adaptZ always_adaptr �e��adaptersrw rx �replace s � zRORMSelectCompileState._get_current_adapter.<locals>._adapt_clause.<locals>.replace)r' Zreplacement_traverse)r� r r� r� rw rx � _adapt_clause s zAORMSelectCompileState._get_current_adapter.<locals>._adapt_clause)r4 r� r� r� r� )r� r� rw r� rx r; � s �� z*ORMSelectCompileState._get_current_adapterc C s� |D �]�\}}}}|d }|d }t |�}|d ur:t |�}t|tj�rb|d urXt�d��|}d }nd|jv rv|jd }|d u r�|js�t|d�s�t� d��t|tj��r�t |dd �} |d u �r| r�| }nH|j }z |j }W n6 t �y } z t� d| �| �W Y d } ~ n d } ~ 0 0 |j} |j } t|tj��s,| }| || jf| jv �rBq|d u�r�|| u�r�|j�dd �| u�r�t�d |j�d|�|f ��n |d u�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# r� r� r� r� r� r� r r6 �AttributeErrorZ _parententityr �QueryableAttributer� �_already_joined_edgesr� �_join_left_to_right) r� �args�entities_collection�right�onclause�from_�flagsr� r� Zof_type�err�leftr rw rw rx rx / s� � � ��� � �� ��� �zORMSelectCompileState._joinc C s |du r,|du sJ �| � ||||�\}}} n| �||�\}} ||u rVt�d||f ��| �||||�\} }}| js~| �| �} nd} |dur�| j| } | jd|� t| ||||| d�g | j|d d� | _nH| dur�t || t �s�J �|| 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 entityrw )r� r� Z_extra_criteriar ) �"_join_determine_implicit_left_side� _join_place_explicit_left_sider# r� � _join_check_and_adapt_right_sider� �_get_extra_criteriarg r r� r� r� ) r� r� r� r� r� r Z outerjoinr� �replace_from_obj_index�use_entity_index�r_infoZextra_criteriaZ left_clauserw rw rx r� � st ��� ���� ��� ��� ��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 } | du r�q�t | �} | |u r�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! rg r Z find_left_clause_to_join_fromr� r� r# r� � enumerate�entity_zero_or_selectabler� r� r� �keys)r� r� r� r� r� r� r� r� �indexesZ potentialZ entity_indexr� r6 Zent_infoZ all_clausesrw rw rx r� � sb � � �� �� ���z8ORMSelectCompileState._join_determine_implicit_left_sidec C s� d }}t |�}| jrHt�| j|j�}t|�dkr<t�d��|rH|d }|du r�|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! rg r Z#find_left_clause_that_matches_givenr� r� r# r� r� r� r� r� �corresponds_to) r� r� r� r� r� �l_infor� �idxr� rw rw rx r� Y s. � � ���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|ju r�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| �� } | du�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�J �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% ZJoinrg r� r Zselectables_overlapr# r� � common_parentr� r� ru �is_clause_elementZ _is_lateralr; Zis_derived_from� descriptionrO r$ rH r& ZFromClauseRoler r# Zwarn_deprecated�warnr 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� �overlapZ right_mapperr� Zright_selectableZright_is_aliasedZ need_adapterr9 Zaliased_entityr rw rw rx r� � 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� r� r r� r� r} r� r� r� r� ) r� r� r� r� r r� r� r} r� r� r� r� r rw rw rx r� X s �z"ORMSelectCompileState._select_argsc C sF | j }|�d�d upD|�d�d upD|�dd�pD|�dd�pD|�dd�S )Nr� r� r� Fr rw r} )r� r� )r� �kwargsrw rw rx r� l s � � � �z-ORMSelectCompileState._should_nest_selectablec s< d� j f�jv r4t� �fdd��jd� j f D ��S dS d S )N�additional_entity_criteriac 3 s2 | ]*}|j s|j� u r|���r|�� �V qd S r| )Zinclude_aliasesr6 Z_should_includeZ_resolve_where_criteria)r} Zae�r� r� rw rx r� | s �z<ORMSelectCompileState._get_extra_criteria.<locals>.<genexpr>rw )r� r� r� )r� r� rw r� rx r� w s ���� z)ORMSelectCompileState._get_extra_criteriac C s | j D ]V}|j�dd�}|r|jjdus8d|jf| jv r|| jvr||jrP|jndf| j|< qt | j� � �}|D ]�\}}|| j v r�qp|jj}| j j r�g }n | �|�}|dur�||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)rg r� r� r� �_single_table_criterionr� r0 r� r� r- r� ru r� r� r� r; �traverser Z_deep_annotatery ) r� Z fromclauser� r� r Z single_critr� r9 ro rw rw rx r� � sB � ���� � z0ORMSelectCompileState._adjust_for_extra_criteria)N)0rl rm rn r� r� rc r4 r rA rB rC rD r� r� ry r{ r� r� rl rh r r� r� r� r� rv r� r� r� r� r� r� r� r� r� r� r: r; rx r� r� r� r� r r� r� r� r� rw rw rw rx rK - sj # [M0]cjB > rK Fz#Union[Query, Select, FromStatement]z Optional[ORMSelectCompileState]r� zList[ORMColumnDescription])�query_or_select_stmtrU r� r� c C s8 |d u rt j| |d�}|}dd� dd� |jD �D �}|S )N)r� c S sJ g | ]B\}}|j |jt|d d�|j|jdur>|js>t|dd�ndd��qS )r� FNr6 )r� �type�aliased�exprr6 )� _label_namer� r� r� r� r� )r} r� Zinsp_entrw rw rx r� � s � ���z(_column_descriptions.<locals>.<listcomp>c S s g | ]}||j f�qS rw �r� )r} Z_entrw rw rx r� � s )rK r r� )r� rU r� r� �drw rw rx r� � s � ��r� z Union[Query[Any], Select[Any]]z"Optional[_InternalEntityType[Any]])�query_or_augmented_selectr� c C sJ | }|j r |j}|d ur |S |jrBd|jd jv rB|jd jd S t|�S )Nr� r )rw �_last_joined_entityrf r� �_entity_from_pre_ent_zero)r� r� r� rw rw rx � _legacy_filter_by_entity_zero� s r c C sX | }|j sd S |j d }d|jv r,|jd S t|t�r<|jS d|jv rP|jd S |S d S )Nr r� �bundle)r+ r� r� r r6 )r� r� r� rw rw rx 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� r6 )r� r� �targetr� r� r� rw rw rx 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.rw r� �supports_single_entityFz Optional[str]r� z!Union[Type[Any], TypeEngine[Any]]r� z.Union[_InternalEntityType, ColumnElement[Any]]r� z Optional[_InternalEntityType]r� rk �None)rU r� c C s t � �d S r| r� �r� rU rw rw rx r1 # s z _QueryEntity.setup_compile_state� Optional[_DMLReturningColFilter]�rU r r� c C s t � �d S r| r� �r� rU r rw rw rx r@ & s z._QueryEntity.setup_dml_returning_compile_statec C s t � �d S r| r� )r� r� r� rw rw rx � row_processor- s z_QueryEntity.row_processorc C s� t |�D ]�\}}|jr8|jr2| �||j||� qn|j}|jr�|jrtd|jv r^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 rF r� r" r� r6 rw rw rx r* 0 sh � �����z _QueryEntity.to_compile_stateN)rl rm rn r� r� r� �_non_hashable_value�_null_column_type�use_id_for_hashr1 r@ r r� r* rw rw rw rx 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 u r | |_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 rA r� � _post_inspectr� r6 r� r� r r� r� r� �class_rl r� r r� Zwith_polymorphic_mappersr Zpolymorphic_onr � _should_select_with_poly_adapterr )r� rU r6 r� r" r� r� rw rw rx r� � s2 �z_MapperEntity.__init__Tc C s | j jS r| )r� r r rw rw rx r� � s z_MapperEntity.typec C s | j S r| r� r rw rw rx r� � s z'_MapperEntity.entity_zero_or_selectablec C s t | j|�S r| )r r� �r� r6 rw rw rx r� � 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� r4 �wrap)r� rU r �retrw rw rx �_get_entity_clauses� s �z!_MapperEntity._get_entity_clausesc C s� |j }| �|�}|jr(|r(|�|j�}n |s2|j}|j| u rL|jj}|j}nd }}tj | | j ||| j |||| j d� }|| j | jfS )N)�only_load_propsra �polymorphic_discriminator)rU r rD r r� r� r� ra r Z_instance_processorr� r r r� r )r� r� r� rU r r ra Z _instancerw rw rx r � s, � z_MapperEntity.row_processorrk 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 rw rw rx r@ � s �z/_MapperEntity.setup_dml_returning_compile_statec C s� | � |�}| jj}|d us*d| jf|jv rJ| 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� r0 r r r r� r r� r� r r. r� r� )r� rU r �single_table_critr� rw rw rx r1 s0 ���� � z!_MapperEntity.setup_compile_stateN)rl rm rn r� r� r� r� r r r r r� r� r� r r r@ r1 rw rw rw rx r� 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 rw )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|jv r�t||||| d� qzt|t�r�t||||| d� qztj||g|d || d� qz| j j| _d S )NTr �� parent_bundle)rA r� r� r� r� r r� r r� r� r r� r� r� r� �exprsr r �_ORMColumnEntityr Z single_entityr )r� rU r� r� r" Zsetup_entitiesr" r rw rw rx r� 1 sR � � �� z_BundleEntity.__init__c C s | j }|d ur|jS d S d S r| )r� r� )r� r� rw rw rx r� l s z_BundleEntity.mapperc C s& | j D ]}|j}|d ur| S qd S r| )r� r� �r� r� r� rw rw rx r� t s z_BundleEntity.entity_zeroc C s dS rq rw r rw rw rx r� } s z _BundleEntity.corresponds_toc C s& | j D ]}|j}|d ur| S qd S r| )r� r� r% rw rw rx r� � s z'_BundleEntity.entity_zero_or_selectablec C s | j D ]}|�|� qd S r| )r� r1 )r� rU r� rw rw rx r1 � s z!_BundleEntity.setup_compile_staterk r r r c C s | � |�S r| �r1 r rw rw rx r@ � s z/_BundleEntity.setup_dml_returning_compile_statec s@ t � �fdd�| jD �� \}}}| j�� j||�}|| j| jfS )Nc s g | ]}|� � ���qS rw )r r� �r� r� rw rx r� � r� z/_BundleEntity.row_processor.<locals>.<listcomp>)r� r� r Zcreate_row_processorrV r� r )r� r� r� Zprocs�labels�extra�procrw r' rx r � s � z_BundleEntity.row_processor)TN)rl rm rn r r� r� r� r r� r� r� r� r1 r@ r rw rw rw rx 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|v r |d } n t�|d�} | rjd|j v rRt|||| |||d� q~t|||| |||d� qt||||||d� qd S )Nr� rf r! )r� r Z extract_first_column_annotation�_IdentityTokenEntityr$ �_RawColumnEntity) r� rU r r� r- r" r" �columnr �_entityrw rw rx r � sF � � � �z_ColumnEntity._for_columnsc C s | j jS r| )r1 r� r rw rw rx r� � s z_ColumnEntity.typec C s | j jj S r| )r1 r� Zhashabler rw rw rx r � s z!_ColumnEntity._non_hashable_valuec C s | j jjS r| )r1 r� Z_isnullr rw rw rx r � s z _ColumnEntity._null_column_typec C s� |j }| jd ur@| j\}}}| jr6||jj| j f7 }|||fS | jd urR| 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+ r1 r4 r r� � _deannotaterD �_getterr� r ) r� r� r� rU �getterZ label_nameZextra_entitiesr1 r rw rw rx r � s4 � � z_ColumnEntity.row_processor)N) rl rm rn r� r� r r r� r r r rw rw rw rx r � s � 1 r c @ sH e Zd ZdZdZdZdZddd�Zdd� Zdd d d �d d �Z dd� Z dS )r0 NF)r� r1 r� r� r c C s� || _ || _|d u| _|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� r1 rO r� r r+ r, )r� rU r1 r� r- r" r" rw rw rx r� & s$ �z_RawColumnEntity.__init__c C s dS rq rw r rw rw rx r� J s z _RawColumnEntity.corresponds_tork r r r c C s | � |�S r| r& r rw rw rx r@ M s z2_RawColumnEntity.setup_dml_returning_compile_statec C s\ |� � }|r&|| jd�}|d u r,d S n| j}|jr:|�� }|j�|� |j�|� || _d S rq ) r; r1 r� r3 r� r� r� r� r+ )r� rU r9 r1 rw rw rx r1 T s z$_RawColumnEntity.setup_compile_state)N) rl rm rn r� r� r r� r� r� r@ r1 rw rw rw rx r0 s � $r0 c @ sD e Zd ZdZdZdZddd�Zdd� Zd d d d �d d�Zdd� Z dS )r$ z Column/expression based entity.F)r� r� r1 r� r� r� r Nc C s |j }|} |�dd �} |�d| �} | r<t| j| �| _d| _n|| _|d u| _|| _|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� r6 r� r. r- r6 r� r� r r� r� r� r� r� rA r1 r+ r, r r r r� )r� rU r1 r� r� r- r" r" r r2 Zorm_keyr8 r� r� rw rw rx r� v s> � �z_ORMColumnEntity.__init__c C s. t |�r|| ju S t | j� o(|�| j�S d S r| )r r� r� r rw rw rx r� � s � �z _ORMColumnEntity.corresponds_tork r r r c C s> | j | _}|r||d�}|d ur:|j�|� |j�|� d S rq )r1 r+ r� r� r� r� )r� rU r r1 rw rw rx r@ � s z2_ORMColumnEntity.setup_dml_returning_compile_statec C s� |� � }|r8|| jd�}|d u r>|js*J �| j| _d S n| j}| j}| jj}|d usdd| jf|jv r~||jrr|j nd f|j |< |j r�|j s�|� � }t| jj��|jj�r�|j�|j� |j�|� |j�|� || _d S )NFr� )r; r1 r� r+ r� r� r� r� r� r� r0 r� Z_expression_labelr3 r- rO � intersectionr� r. r� r� r� r� )r� rU r9 r1 r� r rw rw rx r1 � s8 ���� � z$_ORMColumnEntity.setup_compile_state)N) rl rm rn r� r r� r� r� r@ r1 rw rw rw rx r$ g s � C r$ c @ s e Zd ZdZdd� Zdd� ZdS )r/ Fc C s d S r| rw r rw rw rx r1 � s z(_IdentityTokenEntity.setup_compile_statec s � fdd�}|| j | jfS )Nc s � j jS r| )rY rs )�row�r� rw rx r5 s z2_IdentityTokenEntity.row_processor.<locals>.getter)r� r )r� r� r� r5 rw r; rx r s z"_IdentityTokenEntity.row_processorN)rl rm rn r. r1 r rw rw rw rx r/ � s r/ )NF)N)�� __future__r rQ �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� r_ r� ra rS r� r� r� rk r r r Z plugin_forr r rZ rK r� r r� r� r) r� r r r0 r$ r/ rw rw rw rx �<module> s t�l- +' ;x � � �_ - yN
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????