a h@s*ddlZddlZddlZGdddZdS)Nc@s2eZdZdZd ddZddZdd Zd d ZdS) FileBatonz0A primitive, file-based synchronization utility.皙?NcCs||_||_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 R/var/www/html/assistant/venv/lib/python3.9/site-packages/torch/utils/file_baton.py__init__ s zFileBaton.__init__cCs8zt|jtjtjB|_WdSty2YdS0dS)z Try to atomically create a file under exclusive access. Returns: True if the file could be created, else False. TFN)osopenrO_CREATO_EXCLrFileExistsErrorrr r r try_acquires  zFileBaton.try_acquirecCsld}t}tj|jrht|j|jdur t||jkr |s t d|jd|jdd}q dS)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.T) timer pathexistsrsleeprrwarningswarn)rZ has_warned start_timer r r wait's   zFileBaton.waitcCs&|jdurt|jt|jdS)z'Release the baton and removes its file.N)rr closeremoverrr r r release:s  zFileBaton.release)rN)__name__ __module__ __qualname____doc__r rrrr r r r rs   r)r rrrr r r r s