o lh@sdZddlZddlZddlZddlZddlZddlZzeWn ey)e ZYnwddZ ddZ ddZ d Z d Zdd d Zdd dZdS)z Disk management utilities. NcCsjd}t|dgD]$}ttj||}t|dr#||jd7}q ||jddd7}q t|dS)z& Return the disk usage in a directory.r. st_blocksig@) oslistdirstatpathjoinhasattrrst_sizeint)rsizefilerr=/var/www/html/env/lib/python3.10/site-packages/joblib/disk.py disk_useds  rc Cshd}t||d|dd}zt||dt|dd}W|Sttfy3td|t|fw)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)textkilounitsr rrrmemstr_to_bytes)s" r c CsFzt|WdSty"}z|jtjkrWYd}~dSd}~ww)zgEnsure directory d exists (like mkdir -p on Unix) No guarantee that the directory is writable. N)rmakedirsOSErrorerrnoEEXIST)derrrmkdirp7s r'g?cCsng}zt|}Wntjy"|dur|tj|tnYnw|D]}tj||}t||dq%dS)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_inforr delete_folder)rr)namesnamefullnamerrr rm_subdirsJsr1c Cstj|rD|durt|d|dSd} z t|ddWdSttfyB|d7}|tkr;t d |tt t YnwqdS)zBUtility function to cleanup a temporary folder if it still exists.NFrTrz/Unable to delete folder {} after {} tentatives.)rrisdirshutilrmtreer" WindowsErrorRM_SUBDIRS_N_RETRYwarningswarnformattimesleepRM_SUBDIRS_RETRY_TIME) folder_pathr) err_countrrrr-hs( r-)N)__doc__rr+r:r#r3r7r5 NameErrorr"rr r'r<r6r1r-rrrrs&