관리-도구
편집 파일: strategies.cpython-38.pyc
U -?�fK� � @ s� d Z ddlmZ ddlZddlZddlmZ ddlmZ ddlmZ ddlm Z dd l m Z dd l m Z dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddl m Z ddl m!Z! dd l m"Z" dd l m#Z# dd lm$Z$ dd lm%Z% dd l&m'Z' dd!l(m)Z) dd"l*m+Z+ dd#lm,Z, dd$lm-Z- d%d&l m.Z. d%d l m Z/ d%d'l m0Z0 d%d(l m1Z1 d%d)l m2Z2 d%dl mZ d%dl2mZ3 d%d*l2m4Z4 d%d+l5m6Z6 d%d,l5m7Z7 e �r6dd-lm8Z8 d%d.l9m:Z: dbd0d1�Z;ej<j=d/d/d2�G d3d4� d4e$��Z>e1j?ej<j=d5d/d2�G d6d7� d7e$���Z@e1j?ej<j=d5d8�G d9d:� d:e@���ZAe1j?ej<j=d5d5d;�ej<j=d5d5d5d<�ej<j=d5d=�G d>d?� d?e$�����ZBG d@dA� dA�ZCG dBdC� dCe$�ZDe1j?ej8j=d5d=�G dDdE� dEe$���ZEe1j?ej8j=dFdG�ej8j=ddG�G dHdI� dIeD����ZFe1j?ej8j=d5dG�ej8j=dJdG�ej8j=dKdG�ej8j=dLdG�ej8j=dMdG�G dNdO� dOeDejGe1jH�������ZIG dPdQ� dQ�ZJG dRdS� dSeD�ZKej8j=dTdG�G dUdV� dVeK��ZLe1j?ej8j=dWdG�G dXdY� dYeK���ZMe1j?ej8j=dZdG�ej8j=d/dG�G d[d\� d\eD����ZNe1j?ej8j=d]dG�G d^d_� d_eKejG���ZOd`da� ZPdS )czWsqlalchemy.orm.interfaces.LoaderStrategy implementations, and related MapperOptions.� )� annotationsN)�Any)�Dict)�Tuple)� TYPE_CHECKING� )� attributes)�exc)� interfaces)�loading)� path_registry)� properties)�query)� relationships)� unitofwork)�util)�_DEFER_FOR_STATE)�_RAISE_FOR_STATE)�_SET_DEFERRED_EXPIRED)� ATTR_WAS_SET)�LoaderCallableStatus)� PASSIVE_OFF)� PassiveFlag)�_column_descriptions)�ORMCompileState)�ORMSelectCompileState)� QueryContext)�LoaderStrategy)�StrategizedProperty)�_state_session)� InstanceState)�Load)� _none_set)� AliasedClass� )�event)�inspect)�log)�sql)�visitors)� LABEL_STYLE_TABLENAME_PLUS_COL)�Select)�RelationshipProperty)� ColumnElementFc s" g } |o | j �|r"| jr"| �t� | j| jjkrV| jj| j \��| ���fdd�� |rf| �tj� |r�| j � � r�| j r�| �� �fdd�� |j D ]�} | | j � | j�kr�| j�| j�s�tj| j| jf| �|||o�| jp�| jtjk||||| p�| j | jd� | ��} | D ]} | | | � �q q�d S )Nc s t j| |j� f��S �N)�orm_utilZ_validator_events�key��desc�prop)�fn�opts� �I/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/strategies.py�<lambda>X s ��z%_register_attribute.<locals>.<lambda>c s t �| � ��S r. )r Zbackref_listenersr1 )�backref�uselistr6 r7 r8 f s �) Z parent_tokenr: �compare_function� useobjectZ trackparent� typecallable� callable_�active_history� impl_classZsend_modified_events�doc)r: Z single_parent�append�single_parent_validatorr0 �parentZ validatorsr Ztrack_cascade_eventsZback_populatesZ_effective_sync_backrefZself_and_descendants�_props�get� class_managerZ_attr_has_implr Zregister_attribute_impl�class_� directionr Z ONETOMANYZviewonlyrA )r3 �mapperr< r; r= r> Zproxy_propertyr? r@ �kwZ listen_hooks�mr2 �hookr6 )r9 r4 r5 r: r7 �_register_attributeB sZ � � � �� ��rN )� instrument�deferredc s6 e Zd ZdZdZ� fdd�Zd dd�Zdd � Z� ZS ) �UninstrumentedColumnLoaderz�Represent a non-instrumented MapperProperty. The polymorphic_on argument of mapper() often results in this, if the argument is against the with_polymorphic selectable. ��columnsc s t � �||� | jj| _d S r. )�super�__init__�parent_propertyrS ��selfrD � strategy_key�� __class__r6 r7 rU � s z#UninstrumentedColumnLoader.__init__Nc K s* | j D ] }|r|j | }|�||� qd S r. )rS � _append_dedupe_col_collection) rX � compile_state� query_entity�path�loadopt�adapter�column_collection�kwargs�cr6 r6 r7 � setup_query� s z&UninstrumentedColumnLoader.setup_queryc C s d S r. r6 � rX �contextr^ r_ r` rJ �resultra � populatorsr6 r6 r7 �create_row_processor� s z/UninstrumentedColumnLoader.create_row_processor)N) �__name__� __module__� __qualname__�__doc__� __slots__rU re rj � __classcell__r6 r6 rZ r7 rQ � s � rQ Tc s> e Zd ZdZdZ� fdd�Zd dd�Zdd � Zd d � Z� Z S ) � ColumnLoaderz8Provide loading behavior for a :class:`.ColumnProperty`.)rS � is_compositec s* t � �||� | jj| _t| jd�| _d S )N�composite_class)rT rU rV rS �hasattrrr rW rZ r6 r7 rU � s zColumnLoader.__init__Fc K sv | j D ]<} |r6|r,|�| �} | d kr6 d S n |j | } |�| |� q| j d } |rh|j | } | d krhd S | || j< d S �Nr )rS Zadapt_check_presentr\ rV ) rX r] r^ r_ r` ra rb �memoized_populatorsZcheck_for_adaptrc rd �fetchr6 r6 r7 re � s zColumnLoader.setup_queryc C s` d| _ | jd j}| jjpD| jd jpD|jd k oD|j�|jd �| jk}t | j|d|j |d� d S )NTr F)r< r; r? ) �is_class_levelrS �typerV r? � primary_keyZversion_id_col�_columntopropertyrF rN �compare_values)rX rJ Zcoltyper? r6 r6 r7 �init_class_attribute� s" � �� �z!ColumnLoader.init_class_attributec C sZ | j D ]:} |r|j | } |�| d�} | r|d �| j| f� qVq|d �| jdf� d S �NFZquick�expireT)rS �_getterrB r0 ) rX rg r^ r_ r` rJ rh ra ri �col�getterr6 r6 r7 rj s z!ColumnLoader.create_row_processor)F) rk rl rm rn ro rU re r} rj rp r6 r6 rZ r7 rq � s � !rq )Zquery_expressionc s4 e Zd Z� fdd�Zdd� Zdd� Zdd� Z� ZS ) �ExpressionColumnLoaderc s< t � �||� t�� �d �� t� fdd�| jjD ��| _d S )Nc 3 s | ]}|� � � V qd S r. )�compare��.0rd ��nullr6 r7 � <genexpr>% s z2ExpressionColumnLoader.__init__.<locals>.<genexpr>) rT rU r( r� �label�anyrV rS �_have_default_expressionrW rZ r� r7 rU s �z ExpressionColumnLoader.__init__c K s� d } |r|j r|j } n| jr$| jj} | d kr0d S | D ] } |rF|j| } |�| |� q4| d } |rv|j| } | d krvd S | || j< d S ru )�_extra_criteriar� rV rS r\ ) rX r] r^ r_ r` ra rb rv rc rS rd rw r6 r6 r7 re ) s" z"ExpressionColumnLoader.setup_queryc C sh |rd|j rd|j } | D ]:} |r&|j| } |�| d�} | r|d �| j| f� qdq|d �| jdf� d S r~ )r� rS r� rB r0 ) rX rg r^ r_ r` rJ rh ra ri rS r� r� r6 r6 r7 rj N s z+ExpressionColumnLoader.create_row_processorc C s( d| _ t| j|d| jd jjdd� d S )NTFr )r< r; Zaccepts_scalar_loader)rx rN rV rS ry r| �rX rJ r6 r6 r7 r} h s �z+ExpressionColumnLoader.init_class_attribute)rk rl rm rU re rj r} rp r6 r6 rZ r7 r� s %r� )rP rO )rP rO � raiseload)Z do_nothingc sN e Zd ZdZdZ� fdd�Zdd� Zdd� Zdd d �Zd d � Z dd� Z � Z S )�DeferredColumnLoaderzAProvide loading behavior for a deferred :class:`.ColumnProperty`.)rS �groupr� c sJ t � �||� t| jd�r"td��| j�dd�| _| jj| _| jj | _ d S )Nrs z8Deferred loading for composite types not implemented yetr� F) rT rU rt rV �NotImplementedError� strategy_optsrF r� rS r� rW rZ r6 r7 rU s � z DeferredColumnLoader.__init__c C s� |j rB|jjjrB| j|jjjkrB| j�d��||||||||� nH| jsv| j rX| jj } n| jj } |d � | j| f� n|d � | jdf� d S )N�)rP F�rO T�newr F) � refresh_stater �_compile_options�_only_load_propsr0 rV � _get_strategyrj rx r� Z_raise_column_loaderZ_deferred_column_loaderrB ) rX rg r^ r_ r` rJ rh ra ri Z set_deferred_for_local_stater6 r6 r7 rj � s4 ����� ��z)DeferredColumnLoader.create_row_processorc C s, d| _ t| j|d| jd jj| jdd� d S )NTFr )r< r; r> Zload_on_unexpire)rx rN rV rS ry r| �_load_for_stater� r6 r6 r7 r} � s �z)DeferredColumnLoader.init_class_attributeNc K s� |j jr| jjsT|r(t| j��| jj�sT|rF| j rF|j � d| j d�sT|rx| j |krx| j� d�j|||||||f| � n.| jr�t|| j< n | js�t|| j< n t|| j< d S )Nzundefer_group_%sFr� )�compile_optionsZ_render_for_subqueryrV Z_renders_in_subqueries�setrS � intersectionrD Z_should_undefer_in_wildcardr� � local_optsrF r0 r� re rx r r� r r ) rX r] r^ r_ r` ra rb rv Zonly_load_propsrK r6 r6 r7 re � sR ��� �� � � ������ � z DeferredColumnLoader.setup_queryc s� �j s tjS |tj@ s tjS �jj}� jr@� fdd�|j D �}n� j g}�fdd�|D �}t ��}|d kr�t � dt ���� j f ��� jr�� ��|d� t��j�t|�t� tjS )Nc s4 g | ],}t |t�rt |jt�r|j� jkr|j�qS r6 )� isinstancer �strategyr� r� r0 )r� �p�rX r6 r7 � <listcomp> s �z8DeferredColumnLoader._load_for_state.<locals>.<listcomp>c s g | ]}|� j kr|�qS r6 )Z unmodified)r� �k)�stater6 r7 r� s zfParent instance %s is not bound to a Session; deferred load operation of attribute '%s' cannot proceed�raise)r0 r � ATTR_EMPTYr �SQL_OK�PASSIVE_NO_RESULT�managerrJ r� Ziterate_propertiesr �orm_exc�DetachedInstanceErrorr/ � state_strr� �_invoke_raise_loadr Zload_scalar_attributesr� r r )rX r� �passive� localparentZtoloadr� �sessionr6 )rX r� r7 r� � s6 ��� �z$DeferredColumnLoader._load_for_statec C s t �d| f ��d S )Nz+'%s' is not available due to raiseload=True��sa_exc�InvalidRequestError�rX r� r� �lazyr6 r6 r7 r� # s �z'DeferredColumnLoader._invoke_raise_load)N) rk rl rm rn ro rU rj r} re r� r� rp r6 r6 rZ r7 r� t s 3 � 3(r� c @ s0 e Zd ZdZd ddd�dd�Zejfdd �Zd S ) �LoadDeferredColumnsz7serializable loader object used by DeferredColumnLoaderF�str�bool�r0 r� c C s || _ || _d S r. r� )rX r0 r� r6 r6 r7 rU , s z LoadDeferredColumns.__init__c C s> | j }|jj}|j| }| jr$d}nd}|�|�}|�||�S )N)�rP Tr� )r� T)r� r� )r0 r� rJ rE r� r� r� )rX r� r� r0 r� r3 rY r� r6 r6 r7 �__call__0 s z LoadDeferredColumns.__call__N)F)rk rl rm rn rU r r r� r6 r6 r6 r7 r� ) s r� c s, e Zd ZdZdZ� fdd�Zdd� Z� ZS )�AbstractRelationshipLoaderz0LoaderStratgies which deal with related objects.)rJ �targetr: �entityc s: t � �||� | jj| _| jj| _| jj| _| jj| _d S r. )rT rU rV rJ r� r� r: rW rZ r6 r7 rU F s z#AbstractRelationshipLoader.__init__c C s | j �d��||||||||�S )N))r� � immediate)rV r� rj rf r6 r6 r7 �#_immediateload_create_row_processorM s ��z>AbstractRelationshipLoader._immediateload_create_row_processor)rk rl rm rn ro rU r� rp r6 r6 rZ r7 r� A s r� c @ s e Zd ZdZdS )�DoNothingLoaderz�Relationship loader that makes no change to the object's state. Compared to NoLoader, this loader does not initialize the collection/attribute to empty/none; the usual default LazyLoader will take effect. N)rk rl rm rn r6 r6 r6 r7 r� f s r� Znoload)r� c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�NoLoaderzQProvide loading behavior for a :class:`.Relationship` with "lazy=None". r6 c C s d| _ t| j|d| jjd� d S )NT)r< r= )rx rN rV �collection_classr� r6 r6 r7 r} } s �z NoLoader.init_class_attributec s$ � fdd�} |d � � j| f� d S )Nc s& � j rt�| |� j� n d |� j< d S r. )r: r �init_state_collectionr0 �r� �dict_�rowr� r6 r7 �invoke_no_load� s z5NoLoader.create_row_processor.<locals>.invoke_no_loadr� )rB r0 ) rX rg r^ r_ r` rJ rh ra ri r� r6 r� r7 rj � s z NoLoader.create_row_processorN)rk rl rm rn ro r} rj r6 r6 r6 r7 r� r s r� �selectr� � raise_on_sqlZ baked_selectc s� e Zd ZU dZdZded< ded< ded< ded< d ed < d d d �� fd d� Zdd� Zdd� Zdd� Z dd� Z dddde j fdd�Z dd � Ze �d �d d � �Zd d!� Z� ZS )"� LazyLoaderztProvide loading behavior for a :class:`.Relationship` with "lazy=True", that is loads when first accessed. ) � _lazywhere�_rev_lazywhereZ_lazyload_reverse_option� _order_by�use_get�is_aliased_class� _bind_to_col�_equated_columns�_rev_bind_to_col�_rev_equated_columns�_simple_lazy_clause� _raise_always� _raise_on_sqlzColumnElement[bool]r� z Dict[str, ColumnElement[Any]]r� r� r� zRelationshipProperty[Any]rV zTuple[Any, ...])rD rY c s8 t � �||� | jd dk| _| jd dk| _t| j�j| _| jj }|� � \| _ | _ | _ |j dd�\| _| _| _| jjr�dd� t�| jj�D �| _nd | _| j�d| | j � | j o�| j o�| jjd j| j dd | jjd �| _ | j �r4t | j �D ]4}|| jjkr�| jj| D ]}| j | | j |< �q q�| j�d | � d S ) Nr� r� r� T)Zreverse_directionc S s g | ]}t �|d di��qS )� _orm_adaptT)�sql_util�_deep_annotate�r� �elemr6 r6 r7 r� � s �z'LazyLoader.__init__.<locals>.<listcomp>z%s lazy loading clause %sr F�Z use_proxiesZ compare_keys� equivalentsz4%s will use Session.get() to optimize instance loads) rT rU r� r� r� r& r� r� rV �_join_conditionZcreate_lazy_clauser� r� r� r� r� r� �order_byr �to_listr� �logger�infor: � _get_clauser� rJ �_equivalent_columnsr� �list)rX rD rY Zjoin_conditionr� rd rZ r6 r7 rU � sP � � � � �� �zLazyLoader.__init__c C sv d| _ | jj}| jjr d}d}n4| jjtjk s4| jsL|rBd}d}qTd}d}nd }}t| j|d| j | jj ||d� d S )NTF)r< r> r= r? �_deferred_history) rx rV � _legacy_inactive_history_styler? rI r � MANYTOONEr� rN r� r� )rX rJ r� r? r� r6 r6 r7 r} � s2 � ���z LazyLoader.init_class_attributec sd t �| jddi�}|| j }� g �dd� }t�|i d|i� � �fdd�}t�|i d|i�}|�fS )Nr� Tc S s d| _ d S )NF)�unique�� bindparamr6 r6 r7 �visit_bindparam' s zFLazyLoader._memoized_attr__simple_lazy_clause.<locals>.visit_bindparamr� c sF | j � kr$��| j� | j d f� n | jd krB��| jd | jf� d S r. )Z_identifying_keyrB r0 �callable�valuer� �Z bind_to_col�paramsr6 r7 r� , s �� )r� r� r� r� r) �traverseZcloned_traverse)rX Z lazywhere� criterionr� r6 r� r7 �"_memoized_attr__simple_lazy_clause s � �z-LazyLoader._memoized_attr__simple_lazy_clausec C s� | j \}}|d kr(t�|dd� |D ��S | jj}|�� }t�|�}|tj @ rV|tj N }i }|D ]J\} } } | d k r�|r�|tj @ r�|� ||| |�} n|� ||| |�} | || < q^||fS )Nc S s g | ]\}}}|�qS r6 r6 )r� r0 �identr� r6 r6 r7 r� C s z4LazyLoader._generate_lazy_clause.<locals>.<listcomp>) r� r� Zadapt_criterion_to_nullrV rD �objr � instance_dictr ZINIT_OK�LOAD_AGAINST_COMMITTED�#_get_committed_state_attr_by_column�_get_state_attr_by_column) rX r� r� r� Z param_keysrJ �or� r� r0 r� r� r6 r6 r7 �_generate_lazy_clause> s: � � � z LazyLoader._generate_lazy_clausec C s t �d| |f ��d S )Nz&'%s' is not available due to lazy='%s'r� r� r6 r6 r7 r� ^ s �z LazyLoader._invoke_raise_loadNr6 c C s� |j s | jjs|jr|js tjS |j }d } | jo>| p>|j } |t j @ sN| r\|t j @ sb|rbtj S | jr�|t j@ s�| r�|t j@ r�| �||d� t|�} | s�|t j@ r�tj S t�dt�|�| j f ��| �rz| �| ||�} tj | kr�tj S tj| kr�tjS t�| ��rd S | j |jk�r(|t j @ �s(tj S | j | j | ||d�} | d k �r\| tj k�rVd S | S n |t j @ �rt|t j@ �sztj S | �!| || ||||||� S )Nr� zbParent instance %s is not bound to a Session; lazy load operation of attribute '%s' cannot proceed)r� Zlazy_loaded_from)"r0 rV Zload_on_pendingZ _load_pendingZ session_idr r� r� r� r r� r ZNON_PERSISTENT_OKr� r� �NO_RAISEZRELATED_OBJECT_OKr� r r� r� r/ r� �_get_ident_for_use_getZ NEVER_SETr" � issuperset�dict�DEFERRED_HISTORY_LOADr Z_identity_lookupr� ZPASSIVE_CLASS_MISMATCH�_emit_lazyload) rX r� r� r` �extra_criteria� extra_options�alternate_effective_path�execution_options�pending�primary_key_identityr� r� �instancer6 r6 r7 r� c s� ������� � � �� � ��� ���zLazyLoader._load_for_statec sD �j j}�tj@ r|j�n|j��j� � ����fdd��jjD �S )Nc s" g | ]}��� �j | �d ��qS )�r� �r� �r� �pk�r� Zget_attrr� rX r� r6 r7 r� � s �z5LazyLoader._get_ident_for_use_get.<locals>.<listcomp>)r� rJ r r� r� r� r rz )rX r� r� r� �instance_mapperr6 r r7 r � s �z!LazyLoader._get_ident_for_use_getz sqlalchemy.orm.strategy_optionsc s� t jj�� j�� } tj| g| jtt j d�} t j } | d|d�7 } � j jd k r\| �� j� j j�} |j } | sr|tj@ r�t �ddi�| _� j}|js�|r�|jr�|d kr�|j� j }n |� j }|j}|r�|jr�d}|t�� j|�f7 }|| _ n$|d k�r|jj � j }n |� j }|�r | j |7 _ | j d|i7 _ |�rn� j �rZ|t j!@ �sZ� �"||d� t#j$|| || | d�S � j%�r~� j%| _&� �fdd �}| j'|� j ff7 _'� �(||�\}}| �r�t j)�*| d | i�} nd | i} � j|j+k�r�|t j,@ �s�t-j.S | �rt �/tj0|�1� ��r,d S nt �/tj2|�1� ��r,d S � j �rN|t j!@ �sN� �"||d� |f| _3|j4| || d �}|�5� �6� �7� }� j8�r�|S t9|�}|�r�|d k�r�t �:d � j � |d S d S d S )N)� _raw_columns�_propagate_attrs� _label_styler� F)Z_invoke_all_eagers�_lazy_loaded_fromZ autoflush� _current_pathr� )� load_optionsr c sR � j jD ]D}|jtjkr|jrt|jt�s�j � | j j |j ��|��| � qd S r. )rV Z_reverse_propertyrI r r� Z_use_getr� r� r� r! Z _construct_for_existing_pathr� r rD ZlazyloadZprocess_compile_state)�compile_context�rev�rX �strategy_optionsr6 r7 �_lazyload_reverse7 s �� �����z4LazyLoader._emit_lazyload.<locals>._lazyload_reverseZ_sa_orm_load_options)r r zKMultiple rows returned with uselist=False for lazily-loaded attribute '%s' r );r Z preloadedZorm_strategy_optionsr� �__clause_element__r+ �_create_raw_selectr r* r �default_compile_optionsr Zdefault_load_optionsrV Z secondary� select_fromrJ r0 r Z NO_AUTOFLUSH� immutabledict�_execution_optionsr� r r� Z load_pathr/ �LoaderCriteriaOption� _with_options�_path_registryr� r� r r� r� r Zload_on_pk_identityr� �_order_by_clausesZ_with_context_optionsr� � EMPTY_DICT� merge_withr r r r Zhas_intersectionr" �valuesZ _never_setZ_where_criteria�executer� Zscalars�allr: �len�warn)rX r� r� r r� r` r r r r Z clauseelement�stmtr r r� �effective_pathr5 r Z lazy_clauser� rh �lr6 r r7 r � s� �� � � �� �� � �� � ��zLazyLoader._emit_lazyloadc s� | j � |jjr@|jjjr@| j |jjjkr@| �||||||||�S | jrP|r�|jr�t � |j t � | ||jrp|� |�nd �� �} |d �| j | f� n,|js�|jr�� fdd�} |d �| j | f� d S )Nr� c s | � |� � d S r. )�_resetr� �r0 r6 r7 �reset_for_lazy_callable� s z@LazyLoader.create_row_processor.<locals>.reset_for_lazy_callable)r0 r Z_is_user_refreshr r� r� r� rx r� r Z"_instance_level_callable_processorrG �LoadLazyAttribute�_generate_extra_criteriarB �populate_existingZalways_refresh) rX rg r^ r_ r` rJ rh ra ri Zset_lazy_callabler3 r6 r2 r7 rj � sF ���� �� � � z LazyLoader.create_row_processor)rk rl rm rn ro �__annotations__rU r} r� r� r� r r' r� r Zpreload_moduler rj rp r6 r6 rZ r7 r� � s. <" � p r� c @ s. e Zd ZdZdd� Zdd� Zejfdd�ZdS ) r4 a� semi-serializable loader object used by LazyLoader Historically, this object would be carried along with instances that needed to run lazyloaders, so it had to be serializable to support cached instances. this is no longer a general requirement, and the case where this object is used is exactly the case where we can't really serialize easily, which is when extra criteria in the loader option is present. We can't reliably serialize that as it refers to mapped entities and AliasedClass objects that are local to the current process, which would need to be matched up on deserialize e.g. the sqlalchemy.ext.serializer approach. c C s || _ |j| _|| _|| _d S r. �r0 rY r` r )rX r0 Zinitiating_strategyr` r r6 r6 r7 rU � s zLoadLazyAttribute.__init__c C s( | j d k rt�d� | j| j| jdd�S )NzvCan't reliably serialize a lazyload() option that contains additional criteria; please use eager loading for this caser6 r8 )r r r- r0 rY r` r� r6 r6 r7 � __getstate__� s ��z LoadLazyAttribute.__getstate__c C s: | j }|jj}|j| }|j| j }|j||| j| jd�S )N)r` r ) r0 r� rJ rE Z _strategiesrY r� r` r )rX r� r� r0 r r3 r� r6 r6 r7 r� � s �zLoadLazyAttribute.__call__N) rk rl rm rn rU r9 r r r� r6 r6 r6 r7 r4 � s r4 c @ s e Zd ZdZdZddd�ZdS )� PostLoaderz;A relationship loader that emits a second SELECT statement.r6 Nc C sN |j jptjj| }|�� }t�d|i�}|rF|j� dd �}|dk} nd }d} |d k r�| j j srt � d| j � d���|j� dt| �� �|�}| s�|dk r�|d||fS | s�|�dt| �� �|d i�}tj�||| j �r�|d||fS || j } || j } | �|jd ��sB|�r(| jd |k�rB|d||fS n| �| j��rB|d||fS |d ||fS ) NZsa_top_level_orm_context�recursion_depth���Fz'recursion_depth option on relationship z0 not valid for non-self-referential relationshipZ_recursion_depth_r r �loaderr$ T)r] � current_pathr/ � PathRegistry�rootZ_get_top_level_contextr r! r� rF rV Z_is_self_referentialr� r� r �id�unionr �PostLoadZ path_exists�containsr �length�contains_mapperrJ ) rX rg r_ r` � join_depthr/ Ztop_level_contextr r; Zunlimited_recursion� path_w_propZeffective_path_w_propr6 r6 r7 �_setup_for_recursion sp �� � � � �� � ��z PostLoader._setup_for_recursion)N)rk rl rm rn ro rI r6 r6 r6 r7 r: � s r: r� c s8 e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Z� ZS ) �ImmediateLoader�rG c s t � �||� | jj| _d S r. �rT rU rV rG rW rZ r6 r7 rU L s zImmediateLoader.__init__c C s | j �d��|� d S �N�)r� r� �rV r� r} r� r6 r6 r7 r} P s ��z$ImmediateLoader.init_class_attributec C sl |j jjsd S | �|||| j�\} } } } | s:tjtjB } n tj tjB } t j � || | j | j| j|| | | � d S r. )r] r� �_enable_eagerloadsrI rG r ZPASSIVE_NO_FETCH_RELATEDr r� r r rC �callable_for_pathrD rV �_load_for_path)rX rg r^ r_ r` rJ rh ra ri r/ � run_loaderr r; �flagsr6 r6 r7 rj U s, � �z$ImmediateLoader.create_row_processorc C s� |r.t |jj�} ||�� f| _|�� } | f} n d } |} d} | j} | j�d�} |D ]P\}}|j }|sj| |krP| j ||| | |d�}|t t j fkrP|�| ��|||� qPd S )Nr6 rN )r r r )r! r_ r� Z_recurserg �_truncate_recursiver0 rV r� r r� r r r� �get_impl�set_committed_value)rX rg r_ �states� load_onlyr` rT r; r Znew_optr r r0 � lazyloaderr� � overwriter� r� r6 r6 r7 rR s> � �� �z ImmediateLoader._load_for_path) rk rl rm ro rU r} rj rR rp r6 r6 rZ r7 rJ H s *rJ �subqueryc s� e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z G dd� d�Z dd� Z dd� Zdd� Zdd� Z� ZS ) �SubqueryLoaderrK c s t � �||� | jj| _d S r. rL rW rZ r6 r7 rU � s zSubqueryLoader.__init__c C s | j �d��|� d S rM rO r� r6 r6 r7 r} � s ��z#SubqueryLoader.init_class_attributec s� |}|j }t�|d �}| j�|�rB| j|d krB| j| j � }n||d � }|r�|j| jj| �t |�t �� }|r��t j � || d � �7 �n|�|j } � �fdd�| D �} � | |�fS )Nr r c s& g | ] }t �jd j� j| j��qS �r )�getattrr_ r� r{ r0 r� ��leftmost_mapperZ new_subq_pathr6 r7 r� � s � �z0SubqueryLoader._get_leftmost.<locals>.<listcomp>)r_ r/ Z_class_to_mapperrD �isarV � _entities� entity_zeror% r, r r? �coerce� local_columns) rX �orig_query_entity_index� subq_pathZcurrent_compile_state�is_rootZgiven_subq_pathZ subq_mapperZ leftmost_prop� additionalZ leftmost_cols� leftmost_attrr6 r` r7 � _get_leftmost� s4 � ��� � �z SubqueryLoader._get_leftmostc C s. |� � �d �}tj�tj�}|j�|j� |}|js^d|_|j j |fdd� t ||d�D ��� |� dd� |D �|� � �} | |_|j} | dkr�d|_n6| d kr�dd� | D �D ] } t| ��| j�s�d|_ q�q�|js�d |_|jdk�r |j�r t�| |j�} | �r |�| | � |�t��� } t j || dd �}|S ) NFc S s h | ]}|d dk r|d �qS )r� Nr6 )r� Zentr6 r6 r7 � <setcomp> s �z?SubqueryLoader._generate_from_original_query.<locals>.<setcomp>)r] c S s g | ]}t j�t jj|��qS r6 )r( Z coercions�expectZrolesZColumnsClauseRole)r� r� r6 r6 r7 r� % s �z@SubqueryLoader._generate_from_original_query.<locals>.<listcomp>Tc S s h | ] }|j �qS r6 )�tabler� r6 r6 r7 rm 4 s r6 ��use_mapper_path) Z_cloneZ correlater �Query�__new__�__dict__�updateZ _from_objZ_enable_assertionsr Znon_generativer Z_adapt_col_listZ_get_current_adapterr �distinct_target_keyZ _distinctr� r rz Z_has_row_limiting_clauser& r� Z expand_column_list_from_order_byZ _set_entitiesZset_label_styler* r\ r/ r# )rX �orig_compile_state� orig_queryra rk �leftmost_relationshipZ orig_entity�q�q2Z target_colsrv �tZto_addZembed_q� left_aliasr6 r6 r7 � _generate_from_original_query� s^ � ����� � �z,SubqueryLoader._generate_from_original_queryc s� g }t |�� �}t|�D ]N\}\}}|dkrR||d d j}|�|�rL|n|} n|} |�| |jf� qt|�dk rz|� n.t|d d �} | j r�| j � nt j | j dd�� �j j} � �fdd�| D �} || � fS ) Nr r r$ r< Trp c s g | ]}t � �jj| j��qS r6 )r_ rD r{ r0 r� �� parent_aliasrX r6 r7 r� y s �z2SubqueryLoader._prep_for_joins.<locals>.<listcomp>)r� �pairs� enumeraterJ rb rB r0 r, r& r� r� r/ r# rV rf ) rX r} rh �to_joinr� �irJ r3 Z prev_mapperZ to_appendr� � local_cols� local_attrr6 r r7 �_prep_for_joinsP s, � �z SubqueryLoader._prep_for_joinsc C s0 t |�}|dkr,t||d d ��|�g}n�|dkrft||d d ��|�t||d d ��|�g}n�|dk�rdd� |dd� D �}g }|r�|�d�} t| d | d �} |r�| �|d d �} n | �|�} |�| � q�t||d d ��|d j�g| t||d d ��|�g }|D ]} |�| �}�q|S )Nr r r$ r< c S s: g | ]2}t |d �js$t�|d �n|d j|d f�qS )r r )r& r� r/ r# r� )r� �itemr6 r6 r7 r� � s � ��z/SubqueryLoader._apply_joins.<locals>.<listcomp>)r, r_ �of_type�poprB rD �join) rX rz r� r} r� �effective_entityZltj�middle�innerr� �attrr6 r6 r7 � _apply_joins sD ��� � ����� zSubqueryLoader._apply_joinsc C sB |j }|r*|jr*|t�| j|�|��f7 }|�|�}|j|� }|S r. )r$ r� r/ r# r� r5 Z_with_current_path�options) rX rg rz rh �rewritten_pathrx r� r` � new_optionsr6 r6 r7 �_setup_options� s �� z SubqueryLoader._setup_optionsc s& � j jr"� fdd�}|�|� j �}|S )Nc s | j tt�� jj��7 _ d S r. ��eager_order_by�tupler r� rV r� �r r� r6 r7 �_setup_outermost_orderby� s �zISubqueryLoader._setup_outermost_orderby.<locals>._setup_outermost_orderby)rV r� �_add_context_option)rX rz r� r6 r� r7 r� � s �z'SubqueryLoader._setup_outermost_orderbyc @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd S ) z SubqueryLoader._SubqCollectionsz�Given a :class:`_query.Query` used to emit the "subquery load", provide a load interface that executes the query at the first moment a value is needed. �r� r r r� �subq�_datac C s4 |j | _ |j| _|j| _|jp i | _|| _d | _d S r. r� )rX rg r� r6 r6 r7 rU � s z(SubqueryLoader._SubqCollections.__init__c C s | j d kr| �� | j �||�S r. )r� �_loadrF )rX r0 �defaultr6 r6 r7 rF � s z#SubqueryLoader._SubqCollections.getc C s� t �t�| _| j}|jd ks t�|�| j�}| jj r<|� � }t|� | j ��}t � |dd� �D ]"\}}| j| �dd� |D �� q\d S )Nc S s | dd � S )Nr r6 ��xr6 r6 r7 r8 � z7SubqueryLoader._SubqCollections._load.<locals>.<lambda>c s s | ]}|d V qdS )r Nr6 �r� Zvvr6 r6 r7 r� s z8SubqueryLoader._SubqCollections._load.<locals>.<genexpr>)� collections� defaultdictr� r� r� r� �AssertionErrorZ with_sessionr Z_populate_existingr6 r� � itertools�groupby�extend)rX rz �rowsr� �vr6 r6 r7 r� s z%SubqueryLoader._SubqCollections._loadc C s | j d kr| �� d S r. )r� r� )rX r� r� r� r6 r6 r7 r= s z&SubqueryLoader._SubqCollections.loaderN) rk rl rm rn ro rU rF r� r= r6 r6 r6 r7 �_SubqCollections� s r� c C s� |j }|jjr|jjrd S |j�|�}d|_|| j }|�|j dd �} | d k rT| } n| j } |j j �dt jjt jjf�\} } | t jjk} | | } | | } |j j �dtf|j �}t�|d�}|jr�|jjd kr�t�d| |f � |j}|j|dd�}| �|| || �\}}}} | �||||||�}t � | �}|j j � |j dtf|d| | fi�|_ |� d�}| � || �\}}}|j!|� }| �"||||| �}| �#||| | || |�}| �$|�}|S ) NT�path_with_polymorphic)Zsubquery_pathsNrx �ormz�subqueryloader for "%s" must invoke lambda callable at %r in order to produce a new query, decreasing the efficiency of caching for this statement. Consider using selectinload() for more effective full-lambda cachingF)�legacy)%r] r� rP Z_for_refresh_staterc �indexZloaders_require_bufferingrV rF r r� r r" r/ r? r@ r] r Z _get_plugin_class_for_pluginZ_is_lambda_elementr r r r- Z _resolvedZ_create_entities_collectionrl r~ rr r( r Z_set_enable_single_critr� Z add_columnsr� r� r� )rX rg r^ r_ r� r` ra r] rg �with_poly_entityr� rh r� ri rx Zcompile_state_clsrw ra rk ry r} rz r� r� r� r6 r6 r7 �_setup_query_from_rowproc s� �� �� � � �� � �� � �� � �� z(SubqueryLoader._setup_query_from_rowprocc sj |j r | �||||||� |�S | �|||| j�\} } } } | s@d S t|jt�sPd S | jj| j j j spt � d| ��t|�dkr�t�|j| j�s�d S nt�|d | j�s�d S | �||||d |� �} | d kr�d S | jd ks�t�|| j }| jj} |�|jd�} | d k�r | �|| �} |�|jd| � � �r8� fdd�| D �} | j�rT| � ||| | |� n| � ||| | |� d S )N�J'%s' does not support object population - eager loading cannot be applied.r r< r� c s g | ]}� j | �qS r6 rR r� �ra r6 r7 r� � s z7SubqueryLoader.create_row_processor.<locals>.<listcomp>) r� r� rI rG r� r] r rD rG r0 �impl�supports_populationr� r� r, r/ � _entity_isard r� r� r� rV rf rF r r� r� r: �_create_collection_loader�_create_scalar_loader)rX rg r^ r_ r` rJ rh ra ri �_rS r� r� r� r6 r� r7 rj � s� � � �� � � �z#SubqueryLoader.create_row_processorc sp |� |��� ��fdd����fdd�}|d ��j�f� |d ��j|f� |jrl|d ��j� jf� d S )Nc s* � � �|�d�}| ��j��| ||� d S )Nr6 )rF rV r0 rW )r� r� r� � collection�r� rX � tuple_getterr6 r7 �load_collection_from_subq� s �zKSubqueryLoader._create_collection_loader.<locals>.load_collection_from_subqc s �j |kr� | ||� d S r. r2 r� )r� rX r6 r7 �&load_collection_from_subq_existing_row s zXSubqueryLoader._create_collection_loader.<locals>.load_collection_from_subq_existing_rowr� �existing�eager�Z _tuple_getterrB r0 �invoke_all_eagersr= )rX rg rh r� r� ri r� r6 )r� r� rX r� r7 r� � s �z(SubqueryLoader._create_collection_loaderc sp |� |��� ��fdd����fdd�}|d ��j�f� |d ��j|f� |jrl|d ��j� jf� d S )Nc sL � � �|�d�}t|�dkr*t�d� � |d }| ��j��| ||� d S )Nr. r �LMultiple rows returned with uselist=False for eagerly-loaded attribute '%s' r )rF r, r r- rV r0 rW )r� r� r� r� Zscalarr� r6 r7 �load_scalar_from_subq s ��zCSubqueryLoader._create_scalar_loader.<locals>.load_scalar_from_subqc s �j |kr� | ||� d S r. r2 r� )r� rX r6 r7 �"load_scalar_from_subq_existing_row! s zPSubqueryLoader._create_scalar_loader.<locals>.load_scalar_from_subq_existing_rowr� r� r� r� )rX rg rh r� r� ri r� r6 )r� r� rX r� r7 r� s �z$SubqueryLoader._create_scalar_loader)rk rl rm ro rU r} rl r~ r� r� r� r� r� r� rj r� r� rp r6 r6 rZ r7 r] � s 1d/6"4 Yr] Zjoinedc s� e Zd ZdZdZ� fdd�Zdd� Zd d d �Zd d � Zd d� Z dd� Z dd� Z d dd�Z dd� Z dd� Zdd� Zdd � Z� ZS ) � JoinedLoaderz[Provide loading behavior for a :class:`.Relationship` using joined eager loading. rG c s t � �||� | jj| _d S r. rL rW rZ r6 r7 rU 8 s zJoinedLoader.__init__c C s | j �d��|� d S rM rO r� r6 r6 r7 r} < s ��z!JoinedLoader.init_class_attributeNFc K sR |j js dS | jrd|_|| j }d} |r:| �|||j�nd} | dk r`| �||||| �\} }} n`|�|jd�s�| j r�|j d | j kr�dS n|� | j �r�dS | � ||||||||�\} }} }d|_|�|jdd�}|dk r�t|�j} nd} || j }tj|| j ||| | | | j |d� tj�|j�}|�rN|dk �r<t�d��nd d � |jD �|_dS ) z@Add a left outer join to the statement that's being constructed.NTFr= r$ r� )�with_polymorphic� parentmapper�chained_from_outerjoinz�Detected unaliased columns when generating joined load. Make sure to use aliased=True or flat=True when using joined loading with with_polymorphic().c S s g | ]}|d k r|�qS r. r6 r� r6 r6 r7 r� � s z,JoinedLoader.setup_query.<locals>.<listcomp>)r� rP r: �multi_row_eager_loadersrV � _init_user_defined_eager_procr �$_setup_query_on_user_defined_adapterrD rG rE rF rJ �_generate_row_adapterZeager_adding_joinsrF r& Zwith_polymorphic_mappersr� r Z_setup_entity_queryr ZNONE_SETr� �secondary_columnsr� r� )rX r] r^ r_ r` ra rb r� r� rc r� �user_defined_adapter�clauses�add_to_collectionr� Z has_nonesr6 r6 r7 re A s� � ����� �� �� � ��zJoinedLoader.setup_queryc C s� d|j krdS |jj}|�|jdd�}|dk r2|S |j d }|dd � \}}|d k r�t|t�rj|j�|�}t j t j j |j ||j jdd�}nF|�|jd�r�|�|jd�} t j t j j| |j jd�}n|j�|j d �}|�|d|� |S )NZeager_from_aliasF� user_defined_eager_row_processor���)� selectabler� Zlimit_on_entityr� )r� )r� r_ rD rF r r� r� r� �aliasr/ � ORMAdapter�_TraceAdaptRoleZ JOINEDLOAD_USER_DEFINED_ALIASrJ r� rD Z JOINEDLOAD_PATH_WITH_POLYMORPHICZ_polymorphic_adaptersr� ) rX r` r] Ztarget_attributesr_ ra r� Z root_mapperr3 r� r6 r6 r7 r� � sZ � � � �� ��z*JoinedLoader._init_user_defined_eager_procc C sV |� |�}|r.|r.|�|�}|�|jd|� n|rF|}|�|jd|� |j}|||fS )Nr� )Z_get_entity_clauses�wrapr� r �primary_columns)rX rg r� r_ ra r� r� r6 r6 r7 r� � s" ��z1JoinedLoader._setup_query_on_user_defined_adapterc C s |� |jdd �} | r| } nDt| j�} | jr2| j} nd } tj| j| d k rR| j dd�nd ddd�} t| �} | j d| ftj tj j | | jjdddd�}|js�t�|d k r�|j� d| jj�n| jj}|s�d}|j�| j||||||||r�|jnd f � |j}|�|jd |� ||||fS ) Nr� T)�flat)r� r� rq Zjoinedloader_ormadapterF)r� Zadapt_requiredZallow_label_resolveZanonymize_labels� innerjoinr6 �eager_row_processor)rF r r& r� r� r� r/ r# rJ Z_anonymous_fromclauseZ_memor� r� ZJOINEDLOAD_MEMOIZED_ADAPTERr� r� r� rV r� Zcreate_eager_joinsrB �_create_eager_joinr� r� r� )rX r] r� r_ r` ra rb r� r� r� Zto_adaptZinspZalt_selectableZ to_adapt_inspr� r� r� r6 r6 r7 r� sh � �� � �� ��z"JoinedLoader._generate_row_adapterc s\ |d kr|j } n|} |jo |j} d } ||jkr~| s~|jr~t�|j|j�} t| �dkr^t � d��| r~|j| d }| d | } }| d kr�||j } }|j� | |�}|�rt |dd�r�|j �| �� j�r�| n� j�}t |j� j� j�}n t tj� j|jdd�� j� j�}n� j}|j�st�| �p8| �p8|dk�p8|jj}| }|j�d � j fd �}|�rr|t � fd d �|D ��7 }|�r�tj ||j || �p�|jj�p�|�o�t |t j �� j|� j |d �}n� �!|||||�}||j| < |j|_"|�s*t�#� jj$�D ]4}| j%j&�'|��r�|�r|j(| }|�)||j*� �q� jj+�rX| j,t |j-�.t/�0� jj+���7 _,d S )Nr z�Can't identify which query entity in which to joined eager load from. Please use an exact match when specifying the join path.r r� FTrp Zunnested�additional_entity_criteriar6 c 3 s | ]}|j r|�� j�V qd S r. )Zpropagate_to_loadersZ_resolve_where_criteriarJ )r� Zaer� r6 r7 r� � s �z2JoinedLoader._create_eager_join.<locals>.<genexpr>��isouter� _left_memo� _right_memor� )1rJ r� Z_should_nest_selectableZ eager_joinsZ from_clausesr� Z#find_left_clause_that_matches_givenr� r, r� r� � setdefaultr_ � aliased_inspZ_entity_for_mapperrb rD r� r0 rV r/ r# r� r� rd Zrepresents_outer_joinZglobal_attributesrF r� �_ORMJoinr� r( ZJoin�_splice_nested_inner_joinZstop_onZ _find_columnsZ primaryjoinZpersist_selectablerd Zcontains_columnrS r\ r� r� r� �_target_adapterZcopy_and_processr r� )rX r] r^ r_ ra r� r� r� r� r r� Zshould_nest_selectableZquery_entity_keyZindexesZclauseZdefault_towrapZtowrapZefm�onclauseZattach_on_outsideZextra_join_criteriar� � eagerjoinr� r6 r� r7 r� Y s� ���� � �� � �� �� ��� �� �� � � � ��z JoinedLoader._create_eager_joinc C s� |dkrt |tj�s�t�n�t |tjj�r>| �||j||||�S t |tj�s�|d � |�r�|r�|� |�sjd S t j � |t|�d � �� |�r�d S tj||j|d|||d j |d�S d S | j||j||||jd k r�|jd jnd d�}|d k�r0d} | j||j||||j|jd�}|d k�r4|dk �s*td��d S nd} | �rt|j�sV|j�sV|�|�} n tj|j||j|j|jd �} n tj||j|j|j|jd �} |j| _| S ) NFr� r< r� )�splicing)�detected_existing_pathz9assertion failed attempting to produce joined eager loadsT)r� r� )r� r� )r� r/ r� r� r( r� Z FromGroupingr� �elementrb rF r r? re r, r� rJ �rightr� �leftr� r� Z_splice_into_centerr� r� ) rX r_ Zjoin_objr� r� r r� r� Z target_joinZ right_splicer� r6 r6 r7 r� � s� � �� � �� � �� ��z&JoinedLoader._splice_nested_inner_joinc C s� |j }|r| �|||j�nd}|dk rP|}|jrB|rB|�|j�}qj|jrj|j}n|�|jd�}|d krjdS | j�||�r||S dS d S )NFr� )r] r� r Zcompound_eager_adapterr� rF rJ Z_result_has_identity_key) rX rg rh ra r_ r` r] r� � decoratorr6 r6 r7 �_create_eager_adapteri s4 � �� � �z"JoinedLoader._create_eager_adapterc C s� |j jjsd S | jj| j jjs.t� d| ��| j r:d|_ || j } | � |||| |�} | dk r�| j} t�|| j||| | j | �} | j s�| �|| | |� q�| �|| | |� n | j �d��||||||||� d S )Nr� TFrN )r] r� rP rD rG r0 r� r� r� r� r: Zloaders_require_uniquingrV r� r Z_instance_processorrJ r� r� r� r� rj ) rX rg r^ r_ r` rJ rh ra ri Zour_pathZ eager_adapterr0 � _instancer6 r6 r7 rj � s` �� �� ���z!JoinedLoader.create_row_processorc sr � ��fdd�}� ��fdd�}� fdd�}|d � | j|f� |d � | j|f� �jrn|d � | j|f� d S ) Nc sF t �| |��}t�|d�}|�j | �f< � |�}|d k rB|�|� d S �NZappend_without_event�r r� r ZUniqueAppenderrB )r� r� r� r� � result_list�inst�r� rg r0 r6 r7 �#load_collection_from_joined_new_row� s �zSJoinedLoader._create_collection_loader.<locals>.load_collection_from_joined_new_rowc sd | �f�j kr �j | �f }n(t �| |��}t�|d�}|�j | �f< � |�}|d k r`|�|� d S r� r� )r� r� r� r� r� r� r� r6 r7 �(load_collection_from_joined_existing_row� s � �zXJoinedLoader._create_collection_loader.<locals>.load_collection_from_joined_existing_rowc s � |� d S r. r6 r� �r� r6 r7 � load_collection_from_joined_exec� s zPJoinedLoader._create_collection_loader.<locals>.load_collection_from_joined_execr� r� r� �rB r0 r� )rX rg r0 r� ri r� r� r� r6 r� r7 r� � s ���z&JoinedLoader._create_collection_loaderc sp � �fdd�}� ��fdd�}� fdd�}|d � �j|f� |d � �j|f� |jrl|d � �j|f� d S ) Nc s � |�|�< d S r. r6 r� )r� r0 r6 r7 � load_scalar_from_joined_new_row� s zKJoinedLoader._create_scalar_loader.<locals>.load_scalar_from_joined_new_rowc s8 � |�}�|kr,||� k r4t �d� � n||�< d S )Nr� )r r- )r� r� r� r� �r� r0 rX r6 r7 �$load_scalar_from_joined_existing_row s �� zPJoinedLoader._create_scalar_loader.<locals>.load_scalar_from_joined_existing_rowc s � |� d S r. r6 r� r� r6 r7 � load_scalar_from_joined_exec s zHJoinedLoader._create_scalar_loader.<locals>.load_scalar_from_joined_execr� r� r� r� )rX rg r0 r� ri r� r� r r6 r� r7 r� � s ��z"JoinedLoader._create_scalar_loader)NNF)FN)rk rl rm rn ro rU r} re r� r� r� r� r� r� rj r� r� rp r6 r6 rZ r7 r� - s( � w9O ' � q$>/r� Zselectinc s| e Zd ZdZe�dddddddg�Zd Z� fd d �Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Z� ZS ) �SelectInLoader)rG � omit_join� _parent_alias� _query_info�_fallback_query_infoZ queryinfo�load_only_child�load_with_join�in_expr�pk_cols�zero_idx�child_lookup_colsi� c s� t � �||� | jj| _| jjtjk}| jjd k r>| jj| _n<| j�d�}|rX|j | _n"| j j d j |j dd| j jd�| _| jr�|r�| �� | _| �� | _q�| �� | _n | �� | _d S )NrN r TFr� )rT rU rV rG rI r r� r r� r� rD r� r� r� r� �_init_for_omit_join_m2or �_init_for_joinr �_init_for_omit_join)rX rD rY Zis_m2orZ rZ r6 r7 rU 9 s, � � zSelectInLoader.__init__c s� t �jjj�� � �� �fdd�t� �D �� � fdd��jjD � }}t|�dkrbt j |� }d}n |d }d}�� dd|||d �S ) Nc 3 s0 | ](}�j j�|d �D ]}|� | fV qqdS )r6 N)rD r� rF )r� r� �equiv��pk_to_fkrX r6 r7 r� [ s �z5SelectInLoader._init_for_omit_join.<locals>.<genexpr>c s g | ]}|� kr� | �qS r6 r6 �r� r� )r r6 r7 r� a s z6SelectInLoader._init_for_omit_join.<locals>.<listcomp>r Fr T) r rV r� Zlocal_remote_pairsru r� rD rz r, r( �tuple_� query_info)rX r Zfk_colsr r r6 r r7 r W s �� � z"SelectInLoader._init_for_omit_joinc sb | j j}t|�dkr$tj|� }d}n |d }d}| j�d�� � fdd�|D �}| �dd||||�S )Nr Fr TrN c s g | ]}� j | �qS r6 r r �rZ r6 r7 r� w s z:SelectInLoader._init_for_omit_join_m2o.<locals>.<listcomp>)rJ rz r, r( r rV r� r )rX r r r Z lookup_colsr6 r r7 r m s �z&SelectInLoader._init_for_omit_join_m2oc sj t | jj�| _t| j�� � fdd�| jjD �}t|�dkrJtj|� }d}n |d }d}| � dd|||d �S )Nc s g | ]}� � |��qS r6 ��_adapt_elementr �Zpa_inspr6 r7 r� � s z1SelectInLoader._init_for_join.<locals>.<listcomp>r Fr T) r# rD rH r r& rz r, r( r r )rX r r r r6 r r7 r } s � z SelectInLoader._init_for_joinc C s | j �d��|� d S rM rO r� r6 r6 r7 r} � s ��z#SelectInLoader.init_class_attributec C s |j r | �||||||||�S | j|||| jd�\} } } } | sBd S |jjjsPd S | jj| j j j spt � d| ��t|�dkr�t�|j| j�s�d S nt�|d | j�s�d S | } || j }|�|jdd �}|d k r�t|�}n| j}tj�|| | j| j| j||| | � d S )NrK r� r r< r� )r� r� rI rG r] r� rP rD rG r0 r� r� r� r� r, r/ r� rd rV rF r r& r� r rC rQ rR )rX rg r^ r_ r` rJ rh ra ri r/ rS r r; Z selectin_pathrH r� r� r6 r6 r7 rj � st � �� �� � �z#SelectInLoader.create_row_processorc s� |r�j |krd S �j} | jr�t�t�} g } �j�|D ]l\�} �j�t���fdd�| j D ��} t j | krt�j } q�d | kr�| | � ��| f� q6| � ��| f� q6| js�dd� |D �} | j}| j}| js�jr�fdd�|D �}��|�}tjd|�� }|�� }��� }tj||gttjd�d�d �}| j�s<|���}n$|��j �� t �j �j j �� ���}|�!|�"t#�$d ���}� j%j&}|�j }|� j'k�r�|j(}n&|j(}� j'j(}� fd d�t)||�D �}|�r�|j*�r�|t�+�|�,� ��f7 }|d k �r�|�-� }|j.|� }|�/d |i�}� j0�r&|j1d d�}�j j2�r�| j�sf�j j2}�j�rZ�fdd�|D �}|j2|� }n�fdd�}|�3|�j �}| j�r���4| | | |� |� n��5| | |� |� d S )Nc 3 s" | ]}� j ��|tjd �V qdS )r N)r� r ZPASSIVE_NO_FETCH)r� Zlk)rJ r� � state_dictr6 r7 r� � s ��z0SelectInLoader._load_for_path.<locals>.<genexpr>c S s$ g | ] \}}|j d ||j|f�qS )r )r0 r )r� r� r[ r6 r6 r7 r� s �z1SelectInLoader._load_for_path.<locals>.<listcomp>c s g | ]}� � |��qS r6 r r �r� r6 r7 r� % s r r� )Zcompile_state_pluginZplugin_subject)r r r� r � primary_keysc s g | ]\}}|� � |��qS r6 )Z'_adapt_cached_option_to_uncached_option)r� Zorig_optZ uncached_opt)rg r6 r7 r� w s � �r T)r6 c s g | ]}� � |��qS r6 r r� r r6 r7 r� � s �c s | j tt�� jj��7 _ d S r. r� r� r� r6 r7 r� � s �z?SelectInLoader._load_for_path.<locals>._setup_outermost_orderby)r )6r0 r r r� r� r� rD r r� r r r� r rB r r r r� r r/ ZBundler r+ r r* r r r r r� r_ rV r� �filterZin_r( r� r] Zselect_statementr r$ �zipr� r# r5 rU r� Z_update_compile_optionsr6 r r� r� �_load_via_child�_load_via_parent)rX rg r_ rX rY r� r` r; r r � our_states� none_statesr[ Z related_identr r Z bundle_entZ bundle_sqlZ entity_sqlrz rx r/ r� Zcached_optionsZuncached_optionsr� r� r6 )rg r� rJ rX r� r r7 rR � s� � �� � �� �� �� �� � �� �z SelectInLoader._load_for_pathc s� | j }t|�}|r�|d| j� } || jd � }dd� |jj|d� fdd�| D �i|d��� D �} | D ]T} | �| d �} || D ]:\} }}|s�| j|kr�q|| �| j�� | ||s�| n| g� q|qdq|D ]0\} }}|s�| j|kr�q�| �| j�� | |d � q�d S )Nr c S s i | ]\}}||�qS r6 r6 )r� r� r� r6 r6 r7 � <dictcomp>� s � z2SelectInLoader._load_via_child.<locals>.<dictcomp>r c s g | ]}� j r|d n|�qS r^ �r )r� r0 �r r6 r7 r� � s �z2SelectInLoader._load_via_child.<locals>.<listcomp>�r� r ) r: �sorted� _chunksizer� r* r� rF r0 rV rW )rX r r! r rz rg r r: Zour_keys�chunk�datar0 Z related_objr� r� r[ r6 r$ r7 r � s< ���� � z SelectInLoader._load_via_childc s$ | j }|rdnd }|�r |d| j� }|| jd � }� fdd�|D �} t�t�} t�|jj|d| i|d�� � dd� �D ] \} } | | � d d � | D �� qv|D ]�\} }}}|s�| j |kr�q�| � | |�}|�s|�rt |�d kr�t�d | � |�| j ��|||d � q�|�| j ��|||� q�qd S ) Nr6 r c s& g | ] \}}}}� j r |d n|�qS r^ r# )r� r0 r� r r[ r$ r6 r7 r� � s �z3SelectInLoader._load_via_parent.<locals>.<listcomp>r r% c S s | d S ru r6 r� r6 r6 r7 r8 r� z1SelectInLoader._load_via_parent.<locals>.<lambda>c s s | ]}|d V qdS )r Nr6 r� r6 r6 r7 r� s z2SelectInLoader._load_via_parent.<locals>.<genexpr>r r� )r: r' r� r� r� r� r� r� r* r� r� r0 rF r, r r- rV rW )rX r r rz rg r r: Z _empty_resultr( r r) r� r� r0 r� r r[ r� r6 r$ r7 r � sN � �� �� � �z SelectInLoader._load_via_parent)rk rl rm ro r� � namedtupler r' rU r r r r} rj rR r r rp r6 r6 rZ r7 r s, �� P T7r c sT �fdd�� � fdd�}� fdd�}t j| d|dddd � t j| d |dddd � d S ) Nc sT |d k rP|j � j krP|�t�|��}|rP||k rPtjdt�|�| j� f dd��|S )NzsInstance %s is already associated with an instance of %s via its %s attribute, and is only allowed a single parent.Zbbf1)�code) r0 � hasparentr Zinstance_stater� r� r/ Z instance_strrH )r� r� �oldvalue� initiatorr, )r3 r6 r7 � _do_check s ��z*single_parent_validator.<locals>._do_checkc s � | |d |�S r. r6 )r� r� r. �r/ r6 r7 rB * s z'single_parent_validator.<locals>.appendc s � | |||�S r. r6 )r� r� r- r. r0 r6 r7 �set_- s z%single_parent_validator.<locals>.set_rB T)�raw�retvalr? r� )r% �listen)r2 r3 rB r1 r6 )r/ r3 r7 rC s �rC )NNNNFN)Qrn � __future__r r� r� �typingr r r r � r r r� r r r r r r r r r/ �baser r r r r r r rg r r r r r r r� r r� r r r! r" r# r% r� r& r' r( r� r) Zsql.selectabler* r+ r, Z sql.elementsr- rN ZColumnPropertyZ strategy_forrQ Z class_loggerrq r� r� r� r� r� r� Z MemoizedSlotsZ Identifiedr� r4 r: rJ r] r� r rC r6 r6 r6 r7 �<module> s� � Q+Z W � 0% & � +4L d u }