Current File : //opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/orm/__pycache__/loading.cpython-37.pyc
B
��4]�} � @ s4 d Z ddlmZ ddlZddlmZ ddlmZ ddlmZ ddlm Z dd l
m Z dd
l
m Z dd l
mZ dd l
mZ d
dlmZ d
dlm
Z
d
dlm
Z e
�� Zdd� Ze
�d�d-dd��Zdd� Zd.dd�Zd/dd�Zd0dd �Zd1d d �Z d d � Z d!d"� Z d#d$� Z d%d&� Z d'd(� Z!G d)d*� d*e"�Z#d+d,� Z$dS )2z�private module containing functions used to convert database
rows into object instances and associated state.
the functions here are called primarily by Query, Mapper,
as well as some of the attribute loading strategies.
� )�absolute_importN� )�
attributes)�exc)�
path_registry)�strategy_options)�_DEFER_FOR_STATE)�_SET_DEFERRED_EXPIRED)� _none_set)� state_str� )�utilc
# sz t � � _i � _�j}�j o4t�j�dko4�jd j}|rP|rDt}n �fdd�}y�t t
� ��fdd��jD �� �\�}|s�t � d|��x�i � _
�jr����j�}|s�P n��� }|rЈd ��fdd�|D �}n��fd d�|D �}x"� j�� D ]\} }
|
�� | � q�W |�rt �||�}x|D ] } | V �q W �js�P q�W W n6 tk
�rt } z��� t �| � W d
d
} ~ X Y nX d
S ) z$Return an ORM result as an iterator.r r c s t dd� t� j| �D ��S )Nc s s$ | ] \}}|j rt|�n|V qd S )N)Zuse_id_for_hash�id)�.0�ent�item� r �I/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/orm/loading.py� <genexpr>8 s z/instances.<locals>.filter_fn.<locals>.<genexpr>)�tuple�zip� _entities)�row)�queryr r � filter_fn6 s z instances.<locals>.filter_fnc s g | ]}|� �� ���qS r )Z
row_processor)r � query_entity)�context�cursorr r r �
<listcomp>@ s z instances.<locals>.<listcomp>�resultc s g | ] }� |��qS r r )r r )�procr r r U s c s" g | ]� �� fd d��D ���qS )c s g | ] }|� ��qS r r )r r )r r r r X s z(instances.<locals>.<listcomp>.<listcomp>r )r )� keyed_tuple�process)r r r X s N)�
_new_runid�runid�post_load_pathsZ_has_mapper_entitiesZ_only_return_tuples�lenr Zsupports_single_entityr �listr r
�lightweight_named_tuple�partialsZ
_yield_perZ fetchmanyZfetchall�items�invokeZ unique_list� Exception�closeZraise_from_cause)
r r r Zfiltered�
single_entityr �labelsZfetchZrows�path� post_loadr �errr )r r r! r r" r r � instances# sP
r3 zsqlalchemy.orm.queryTc
s |j �� r��� �j}z�d�_t|j�dk}|rdt|jd �j�rZ� �fdd�|D �}nt|�}n��fdd�t|j�D �}g }dd� |jD �}t �
d|�} xh|D ]`}
t|
�} xD|D ]<} | | d k r��j t �
| | �t �| | �� i i d
�| | < q�W |�| | �� q�W t|�S |�_X d S ) z:Merge a result into this :class:`.Query` object's Session.Fr r c s, g | ]$}�j t�|�t�|�� i i d ��qS ))�load�
_recursive�_resolve_conflict_map)�_merger �instance_state�
instance_dict)r �instance)r4 �sessionr r r | s z merge_result.<locals>.<listcomp>c s g | ]\}}t |� j�r|�qS r )�
isinstance�
_MapperEntity)r �i�e)�querylibr r r � s c S s g | ]
}|j �qS r )Z _label_name)r r r r r r � s r N)r4 r5 r6 )r; Z
_autoflush� autoflushr& r r< r= r' � enumerater
r( r7 r r8 r9 �append�iter)
r@ r �iteratorr4 rA r. r Zmapped_entities�keysr! r Znewrowr> r )r4 r@ r; r � merge_resultl s<
rG c C s� | j �|�}|dk rzt�|�}|jrv|tj@ s4tjS |tj@ sB|S y|�||� W n" t j
k
rt | � |g� dS X |S dS dS )zqLook up the given key in the given session's identity map,
check the object for expired state if found.
N) � identity_map�getr r8 ZexpiredZSQL_OKZPASSIVE_NO_RESULTZRELATED_OBJECT_OKZ
_load_expired�orm_exc�ObjectDeletedErrorZ_remove_newly_deleted)r; �keyZpassiver: �stater r r �get_from_identity� s
rN c C s6 |dk r|d }|d }nd }}t | |||||d�S )z.Load the given identity key from the database.Nr r )�
refresh_state�with_for_update�only_load_props�identity_token)�load_on_pk_identity)r rL rO rP rQ ZidentrR r r r �
load_on_ident� s
rT c s |dkr| � � }|�� n| � � }|dk r�| �� }|j\}� d|krnt� fdd�t|j|�D ��} t�|| �}|� |dd�}||_
t � fdd�t||j�D ��}
|
|_ |dk r�d} ||_
n | j
dk r�d} | j
|_
nd} |jt|�| |||d� d|_y|�� S tjk
�r dS X dS )z6Load the given primary key identity from the database.Nc s" g | ]\}}|d kr� | j �qS )N)rL )r �col�value)� _get_paramsr r r � s z'load_on_pk_identity.<locals>.<listcomp>TFc s g | ]\}}� | j |f�qS r )rL )r Zid_val� primary_key)rW r r r � s )�populate_existing�
version_checkrQ rO rR )Z_cloneZ_get_conditionZ _mapper_zero� _get_clause�setr rX �sql_utilZadapt_criterion_to_nullZ
_adapt_clauseZ
_criterion�dictZ_paramsZ_for_update_argZ _get_options�boolZ _order_byZonerJ Z
NoResultFound) r Zprimary_key_identityrO rP rQ rR �q�mapperr[ Znones�paramsrZ r )rW r rS � sL
rS c K s� |r|� |�}
n|j}
i } |�| jd| � x<|
D ]4} |rD| j|krDq0| j| |||f||| d�| �� q0W |d k r�||jk r�|r�|j| }
n|}
|�|
� d S )N�memoized_setups)rQ �column_collectionZmemoized_populators) Z _iterate_polymorphic_propertiesZ_polymorphic_propertiesr\ r rL Zsetup�polymorphic_on�columnsrC )r ra r r0 �adapterrd Zwith_polymorphicrQ �polymorphic_discriminator�kwZpoly_properties�quick_populatorsrV Zpdr r r �_setup_entity_query s0
rk c s$ �
j �
� r � fdd��
D ��
�
j�t�t���
j} � dk rR| ��
fdd�� D ��} |��jdt �}
x�| D ]�} | |
k�r$|
| } | t
kr��d � | j | j
f� n�| tkr��d � | j d f� nfd}
� r� j| }|dk r�|�|d �}
|
s�|�| d �}
|
�r�d
� | j |
f� n| ��|�
|� �� qh| ��|�
|� �� qhW �j��jjj�rZ�jj| n|��jj��j�pr�
j�t�
jj j ��t�
jj j ��t�jj j �� � �r��jj j � tj �tj!��jj"��j#��j$��j%���sr|dk �rrd |jf}|�jk�r&�j| j&d k�r&�
�'�j| j(d
|�}n �
�'d|�}|�rr||k �rr� dk�sPt)�t*�||�}t+�,��|j-|||� t+�.��� ����r��j ��dk�r��
�/���nd��
j0�r�t j1�nt j�� ��������� �
� � �
�����������fdd�}�
j2�r |�s ��s t3|��
|||� �}|S )z`Produce a mapper level row processor callable
which processes rows into mapped instances.c s g | ]}� j | �qS r )rf )r �c)rg r r r h s z'_instance_processor.<locals>.<listcomp>Nc 3 s | ]}� j | V qd S )N)Z_props)r �k)ra r r r p s z&_instance_processor.<locals>.<genexpr>rc �new�expireF�quick�loader))Zselectinload_polymorphicTZentitiesc
sJ �r,�}|� � }�|�}|j�k}d}d}n��t� fdd��D ���f}��|�}|d k r��|�}�|�}|j�k}| }d}�r�|s�t� ||� �� nT�|d �r�d S d}d}d}� j�� }�|�}�|�}||_�|_�|_ ��
||� |s���r�|�r��s��s�|_ � |_ t
�� |||� |��� |�r�|�rf��rN|jj�|�� �
�r~�
�j|� � � n��r~|jj�|�� � ��s�|j�r���r�� �r�|�|� � n |�|�� ��rF��|d� n�|j}|�jk}|�r�|�s�d �r,t�� |||� |��} |�r,��r |jj�|�| � |�|| � ��rF�j�rF��|d� |S )NTFc s g | ] }� | �qS r r )r �column)r r r r � s z:_instance_processor.<locals>._instance.<locals>.<listcomp>r �eager)�objr$ r rI �_validate_version_id�
class_managerZ new_instancerL rR �
session_idZ_add_unpresentZ load_options� load_path�_populate_full�manager�dispatchr4 r; �refreshZmodifiedZ_commitZ _commit_all� add_state�unloadedr) �_populate_partialZinvoke_all_eagers)
r rM r: �dict_�isnewZ currentload�loaded_instanceZ identitykeyr~ �to_load)rg r �identity_classrR r9 r8 �is_not_primary_key�load_evtrx �loaded_as_persistentra rQ �persistent_evt�pk_colsrY �
populatorsr1 �propagate_options� refresh_evt�refresh_identity_keyrO r$ rw �session_identity_maprZ )r r � _instance� s�
z&_instance_processor.<locals>._instance)4rX Z_identity_class� collections� defaultdictr' Z _prop_set� intersectionrI r r
r rC rL Z_deferred_column_loaderr rf Z_getterZcreate_row_processorr� r Z
_current_pathr0 r; rH rY Zalways_refreshr_ rv r{ r4 r| r� r8 r9 Zhash_keyrZ r$ rR ZstrategyZ_should_selectin_loadZ
local_opts�AssertionError�_load_subclass_via_in�PostLoad�callable_for_pathra � for_context�_identity_key_from_state�allow_partial_pks�
issuperset�polymorphic_map� _decorate_polymorphic_switch)ra r r r0 rg rQ rO rh �_polymorphic_fromZpropsrj ZproprU �getterZ adapted_colrL Zselectin_load_viaZ callable_r� r )rg r r� rR r9 r8 r� r� rx r� ra rQ r� r� rY r� r1 r� r� r� rO r$ rw r� rZ r �_instance_processorQ s�
<
r� c sP |j }t|jj�dk�|jr.|�|�\��� n |j\��� � ���fdd�}|S )Nr c sb | j }�j�f|j � f |j|d�}|jr:|�dd� � || j�j�fdd�|D �d��� d S )N)�
cache_pathc S s | � � S )N)rY )r` r r r �<lambda>� � z8_load_subclass_via_in.<locals>.do_load.<locals>.<lambda>c s, g | ]$\}}� r |j d d n|j d �qS )r r )rL )r rM Z
load_attrs)�zero_idxr r r � s z:_load_subclass_via_in.<locals>.do_load.<locals>.<listcomp>)Z primary_keys) r Z_with_lazyload_optionsZ
_with_options�parentZ_populate_existingZ add_criteriar; rb �all)r r0 �statesZ load_onlyZeffective_entityZ
orig_queryZq2)� disable_opt�
enable_optr` r� r r �do_load� s
z&_load_subclass_via_in.<locals>.do_load)ra r& Z base_mapperrX Zis_aliased_classZ_subclass_load_via_inZ _subclass_load_via_in_mapper)r r0 Zentityra r� r )r� r� r` r� r r� � s r� c
C s\ |r�| j |_ x |d D ]\} }
|
|�|| < qW |rfxX|d D ]$\} } |�| d � | r<|j�| � q<W n&x$|d D ]\} } | rp|j�| � qpW x |d D ]\} } | |||� q�W x�|d D ]\} } | |||� q�W n�||jk�r4||_x(|d D ] \} }
| |kr�|
|�|| < q�W xH|d D ]\} } | |||� �qW n$x"|d D ]\} } | |||� �q>W d S )Nrp ro rn �delayed�existing)r$ �pop�expired_attributes�addrx )
r r rM r� r� rx r� rY r� rL r� � set_callable� populatorr r r ry � s2 ry c
C s2 |s:| j | }x�|d D ] \} }
| |kr|
|||� qW n�|}|| j |< x(|d D ] \} } | |krR| |�|| < qRW x8|d D ],\} } | |kr||�| d � | r||j�| � q|W x(|d D ] \} }
| |kr�|
|||� q�W x(|d D ] \} }
| |kr�|
|||� q�W x,|d D ] \} }
| |k�r
|
|||� �q
W |S )Nr� rp ro rn r� rs )r) r� r� r� )
r r rM r� r� rx r~ r� r� rL r� r� r� r r r r � s2
r c C sd | j }|d krd S |r |j| }| �||| j �|| kr`t�dt|�| �||| j �|| f ��d S )NzWInstance '%s' has version id '%s' which does not match database-loaded version id '%s'.)�version_id_colrf Z_get_state_attr_by_columnrJ ZStaleDataErrorr )ra rM r� r rg r� r r r ru s
ru c s` |d k r|�n�j ��d kr �S � r.� j� �� ����fdd�}t�|�����fdd�}|S )Nc sT y�j | }W n tk
r. td| ��Y n"X |�kr<d S t|���� �d�S d S )Nz*No such polymorphic_identity %r is defined)r� )r� �KeyErrorr� r� )�
discriminatorZ
sub_mapper)rg r ra r0 r r r �configure_subclass_mapper. s z?_decorate_polymorphic_switch.<locals>.configure_subclass_mapperc s, | � }|d k r$�| }|r$|| �S � | �S )Nr )r r� r� )� instance_fn�polymorphic_instancesre r r �polymorphic_instanceD s z:_decorate_polymorphic_switch.<locals>.polymorphic_instance)re rf r
Z PopulateDict) r� r ra r r0 rh rg r� r� r )rg r r� ra r0 r� re r r r� s
r� c @ sP e Zd ZdZdZdd� Zdd� Zdd� Zed d
� �Z ed d � �Z
ed
d� �Z dS )r� z:Track loaders and states for "post load" operations.
)�loadersr� � load_keysc C s i | _ t�� | _d | _d S )N)r� r
� OrderedDictr� r� )�selfr r r �__init__V s
zPostLoad.__init__c C s || j |< d S )N)r� )r� rM � overwriter r r r} [ s zPostLoad.add_statec sv | j s
d S tj�|�}xP| j�� D ]B\}� }}}� fdd�| j �� D �}|r"||||| jf|�|� q"W | j �� d S )Nc s& g | ] \}}|j j�� �r||f�qS r )rz ra Zisa)r rM r� )�limit_to_mapperr r r h s z#PostLoad.invoke.<locals>.<listcomp>) r� r Z PathRegistryZcoercer� �valuesr* r� �clear)r� r r0 �tokenrq �argri r� r )r� r r+ b s
zPostLoad.invokec C s$ |j �|j�}|d k r |r ||_|S )N)r% rI r0 r� )�clsr r0 rQ �plr r r r� p s zPostLoad.for_contextc C s |j |jko ||j|j jkS )N)r0 r% r� )r� r r0 rL r r r � path_existsw s zPostLoad.path_existsc O sD |j |jkr|j|j }nt� }|j|j < |||||f|j|< d S )N)r0 r% r� r� ) r� r r0 r� r� Zloader_callabler� ri r� r r r r� ~ s zPostLoad.callable_for_pathN) �__name__�
__module__� __qualname__�__doc__� __slots__r� r} r+ � classmethodr� r� r� r r r r r� O s r� c sB |j }|s t�dt|� ��t|j�}d}|r>|�� j�� �}� j r�� j
s�� � ||�}|dk r�t |�
� ��t�� ��d���|�d||d�}|dk�r$|r�|j}n<� fdd�� jD �}|j�|�r�t�dt|� ��� �|�}t�|�r� jr�t�|��rt � d t|�� dS t |�
� �|||d
�}|�r>|dk�r>t� |��dS ) z4initiate a column-based attribute refresh operation.zQInstance %s is not bound to a Session; attribute refresh operation cannot proceedFN�*)rQ rO c s g | ]}� j | j�qS r )Z_columntopropertyrL )r rU )ra r r r � s z*load_scalar_attributes.<locals>.<listcomp>z_Instance %s cannot be refreshed - it's not persistent and does not contain a full primary key.zwInstance %s to be refreshed doesn't contain a full primary key - can't be refreshed (and shouldn't be expired, either).)rO rQ ) r; rJ ZDetachedInstanceErrorr r_ rL r� ZattrsrF ZinheritsZconcreteZ_optimized_get_statementrT r Zoptionsr ZLoadZundeferZfrom_statementrX r� �sa_excZInvalidRequestErrorr� r
�issubsetr� r� r
Z warn_limitedrK ) ra rM Zattribute_namesr; Zhas_keyr Z statementZ identity_keyZpk_attrsr )ra r �load_scalar_attributes� sR
r� )T)NNN)NNNN)NNN)NNNN)%r� Z
__future__r r� � r r rJ r r �baser r r
r
r r� Zsqlr] Zcounterr# r3 Z dependenciesrG rN rT rS rk r� r� ry r ru r� �objectr� r� r r r r �<module> sN I 5
L
/
- 9"4:
Mini Shell