o i\@s*ddlZddlZddlZGdddZdS)Nc@sBeZdZdZd dddZdedBfddZdd d Zdd d ZdS) FileBatonz0A primitive, file-based synchronization utility.皙?NreturncCs||_||_d|_||_dS)an Create a new :class:`FileBaton`. Args: lock_file_path: The path to the file used for locking. wait_seconds: The seconds to periodically sleep (spin) when calling ``wait()``. warn_after_seconds: The seconds to wait before showing lock file path to warn existing lock file. N)lock_file_path wait_secondsfdwarn_after_seconds)selfrrrr X/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/torch/utils/file_baton.py__init__ s  zFileBaton.__init__cCs4zt|jtjtjB|_WdStyYdSw)z Try to atomically create a file under exclusive access. Returns: True if the file could be created, else False. TF)osopenrO_CREATO_EXCLrFileExistsErrorr r r r try_acquires  zFileBaton.try_acquirecCsd}t}tj|jr>t|j|jdur5t||jkr5|s5tj d|jd|jdddd}tj|js dSdS) z Periodically sleeps for a certain amount until the baton is released. The amount of time slept depends on the ``wait_seconds`` parameter passed to the constructor. FNzWaited on lock file "z" for z seconds.) stacklevelT) timer pathexistsrsleeprrwarningswarn)r has_warned start_timer r r wait(s  zFileBaton.waitcCs&|jdur t|jt|jdS)z'Release the baton and removes its file.N)rr closeremoverrr r r release;s  zFileBaton.release)rN)rN) __name__ __module__ __qualname____doc__r boolrrr!r r r r rs   r)r rrrr r r r s