U
|
¸ý°d㶠ã @ 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<lEmFZF e rªdd=lGmHZH dd>lGmIZI dd?lmJZJ dd@lKmLZL ddAlMmNZN ddBlOmPZP ddClOmQZQ d dDlRmSZS d dElTmUZU d dFl,mVZV d dGlWmXZX d dHl8mYZY d dIl:mZZZ d dJl=m[Z[ d dKl=m\Z\ d dLl=m]Z] d dMl=m^Z^ d dNl=m_Z_ d dOl`maZa edPedQZbejcZde e¡ ZfeBjgZgG dRdS dSZhe edTdUi¡ZiG dVdW dWe4ZjG dXdY dYejZkG dZd[ d[Zle&jj4 md\d]¡G d^d_ d_ekZnG d`da dae;e6eDe- Zoe&jj4 md\db¡G dcdd ddekeCZpddfdgdhdidjdkdlZqdmdndodpdqZrdmdndodrdsZsddtdndudvdwdxZtG dydz dzZuG d{d| d|euZvG d}d~ d~euZwG dd deuZxG dd dexZyG dd dexZzG d
d dezZ{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)Ú 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 eZdd d d dddddddZd dddZd
|
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Úintrb z Dict[PathRegistry, PostLoad]rc ÚORMCompileStaterT 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© ru ru úMd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/orm/context.pyÚdefault_load_options s rw Nr. ú&Union[Select[Any], FromStatement[Any]]rE rC zQUnion[Type[QueryContext.default_load_options], QueryContext.default_load_options]z$Optional[OrmExecuteOptionsParameter]zOptional[_BindArguments])rT Ú statementrV rZ rW rY rX c
|
sº | _ |p t _|pt _| _| _| _d _d _| _ |j
|
_ |j j }|j } t fddt|| 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©Úselfru rv Ú <genexpr>¸ s þz(QueryContext.__init__.<locals>.<genexpr>)!rW Ú _EMPTY_DICTrY rX rT rU rZ rf rg rV rt rS Úselect_statementÚ _with_optionsÚtupleÚzipra Údictr rp r[ rm r\ ro r] rn r^ rs r_ rr re rq rd )
|
r~ rT ry rV rZ rW rY rX Zcached_optionsZuncached_optionsru r} rv Ú__init__ s.
|
|
þ
|
zQueryContext.__init__©Úreturnc C s
|
| j p| S rz )rS r} ru ru rv Ú_get_top_level_contextÈ s z#QueryContext._get_top_level_context)NN) rj rk rl Ú __slots__Ú__annotations__r1 rw r r ru ru ru rv rR e s
|
õ 6rR Z _result_disable_adapt_to_contextTc sb e Zd ZdZdd Zedddd d fdd Zed
|
d Zed d ddZedd Z Z
|
S )ÚAbstractORMCompileStateFc C sÊ i | _ |d kr$i | _}|s td 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~ ry ÚcompilerÚtoplevelÚ process_criteria_for_toplevelÚgar| Zstack_0Z toplevel_stmtru ru rv Ú_init_global_attributesÔ s0
|
|
|
|
z/AbstractORMCompileState._init_global_attributesú Union[Select, FromStatement]úOptional[SQLCompiler]r ©ry 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©Úclsry r r ©Ú __class__ru rv r û s z,AbstractORMCompileState.create_for_statementc C s
|
t d S rz ©ÚNotImplementedError)r¡ rZ ry rV rY rX Ú is_pre_eventru ru rv Úorm_pre_session_exec s
|
z,AbstractORMCompileState.orm_pre_session_execrD r c C s( |j ||p i |d}| ||||||¡S )N)rY )ÚexecuteÚorm_setup_cursor_result)r¡ rZ ry rV rY rX ÚconnÚresultru ru rv Úorm_execute_statement s
|
ÿúz-AbstractORMCompileState.orm_execute_statementc C s
|
t d S rz r¤ )r¡ rZ ry rV rY rX r« ru ru rv r© 0 s
|
z/AbstractORMCompileState.orm_setup_cursor_result) rj rk rl Úis_dml_returningr Ú classmethodr r§ r¬ r© Ú __classcell__ru ru r¢ rv r Ñ s '
|
r c @ s, e Zd ZU G dd deZded< 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 )<ri 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)rj rk rl 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º ru ru ru rv Údefault_compile_options> s. õr¾ zDict[Any, Any]r r rx ry 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_stylerK Ú_label_conventionzList[ColumnElement[Any]]Úprimary_columnsÚsecondary_columnszSet[ColumnElement[Any]]Údedupe_columnszList[Tuple[Any, ...]]r` r Ú current_pathFc O s
|
t d S rz r¤ )r~ Úargr ru ru rv r z s zORMCompileState.__init__r r r r c K s d S rz ru r ru ru rv r s z$ORMCompileState.create_for_statementc C s& | j }||kr"| |¡ | |¡ d S rz )rÉ ÚaddÚappend)r~ ÚobjZcol_collectionZdeduperu ru rv Ú _append_dedupe_col_collection s
|
z-ORMCompileState._append_dedupe_col_collectionr9 )Ú label_styleÚlegacyr c C s |rddd}|S t |¡S d S )Nc S s |r|S t | dS d S )NÚkey)Úgetattr)ÚcolÚcol_nameru ru rv Úname s z7ORMCompileState._column_naming_convention.<locals>.name)N)r: Ú_column_naming_convention)r¡ rÐ rÑ rÖ ru ru rv r× s
|
z)ORMCompileState._column_naming_conventionc C s t |S rz )Ú_column_descriptions©r¡ ry ru ru rv Úget_column_descriptions s z'ORMCompileState.get_column_descriptionsc C sò t j ddddddh||j¡\}}|s,t}n
|
| t¡}|jrL| d|ji¡}t|jdd rt |jj
|
dkr| d d
|
¡d k rt d ¡ | d d i¡}||d < z|jd }W n tk
|
rÈ dsÄtdY nX |rØ|j|d< |sê|jrê| ¡ ||fS )NÚ_sa_orm_load_optionsr\ r[ re rd Zsa_top_level_orm_contextr³ é
|
Zcompiled_cacheTz³Loader depth for query is excessively deep; caching will be disabled for additional loaders. Consider using the recursion_depth feature for deeply nested recursive eager loaders.ÚclauseÚplugin_subjectFz0statement had 'orm' plugin but no plugin_subjectÚmapper)rR rw Zfrom_execution_optionsÚ_execution_optionsÚ_orm_load_exec_optionsÚunionrr rÓ Ú_compile_optionsÚlenr³ r r# ÚwarnÚ_propagate_attrsr r rß rp ) r¡ rZ ry rV rY rX r¦ rW rÞ ru ru rv r§ £ sV ûöý
|
ÿ ÿþýÿÿ
|
|
z$ORMCompileState.orm_pre_session_execc C sH |j }|jj}| dtj¡} |jjr(|S t||||| ||}
|
t ||
|
¡S )NrÛ )
|
ÚcontextZcompiledrT r rR rw r rº r Z instances) r¡ rZ ry rV rY rX r« Zexecution_contextrT rW Z querycontextru ru rv 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 |tr|qS ru ©Ú
|
isinstanceÚ _MapperEntity©r{ Úentru ru rv Ú
|
<listcomp># s
|
z9ORMCompileState._lead_mapper_entities.<locals>.<listcomp>)rÀ r} ru ru rv Ú_lead_mapper_entities 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 Úmpru ru rv Ú _create_with_polymorphic_adapter' s ÿÿþüþz0ORMCompileState._create_with_polymorphic_adapterc C s: |j p
|
|gD ](}|| j|< | ¡ D ]}|| j|j< q"q d S rz )Ú_with_polymorphic_mappersrÁ rò Ú local_table)r~ rß ÚadapterÚm2Úmru ru rv ró D s
|
z2ORMCompileState._mapper_loads_polymorphically_withc C s t dd S )Nz0this method only works for ORMSelectCompileStater¤ )r¡ rU rÑ ru ru rv Ú_create_entities_collectionK s ÿz+ORMCompileState._create_entities_collectionN)rj rk rl r- r¾ r r½ rÊ Ú_has_mapper_entitiesr r r® r rÏ r× rÚ r§ r© Úpropertyrî r÷ ró rý ru ru ru rv ri = sH
|
%
|
|
S
|
#
|
ri 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 rz )rù rß r# ZWeakPopulateDictÚadapt_check_presentr )r~ Ú target_mapperZimmediate_dml_mapperru ru rv r ^ s ÿÿþÿz DMLReturningColFilter.__init__c C s. t |¡D ] }| |¡}|d k r
|
| S q
|
d S rz )Úsql_utilZ _find_columnsr )r~ rÔ Ú as_filterÚccÚc2ru ru rv Ú__call__m s
|
|
|
z DMLReturningColFilter.__call__c C s. | j }|j |d ¡}|d kr d S |jj |¡S rz )rß Z_columntopropertyr rù ÚcZcorresponding_column)r~ rÔ rß Úpropru ru rv r u s
|
z)DMLReturningColFilter.adapt_check_presentN)rj rk rl Ú__doc__r r r r ru ru ru rv r 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 dZdd 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 |tst|d k r&|jr&t d¡| | ¡}d |_|jj |_
|
| |_ |_ |j |_}|jrl|j|_d|_g |_i |_|j|_|j
|
r¶t |tjr¶|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 tj1râi |_2|jD ]} | 3|¡ qd |_4|_5d|_6|j*D ]}
|
|j7|
|
d|j8d qÆnt9t:j;|j |j<d|_=|S )NzThe 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% rN Z _is_textualÚ _label_styler6 Zset_label_styler7 ry r× rÆ Ú _QueryEntityÚto_compile_stateÚ _raw_columnsr³ rÊ r r r r Ú_with_context_optionsrÇ rÈ ÚsetrÉ r` Ú_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~ ry r| ÚfnrÒ Úentityr
|
ru ru rv r sª ÿ
|
ÿ ÿ
|
þýüûÿÿÿüü
|
ü
|
|
þ
|
ý ýz1ORMFromStatementCompileState.create_for_statementc C s |S rz ru ©r~ ÚcolsÚcurrent_adapterru ru rv Ú_adapt_col_list s z,ORMFromStatementCompileState._adapt_col_listc C s d S rz ru r} ru ru rv Ú_get_current_adapter s z1ORMFromStatementCompileState._get_current_adapterc C s6 | j j dd¡}t||}| jD ]}| | |¡ q dS )zused by BulkORMInsert (and Update / Delete?) to set up a handler
|
for RETURNING to return ORM objects and expressions
|
|
rÞ N)ry ræ r r rÀ Ú!setup_dml_returning_compile_state)r~ Z
|
dml_mapperr rú r( ru ru rv r. s ÿ
|
|
z>ORMFromStatementCompileState.setup_dml_returning_compile_state)rj rk rl r& rþ r r Ú_has_orm_entitiesÚmulti_row_eager_loadersÚeager_adding_joinsÚcompound_eager_adapterr r" Ú eager_joinsr® r r, r- r. ru ru ru rv r } s"
|
r c sÐ e Zd ZU dZdZejZejZ dZ
|
de d< de d< de j fde jfgej Zed e jfg Zd d ddd d dZdd Zedd Zdd Z fddZedd Zedd Zedd 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 fddt |¡D _| _|j _t|r6|jnd _| _d S )Nc s g | ]}t jtj| d dqS )T)Úapply_propagate_attrsZ post_inspect)r$ Úexpectr& ZColumnsClauseRolerë r} ru rv rí G s úüz*FromStatement.__init__.<locals>.<listcomp>)r# Zto_listr r r) r+ r r% )r~ r4 r r% ru r} rv r A s
|
ù ÿzFromStatement.__init__c K s2 | j | |f|}|j }|r"||_|j|jf|S )zprovide a fixed _compiler_dispatch method.
|
|
This is roughly similar to using the sqlalchemy.ext.compiler
|
``@compiles`` extension.
|
|
)Ú_compile_state_factoryr rT r$ ry )r~ r r rT r ru ru rv Ú_compiler_dispatchW 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~ Úmethru ru rv Úcolumn_descriptionsi s
|
ÿz!FromStatement.column_descriptionsc C s | S rz ru r} ru ru rv Ú_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 rz )Ú _from_objects©r{ r ru ru rv r s z-FromStatement.get_children.<locals>.<genexpr>)Ú itertoolsÚchainÚ from_iterabler r Ú get_children)r~ r r¢ ru rv rB ~ s ÿzFromStatement.get_childrenc C s | j jS rz )r Ú_all_selected_columnsr} ru ru rv rC s z#FromStatement._all_selected_columnsc C s t | jr| jjS d S rz )r) r Ú_return_defaultsr} ru ru rv rD s z FromStatement._return_defaultsc C s t | jr| jjS d S rz )r) r Ú
|
_returningr} ru ru rv rE s zFromStatement._returningc C s t | jr| jjS d S rz )r* r Ú_inliner} ru ru rv rF s zFromStatement._inline)T) rj rk rl r Z__visit_name__r r¾ rã r r7 Ú_for_update_argr r< Zdp_clauseelement_listZdp_clauseelementr/ Z _executable_traverse_internalsZ_traverse_internalsr» r¼ r r8 rÿ r; r< rB rC rD rE rF r¯ ru ru r¢ rv r " s<
|
þýÿ ü
|
|
|
|
r Úselectc @ sZ e Zd ZdZeZdZdZdZdZ dZ
|
dZ dZ dZ dZdZedddd ddd 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 )Er9 ru NFr r r r c
|
s4 | | ¡ |dk r |j }nd}|}| j |j¡|_|jrR| ¡ _t ¡ j_n| _|jj
|
_ |jj _ g _d _i _|j _|s jddd7 _ 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|jr®|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 dqS )Fr )r r r )r{ Úmemoized_entitiesr} ru rv Ú
|
<dictcomp>ï s ú üz>ORMSelectCompileState.create_for_statement.<locals>.<dictcomp>r ru )r r c S s g | ]}t |tr|qS ru rè rë ru ru rv rí ! s
|
üz>ORMSelectCompileState.create_for_statement.<locals>.<listcomp>c s s | ] }|j V qd S rz )r )r{ Úinforu ru rv r D 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_ORMr7 rÐ r5 Ú_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É r3 r" r` r Ú_normalize_fromsÚ _from_objÚ from_clausesÚ_setup_for_generater: r ry )
|
r¡ ry r r r r rI r| r' rÒ ru r} rv r ¨ s¶
|
|
ÿ
|
|
ÿþ
|
|
ÿ
|
þ
|
|
ø ÿü
|
üÿþþ
|
ÿþþ
|
ÿ
|
z*ORMSelectCompileState.create_for_statementc C sÄ t d t d| j | jD ] }t|tr||d dkr|t dt |d ¡ t d| j| t d| j| j q t|tr |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Ò ru ru rv Ú_dump_option_structP s
|
z)ORMSelectCompileState._dump_option_structc sÞ | j }d | _d| _| jjr"| ¡ |jD ] }|jr(| |j| j | ¡ q(|jr^| |j| j
|
¡ | ¡ |j r|j | _ rt fdd| j D | _ r®|jdkr®| |j ¡n|j| _|jrÖt fdd|jD | _ rø|jdkrø| t |j¡ ¡n
|
|jpd | _| jr(| jd j}| | j¡| _|jrB| |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
|
krÚt& 'd ¡| j
|
D ]}| (| ¡ qà| j)D ]$}|d }|| f|d
|
d
qø| jj*r0| +¡ | j,sZ| jj-rPd sPt.d t& /d¡| jsnt0| j1| _| jd krd | _| j2r¤| j3r¤| j4r¤| 5¡ | _n
|
| 6¡ | _| j7rÚ| 8¡ }|d k rÚ| j 9d|i¡| _d S )Nru c 3 s | ]} |d V qdS ©TNru ©r{ Úcrit©r+ ru rv r v s ÿz<ORMSelectCompileState._setup_for_generate.<locals>.<genexpr>©NFc 3 s | ]} r |d n|V qdS rZ ru r[ r] ru rv r
s ÿr c s s$ | ] }|d k rt |¡nd V qd S rz ©r Zsurface_selectables©r{ Úsru ru rv r ¤ s ÿc s s$ | ] }|d k rt |¡nd V qd S rz r_ r` ru ru rv r « s ÿrz r zsCan't generate ORM query that includes multiple expressions at the same time as '*'; query for '*' alone if presentFz,no columns were included in _only_load_propsz4Query contains no columns with which to SELECT from.Z
|
deepentity):r ry Ú_join_entitiesr r· Ú_set_select_from_aliasrO Ú _setup_joinsÚ_joinrP rÀ r- Ú_where_criteriar Ú_order_by_clausesr, r! Ú_having_criteriaÚ_group_by_clausesr# Zflatten_iteratorÚgroup_byrQ rT Z_target_adapterÚcopy_and_processZ _distinct_onÚ distinct_onZ _distinctÚdistinctZ
|
_correlateÚ correlateZ_correlate_exceptÚcorrelate_exceptZ_auto_correlaterG rº rä r r r# r` r´ Ú_adjust_for_extra_criteriarÇ r¶ r ÚInvalidRequestErrorÚlistr r0 r1 Ú_should_nest_selectableÚ_compound_eager_statementÚ_simple_statementrM Ú _mapper_zeroZ _annotate)r~ rU rI rú r( ZrecZstrategyÚezeroru r] rv rU \ 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¡ rU rÑ r~ ru ru rv rý é s
|
ÿ ÿz1ORMSelectCompileState._create_entities_collectionc C s |j }t|d S rz )rd Ú _determine_last_joined_entity)r¡ ry Ú setup_joinsru ru rv Ú determine_last_joined_entity s z2ORMSelectCompileState.determine_last_joined_entityc c sj |j D ]^}|jrTd|jkrT|jd }|jsB|jsBt|gE d H qdt|jE d H qt|gE d H qd S )NZentity_namespace)r Ú is_selectableÚ _annotationsZ is_mapperrð r, Z_all_column_expressions)r¡ ry r Zensru ru rv Úall_selected_columns 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| r= Ú__clause_element__r> ru ru rv r s
|
üÿ ÿzAORMSelectCompileState.get_columns_clause_froms.<locals>.<genexpr>)rR r? r@ rA r rÙ ru ru rv Ú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)r5 )r r rà ræ ) r$ r6 r& ZReturnsRowsRoler r rX Úupdater r rà ræ )r¡ ry Úfrom_statementZstmtru ru rv r % s ý üz$ORMSelectCompileState.from_statementc C sT | j }| jjstt|jdks$t| |jd ¡}|rP| jddi7 _|| _dS )ú used only for legacy Query casesr r r´ FN)r r r· r rä rS Ú _get_select_from_alias_from_objr& )r~ rU rú ru ru rv rc 8 s z,ORMSelectCompileState._set_select_from_aliasc C s| |}d|j kr|j d }t|dr<|js4t d¡qx|jS n<t|jtjj rt|
|
¡ }||jksbt t t j|j|dS 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_objrK Úequivsru ru rv r E 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} ru ru rv rv a s z"ORMSelectCompileState._mapper_zeroc C sD | j D ] }d|jkr|jd S q| jD ]}|jr*|j S q*dS )zReturn the 'entity' (mapper or AliasedClass) associated
|
with the first QueryEntity, or alternatively the 'select from'
|
entity if specified.r~ N)rT r| rÀ Ú entity_zero)r~ rì Zqentru ru rv Ú _entity_zeroe 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 rq r )r~ Zmethnameru ru rv Ú_only_full_mapper_zeros s ÿÿz,ORMSelectCompileState._only_full_mapper_zeroc C s$ t | jdkr t |pd¡| ¡ S )Nr z8This operation requires a Query against a single mapper.)rä rÀ r rq r )r~ Z rationaleru ru rv Ú_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 |tr|qS ru rè rë ru ru rv rí s
|
þz5ORMSelectCompileState._all_equivs.<locals>.<listcomp>c S s g | ]}t |tr|qS ru rè rë ru ru rv rí s
|
)rP Úvaluesr rß rô rÀ )r~ r rI rì ru ru rv 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 |tjjr |jn|qS ru )ré r" ÚelementsZ_label_referencer )r{ Úelemru ru rv rí ¡ s þ ÿzCORMSelectCompileState._compound_eager_statement.<locals>.<listcomp>c s g | ]}| j kr|qS ru )rÉ )r{ r
|
r} ru rv rí ³ s
|
©Ú
|
for_updateÚhintsÚstatement_hintsrn ro r
)#r! r Z expand_column_list_from_order_byrÇ Ú_select_statementrT rf rh rÐ rG r Ú_hintsÚ_statement_hintsrn ro Ú _select_argsÚaliasr r r ZCOMPOUND_EAGER_STATEMENTr2 r rH rÈ r Zofr3 r Z splice_joinsZstop_onÚ select_fromÚnon_generativerk rQ )r~ Zunwrapped_order_byZorder_by_col_exprÚinnerr ry Z from_clauseZ
|
eager_joinru r} rv rt s| ü ÿÿùô ó ÿ
|
ÿÿ
|
þ ÿÿÿþz/ORMSelectCompileState._compound_eager_statementc C sz | j | j| j t| jt| 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 rT r3 r rf rh rÐ r! rG r r r rn ro r rQ r )r~ ry ru ru rv ru î 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 rS r )r8 Z_create_raw_selectrf rh rg r rm r ri Ú _limit_clauseÚ_offset_clauseÚ _fetch_clauseÚ_fetch_clause_optionsÚ _prefixesÚ _suffixesrG r r rn ro )r~ Z raw_columnsr Zwhere_criteriaZhaving_criteriarÐ r! r r r rn ro Ú limit_clauseÚ offset_clauseÚ fetch_clauseÚfetch_clause_optionsrm rl ÚprefixesÚsuffixesrj ry ru ru rv r sN ý ÿÿz'ORMSelectCompileState._select_statementc C sx d|j kr0|j d }| j |d ¡}|r0| |¡S t|tjrB|}nt|drT|j}nd S | j |d ¡}|rt| |¡S d S )Nr~ r ) r| rÁ r Z adapt_clauseré r% Z
|
FromClauser r )r~ r Úsearchr ru ru rv Ú_adapt_polymorphic_elementM s
|
|
|
|
z0ORMSelectCompileState._adapt_polymorphic_elementc s r fdd|D S |S d S )Nc s g | ]} |d qS )Tru )r{ Úor] ru rv rí a s z9ORMSelectCompileState._adapt_col_list.<locals>.<listcomp>ru r) ru r] rv 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 kpd| j k} D ](\}}|s(|r|| }|d k r| S qd S )NÚ
|
_orm_adaptr~ )r| )r Z is_orm_adaptZ always_adaptrú Úe©Úadaptersru rv Úreplace s
|
þ zRORMSelectCompileState._get_current_adapter.<locals>._adapt_clause.<locals>.replace)r' Zreplacement_traverse)rÝ r r² r° ru rv Ú _adapt_clause s zAORMSelectCompileState._get_current_adapter.<locals>._adapt_clause)r& rÍ r² rÁ r¬ )r~ r³ ru r° rv r- e s
|
þÿ
|
z*ORMSelectCompileState._get_current_adapterc C s |D ]¶\}}}}|d }|d }t |}|d k r:t |}t|tjrb|d k rXt d¡|}d }nd|jkrv|jd }|d kr|jst|dst d¡d } t|tjrt
|
|dd } |d kr| rÊ| }nF|j }z
|
|j }W n4 t k
|
r }
|
zt d| ¡|
|
W 5 d }
|
~
|
X Y nX |j} |j } t|tjs.| }| || jf| jkrDq|d k r¦|| k r¦|j dd ¡| k r¦t d |j d|¡|f ¡n |d k rd } |} 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 rq 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 ru ru rv re s ÿ
|
|
ÿ
|
|
ÿÿý
|
ÿ ÿþ þþÿ
|
ùzORMSelectCompileState._joinc C s |dkr,|dkst | ||||¡\}}} n| ||¡\}} ||krVt d||f ¡| ||||¡\}
|
}}|
|
js~| |
|
¡} nd} |dk rÎ| j| } | jd|
t | ||||| dg | j|d d
| _nH| dk rôt
|
|| t sèt || j } n|} | jt | |
|
|||| dg | _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 entityru )r´ rµ Z_extra_criteriar ) r Ú"_join_determine_implicit_left_sideÚ _join_place_explicit_left_sider rq Ú _join_check_and_adapt_right_sider{ Ú_get_extra_criteriarT r ré rê r ) r~ r¼ r r½ r¾ r Z outerjoinrµ Úreplace_from_obj_indexÚuse_entity_indexÚr_infoZextra_criteriaZ left_clauseru ru rv 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 } | dkrqt | } | |kr®qt
|
| t rÈ|
|
| f| | j< qd| f| | j< qt | ¡ }t ||j|¡}t|dkr| ||d \}}n*t|dkr2t d¡nt d|f ¡n
|
t d¡|||fS )zWhen 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! rT r Z find_left_clause_to_join_fromr rä r rq Ú enumerateÚentity_zero_or_selectableré rê rr Úkeys)r~ r¼ r r½ r¾ rÉ rÇ rÈ ÚindexesZ potentialZ entity_indexrì r( Zent_infoZ all_clausesru ru rv rà X sj ÿ ÿ üÿ
|
ÿÿ üÿÿz8ORMSelectCompileState._join_determine_implicit_left_sidec C s d }}t |}| jrHt | j|j¡}t|dkr<t d¡|rH|d }|dkr|rt|drt |D ]$\}}t
|
|t rf| |¡rf|} qqf||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! rT r Z#find_left_clause_that_matches_givenr rä r rq r rÊ ré rê Úcorresponds_to) r~ r¼ r rÇ rÈ Úl_inforÍ Úidxrì ru ru rv rÄ Â s0 ÿ ÿ
|
ÿþýz4ORMSelectCompileState._join_place_explicit_left_sidec C s t |}t |}d}t|dd}|rp|js8t|jtjrp| jpD|jgD ](} t
|
|j| ¡rFt
|
| |j¡rFd} qpqF|r|j|jkrt d|j ¡t|dd|jt|dd }}
|
} |rØ|rØ| |j¡sØt d||f ¡t|drò| j|f7 _d} |jr|r
|
|j}|
|
jr0| ¡ } | dk r| |d}nj|r|
|
|j¡s\t d|
|
j|jjf ¡t|
|
tjr|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|jsB| 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% ZJoinrT r r Zselectables_overlapr rq Ú common_parentrß r rb Úis_clause_elementZ _is_lateralr- Zis_derived_fromÚ descriptionrN r$ r6 r& ZFromClauseRoler r# Zwarn_deprecatedrå r 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É ÚoverlapZ right_mapperr Zright_selectableZright_is_aliasedZ need_adapterr+ Zaliased_entityrú ru ru rv rÅ sê ÿ þ ÿ ÿþÿÿ
|
|
ý
|
ÿþ
|
ýÿÿ
|
|
ÿþÿÿ
|
ÿ
|
ûø ûù
|
|
ý
|
ÿþýüþ
|
z6ORMSelectCompileState._join_check_and_adapt_right_sidec
|
C s: | j j| j j| j| j| j j| j j| jp(d | j j| j j d S )N) r¥ r¦ rm rl r© rª rj r§ r¨ )
|
r r r rm rl r£ r¤ rj r¡ r¢ r} ru ru rv 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¦ rm Frl ru rj )r r )r~ Úkwargsru ru rv rs Ò s ÿ
|
þ
|
ý
|
ûz-ORMSelectCompileState._should_nest_selectablec s< d j fjkr4t fddjd j f D S dS d S )NÚadditional_entity_criteriac 3 s2 | ]*}|j s|j kr| ¡r| ¡V qd S rz )Zinclude_aliasesr( Z_should_includeZ_resolve_where_criteria)r{ Zae©rõ r~ ru rv r â s
|
|
|
ûz<ORMSelectCompileState._get_extra_criteria.<locals>.<genexpr>ru )rß r r )r~ rõ ru rÙ rv 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 krqp|jj}| j j rg }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)rT r| r rß Ú_single_table_criterionr r" rð r r r rb r r¸ rÆ r- Ztraverser Z_deep_annotaterf ) r~ Z
|
fromclauserõ r« rú Z single_critrØ r+ r\ ru ru rv rp í sB
|
ÿ
|
ýÿüù þ
|
|
|
|
|
z0ORMSelectCompileState._adjust_for_extra_criteria)N)0rj rk rl r¹ r rP r& rþ r/ r0 r1 r2 rn ro rf rh r® r rY rU rý rz r} r r rc r rv r r r r rt ru r r¬ r, r- re rº rà rÄ rÅ rÿ r rs rÆ rp ru ru ru rv r9 sj (
|
|
|
|
|
|
YJ1_cjB ?
|
|
|
r9 Fz#Union[Query, Select, FromStatement]z Optional[ORMSelectCompileState]rÄ zList[ORMColumnDescription])Úquery_or_select_stmtrT 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|ddnddqS )rð FNr( )rÖ ÚtypeÚaliasedÚexprr( )Ú _label_namerÜ rÓ rÞ r rÔ )r{ rì Zinsp_entru ru rv rí 2 s
|
ø
|
ÿÿùz(_column_descriptions.<locals>.<listcomp>c S s g | ]}||j fqS ru ©r )r{ Z_entru ru rv rí < s )r9 rý rÀ )rÛ rT rÑ ÚctxÚdru ru rv rØ ( 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 )rd Ú_last_joined_entityrS r| Ú_entity_from_pre_ent_zero)rã r~ rä ru ru rv Ú _legacy_filter_by_entity_zeroC s ræ c C sX | }|j sd S |j d }d|jkr,|jd S t|tr<|jS d|jkrP|jd S |S d S )Nr r~ Úbundle)r r| ré r r( )rã r~ rì ru ru rv rå R s
|
|
|
|
|
|
rå z Tuple[_SetupJoinsElement, ...]z=Optional[Union[_InternalEntityType[Any], _JoinTargetElement]])ry r r c C s2 | sd S | d \}}}}t |tjr*|jS |S d S )Néÿÿÿÿ)ré r r¸ r( )ry r Útargetr¾ r¿ rÀ ru ru rv rx e s þrx c @ s e Zd ZU dZdZded< dZdZdZded< ded < d
|
ed < d ed < dddddZ ddddddZ
|
dd Z e dd Z dS ) r z:represent an entity column returned within a Query result.ru rÄ Úsupports_single_entityFz Optional[str]rß z!Union[Type[Any], TypeEngine[Any]]rÜ z.Union[_InternalEntityType, ColumnElement[Any]]rÞ z Optional[_InternalEntityType]r ri ÚNone)rT r c C s
|
t d S rz r¤ ©r~ rT ru ru rv r# s z _QueryEntity.setup_compile_stater ©rT rú r c C s
|
t d S rz r¤ ©r~ rT rú ru ru rv r. s z._QueryEntity.setup_dml_returning_compile_statec C s
|
t d S rz r¤ )r~ rç r« ru ru rv Ú 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¡rt |||| 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¡ rT r4 r¼ r rÐ r( ru ru rv r sh ü
|
üûüûûz _QueryEntity.to_compile_stateN)rj rk rl r r r Ú_non_hashable_valueÚ_null_column_typeÚuse_id_for_hashr# r. rï r® r ru ru ru rv r w 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þ r/ r| Ú _post_inspectr r( rÞ rß r÷ rð rÖ rß Úclass_rj r½ rö r Zwith_polymorphic_mappersrø Zpolymorphic_onrø Ú _should_select_with_poly_adapterr÷ )r~ rT r( r¼ r rõ rß ru ru rv r î s4
|
|
|
|
|
|
|
ÿz_MapperEntity.__init__Tc C s | j jS rz )rß rû r} ru ru rv rÜ
|
s z_MapperEntity.typec C s | j S rz rà r} ru ru rv rË
|
s z'_MapperEntity.entity_zero_or_selectablec C s t | j|S rz )r r ©r~ r( ru ru rv rÎ
|
s z _MapperEntity.corresponds_toc C sR d }| j s"|jr*|j | jd ¡}n| jj}|rH|jrB| |j¡}qN|}n|j}|S rz )rð rÁ r rß r r r& Úwrap)r~ rT rú Úretru ru rv Ú_get_entity_clauses#
|
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)rT r r2 rþ rà r r¶ r_ r Z_instance_processorrß rö rø rß r÷ )r~ rç r« rT rú r r_ Z _instanceru ru rv rï 9
|
s,
|
|
|
÷ z_MapperEntity.row_processorri 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î ru ru rv r. V
|
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~ rT rú Úsingle_table_critrõ ru ru rv r# g
|
s0
|
ÿÿþþ
|
÷ z!_MapperEntity.setup_compile_stateN)rj rk rl r r r r rê ró rõ rÿ rÜ rË rÎ r rï r. r# ru ru ru rv rê × s*
|
%
|
|
rê c @ s e Zd ZU dZdZded< ded< ded< d ed
|
< d ed < ded < d ddZedd Zedd Z dd Z
|
edd Z dd Z d d Z dS ) rò ru )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|tjtjfrH| ¡ }n|}| | _
|
| _ t || _ |j | _g | _|rÔ|jD ]X}d|jkrt||||| d qzt|trºt||||| d qztj||g|d || d qz| j
|
j| _d S )NTrç ©Ú parent_bundle)r/ r| rÀ rÍ ré r r¸ r r¶ r rç rÞ rÜ rÖ rß Úexprsrò r Ú_ORMColumnEntityrñ Z single_entityrê )r~ rT rÞ r¼ r Zsetup_entitiesr rç ru ru rv r
|
sT
|
|
|
ÿ
|
|
|
|
û
|
ûú z_BundleEntity.__init__c C s | j }|d k r|jS d S d S rz )r rß )r~ rw ru ru rv rß Ó
|
s z_BundleEntity.mapperc C s& | j D ]}|j}|d k r| S qd S rz )rÀ r ©r~ rì rw ru ru rv r Û
|
s
|
|
|
z_BundleEntity.entity_zeroc C s dS r^ ru rý ru ru rv rÎ ä
|
s z _BundleEntity.corresponds_toc C s& | j D ]}|j}|d k r| S qd S rz )rÀ rË r
|
ru ru rv rË é
|
s
|
|
|
z'_BundleEntity.entity_zero_or_selectablec C s | j D ]}| |¡ qd S rz )rÀ r# )r~ rT rì ru ru rv r# ò
|
s
|
z!_BundleEntity.setup_compile_statec s@ t fdd| jD \}}}| j j||¡}|| j| jfS )Nc s g | ]}| ¡qS ru )rï rë ©rç r« ru rv rí ø
|
s z/_BundleEntity.row_processor.<locals>.<listcomp>)r rÀ rç Zcreate_row_processorrU rß r÷ )r~ rç r« ZprocsÚlabelsÚextraÚprocru r rv rï ö
|
s
|
ÿ
|
z_BundleEntity.row_processor)TN)rj rk rl r÷ r r r rÿ rß r rÎ rË r# rï ru ru ru rv rò
|
s(
|
ù
|
;
|
|
|
rò c @ sJ e Zd ZdZed ddZedd 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~ rd r )r| r Z extract_first_column_annotationÚ_IdentityTokenEntityr Ú_RawColumnEntity)
|
r¡ rT r r¼ r r r Úcolumnr Ú_entityru ru rv rñ sH
|
|
ÿ
|
ù
|
ù
|
úz_ColumnEntity._for_columnsc C s | j jS rz )r rÜ r} ru ru rv rÜ : s z_ColumnEntity.typec C s | j jj S rz )r rÜ Zhashabler} ru ru rv ró > s z!_ColumnEntity._non_hashable_valuec C s
|
| j jjS rz )r rÜ Z_isnullr} ru ru rv rô B 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 rz )rT r r rU r r r r r& r r| Ú _deannotater2 Z_getterrß r÷ ) r~ rç r« rT ÚgetterZ
|
label_nameZextra_entitiesr rÿ ru ru rv rï F s4
|
ÿ
|
|
|
ÿ z_ColumnEntity.row_processor)N) rj rk rl r r® rñ rÿ rÜ ró rô rï ru ru ru rv rð s ù 1
|
|
|
rð c @ s6 e Zd ZdZdZdZdZd
|
ddZdd Zdd Z dS ) r NF)rÞ r rß rË r÷ c C s¦ || _ || _|d k | _|jr.| jddi7 _|r8|jr@d | _n | |¡| _|r^|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ß rÆ rÀ rÍ r r= rË r÷ r r )r~ rT r r¼ r r r ru ru rv r s
|
|
|
ÿz_RawColumnEntity.__init__c C s dS r^ ru rý ru ru rv rÎ ¨ s z _RawColumnEntity.corresponds_toc C s\ | ¡ }|r&|| jd}|d kr,d S n| j}|jr:| ¡ }|j |¡ |j |¡ || _d S r^ ) r- r r| r rÉ rÌ rÇ rÍ r )r~ rT r+ r ru ru rv r# « s z$_RawColumnEntity.setup_compile_state)N)
|
rj rk rl r rß rê r r rÎ r# ru ru ru rv r z s ù
|
!r c @ sD e Zd ZdZdZdZdddZdd Zd d
|
d d d dZdd 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 | _|| _|rh|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/ r r r r÷ rü r÷ r )r~ rT r r¼ r~ r r r r r Zorm_keyr rw rß ru ru rv r Í s>
|
|
ÿ
|
|
ÿz_ORMColumnEntity.__init__c C s. t |r|| jkS t | j o(| | j¡S d S rz )r r rÓ rý ru ru rv rÎ s
|
ÿ
|
þz _ORMColumnEntity.corresponds_tori r rë rí c C s8 | j | _|| j d}|d k r4|j |¡ |j |¡ d S r^ )r r rÉ rÌ rÇ rÍ )r~ rT rú r ru ru rv r. 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Ø )r- r r r r r rß rÚ r rð r r" r| Z_expression_labelr r r= Ú intersectionr r rÍ rÉ rÌ rÇ )r~ rT r+ r rw r ru ru rv r# ! s8
|
ÿÿþþ
|
ÿ z$_ORMColumnEntity.setup_compile_state)N)
|
rj rk rl r rê r r rÎ r. r# ru ru ru rv r ¾ s ø
|
@ r c @ s e Zd ZdZdd Zdd ZdS )r Fc C s d S rz ru rì ru ru rv r# Q s z(_IdentityTokenEntity.setup_compile_statec s fdd}|| j | jfS )Nc s j jS rz )rW rq )Úrow©rç ru rv r U s z2_IdentityTokenEntity.row_processor.<locals>.getter)rß r÷ )r~ rç r« r ru r rv rï T s z"_IdentityTokenEntity.row_processorN)rj rk rl r r# rï ru ru ru rv r N s r )NF)N)|Ú
|
__future__r r? Ú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; Z sql.visitorsr< Ú_typingr= r> r? rß r@ rU rA rZ rB rC ZenginerD Zengine.interfacesrE rF Z sql.compilerrG rH rI rJ rK rL rM rN Z sql.type_apirO rP Úrootr½ rL r rN rR rá r ri r Z
|
plugin_forr r r9 rØ ræ rå rx r rê rò rð r r r ru ru ru rv Ú<module> sø gÿl + %s ý þ` .|zD
|
