* implement a function in util to calculate the total file size under a dir
This commit is contained in:
@@ -34,6 +34,17 @@ def clean_dir(top):
|
|||||||
for name in dirs:
|
for name in dirs:
|
||||||
os.rmdir(os.path.join(root, name))
|
os.rmdir(os.path.join(root, name))
|
||||||
|
|
||||||
|
|
||||||
|
# calculate the size of files under a dir
|
||||||
|
# based on the tutorial example
|
||||||
|
def dir_size(dir):
|
||||||
|
from os.path import join, getsize
|
||||||
|
def sizes():
|
||||||
|
for root, dirs, files in os.walk(dir):
|
||||||
|
print sum(getsize(join(root, name)) for name in files)
|
||||||
|
yield sum(getsize(join(root, name)) for name in files)
|
||||||
|
return sum( sizes() )
|
||||||
|
|
||||||
def copy_file(s,d):
|
def copy_file(s,d):
|
||||||
check_file(s)
|
check_file(s)
|
||||||
check_dir(os.path.dirname(d))
|
check_dir(os.path.dirname(d))
|
||||||
|
|||||||
Reference in New Issue
Block a user