Current File : //opt/alt/python37/lib/python3.7/site-packages/beaker/ext/__pycache__/database.cpython-37.pyc
B
��[� � @ s� d dl mZ d dlZd dlZd dlmZ d dlmZmZ d dlmZm Z d dl
m Z m Z d dl
mZmZ e�e�ZdadadaG dd � d e�ZG d
d � d e�ZdS ) � )�pickleN)�datetime)� OpenResourceNamespaceManager� Container)�InvalidCacheBackendError�MissingCacheParameter)�file_synchronizer�null_synchronizer)�verify_directory�SyncDictc @ s~ e Zd Ze� Ze� Zedd� �Zd dd�Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� ZdS ) �DatabaseNamespaceManagerc C sR t d k r d S y$dd la dd lma ddlma W n tk
rL td��Y nX d S )Nr )�typesz8Database cache backend requires the 'sqlalchemy' library)�sa�
sqlalchemyZsqlalchemy.pool�poolr
� ImportErrorr )�cls� r �D/opt/alt/python37/lib/python3.7/site-packages/beaker/ext/database.py�_init_dependencies s z+DatabaseNamespaceManager._init_dependenciesNF� beaker_cachec s� t �| |� � dkri � d| _|r*|| _n|r8|d | _| jrHt| j� �pR� d ��� }
� ���fdd�} i | _d| _d| _tj� |
| �| _
dS )aT Creates a database namespace manager
``url``
SQLAlchemy compliant db url
``sa_opts``
A dictionary of SQLAlchemy keyword options to initialize the engine
with.
``optimistic``
Use optimistic session locking, note that this will result in an
additional select when updating a cache value to compare version
numbers.
``table_name``
The table name to use in the database for the cache.
``schema_name``
The schema name to use in the database for the cache.
Nz/container_db_lockzsa.urlc s� �� } � �fdd�}t j�| |�}tj�|tjdtjdd�tjdt�d�dd �tjd
tj dd �tjd tj dd �tjd tj
dd �t� d��r��n|j d
� }|j
dd� |S )Nc s& �� d< t �� d�} t �� }| |_|S )Nzsa.urlzsa.)r Zengine_from_configZMetaData�bind)Zengine�meta)�sa_opts�urlr r � make_metaM s
zHDatabaseNamespaceManager.__init__.<locals>.make_cache.<locals>.make_meta�idT)Z primary_key� namespace� F)Znullable�accessed�created�data)�schema)Z
checkfirst)r � metadatas�getr ZTableZColumnr
ZInteger�StringZDateTimeZ
PickleTypeZUniqueConstraintr" �create)Zmeta_keyr r �cache)r � schema_name�
table_namer r r �
make_cacheI s z5DatabaseNamespaceManager.__init__.<locals>.make_cacheF) r �__init__�lock_dirr
�hash�_is_new�loadedr �tablesr$ r' ) �selfr r r Z
optimisticr) Zdata_dirr, r( �paramsZ table_keyr* r )r r( r) r r r+ $ s"
z!DatabaseNamespaceManager.__init__c C s t � S )N)r )r1 r r r �get_access_lockg s z(DatabaseNamespaceManager.get_access_lockc C s t d| j|f | jd�S )Nz databasecontainer/funclock/%s/%s)�
identifierr, )r r r, )r1 �keyr r r �get_creation_lockj s
z*DatabaseNamespaceManager.get_creation_lockc C s� | j r|| _d S | j}t�|jjg|jj| jk��� }|� � }|�
� |sXd| _ i | _ nNd| _ y|d | _ W n8 t
tttjtjfk
r� t�d� i | _ d| _ Y nX || _d| _ d S )NTFr! z.Couln't load pickle data, creating new storage)r/ �flagsr' r �select�cr! r �execute�fetchone�closer. r- �IOError�OSError�EOFErrorr Z PickleError�log�debug)r1 r7 �replacer' Z result_proxy�resultr r r �do_openq s*
z DatabaseNamespaceManager.do_openc C s� | j d k rx| j dks | j dkrx| j}| jrT|�� j| j| jt�� t�� d� d| _n$|� |j
j| jk�j| jt�� d� d | _ d S )Nr9 �w)r r! r r F)r! r ) r7 r' r. �insertr: r r- r �now�updater9 )r1 r' r r r �do_close� s z!DatabaseNamespaceManager.do_closec C s. | j }|�|jj| jk��� i | _d| _d S )NT)r' �deleter9 r r: r- r. )r1 r' r r r � do_remove� s z"DatabaseNamespaceManager.do_removec C s
| j | S )N)r- )r1 r5 r r r � __getitem__� s z$DatabaseNamespaceManager.__getitem__c C s
|| j kS )N)r- )r1 r5 r r r � __contains__� s z%DatabaseNamespaceManager.__contains__c C s || j |< d S )N)r- )r1 r5 �valuer r r � __setitem__� s z$DatabaseNamespaceManager.__setitem__c C s | j |= d S )N)r- )r1 r5 r r r � __delitem__� s z$DatabaseNamespaceManager.__delitem__c C s
| j �� S )N)r- �keys)r1 r r r rQ � s z DatabaseNamespaceManager.keys)NNFr NNN)�__name__�
__module__� __qualname__r r# r0 � classmethodr r+ r3 r6 rD rI rK rL rM rO rP rQ r r r r r s
A
r c @ s e Zd ZeZdS )�DatabaseContainerN)rR rS rT r Znamespace_managerr r r r rV � s rV )Zbeaker._compatr Zloggingr Zbeaker.containerr r Zbeaker.exceptionsr r Zbeaker.synchronizationr r Z beaker.utilr
r Z getLoggerrR r@ r r r
r rV r r r r �<module> s
!
Mini Shell