B Wo$c%@sdZddlZddlZddlZddlZddlZddlmZyeWne k rXe ZYnXddZ ddZ dd Z d Zd Zdd d ZdddZdS)z Disk management utilities. N)utilcCsnd}x\t|dgD]H}ttj||}t|drH||jd7}q||jddd7}qWt|dS)z& Return the disk usage in a directory.r. st_blocksig@) oslistdirstatpathjoinhasattrrst_sizeint)r sizefilerr*lib/python3.7/site-packages/joblib/disk.py disk_useds rc Csd}t||d|dd}y$t||dt|dd}Wn<ttfk rz}ztd|t|f|Wdd}~XYnX|S)z2 Convert a memory text to its value in bytes. i)KMGNzQInvalid literal for size give: %s (type %s) should be alike '10G', '500M', '50K'.)dictr floatKeyError ValueErrortype)textZkiloZunitsrerrrmemstr_to_bytes*s$$r c CsDyt|Wn0tk r>}z|jtjkr.Wdd}~XYnXdS)zgEnsure directory d exists (like mkdir -p on Unix) No guarantee that the directory is writable. N)rmakedirsOSErrorerrnoZEEXIST)drrrrmkdirp8s  r%g? cCstg}yt|}Wn4tjk rF|dk r@|tj|tnYnXx&|D]}tj||}t||dqNWdS)aRemove all subdirectories in this path. The directory indicated by `path` is left in place, and its subdirectories are erased. If onerror is set, it is called to handle the error with arguments (func, path, exc_info) where func is os.listdir, os.remove, or os.rmdir; path is the argument to that function that caused it to fail; and exc_info is a tuple returned by sys.exc_info(). If onerror is None, an exception is raised. N)onerror)rrerrorsysexc_infor r delete_folder)r r'namesnamefullnamerrr rm_subdirsKs r/Tc Cstj|r|dk r$t|d|nd}xt|}yLt|dksF|rjtj|dddtd |Pnt d |t|Wn*t t fk r|d7}|t krYnXt tq*WdS)zBUtility function to cleanup a temporary folder if it still exists.NFr) ignore_errorsr'zSuccessfully deleted {}z*Expected empty folder {} but got {} files.r)rr isdirshutilZrmtreerlenrdebugformatr" WindowsErrorRM_SUBDIRS_N_RETRYtimeZsleepRM_SUBDIRS_RETRY_TIME)Z folder_pathr'Zallow_non_emptyZ err_countfilesrrrr+is*    r+)N)NT)__doc__rr)r8r#r2Zmultiprocessingrr6 NameErrorr"rr r%r9r7r/r+rrrrs"