(enhance)Add env variable for training workspace download limit

hotosm · Apr 29, 2024 · 06d8c8d · 06d8c8d
1 parent c060cfa
commit 06d8c8d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/backend/aiproject/settings.py b/backend/aiproject/settings.py
@@ -56,6 +56,7 @@
 # Limiter
 EPOCHS_LIMIT = env("EPOCHS_LIMIT", default=30)
 BATCH_SIZE_LIMIT = env("BATCH_SIZE_LIMIT", default=8)
+TRAINING_WORKSPACE_DOWNLOAD_LIMIT = env("TRAINING_WORKSPACE_DOWNLOAD_LIMIT", default=200)
 
 
 # Application definition

diff --git a/backend/core/views.py b/backend/core/views.py
@@ -689,9 +689,9 @@ def get(self, request, lookup_dir):
             if os.path.isdir(base_dir)
             else os.path.getsize(base_dir)
         ) / (1024**2)
-        if size > 200:  # if file is greater than 200 mb exit
+        if size > settings.TRAINING_WORKSPACE_DOWNLOAD_LIMIT:  # if file is greater than 200 mb exit
             return Response(
-                {f"Errr: File Size {size} MB Exceed More than 200 MB"}, status=403
+                {f"Errr: File Size {size} MB Exceed More than {settings.TRAINING_WORKSPACE_DOWNLOAD_LIMIT} MB"}, status=403
             )
 
         if os.path.isfile(base_dir):