Edit File: 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 �filtered� single_entityr �labelsZfetchZrows�path� post_loadr �errr )r r r! r r" r r � instances# sP r4 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)r5 �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)r5 r6 r7 )r< Z _autoflush� autoflushr& r r= r> r' � enumerater r( r8 r r9 r: �append�iter) rA r �iteratorr5 rB r/ r Zmapped_entities�keysr! r Znewrowr? r )r5 rA r< r � merge_resultl s< rH 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 r9 ZexpiredZSQL_OKZPASSIVE_NO_RESULTZRELATED_OBJECT_OKZ _load_expired�orm_exc�ObjectDeletedErrorZ_remove_newly_deleted)r< �keyZpassiver; �stater r r �get_from_identity� s rO 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 rM rP rQ rR �identrS r r r � load_on_ident� s rV 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)rM )r �col�value)� _get_paramsr r r � s z'load_on_pk_identity.<locals>.<listcomp>TFc s g | ]\}}� | j |f�qS r )rM )r Zid_val� primary_key)rY r r r � s )�populate_existing� version_checkrR rP rS )Z_cloneZ_get_conditionZ _mapper_zero� _get_clause�setr rZ �sql_utilZadapt_criterion_to_nullZ _adapt_clauseZ _criterion�dict�_paramsZ_for_update_argZ _get_options�boolZ _order_by�onerK Z NoResultFound) r Zprimary_key_identityrP rQ rR rS �q�mapperr] Znones�paramsr\ r )rY r rT � sL rT 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)rR �column_collectionZmemoized_populators) Z _iterate_polymorphic_propertiesZ_polymorphic_propertiesr^ r rM �setup�polymorphic_on�columnsrD )r re r r1 �adapterrh Zwith_polymorphicrR �polymorphic_discriminator�kwZpoly_properties�quick_populatorsrX �pdr r r �_setup_entity_query s0 rq 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 )rk )r �c)rl r r r h s z'_instance_processor.<locals>.<listcomp>Nc 3 s | ]}� j | V qd S )N)Z_props)r �k)re r r r p s z&_instance_processor.<locals>.<genexpr>rg �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 rJ �_validate_version_id� class_managerZ new_instancerM rS � session_idZ_add_unpresentZ load_options� load_path�_populate_full�manager�dispatchr5 r< �refresh�modifiedZ_commitZ _commit_all� add_state�unloadedr) �_populate_partialZinvoke_all_eagers) r rN r; �dict_�isnewZ currentload�loaded_instanceZ identitykeyr� �to_load)rl r �identity_classrS r: r9 �is_not_primary_key�load_evtr~ �loaded_as_persistentre rR �persistent_evt�pk_colsr[ � populatorsr2 �propagate_options� refresh_evt�refresh_identity_keyrP r$ r} �session_identity_mapr\ )r r � _instance� s� z&_instance_processor.<locals>._instance)4rZ Z_identity_class� collections� defaultdictr' Z _prop_set� intersectionrJ r r r rD rM Z_deferred_column_loaderr rk Z_getterZcreate_row_processorr� r Z _current_pathr1 r< rI r[ Zalways_refreshrb r| r� r5 r� r� r9 r: Zhash_keyr\ r$ rS ZstrategyZ_should_selectin_loadZ local_opts�AssertionError�_load_subclass_via_in�PostLoad�callable_for_pathre � for_context�_identity_key_from_state�allow_partial_pks� issuperset�polymorphic_map� _decorate_polymorphic_switch)re r r r1 rl rR rP rm �_polymorphic_from�propsro ZproprW �getterZ adapted_colrM Zselectin_load_viaZ callable_r� r )rl r r� rS r: r9 r� r� r~ r� re rR r� r� r[ r� r2 r� r� r� rP r$ r} r� r\ 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)r[ )rd 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 )rM )r rN 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< rf �all)r r1 �statesZ load_onlyZeffective_entityZ orig_queryZq2)� disable_opt� enable_optrd r� r r �do_load� s z&_load_subclass_via_in.<locals>.do_load)re r& Z base_mapperrZ Zis_aliased_classZ_subclass_load_via_inZ _subclass_load_via_in_mapper)r r1 �entityre r� r )r� r� rd 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 )Nrv ru rt �delayed�existing)r$ �pop�expired_attributes�addr~ ) r r rN r� r� r~ r� r[ r� rM r� � set_callable� populatorr r r r � s2 r 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� rv ru rt r� ry )r) r� r� r� ) r r rN r� r� r~ r� r� r� rM 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_colrk Z_get_state_attr_by_columnrK ZStaleDataErrorr )re rN r� r rl r� r r r r{ s r{ 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)rl r re r1 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_instancesrj r r �polymorphic_instanceD s z:_decorate_polymorphic_switch.<locals>.polymorphic_instance)rj rk r Z PopulateDict) r� r re r r1 rm rl r� r� r )rl r r� re r1 r� rj 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� rN � 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 )r� re Zisa)r rN r� )�limit_to_mapperr r r h s z#PostLoad.invoke.<locals>.<listcomp>) r� r Z PathRegistryZcoercer� �valuesr* r� �clear)r� r r1 �tokenrw �argrn r� r )r� r r+ b s zPostLoad.invokec C s$ |j �|j�}|d k r |r ||_|S )N)r% rJ r1 r� )�clsr r1 rR �plr r r r� p s zPostLoad.for_contextc C s |j |jko ||j|j jkS )N)r1 r% r� )r� r r1 rM r r r � path_existsw s zPostLoad.path_existsc O sD |j |jkr|j|j }nt� }|j|j < |||||f|j|< d S )N)r1 r% r� r� ) r� r r1 r� r� Zloader_callabler� rn 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�*)rR rP c s g | ]}� j | j�qS r )Z_columntopropertyrM )r rW )re 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).)rP rR ) r< rK ZDetachedInstanceErrorr rb rM r� �attrsrG ZinheritsZconcreteZ_optimized_get_statementrV r �optionsr �LoadZundeferZfrom_statementrZ r� �sa_excZInvalidRequestErrorr� r �issubsetr� r� r Z warn_limitedrL ) re rN Zattribute_namesr< Zhas_keyr � statementZ identity_keyZpk_attrsr )re r �load_scalar_attributes� sR r� )T)NNN)NNNN)NNN)NNNN)%r� � __future__r r� � r r rK r r �baser r r r r r� Zsqlr_ �counterr# r4 Z dependenciesrH rO rV rT rq r� r� r r� r{ r� �objectr� r� r r r r �<module> sN I 5 L / - 9"4: