Show available disk space in stats

transcribe-batch/processedfiles.py

@@ -98,20 +98,30 @@ def get_num_of_files_stored(directory=PROCESSED):
         files = ProcessedFiles._find_files(directory, "*")
         return len(files)
 
+    def _get_human_readable_size(size):
+        GB_IN_BYTES = 1024**3
+        if size > GB_IN_BYTES:
+            gbs = size / GB_IN_BYTES
+            text = f"{gbs:.2f} GB"
+        else:
+            text = f"{size} bytes"
+
+        return text
+
     def get_num_of_files_stored_size(directory=PROCESSED):
         files = ProcessedFiles._find_files(directory, "*")
         total_size = 0
         for _file in files:
             total_size += os.stat(_file).st_size
 
-        GB_IN_BYTES = 1024 * 1024 * 1024
-        if total_size > GB_IN_BYTES:
-            gbs = total_size / GB_IN_BYTES
-            size = f"{gbs:.2f} GB"
-        else:
-            size = f"{total_size} bytes"
+        return ProcessedFiles._get_human_readable_size(total_size)
+
+    def get_free_space_in_directory(directory=PROCESSED):
+        statvfs = os.statvfs(directory)
 
-        return size
+        # Available blocks * block size gives the available space in bytes
+        free_space_bytes = statvfs.f_frsize * statvfs.f_bavail
+        return ProcessedFiles._get_human_readable_size(free_space_bytes)
 
     def purge_files(days, directory=PROCESSED):
         HOURS_DAY = 24

transcribe-service/transcribe-service.py

@@ -70,6 +70,7 @@ def stats():
 
     result["files_stored"] = ProcessedFiles.get_num_of_files_stored()
     result["files_stored_size"] = ProcessedFiles.get_num_of_files_stored_size()
+    result["free_storage_space"] = ProcessedFiles.get_free_space_in_directory()
     queue["items"] = len(records)
     queue["who"] = print_who
     result["queue"] = queue