3 �Pf��@srddlmZmZddlZddlZyeWnek r@eZYnXddlm Z m Z mZmZm Z Gdd�de �ZdS)�)�absolute_import�divisionN�)�LockBase� NotLocked� NotMyLock�LockTimeout� AlreadyLockedc@sPeZdZdZdZddd�Zddd�Zdd �Zd d�Zdd �Z dd�Z dd�ZdS)�SQLiteLockFilezDemonstrate SQL-based locking.NTc Cs�tj||||�t|j�|_t|j�|_tjdkrdddl}|j�\}}t j |�t j|�~~|t_ddl}|j tj�|_|jj�}y|jd�Wn|jk r�Yn$X|jj�ddl} | jt jtj�dS)zu >>> lock = SQLiteLockFile('somefile') >>> lock = SQLiteLockFile('somefile', threaded=False) NrzGcreate table locks( lock_file varchar(32), unique_name varchar(32)))r�__init__�unicode� lock_file�unique_namer �testdb�tempfileZmkstemp�os�close�unlink�sqlite3Zconnect� connection�cursor�executeZOperationalError�commit�atexit�register) �self�pathZthreaded�timeoutrZ_fdrr�cr�r�$/usr/lib/python3.6/sqlitelockfile.pyrs( zSQLiteLockFile.__init__cCsH|dk r|n|j}tj�}|dk r2|dkr2||7}|dkr@d}n|dkrNd}n|d}|jj�}x�|j�s�|jd|j|jf�|jj�|jd|jf�|j �}t |�dkr�|jd|jf�|jj�q�dSn(|jd|jf�|j �}t |�dkr�dS|dk �r6tj�|k�r6|dk�r(td|j��nt d |j��tj|�qbWdS) Nrg��?� z;insert into locks (lock_file, unique_name) values (?, ?)z*select * from locks where unique_name = ?rz(delete from locks where unique_name = ?z&Timeout waiting to acquire lock for %sz%s is already locked)r�timerr� is_lockedrr rr�fetchall�lenrrr Zsleep)rrZend_time�waitr�rowsrrr �acquire5sD zSQLiteLockFile.acquirecCs\|j�std|j��|j�s4td|j|j�f��|jj�}|j d|jf�|jj �dS)Nz%s is not lockedz#%s is locked, but not by me (by %s)z(delete from locks where unique_name = ?)r#rr�i_am_lockingrr�_who_is_lockingrrrr)rrrrr �releasets zSQLiteLockFile.releasecCs&|jj�}|jd|jf�|j�dS)Nz2select unique_name from locks where lock_file = ?r)rrrr Zfetchone)rrrrr r*�s zSQLiteLockFile._who_is_lockingcCs*|jj�}|jd|jf�|j�}|S)Nz(select * from locks where lock_file = ?)rrrr r$)rrr'rrr r#�s zSQLiteLockFile.is_lockedcCs*|jj�}|jd|j|jf�|j�S)Nz?select * from locks where lock_file = ? and unique_name = ?)rrrr rr$)rrrrr r)�s zSQLiteLockFile.i_am_lockingcCs(|jj�}|jd|jf�|jj�dS)Nz&delete from locks where lock_file = ?)rrrr r)rrrrr � break_lock�s zSQLiteLockFile.break_lock)TN)N)�__name__� __module__�__qualname__�__doc__rrr(r+r*r#r)r,rrrr r s " ?r )Z __future__rrr"rr� NameError�str�rrrrr r rrrr �s