B \@sdZddlZddlZddlZddlZddlZddlZyeWnek rTe ZYnXddZ ddZ ddZ d Z d Zdd d Zdd dZdS)z Disk management utilities. NcCsnd}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)rsizefilerr}z|jtjkr.Wdd}~XYnXdS)zgEnsure directory d exists (like mkdir -p on Unix) No guarantee that the directory is writable. N)rmakedirsOSErrorerrnoZEEXIST)derrrmkdirp7s  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_inforr delete_folder)rr&namesnamefullnamerrr rm_subdirsJs r.c Cstj|r|dk r$t|d|nfd}x`yt|ddPWq*ttfk r|d7}|tkrvt d |tt t Yq*Xq*WdS)zBUtility function to cleanup a temporary folder if it still exists.NFrrz/Unable to delete folder {} after {} tentatives.)rrisdirshutilZrmtreer WindowsErrorRM_SUBDIRS_N_RETRYwarningswarnformattimeZsleepRM_SUBDIRS_RETRY_TIME)Z folder_pathr&Z err_countrrrr*hs   r*)N)N)__doc__rr(r6r!r0r3r1 NameErrorr rrr$r7r2r.r*rrrrs"