U
UV @ sr d dl mZmZ d dlZd dlZze W n ek
r@ eZY nX ddlm Z m
Z
m Z m Z m
Z
G dd de ZdS ) )absolute_importdivisionN )LockBase NotLocked NotMyLock LockTimeout
AlreadyLockedc @ sP e Zd ZdZdZdddZdddZdd Zd
d Zd d
Z dd Z
dd Z dS )SQLiteLockFilez Demonstrate SQL-based locking.NTc
C s t | ||| t| j| _t| j| _tjdkrdddl}| \}}t
| t | ~~|t_ddl }|
tj| _| j }z|d W n |jk
r Y n$X | j ddl} | t j tj dS )zu
>>> lock = SQLiteLockFile('somefile')
>>> lock = SQLiteLockFile('somefile', threaded=False)
Nr zGcreate table locks( lock_file varchar(32), unique_name varchar(32)))r __init__unicode lock_file unique_namer
testdbtempfileZmkstemposcloseunlinksqlite3Zconnect
connectioncursorexecuteZOperationalErrorcommitatexitregister)
selfpathZthreadedtimeoutr Z_fdr r cr r /sqlitelockfile.pyr s(
zSQLiteLockFile.__init__c C sD |d k r |n| j }t }|d k r2|dkr2||7 }|d kr@d}n|dkrNd}n|d }| j }| s|d| j| jf | j |d| jf | }t
|dkr|d| jf | j qd S n(|d| jf | }t
|dkrd S |d k r4t |kr4|dkr&t d| j nt
d | j t| q`d S )
Nr g?
z;insert into locks (lock_file, unique_name) values (?, ?)z*select * from locks where unique_name = ?r (delete from locks where unique_name = ?z&Timeout waiting to acquire lock for %sz%s is already locked)r timer r is_lockedr r
r r fetchalllenr r r sleep)r r Zend_timewaitr rowsr r r acquire5 sL
zSQLiteLockFile.acquirec C s\ | std| j | s4td| j| f | j }| d| jf | j
d S )Nz%s is not lockedz#%s is locked, but not by me (by %s)r" ) r$ r r i_am_lockingr r _who_is_lockingr r r r r r r r r releaset s
zSQLiteLockFile.releasec C s& | j }|d| jf | d S )Nz2select unique_name from locks where lock_file = ?r )r r r r
Zfetchoner- r r r r, s
z SQLiteLockFile._who_is_lockingc C s* | j }|d| jf | }| S )Nz(select * from locks where lock_file = ?)r r r r
r% )r r r) r r r r$ s
zSQLiteLockFile.is_lockedc C s* | j }|d| j| jf | S )Nz?select * from locks where lock_file = ? and unique_name = ?)r r r r
r r% r- r r r r+ s
zSQLiteLockFile.i_am_lockingc C s( | j }|d| jf | j d S )Nz&delete from locks where lock_file = ?)r r r r
r r- r r r
break_lock s
zSQLiteLockFile.break_lock)TN)N) __name__
__module__ __qualname____doc__r r r* r. r, r$ r+ r/ r r r r r
s
"
? r
)Z
__future__r r r# r r NameErrorstr r r r r r r
r r r r <module> s