3 QfX @sVdZddlZddlZddlZddlZddlZddZddZddZd Z d d d Z dS) z Disk management utilities. NcCsnd}x\tj|dgD]H}tjtjj||}t|drH||jd7}q||jddd7}qWt|dS)z& Return the disk usage in a directory.r. st_blocksig@) oslistdirstatpathjoinhasattrrst_sizeint)rsizefilerrR/home/psgendb/BIRCHDEV/pkg/SPAdes-3.15.4/linux-x86_64/share/spades/joblib3/disk.py disk_useds rc Csld}td||dd}y$t||dt|dd}Wn,ttfk rftd|t|fYnX|S) z7 Convert a memory text to it's value in kilobytes. ir)KMGNzQInvalid literal for size give: %s (type %s) should be alike '10G', '500M', '50K'.r)dictr floatKeyError ValueErrortype)textZkiloZunitsr rrrmemstr_to_kbytes"s$rcCsDytj|Wn0tk r>}z|jtjkr.WYdd}~XnXdS)zgEnsure directory d exists (like mkdir -p on Unix) No guarantee that the directory is writable. N)rmakedirsOSErrorerrnoEEXIST)derrrmkdirp1s  r$g?cCsg}ytj|}WnBtjk rT}z$|dk rB|tj|tjnWYdd}~XnXx|D]}tjj||}tjj|r\|dk rtj |d|q\d}xLytj |ddPWqtjk r|dkrȂ|d7}t j t YqXqWq\WdS)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. NFrr) rrerrorsysexc_inforr isdirshutilrmtreetimesleepRM_SUBDIRS_RETRY_TIME)ronerrornameserrnamefullnameZ err_countrrr rm_subdirsAs,  r3)N) __doc__r rr)r&r+rrr$r-r3rrrrs