a &j=@sddlmZddlZddlZddlZddlZddlmZddlZddl Z ddl m Z ddl m Z ddl m Z ddlmZddlmZdd lmZeeZd d ZGd d d ZGdddZe eefdddZe eefdddZe eefdddZdS))contextmanagerN)Path)Callable)Optional)Union)_utils)_interprocess_mechanism)%_interprocess_reader_writer_mechanismc Cszt|Wnltyz}zT|jtjkrJtj|s:qfWYd}~dSn|jtjkrdWYd}~dSWYd}~nd}~00dSdS)zfCreate a directory (and any ancestor directories required). :param path: Directory to create NFT)osmakedirsOSErrorerrnoEEXISTpathisdirZEISDIR)rer|rLWYd}~dStnt d|j |dWYd}~nd}~00dSdS)NFz9Unable to acquire lock on `%(path)s` due to %(exception)s)r exceptionT) trylockIOErrorr r EACCESEAGAINexpiredr RetryAgain threading ThreadErrorr)r!blockingwatchrrrr _try_acquireSs   zInterProcessLock._try_acquirecCsTtj|j}|r0t|}|r0|jtjd||jdusB|jj rPt |jd|_dS)NCreated lock base path `%s`a) r rdirnamerrlogrBLATHERrclosedopenr!ZbasedirZ made_basedirrrr_do_opengs zInterProcessLock._do_openTr,delay max_delaytimeoutreturncCs|dkrtd|dur(|dkr(td||kr4|}|tj|d}tj|||j|d}|||j||}Wdn1s0Y|sdSd|_|j tj d |j | |j dSdS) a,Attempt to acquire the lock. Args: blocking: Whether to wait to try to acquire the lock. delay: When `blocking`, starting delay as well as the delay increment (in seconds). max_delay: When `blocking` the maximum delay in between attempts to acquire (in seconds). timeout: When `blocking`, maximal waiting time (in seconds). Returns: whether or not the acquisition succeeded r+Delay must be greater than or equal to zeroN-Timeout must be greater than or equal to zerodurationrr-FTHAcquired file lock `%s` after waiting %0.3fs [%s attempts were required]) ValueErrorr7r StopWatchRetryrr.rrr2r3relapsedattempts)r!r,r9r:r;r-rgottenrrracquirets, ,  zInterProcessLock.acquirecCs|jdur|jd|_dSN)rcloser!rrr _do_closes  zInterProcessLock._do_closecCs |}|std|j|S)NFUnable to acquire a file lock on `%s` (when used as a context manager))rJr*r+r)r!rIrrr __enter__s zInterProcessLock.__enter__c Cs|jstdz |WnFtyb}z.d|jf}|j|t||WYd}~nTd}~00d|_z |Wn"t y|jd|jYn0|j t j d|jdS)z%Release the previously acquired lock.z#Unable to release an unaquired lock1Could not unlock the acquired lock opened on `%s`NF.Could not close the file handle opened on `%s`*Unlocked and closed file lock open on `%s`) rr*r+unlock Exceptionrrr#rNr%r2rr3)r!rmsgrrrreleases&    "    zInterProcessLock.releasecCs |dSrK)rW)r!exc_typeexc_valexc_tbrrr__exit__szInterProcessLock.__exit__cCstj|jSrK)r rexistsrMrrrr\szInterProcessLock.existscCst|jdSrK)rr$rrMrrrr$szInterProcessLock.trylockcCst|jdSrK)rrTrrMrrrrTszInterProcessLock.unlock)TrrN)__name__ __module__ __qualname____doc__ MAX_DELAYDELAY_INCREMENTtimesleeprrstrrfloatrloggingLoggerr"r.r7boolrJrNrPrWr[r\r$rTrrrrr:s:    , rc@seZdZdZdZdZejdfee e fe e gdfe ejdddZeddd Zedd d Zd d ZddZd ee e e edddZd!ee e e edddZd"ddZddZddZddZdS)#InterProcessReaderWriterLockz$An interprocess readers writer lock.rrNrcCs*d|_t||_||_t|t|_dS)a Args: path: Path to the file that will be used for locking. sleep_func: Optional function to use for sleeping. logger: Optional logger to use for logging. N)rrrrrrrrr rrrr"s  z%InterProcessReaderWriterLock.__init__ccs4|jd||ddzdVW|n |0dS)z&Context manager that grans a read lockTNr,r9r:r;)acquire_read_lockrelease_read_lock)r!r9r:rrr read_locks z&InterProcessReaderWriterLock.read_lockccsH|jd||dd}|s&td|jzdVW|n |0dS)z'Context manager that grans a write lockTNrkrO)acquire_write_lockr*r+rrelease_write_lock)r!r9r:rIrrr write_locksz'InterProcessReaderWriterLock.write_lockc Cspzt|j|}Wn8tyJ}z td|j|WYd}~n d}~00|rTdS|r`|rddSt dS)Nz'Unable to acquire lock on {} due to {}!TF) r r$rrUr*r+formatrr(rr))r!r,r- exclusiverIrrrrr.s  z)InterProcessReaderWriterLock._try_acquirecCsLtj|j}|r0t|}|r0|jtjd||jdurHt |j|_dS)Nr/) r rr1rrr2rr3rr Z get_handler6rrrr7s  z%InterProcessReaderWriterLock._do_openTr8cCs|j||||ddS)a3Attempt to acquire a reader's lock. Args: blocking: Whether to wait to try to acquire the lock. delay: When `blocking`, starting delay as well as the delay increment (in seconds). max_delay: When `blocking` the maximum delay in between attempts to acquire (in seconds). timeout: When `blocking`, maximal waiting time (in seconds). Returns: whether or not the acquisition succeeded Frs_acquirer!r,r9r:r;rrrrlsz.InterProcessReaderWriterLock.acquire_read_lockcCs|j||||ddS)a3Attempt to acquire a writer's lock. Args: blocking: Whether to wait to try to acquire the lock. delay: When `blocking`, starting delay as well as the delay increment (in seconds). max_delay: When `blocking` the maximum delay in between attempts to acquire (in seconds). timeout: When `blocking`, maximal waiting time (in seconds). Returns: whether or not the acquisition succeeded Trtrurwrrrro6sz/InterProcessReaderWriterLock.acquire_write_lockc Cs|dkrtd|dur(|dkr(td||kr4|}|tj|d}tj|||j|d}| ||j|||}Wdn1s0Y|sdS|jtj d|j | |j dSdS) Nrr=r>r?rAFrBT) rCr7rrDrErr.rr2r3rrFrG) r!r,r9r:r;rsr-rHrIrrrrvNs* .  z%InterProcessReaderWriterLock._acquirecCs |jdurt|jd|_dSrK)rr close_handlerMrrrrNgs  z&InterProcessReaderWriterLock._do_closecCs|zt|jWn"ty2|jd|jYnF0z |Wn"tyb|jd|jYn0|jt j d|jdS)zRelease the writer's lock.rQrRrSN r rTrr%rr#rrNr2rr3rMrrrrpls       z/InterProcessReaderWriterLock.release_write_lockcCs|zt|jWn"ty2|jd|jYnF0z |Wn"tyb|jd|jYn0|jt j d|jdS)zRelease the reader's lock.rQrRrSNryrMrrrrm~s       z.InterProcessReaderWriterLock.release_read_lock)rr)rr)TrrN)TrrN)TrrNT)r]r^r_r`rarbrcrdrrrerrfrrgrhr"rrnrqr.r7rirlrorvrNrprmrrrrrjsV         rj)rcst|fdd}|S)zAcquires & releases an interprocess **write** lock around the call into the decorated function Args: path: Path to the file used for locking. cstfdd}|S)Ncs:|i|WdS1s,0YdSrK)rqargskwargsflockrrwrappers z=interprocess_write_locked..decorator..wrapper functoolswrapsr~rrr~r decoratorsz,interprocess_write_locked..decoratorrjrrrrrinterprocess_write_lockeds rcst|fdd}|S)zAcquires & releases an interprocess **read** lock around the call into the decorated function Args: path: Path to the file used for locking. cstfdd}|S)Ncs:|i|WdS1s,0YdSrK)rnrzr}rrrs z.decorator..wrapperrrrrrrsz+interprocess_read_locked..decoratorrrrrrinterprocess_read_lockeds rcst|fdd}|S)zAcquires & releases an interprocess lock around the call to the decorated function. Args: path: Path to the file used for locking. cstfdd}|S)Ncs6|i|WdS1s(0YdSrKrrzr}rrrsz7interprocess_locked..decorator..wrapperrrrrrrsz&interprocess_locked..decorator)rrrrrinterprocess_lockeds r) contextlibrr rrgr pathlibrr*rctypingrrrZ fastenersrZfasteners.process_mechanismrr getLoggerr]rrrrjrerrrrrrrs,         A