adding import of calculation of folder size from utils

Author: vsoch

singularity/package.py

@@ -8,6 +8,7 @@
 from singularity.logman import bot
 
 from singularity.utils import (
+    calculate_folder_size,
     format_container_name,
     read_file, 
     zip_up

singularity/tests/test_package.py

@@ -62,5 +62,6 @@ def test_packaging(self):
         self.assertTrue(os.path.exists(image1_extraction))
         shutil.rmtree(os.path.dirname(image1_extraction))
 
+
 if __name__ == '__main__':
     unittest.main()

singularity/utils.py

@@ -229,6 +229,24 @@ def read_file(filename,mode="r"):
     return content
 
 
+############################################################################
+## OTHER MISC. #############################################################
+############################################################################
+
+
+def calculate_folder_size(folder_path):
+    '''calculate_folder size recursively walks a directory to calculate
+    a total size (in)
+    :param folder_path: the path to calculate size for
+    '''
+    total_size = 0
+    for dirpath, dirnames, filenames in os.walk(folder_path):
+        for filey in filenames:
+            fp = os.path.join(dirpath, filey)
+            total_size += os.path.getsize(fp)
+    return total_size
+
+
 def remove_unicode_dict(input_dict):
     '''remove unicode keys and values from dict, encoding in utf8
     '''