o wOi@s6ddlZddlZddlZeeZGdddeZdS)Nc@seZdZdZddZeddZejddZeddZdd d Z ddZ ddZ ddZ d ddZ eddZddZddZddZdS)!LockzM Locking recipe for etcd, inspired by the kazoo recipe for zookeeper cCsJ||_||_tj|_d|j||_d|_ d|_ t d|j|jdS)Nz{}/{}Fz#Initiating lock for %s with uuid %s) clientnameuuiduuid4hex_uuidformat lock_prefixpathis_taken _sequence_logdebug)selfr lock_namer=/home/ubuntu/.local/lib/python3.10/site-packages/etcd/lock.py__init__ s z Lock.__init__cCs|jS)z+ The unique id of the lock )rrrrrrsz Lock.uuidcCs0|j}||_|std||_tddS)Nz)The hand-set uuid was not found, refusingzInexistent UUID)r _find_lockrwarn ValueError)rvalueold_uuidrrrr s cCsP|js tddSz |j|jWdStjy'tdd|_YdSw)z2 tells us if the lock is acquired zLock not takenFTz0Lock was supposedly taken, but we cannot find it) r rrrreadlock_keyetcdEtcdKeyNotFoundrrrrr is_acquired)s  zLock.is_acquiredTrcCsv|s'td|j|jj|j|j|dd}||jtd|j|j n |r4|jj|j |j|d|j ||dS)a Acquire the lock. :param blocking Block until the lock is obtained, or timeout is reached :param lock_ttl The duration of the lock we acquired, set to None for eternal locks :param timeout The time to wait before giving up on getting a lock z Lock not found, writing it to %sT)ttlappendz#Lock key %s written, sequence is %s)r!blockingtimeout) rrrr rwriter _set_sequencekeyr r _acquired)rr$lock_ttlr%resrrracquire9s  z Lock.acquirecCst|js|z.ztd|j|j|jWntjy)t d|jYnwWd|_ dSWd|_ dSd|_ w)z" Release the lock zReleasing existing lock %sz%Lock %s not found, nothing to releaseFN) r rrrrrdeleterrinfor rrrrreleaseNs z Lock.releasecCs|jddd|S)z: You can use the lock as a contextmanager TN)r$r*)r,rrrr __enter__]szLock.__enter__cCs |dS)NF)r/)rtyper tracebackrrr__exit__dsz Lock.__exit__c Cs|\}}d|_|j|krtdd|_dSd|_|sdS|j}td|td|} z|jj|||j dd}td|j|j |j d|d WSt j ybtd ||j d|d YSt jyp}z|d}~wt jy~td Ynwq.) NFzLock acquired!Tz"Lock not acquired, now watching %sr)r%indexzDetected variation for %s: %sr#z%Key %s not present anymore, moving onzUnexpected exception) _get_lockerr rrrr(maxrwatch modifiedIndexactionr)rrEtcdLockExpired EtcdException exception) rr$r%lockernearest watch_keytrerrrr)hs6      zLock._acquiredcCs"|jstd|jdt|jS)NzNo sequence present./)r rr strrrrrrsz Lock.lock_keycCs||jdd|_dS)NrD)replacer lstripr )rr(rrrr'szLock._set_sequencecCs|jrz|j|j}|j|_WdStjyYdSw|jrLz |jj|jddj D]}|j|jkr=| |j WdSq+WdStjyKYdSwdS)NTF recursive) r rrrrrrrr leavesr'r()rr+rBrrrrs,  zLock._find_lockcsdd|jj|jddjD}|js|tdd|Dtdz4 |j dkrsz$Lock._get_locker..TrIcSsg|]}|jqSrr()rLrBrrrrMszLock keys found: %srz(No key before our one, we are the lockerzLocker: %s, key to watch: %sr4c3s&|]}|jdkr|VqdS)r4NrN)rLxilrr s$z#Lock._get_locker..zLock not found)rrr rKr rsortedrrr5rnextrrr;)rresultsrrPrr6s"   "  zLock._get_lockerN)Tr r)Tr)__name__ __module__ __qualname____doc__rpropertyrsetterrr,r/r0r3r)rr'rr6rrrrrs&       r)loggingrr getLoggerrWrobjectrrrrrs