From 82c039171f5247bf87d8439e6f41864a5820c467 Mon Sep 17 00:00:00 2001
From: Daniel Markstedt <markstedt@gmail.com>
Date: Mon, 20 Sep 2021 18:55:09 -0700
Subject: [PATCH] Display available disk space on the Pi

---
 src/web/pi_cmds.py            | 6 ++++++
 src/web/templates/drives.html | 1 +
 src/web/templates/index.html  | 1 +
 src/web/web.py                | 6 ++++++
 4 files changed, 14 insertions(+)

diff --git a/src/web/pi_cmds.py b/src/web/pi_cmds.py
index 6d865009..9de516cf 100644
--- a/src/web/pi_cmds.py
+++ b/src/web/pi_cmds.py
@@ -39,3 +39,9 @@ def is_bridge_setup():
     if "rascsi_bridge" in output:
         return True
     return False
+
+
+def disk_space():
+    from shutil import disk_usage
+    total, used, free = disk_usage(__file__)
+    return {"total": total, "used": used, "free": free}
diff --git a/src/web/templates/drives.html b/src/web/templates/drives.html
index f8d335c7..85418b1d 100644
--- a/src/web/templates/drives.html
+++ b/src/web/templates/drives.html
@@ -135,5 +135,6 @@
         {% endfor %}
         </tbody>
     </table>
+    <p><small>Available disk space on the Pi: {{free_disk}} MB</small></p>
 
 {% endblock %}
diff --git a/src/web/templates/index.html b/src/web/templates/index.html
index f62b63cd..ab544db6 100644
--- a/src/web/templates/index.html
+++ b/src/web/templates/index.html
@@ -126,6 +126,7 @@
         {% endfor %}
         </tbody>
     </table>
+    <p><small>Available disk space on the Pi: {{free_disk}} MB</small></p>
 
     <hr/>
     <h2>Attach Ethernet Adapter</h2>
diff --git a/src/web/web.py b/src/web/web.py
index 78bc57e2..7f6b6c65 100644
--- a/src/web/web.py
+++ b/src/web/web.py
@@ -28,6 +28,7 @@ from pi_cmds import (
     running_env,
     rascsi_service,
     is_bridge_setup,
+    disk_space,
 )
 from ractl_cmds import (
     attach_image,
@@ -50,6 +51,7 @@ app = Flask(__name__)
 @app.route("/")
 def index():
     server_info = get_server_info()
+    disk = disk_space()
     devices = list_devices()
     files=list_files()
     config_files=list_config_files()
@@ -60,6 +62,7 @@ def index():
     reserved_scsi_ids = server_info["reserved_ids"]
     formatted_devices = sort_and_format_devices(devices["device_list"])
     scsi_ids = get_valid_scsi_ids(devices["device_list"], reserved_scsi_ids)
+
     return render_template(
         "index.html",
         bridge_configured=is_bridge_setup(),
@@ -73,6 +76,7 @@ def index():
         running_env=running_env(),
         server_info=server_info,
         netinfo=get_network_info(),
+        free_disk=int(disk["free"] / 1024 / 1024),
         valid_file_suffix="."+", .".join(VALID_FILE_SUFFIX),
         removable_device_types=REMOVABLE_DEVICE_TYPES,
         harddrive_file_suffix=HARDDRIVE_FILE_SUFFIX,
@@ -88,6 +92,7 @@ def drive_list():
     Sets up the data structures and kicks off the rendering of the drive list page
     """
     server_info = get_server_info()
+    disk = disk_space()
 
     # Reads the canonical drive properties into a dict
     # The file resides in the current dir of the web ui process
@@ -136,6 +141,7 @@ def drive_list():
         rm_conf=rm_conf,
         running_env=running_env(),
         server_info=server_info,
+        free_disk=int(disk["free"] / 1024 / 1024),
         cdrom_file_suffix=CDROM_FILE_SUFFIX,
     )