B °F.\‚ã@s®ddlmZmZmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZe e¡ZdZdd„Zd d „ZGd d „d eƒZGd d„deƒZdS)é)Úprint_functionÚdivisionÚabsolute_importNé)Úlocket)Úfinalizez.dirlockcCs tj d¡S)Nz#distributed.worker.use-file-locking)ÚdaskZconfigÚget©r r ú4lib/python3.7/site-packages/distributed/diskutils.pyÚis_locking_enabledsr c CsRyt |¡Wn>tk rL}z |jtjkrtj |¡|_| ¡tj |jd¡|_tj |jd¡|_dS)Nz global.lockz purge.lock)r rÚabspathrÚ_init_workspacer Ú_global_lock_pathÚ_purge_lock_path)r0rr r r r3mszWorkSpace.__init__c CsFyt |j¡Wn0tk r@}z|jtjkr0‚Wdd}~XYnXdS)N)r r!rrrZEEXIST)r0rr r r r@ss  zWorkSpace._init_workspacecKstj|jf|ŽS)N)rr&rA)r0Úkwargsr r r r%zszWorkSpace._global_lockcKstj|jf|ŽS)N)rr&rB)r0rCr r r Ú _purge_lock}szWorkSpace._purge_lockcCsÀtƒs gS|jdd}y | ¡Wntjk r8gSXzt| ¡ƒ}Wd| ¡Xg}|jdd}y | ¡Wntjk rˆYn4Xz&x |D]}|  |¡r’|  |¡q’WWd| ¡X|S)Nr)Útimeout) r r%r'rÚ LockErrorÚlistÚ_list_unknown_locksr4rDÚ_check_lock_or_purgeÚappend)r0ÚlockZ candidatesZpurgedrr r r Ú_purge_leftovers€s,        zWorkSpace._purge_leftoversc cs^xXt tj |jdt¡¡D]:}yt |¡}Wntk rBYqXt |j ¡r|VqWdS)NÚ*) Úglobr rr rr"ÚstatrÚS_ISREGÚst_mode)r0ÚpÚstr r r rH§s  zWorkSpace._list_unknown_lockscCstj||jddS)N)Úonerror)r*r+Ú_on_remove_error)r0rr r r r5³szWorkSpace._purge_directorycCsª| t¡st‚||jkrdSt d|¡tj|dd}y | ¡Wntj k rXdSXz8|dt tƒ …}t j   |¡rt d|¡| |¡Wd| ¡Xt|ƒ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...r)rENz/Found stale lock file and directory %r, purgingT)Úendswithr"rr,rr$rr&r'rFÚlenr rr#Úinfor5r4r)r0r8rKrr r r rI¶s$     zWorkSpace._check_lock_or_purgecCs"|\}}}t d||t|ƒ¡dS)Nz&Failed to remove %r (failed in %r): %s)rrr)r0ÚfuncrrÚtypÚexcÚtbr r r rUÕs zWorkSpace._on_remove_errorcKs<y | ¡Wn"tk r.tjdddYnXt|f|Ž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 zbs' r>)Z __future__rrrrrNZloggingr r*rOrrÚrZ compatibilityrZ getLoggerr9rr"r rÚobjectrr>r r r r Ús     ?