utils: add block size arg to disk usage function

reanahub · Jan 10, 2020 · 9723440 · 9723440
1 parent fd9e20f
commit 9723440
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/reana_commons/utils.py b/reana_commons/utils.py
@@ -181,9 +181,13 @@ def build_caching_info_message(job_spec,
     return caching_info_message
 
 
-def get_workspace_disk_usage(workspace, summarize=False):
+def get_workspace_disk_usage(workspace, summarize=False, block_size=None):
     """Retrieve disk usage information of a workspace."""
-    command = ['du', '-h']
+    command = ['du']
+    if block_size in ['b', 'k']:
+        command.append('-{}'.format(block_size))
+    else:
+        command.append('-h')
     if summarize:
         command.append('-s')
     else:

diff --git a/reana_commons/version.py b/reana_commons/version.py
@@ -14,4 +14,4 @@
 
 from __future__ import absolute_import, print_function
 
-__version__ = "0.7.0.dev20191220"
+__version__ = "0.7.0.dev20200110"