Current File : //opt/alt/python37/lib/python3.7/site-packages/beaker/ext/__pycache__/mongodb.cpython-37.pyc
B
��[? � @ s� d dl Z d dlZd dlZd dlZd dlZy d dlZd dlZd dlZW n ek
r` dZdZY nX d dl m
Z
d dl m Z d dl
mZmZ d dlmZ d dlmZmZ G dd� de
�ZG d d
� d
e �ZdS ) � N)�NamespaceManager)�SynchronizerImpl)�SyncDict�machine_identifier)�sha1)� string_type�PY2c s� e Zd ZdZdZe� Z� fdd�Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z d dd�Z
dd� Zdd� Zdd� Zdd� Zdd� Z� ZS ) �MongoNamespaceManagera^ Provides the :class:`.NamespaceManager` API over MongoDB.
Provided ``url`` can be both a mongodb connection string or
an already existing MongoClient instance.
The data will be stored into ``beaker_cache`` collection of the
*default database*, so make sure your connection string or
MongoClient point to a default database.
i c s\ t t| ��|� d | _td kr&td��t|t�rFtj� |tj
|�| _ n|| _ | j � � | _
d S )Nzpymongo3 is not available)�superr �__init__Zlock_dir�pymongo� RuntimeError�
isinstancer �clients�get� MongoClient�client�get_default_database�db)�self� namespace�url�kw)� __class__� �C/opt/alt/python37/lib/python3.7/site-packages/beaker/ext/mongodb.pyr $ s
z MongoNamespaceManager.__init__c C sX t |t�s|�d�}t|�| jt| j� d krJts>|�d�}t|�� � }d| j|f S )N�ascii� zutf-8z%s:%s)
r �str�decode�len�MAX_KEY_LENGTHr r �encoder � hexdigest)r �keyr r r � _format_key1 s
z!MongoNamespaceManager._format_keyc C s t | �|�| j�S )N)�MongoSynchronizerr% r )r r$ r r r �get_creation_lock: s z'MongoNamespaceManager.get_creation_lockc C s> | � � | jj�d| �|�i�}|d kr0t|��t�|d �S )N�_id�value)�_clear_expiredr � backer_cache�find_oner% �KeyError�pickle�loads)r r$ �entryr r r � __getitem__= s
z!MongoNamespaceManager.__getitem__c C s( | � � | jj�d| �|�i�}|d k S )Nr( )r* r r+ r, r% )r r$ r0 r r r � __contains__D s z"MongoNamespaceManager.__contains__c C s || kS )Nr )r r$ r r r �has_keyI s z MongoNamespaceManager.has_keyNc C s\ | � � d }|d k r t�� | }t�|�}| jjjd| �|�idt� |�|d�idd� d S )Nr( z$set)r) �
expirationT)�upsert)
r* �timer. �dumpsr r+ �
update_oner% �bson�Binary)r r$ r) Z
expiretimer4 r r r � set_valueL s
z MongoNamespaceManager.set_valuec C s | � ||� d S )N)r; )r r$ r) r r r � __setitem__Y s z!MongoNamespaceManager.__setitem__c C s$ | � � | jj�d| �|�i� d S )Nr( )r* r r+ � delete_manyr% )r r$ r r r � __delitem__\ s z!MongoNamespaceManager.__delitem__c C s | j j�ddd| j ii� d S )Nr( z$regexz^%s)r r+ r= r )r r r r � do_remove` s z MongoNamespaceManager.do_removec C s& dd� | j j�ddd| j ii�D �S )Nc S s g | ]}|d � dd�d �qS )r$ �:r ���)�split)�.0�er r r �
<listcomp>d s z.MongoNamespaceManager.keys.<locals>.<listcomp>r( z$regexz^%s)r r+ Zfind_allr )r r r r �keysc s zMongoNamespaceManager.keysc C s0 t � � }| jj�dd| j id |d�d�� d S )Nz$regexz^%s)z$nez$lte)r( r4 )r6 r r+ r= r )r �nowr r r r* h s z$MongoNamespaceManager._clear_expired)N)�__name__�
__module__� __qualname__�__doc__r! r r r r% r' r1 r2 r3 r; r< r> r? rF r* �
__classcell__r r )r r r s
r c sZ e Zd ZdZdZe� Z� fdd�Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
� ZS )r& a� Provides a Writer/Reader lock based on MongoDB.
Provided ``url`` can be both a mongodb connection string or
an already existing MongoClient instance.
The data will be stored into ``beaker_locks`` collection of the
*default database*, so make sure your connection string or
MongoClient point to a default database.
Locks are identified by local machine, PID and threadid, so
are suitable for use in both local and distributed environments.
i� c sJ t t| ��� || _t|t�r4tj�|t j
|�| _ n|| _ | j � � | _
d S )N)r
r& r �
identifierr r r r r r r r r r )r rM r )r r r r � s
zMongoSynchronizer.__init__c C s: t j �� }|t j| jd� }| jj�| jd|id�� |S )N)�secondsz$lte)r( � timestamp)�datetime�utcnow� timedelta�LOCK_EXPIRATIONr � beaker_locksr= rM )r rG Zexpiredr r r �_clear_expired_locks� s
z&MongoSynchronizer._clear_expired_locksc C s d| j t�� t�� jf S )Nz%s-%s-%s)�
MACHINE_ID�os�getpid� threading�current_thread�ident)r r r r �
_get_owner_id� s z MongoSynchronizer._get_owner_idc C s, | � � }| jj�| j|d�dd|ii� d S )N)r( �readersz$pullr] )r\ r rT r8 rM )r �owner_idr r r �do_release_read_lock� s z&MongoSynchronizer.do_release_read_lockc C st | � � }| �� }x^y.| jjj| jd d�d|id|id�dd� dS tjjk
rj |s\dS t �
d� Y qX qW d S ) N)r( �ownerrO r] )z$setz$pushT)r5 Fg�������?) rU r\ r rT r8 rM r �errors�DuplicateKeyErrorr6 �sleep)r �waitrG r^ r r r �do_acquire_read_lock� s
z&MongoSynchronizer.do_acquire_read_lockc C s | j j�| j| �� d�� d S )N)r( r` )r rT Z
delete_onerM r\ )r r r r �do_release_write_lock� s z'MongoSynchronizer.do_release_write_lockc C sr | � � }| �� }x\y,| jjj| jd g d�d||d�idd� dS tjjk
rh |sZdS t �
d� Y qX qW d S )N)r( r` r] z$set)r` rO T)r5 Fg�������?) rU r\ r rT r8 rM r ra rb r6 rc )r rd rG r^ r r r �do_acquire_write_lock� s z'MongoSynchronizer.do_acquire_write_lock)rH rI rJ rK rS r rV r rU r\ r_ re rf rg rL r r )r r r& n s r& )rP rW rY r6 r. r Zpymongo.errorsr9 � ImportErrorZbeaker.containerr Zbeaker.synchronizationr Z beaker.utilr r Zbeaker.crypto.utilr Zbeaker._compatr r r r& r r r r �<module> s$
X
Mini Shell