add quota setting to env

ephios-dev · Oct 21, 2024 · 693e387 · 693e387
1 parent bba2c7d
commit 693e387
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/docs/admin/configuration/index.rst b/docs/admin/configuration/index.rst
@@ -74,6 +74,9 @@ Data storage and Logging
     Defaults to `PRIVATE_DIR/media`.
     You should backup this folder regularly.
 
+`MEDIA_FILES_DISK_QUOTA`:
+    Quota in MB that is enforced for the media files. Defaults to 0, meaning no quota is enforced.
+
 `LOG_DIR`:
     Path to the folder where log files are put. Files inside are rotated daily.
     Defaults to `PRIVATE_DIR/logs`.

diff --git a/ephios/settings.py b/ephios/settings.py
@@ -328,7 +328,9 @@ def GET_USERCONTENT_URL():
 def GET_USERCONTENT_QUOTA():
     """Returns a tuple (used, free) of the user content quota in bytes"""
     used = sum(p.stat().st_size for p in Path(MEDIA_ROOT).rglob("*"))
-    free = shutil.disk_usage(MEDIA_ROOT).free
+    quota = env.int("MEDIA_FILES_DISK_QUOTA", default=0)
+    disk_free = shutil.disk_usage(MEDIA_ROOT).free
+    free = min(disk_free, quota * 1024 * 1024 - used) if quota else disk_free
     return used, free