Edit File: scoping.cpython-37.pyc
B ��4]� � @ s� d dl mZ d dl mZ d dlmZ ddl mZ ddlmZ ddlm Z ddlm Z dgZ G d d� de �Z e Zd d � Zx ejD ]Zee eee�� q~W d d � Zx dD ]Zee eee�� q�W dd� Zx dD ]Zee eee�� q�W dS )� )� class_mapper)�exc)�Session� )�ScopedRegistry)�ThreadLocalRegistry)�warn�scoped_sessionc @ s@ e Zd ZdZdZd dd�Zdd� Zdd� Zd d � Zdd d �Z dS )r ztProvides scoped management of :class:`.Session` objects. See :ref:`unitofwork_contextual` for a tutorial. Nc C s& || _ |rt||�| _n t|�| _dS )a� Construct a new :class:`.scoped_session`. :param session_factory: a factory to create new :class:`.Session` instances. This is usually, but not necessarily, an instance of :class:`.sessionmaker`. :param scopefunc: optional function which defines the current scope. If not passed, the :class:`.scoped_session` object assumes "thread-local" scope, and will use a Python ``threading.local()`` in order to maintain the current :class:`.Session`. If passed, the function should return a hashable token; this token will be used as the key in a dictionary in order to store and retrieve the current :class:`.Session`. N)�session_factoryr �registryr )�selfr Z scopefunc� r �I/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/orm/scoping.py�__init__! s zscoped_session.__init__c K sD |r8| j �� rt�d��q@| jf |�}| j �|� |S n| � � S dS )a� Return the current :class:`.Session`, creating it using the :attr:`.scoped_session.session_factory` if not present. :param \**kw: Keyword arguments will be passed to the :attr:`.scoped_session.session_factory` callable, if an existing :class:`.Session` is not present. If the :class:`.Session` is present and keyword arguments have been passed, :exc:`~sqlalchemy.exc.InvalidRequestError` is raised. zEScoped session is already present; no new arguments may be specified.N)r �has�sa_excZInvalidRequestErrorr �set)r �kwZsessr r r �__call__8 s zscoped_session.__call__c C s$ | j �� r| � � �� | j �� dS )a� Dispose of the current :class:`.Session`, if present. This will first call :meth:`.Session.close` method on the current :class:`.Session`, which releases any existing transactional/connection resources still being held; transactions specifically are rolled back. The :class:`.Session` is then discarded. Upon next usage within the same scope, the :class:`.scoped_session` will produce a new :class:`.Session` object. N)r r �close�clear)r r r r �removeP s zscoped_session.removec K s$ | j �� rtd� | jjf |� dS )z�reconfigure the :class:`.sessionmaker` used by this :class:`.scoped_session`. See :meth:`.sessionmaker.configure`. ztAt least one scoped session is already present. configure() can not affect sessions that have already been created.N)r r r r � configure)r �kwargsr r r r a s zscoped_session.configurec s G � �fdd�dt �}|� S )a" return a class property which produces a :class:`.Query` object against the class and the current :class:`.Session` when called. e.g.:: Session = scoped_session(sessionmaker()) class MyClass(object): query = Session.query_property() # after mappers are defined result = MyClass.query.filter(MyClass.name=='foo').all() Produces instances of the session's configured query class by default. To override and use a custom implementation, provide a ``query_cls`` callable. The callable will be invoked with the class's mapper as a positional argument and a session keyword argument. There is no limit to the number of query properties placed on a class. c s e Zd Z� �fdd�ZdS )z,scoped_session.query_property.<locals>.queryc sN y2t |�}|r0� r"� |��� d�S ��� �|�S W n tjk rH d S X d S )N)�session)r r �query�orm_excZUnmappedClassError)�s�instance�ownerZmapper)� query_clsr r r �__get__� s z4scoped_session.query_property.<locals>.query.__get__N)�__name__� __module__� __qualname__r! r )r r r r r � s r )�object)r r r r )r r r �query_propertyr s z scoped_session.query_property)N)N) r" r# r$ �__doc__r r r r r r& r r r r r s c s � fdd�}|S )Nc s t | �� � �||�S )N)�getattrr )r �argsr )�namer r �do� s zinstrument.<locals>.dor )r* r+ r )r* r � instrument� s r, c s" � fdd�}� fdd�}t ||�S )Nc s t | �� � |� d S )N)�setattrr )r �attr)r* r r �set_� s zmakeprop.<locals>.set_c s t | �� � �S )N)r( r )r )r* r r �get� s zmakeprop.<locals>.get)�property)r* r/ r0 r )r* r �makeprop� s r2 ) �bindZdirtyZdeleted�newZ identity_mapZ is_activeZ autoflushZ no_autoflush�infoZ autocommitc s � fdd�}t |�S )Nc s t t� �||�S )N)r( r )�clsr) r )r* r r r+ � s zclslevel.<locals>.do)� classmethod)r* r+ r )r* r �clslevel� s r8 )Z close_allZobject_sessionZ identity_keyN)� r r r r r r �utilr r r �__all__r% r Z ScopedSessionr, Zpublic_methods�methr- r2 Zpropr8 r r r r �<module> s(