a j@sdZddlZddlZddlZddlZddlZddlmZzeWne yVe ZYn0ddZ ddZ dd Z d Zd Zdd d ZdddZdS)z Disk management utilities. N)utilcCsjd}t|dgD]H}ttj||}t|drF||jd7}q||jddd7}qt|dS)z%Return the disk usage in a directory.r. st_blocksig@) oslistdirstatpathjoinhasattrrst_sizeint)r sizefilerr1E:\soccer\ml_env\Lib\site-packages\joblib/disk.py disk_useds rc Csd}t||d|dd}z$t||dt|dd}Wn>ttfy|}z"td|t|f|WYd}~n d}~00|S)z,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 CsFzt|Wn2ty@}z|jtjkr,WYd}~n d}~00dS)zgEnsure directory d exists (like mkdir -p on Unix) No guarantee that the directory is writable. N)rmakedirsOSErrorerrnoEEXIST)drrrrmkdirp5s  r&g? cCsng}zt|}Wn2tjyD|dur>|tj|tnYn0|D]}tj||}t||dqJdS)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_subdirsHsr0Tc Cstj|r|dur$t|d|nd}t|}zNt|dksD|rjtj|dddtd |Wqnt d |t|Wn(t t fy|d7}|t krYn0t tq(dS)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 isdirshutilrmtreerlenrdebugformatr" WindowsErrorRM_SUBDIRS_N_RETRYtimesleepRM_SUBDIRS_RETRY_TIME)Z folder_pathr(Zallow_non_emptyZ err_countfilesrrrr,fs(  r,)N)NT)__doc__r#rr3r*r:multiprocessingrr8 NameErrorr"rr r&r<r9r0r,rrrrs"