`idZddlmZddlZddlZddlZddlmZmZej dkrddl Z nddl Z GddZ dS)z3Simple cross-platform advisory file lock utilities.) annotationsN)AnyLiteralwin32cDeZdZdZddZddZdd ZddZdddZddZ dS)FileLockaProvide a cross-platform file locking mechanism using Python's stdlib. This class implements an advisory lock, which must be respected by all cooperating processes. Please note that this lock does not prevent the same process from acquiring the lock multiple times; it is the caller's responsibility to manage this. Examples -------- .. code-block:: python from tvm_ffi.utils import FileLock with FileLock("/tmp/my.lock"): # Critical section guarded by the lock. # Other processes attempting to acquire the same lock will block # (or fail, if using ``acquire()``) until this context exits. do_work() lock_file_pathstrreturnNonec"||_d|_dS)z6Initialize a file lock using the given lock file path.N)r _file_descriptor)selfr s j/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/tvm_ffi/utils/lockfile.py__init__zFileLock.__init__7s,,0c.||S)zlAcquire the lock upon entering the context. This method blocks until the lock is acquired. )blocking_acquirers r __enter__zFileLock.__enter__<s  rexc_typerexc_valexc_tbLiteral[False]c.|dS)zIContext manager protocol: release the lock upon exiting the 'with' block.F)release)rrrrs r__exit__zFileLock.__exit__Ds urboolc |jdStjdkritj|jtjtjztjz|_tj |jtj dngtj|jtj tjz|_tj|jtjtjzdS#t"t$f$r+|j tj|jd|_YdSt($r>}|j tj|jd|_t+d|d}~wwxYw)a$Acquire an exclusive, non-blocking lock on the file. Returns ------- ret: bool True if the lock was acquired, False otherwise. Raises ------ RuntimeError If an unexpected error occurs during lock acquisition. NFrTzAn unexpected error occurred: )rsysplatformosopenr O_RDWRO_CREATO_BINARYmsvcrtlockingLK_NBLCKO_WRONLYfcntlflockLOCK_EXLOCK_NBOSErrorBlockingIOErrorclose Exception RuntimeError)res racquirezFileLock.acquireIsR E$0u|w&&(*'RZ)?"+)M))%t4foqIIII(*0CR[SUS]E](^(^% D15=5=3PQQQ4)   $0.///(,%55 E E E$0.///(,%CCCDD D  Es#C-C C--8E/( E/19E**E/N皙?timeout float | None poll_intervalfloatc|jtdtj} |rdS|5tj|z |krt d|jd|dtj|b)aWait until an exclusive lock can be acquired, with an optional timeout. Parameters ---------- timeout: float, optional The maximum time to wait for the lock in seconds. A value of None means wait indefinitely. poll_interval: float The time to wait between lock attempts in seconds. Returns ------- ret: bool True if the lock was acquired. Raises ------ TimeoutError If the lock is not acquired within the timeout period. RuntimeError If the lock is already held by this instance. Nz&Lock is already held by this instance.TzFailed to acquire lock on 'z' after z seconds.)rr4timer6 TimeoutErrorr sleep)rr8r: start_times rrzFileLock.blocking_acquirens.  ,GHH HY[[  &||~~ t" j(@G'K'K"a$2Eaawaaa J} % % % &rc |j|tjdkr&tj|jtjdn$t j|jt jtj |jd|_dSdS)z1Releases the lock and closes the file descriptor.Nrr ) rr!r"r(r)LK_UNLCKr,r-LOCK_UNr#r2rs rrzFileLock.releasest  ,|w&&t4foqIIII D15=AAA HT* + + +$(D ! ! ! - ,r)r r r r )r r)rrrrrrr r)r r)Nr7)r8r9r:r;r r)r r ) __name__ __module__ __qualname____doc__rrrr6rrrrrr!s*1111  #E#E#E#EJ$&$&$&$&$&L))))))rr) rG __future__rr#r!r=typingrrr"r(r,rrHrrrKs":9"""""" <7MMMMLLL{){){){){){){){){){)r