a 2`/bw@sddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl Z ddl m Z eeZdZddZd d ZGd d d ZGd ddZdS)) annotationsN)ClassVar)locketz.dirlockcCs tjdS)Nz#distributed.worker.use-file-locking)daskZconfiggetrr4lib/python3.9/site-packages/distributed/diskutils.pyis_locking_enabledsr c Cs^zt|WnJty Yn:tyX}z"td|d|WYd}~n d}~00dS)NzFailed to remove z: )osunlinkFileNotFoundErrorOSErrorloggererror)patherrr safe_unlinks  rc@sHeZdZUdZded<ded<ded<ddd Zd d Zed d ZdS)WorkDirz8 A temporary work directory inside a WorkSpace. strdir_path _lock_pathzweakref.finalize _finalizerNcCsr|dus|dusJ|dur0tj||jd|_ntj|j||_t|jtrTztj|jt |_ tj |j r~Jt d|j zD|(t|j |_|jWdn1s0YWn*tyt jd|j ddd|_Yn0Wn&ty$tj|jddYn0|j|j t||j||j |j|j|_nt||j|dd|j|_dS)N)prefixdirz Locking %r...zyCould not acquire workspace lock on path: %s .Continuing without lock. This may result in workspaces not being cleaned upTexc_info) ignore_errors)tempfileZmkdtempbase_dirrr rjoinmkdirr DIR_LOCK_EXTrexistsrdebug _global_lockr lock_fileZ _lock_fileacquirer exception Exceptionshutilrmtree _known_locksaddweakreffinalize _finalizer)self workspacenamerrrr __init__,sH  ,   zWorkDir.__init__cCs |dS)z, Dispose of this directory. N)rr1rrr release]szWorkDir.releasec Csnz:||W|dur||durj|j|t|n.|durL||durh|j|t|0dSN)_purge_directoryr6r,remover)clsr2 lock_pathr&rrrr r0cs    zWorkDir._finalize)NN) __name__ __module__ __qualname____doc____annotations__r4r6 classmethodr0rrrr r#s  1rc@speZdZUdZeZded<ddZddZdd Z d d Z d d Z ddZ ddZ ddZddZddZdS) WorkSpacez An on-disk workspace that tracks disposable work directories inside it. If a process crashes or another event left stale directories behind, this will be detected and the directories purged. zClassVar[set[str]]r,cCs>tj||_|tj|jd|_tj|jd|_dS)Nz global.lockz purge.lock)r rabspathr_init_workspacer _global_lock_path_purge_lock_path)r1rrrr r4zszWorkSpace.__init__cCs(zt|jWnty"Yn0dSr7)r r!rFileExistsErrorr5rrr rDs zWorkSpace._init_workspacecKstj|jfi|Sr7)rr&rEr1kwargsrrr r%szWorkSpace._global_lockcKstj|jfi|Sr7)rr&rFrHrrr _purge_lockszWorkSpace._purge_lockcCsts gS|jdd}z |Wntjy:gYS0zt|}W|n |0g}|jdd}z |WntjyYn80z*|D]}| |r| |qW|n |0|S)Nrtimeout) r r%r'r LockErrorlist_list_unknown_locksr6rJ_check_lock_or_purgeappend)r1lockZ candidatesZpurgedrrrr _purge_leftoverss,      zWorkSpace._purge_leftoversc csXttj|jdtD]8}zt|}Wnty>Yq0t|j r|VqdS)N*) globr rr rr"statrS_ISREGst_mode)r1pstrrr rOs  zWorkSpace._list_unknown_lockscCstj||jddS)N)onerror)r*r+_on_remove_error)r1rrrr r8szWorkSpace._purge_directorycCs|tsJ||jvrdStd|tj|dd}z |WntjyXYdS0z@|dt t }t j |rt d|||W|n |0t|dS)z Try locking the given path, if it fails it's in use, otherwise the corresponding directory is deleted. Return True if the lock was stale. FzChecking lock file %r...rrKNz/Found stale lock file and directory %r, purgingT)endswithr"r,rr$rr&r'rMlenr rr#infor8r6r)r1r;rRrrrr rPs"      zWorkSpace._check_lock_or_purgecCs"|\}}}td||t|dS)Nz&Failed to remove %r (failed in %r): %s)rrr)r1funcrrtypexctbrrr r\s zWorkSpace._on_remove_errorcKs>z |Wn ty,tjdddYn0t|fi|S)a Create and return a new WorkDir in this WorkSpace. Either the *prefix* or *name* parameter should be given (*prefix* is preferred as it avoids potential collisions) Parameters ---------- prefix : str (optional) The prefix of the temporary subdirectory name for the workdir name : str (optional) The subdirectory name for the workdir zr?setr,r@r4rDr%rJrSrOr8rPr\rdrrrr rBos ' rB)Z __future__rrUloggingr r*rVrr.typingrrrZ getLoggerr<rr"r rrrBrrrr s     L