관리-도구
편집 파일: baked.cpython-38.pyc
U -?�f�E � @ s� d Z ddlmZ ddlZddlmZ ddlmZ ddl mZ ddl m Z ddl mZ ddlmZ dd lmZ ddlmZ e�e�ZG d d � d �ZG d d � d �ZG dd� d�ZejZdS )z�Baked query extension. Provides a creational pattern for the :class:`.query.Query` object which allows the fully constructed object, Core select statement, and string compiled result to be fully cached. � N� )�exc)�util)�Query)�Session)�func)�literal_columnc @ s$ e Zd ZdZdZdd� Zdd� ZdS )�Bakeryz�Callable which returns a :class:`.BakedQuery`. This object is returned by the class method :meth:`.BakedQuery.bakery`. It exists as an object so that the "cache" can be easily inspected. .. versionadded:: 1.2 ��cls�cachec C s || _ || _d S �Nr )�selfZcls_r � r �D/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/ext/baked.py�__init__1 s zBakery.__init__c G s | � | j||�S r r )r � initial_fn�argsr r r �__call__5 s zBakery.__call__N)�__name__� __module__� __qualname__�__doc__� __slots__r r r r r r r # s r c @ s� e Zd ZdZdZd+dd�Zed,dd ��Zd d � Zd-d d �Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd.dd �Zd d � Zd/d d �Zd0d!d"�Zd#d$� Zd%d&� Zd'd(� Zd)d*� ZdS )1� BakedQueryz3A builder object for :class:`.query.Query` objects.)�steps�_bakery� _cache_key�_spoiledr c C s* d| _ | �||� |g| _d| _|| _d S )Nr F)r �_update_cache_keyr r r )r �bakeryr r r r r r > s zBakedQuery.__init__�� Nc C s t | tj||d��S )zSConstruct a new bakery. :return: an instance of :class:`.Bakery` )Z size_alert)r r ZLRUCache)r �sizeZ _size_alertr r r r E s zBakedQuery.bakeryc C s2 t �t �}| j|_t| j�|_| j|_| j|_|S r )r �__new__r �listr r r )r �b1r r r �_cloneO s zBakedQuery._clonec C s | j |jf| 7 _ d S r )r �__code__�r �fnr r r r r W s z BakedQuery._update_cache_keyc C s$ t |t�r| j|� n | �|� | S r )� isinstance�tuple� add_criteria�r �otherr r r �__iadd__Z s zBakedQuery.__iadd__c C s" t |t�r| j|� S | �|�S d S r )r* r+ � with_criteriar- r r r �__add__a s zBakedQuery.__add__c G s | � ||� | j�|� | S )z�Add a criteria function to this :class:`.BakedQuery`. This is equivalent to using the ``+=`` operator to modify a :class:`.BakedQuery` in-place. )r r �appendr( r r r r, g s zBakedQuery.add_criteriac G s | � � j|f|�� S )z�Add a criteria function to a :class:`.BakedQuery` cloned from this one. This is equivalent to using the ``+`` operator to produce a new :class:`.BakedQuery` with modifications. )r& r, r( r r r r0 r s zBakedQuery.with_criteriac C s t | |�S )z�Return a :class:`_baked.Result` object for this :class:`.BakedQuery`. This is equivalent to calling the :class:`.BakedQuery` as a Python callable, e.g. ``result = my_baked_query(session)``. )�Result�r �sessionr r r � for_session| s zBakedQuery.for_sessionc C s | � |�S r )r6 r4 r r r r � s zBakedQuery.__call__Fc C s4 |s*| j s*| �� }| jd7 _|jg| _d| _ | S )a� Cancel any query caching that will occur on this BakedQuery object. The BakedQuery can continue to be used normally, however additional creational functions will not be cached; they will be called on every invocation. This is to support the case where a particular step in constructing a baked query disqualifies the query from being cacheable, such as a variant that relies upon some uncacheable value. :param full: if False, only functions added to this :class:`.BakedQuery` object subsequent to the spoil step will be non-cached; the state of the :class:`.BakedQuery` up until this point will be pulled from the cache. If True, then the entire :class:`_query.Query` object is built from scratch each time, with all creational functions being called on each invocation. )Z _query_onlyT)r r& r �_retrieve_baked_queryr )r �fullZ _spoil_pointr r r �spoil� s zBakedQuery.spoilc C s | j |jf S )a� Return the key that actually goes into the cache dictionary for this :class:`.BakedQuery`, taking into account the given :class:`.Session`. This basically means we also will include the session's query_class, as the actual :class:`_query.Query` object is part of what's cached and needs to match the type of :class:`_query.Query` that a later session will want to use. )r Z _query_clsr4 r r r �_effective_key� s zBakedQuery._effective_keyc C s | � � }|j|||d� |S )z)Cloning version of _add_lazyload_options.)� cache_path)r& �_add_lazyload_options)r �options�effective_pathr; �qr r r �_with_lazyload_options� s z!BakedQuery._with_lazyload_optionsc sz d}|s � }�D ]J}|j s |jr|�� }|dkr>| jdd� q|d rNtd��||d 7 }q| �� �fdd �|j|� dS ) a* Used by per-state lazy loaders to add options to the "lazy load" query from a parent query. Creates a cache key based on given load path and query options; if a repeatable cache key cannot be generated, the query is "spoiled" so that it won't use caching. r NT)r8 � z�loader options with variable bound parameters not supported with baked queries. Please use new-style select() statements for cached ORM queries.r c s | � � �j�� S r )Z_with_current_pathr= �r? �r> r= r r �<lambda>� � z2BakedQuery._add_lazyload_options.<locals>.<lambda>)Z_is_legacy_optionZ_is_compile_stateZ_generate_cache_keyr9 �AssertionErrorr, �path)r r= r> r; �key�optZckr rC r r< � s" � �z BakedQuery._add_lazyload_optionsc C sF | j �| �|�d �}|d kr<| �|�}|�d �| j | �|�< |�|�S r )r �getr: � _as_queryZ with_session)r r5 �queryr r r r7 � s �z BakedQuery._retrieve_baked_queryc C s< | � |�}d |_|�� }|jjr4||f| j| �|�< ||fS r )rK r5 Z _statement_20Z_compile_optionsZ_bake_okr r: )r r5 rL � statementr r r �_bake� s �zBakedQuery._bakec C sN t |t�r|}n4t |t�r4|j}|dkrDt�d��ntdt|� ��| �|�S )aL Return the :class:`_query.Query` object for use as a subquery. This method should be used within the lambda callable being used to generate a step of an enclosing :class:`.BakedQuery`. The parameter should normally be the :class:`_query.Query` object that is passed to the lambda:: sub_bq = self.bakery(lambda s: s.query(User.name)) sub_bq += lambda q: q.filter( User.id == Address.user_id).correlate(Address) main_bq = self.bakery(lambda s: s.query(Address)) main_bq += lambda q: q.filter( sub_bq.to_query(q).exists()) In the case where the subquery is used in the first callable against a :class:`.Session`, the :class:`.Session` is also accepted:: sub_bq = self.bakery(lambda s: s.query(User.name)) sub_bq += lambda q: q.filter( User.id == Address.user_id).correlate(Address) main_bq = self.bakery( lambda s: s.query( Address.id, sub_bq.to_query(q).scalar_subquery()) ) :param query_or_session: a :class:`_query.Query` object or a class :class:`.Session` object, that is assumed to be within the context of an enclosing :class:`.BakedQuery` callable. .. versionadded:: 1.3 Nz1Given Query needs to be associated with a Sessionz)Query or Session object expected, got %r.) r* r r r5 �sa_exc� ArgumentError� TypeError�typerK )r Zquery_or_sessionr5 r r r �to_query� s & ���zBakedQuery.to_queryc C s. | j d |�}| j dd � D ] }||�}q |S �Nr rA )r )r r5 rL �stepr r r rK 1 s zBakedQuery._as_query)r )r! N)r )F)N)N)r r r r r r � classmethodr r& r r/ r1 r, r0 r6 r r9 r: r@ r< r7 rN rS rK r r r r r 9 s* # 5r c @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$r3 a Invokes a :class:`.BakedQuery` against a :class:`.Session`. The :class:`_baked.Result` object is where the actual :class:`.query.Query` object gets created, or retrieved from the cache, against a target :class:`.Session`, and is then invoked for results. ��bqr5 �_params�_post_criteriac C s || _ || _i | _g | _d S r rW )r rX r5 r r r r E s zResult.__init__c O sB t |�dkr |�|d � nt |�dkr2t�d��| j�|� | S )z@Specify parameters to be replaced into the string SQL statement.rA r zFparams() takes zero or one positional argument, which is a dictionary.)�len�updaterO rP rY )r r �kwr r r �paramsK s � z Result.paramsc C s |r| j �|� | S r )rZ �extend)r Zfnsr r r �_using_post_criteriaX s zResult._using_post_criteriac C s | � |g�S )a� Add a criteria function that will be applied post-cache. This adds a function that will be run against the :class:`_query.Query` object after it is retrieved from the cache. This currently includes **only** the :meth:`_query.Query.params` and :meth:`_query.Query.execution_options` methods. .. warning:: :meth:`_baked.Result.with_post_criteria` functions are applied to the :class:`_query.Query` object **after** the query's SQL statement object has been retrieved from the cache. Only :meth:`_query.Query.params` and :meth:`_query.Query.execution_options` methods should be used. .. versionadded:: 1.2 )r` )r r) r r r �with_post_criteria] s zResult.with_post_criteriac C s. | j �| j��| j�}| jD ] }||�}q |S r )rX rK r5 r^ rY rZ )r r? r) r r r rK v s zResult._as_queryc C s t | �� �S r )�strrK �r r r r �__str__| s zResult.__str__c C s | � � �� S r )�_iter�__iter__rc r r r rf s zResult.__iter__c C s� | j }| jjr|jr | �� �� S |j�|�| j�d�\}}|d krR|� | j�\}}| j rf|� | j �}n|}| j D ] }||�}qp|j }t |j�}|�|j|jd�� | jj|||d�}|j�dd�r�|�� }|j�dd�r�|�� }|S )N)NN)Z_sa_orm_load_optionsZcompiled_cache)�execution_optionsZis_single_entityF�filtered)rX r5 Zenable_baked_queriesr rK re r rJ r: rN rY r^ rZ �dictZ_execution_optionsr\ Z load_options�execute� _attributesZscalars�unique) r rX rL rM r? r) r^ rg �resultr r r re � s> � �� �z Result._iterc s: t �td��� | j�� fdd��}|�| j��| j�� � S )z�return the 'count'. Equivalent to :meth:`_query.Query.count`. Note this uses a subquery to ensure an accurate count regardless of the structure of the original statement. �*c s | � � �S r )Z_legacy_from_selfrB ��colr r rD � rE z Result.count.<locals>.<lambda>) r �countr rX r0 r6 r5 r^ rY �scalar�r rX r ro r rq � s z Result.countc C sB z$| � � }t|tj�s |W S |d W S tjk r< Y dS X dS )z�Return the first element of the first result or None if no rows present. If multiple rows are returned, raises MultipleResultsFound. Equivalent to :meth:`_query.Query.scalar`. r N)�oner* �collections_abc�Sequence�orm_excZ NoResultFound)r �retr r r rr � s z Result.scalarc C s4 | j �dd� �}|�| j��| j��| j��� � � S )zRReturn the first row. Equivalent to :meth:`_query.Query.first`. c S s | � dd�S rT )�slicerB r r r rD � rE z Result.first.<locals>.<lambda>) rX r0 r6 r5 r^ rY r` rZ re �firstrs r r r rz � s �� �z Result.firstc C s | � � �� S )zkReturn exactly one result or raise an exception. Equivalent to :meth:`_query.Query.one`. )re rt rc r r r rt � s z Result.onec C s | � � �� S )z�Return one or zero results, or raise an exception for multiple rows. Equivalent to :meth:`_query.Query.one_or_none`. )re � one_or_nonerc r r r r{ � s zResult.one_or_nonec C s | � � �� S )zKReturn all rows. Equivalent to :meth:`_query.Query.all`. )re �allrc r r r r| � s z Result.allc C s | j jd | j�}|�|| j�S )z`Retrieve an object based on identity. Equivalent to :meth:`_query.Query.get`. r )rX r r5 Z _get_impl�_load_on_pk_identity)r �identrL r r r rJ � s z Result.getc s� |j d jd ��j\� �� ����fdd�}�j}|�� }| j� f7 _|�|tdd� �D ���}�fdd�t��j �D �}t |� �j �j f |��}t|�} | d kr�t�� �n| r�|d S d S d S ) z6Load the given primary key identity from the database.r Z parententityc st � }| � � }|�� d |_d �krH�fdd�t�j��D �}t�||�}t�|ddi�f|_�j D ] }||�}qb|S )Nc s" h | ]\}}|d kr� | j �qS r �rH )�.0rp �value�� _get_paramsr r � <setcomp> s �z=Result._load_on_pk_identity.<locals>.setup.<locals>.<setcomp>Z _orm_adaptT) r& Z_get_conditionZ _order_by�zip� primary_key�sql_utilZadapt_criterion_to_nullZ_deep_annotateZ_where_criteriarZ )rL Z_lcl_get_clauser? Znonesr) �� _get_clauser� Zmapper�primary_key_identityr r r �setup s( �� �� z*Result._load_on_pk_identity.<locals>.setupc s s | ]}|d kV qd S r r )r� �elemr r r � <genexpr>* s z.Result._load_on_pk_identity.<locals>.<genexpr>c s i | ]\}}� | j |�qS r r )r� Zid_valr� r� r r � <dictcomp>- s � z/Result._load_on_pk_identity.<locals>.<dictcomp>rA N)Z _raw_columnsZ _annotationsr� rX r& r r0 r+ r� r� r$ r6 r5 r^ r[ rw ZMultipleResultsFound) r r5 rL r� r] r� rX r^ rm �lr r� r r} � s. � �� zResult._load_on_pk_identityN)r r r r r r r^ r` ra rK rd rf re rq rr rz rt r{ r| rJ r} r r r r r3 : s$ ' r3 )r �collections.abc�abcru �logging� r rO r Zormrw Z orm.queryr Z orm.sessionr �sqlr r r� � getLoggerr �logr r r3 r r r r r �<module> s&