From 9a183fe7cd77d3de953fd7c6916f6d971331b499 Mon Sep 17 00:00:00 2001
From: Nick Mowen <nickmowen213@gmail.com>
Date: Fri, 1 Jul 2022 10:51:04 -0600
Subject: [PATCH] Get storage output stats for each camera

---
 frigate/http.py | 33 +++++++++++++++++++++++++++++++--
 1 file changed, 31 insertions(+), 2 deletions(-)

diff --git a/frigate/http.py b/frigate/http.py
index 57517f168..971e2a891 100644
--- a/frigate/http.py
+++ b/frigate/http.py
@@ -27,8 +27,7 @@ from flask import (
 from peewee import SqliteDatabase, operator, fn, DoesNotExist
 from playhouse.shortcuts import model_to_dict
 
-from frigate.config import CameraConfig
-from frigate.const import CLIPS_DIR
+from frigate.const import CLIPS_DIR, RECORD_DIR
 from frigate.models import Event, Recordings
 from frigate.object_processing import TrackedObject
 from frigate.stats import stats_snapshot
@@ -690,6 +689,36 @@ def latest_frame(camera_name):
         return "Camera named {} not found".format(camera_name), 404
 
 
+@bp.route("/recordings/storage")
+def recordings_storage_usage():
+    total_bytes = 0
+    storage = {}
+
+    for camera_name in current_app.frigate_config.cameras.keys():
+        camera_bytes = 0
+        du_cmd = [
+            "du",
+            "-h",
+            RECORD_DIR,
+        ]
+        p: sp.CompletedProcess = sp.run(
+            du_cmd,
+            encoding="ascii",
+            capture_output=True,
+        )
+        summary = sp.check_output(('grep', camera_name), stdin=p.stdout)
+
+        for line in summary.split("\n"):
+            logger.error(F"Found the line {line}")
+            camera_bytes += 1
+
+        storage[camera_name] = camera_bytes
+        total_bytes += camera_bytes
+
+    storage["total"] = total_bytes
+    return jsonify(storage)
+
+
 # return hourly summary for recordings of camera
 @bp.route("/<camera_name>/recordings/summary")
 def recordings_summary(camera_name):