Refactor WSGI and manage.py to conditionally download and copy db.sqlite3 when running inside Docker

BridgeMentor/wsgi.py

@@ -9,22 +9,24 @@ https://docs.djangoproject.com/en/4.2/howto/deployment/wsgi/
 
 import os
 import shutil
-
-from huggingface_hub import hf_hub_download
 from django.core.wsgi import get_wsgi_application
-custom_cache_dir = os.path.expanduser("/app/.cache/huggingface")
-
-file_path = hf_hub_download(
-    repo_id="SushantGautam/BridgeMentor",
-    filename="db.sqlite3",
-    repo_type="dataset",
-    cache_dir=custom_cache_dir
-)
-print(f"Downloaded to: {file_path}")
-destination_path = os.path.join(os.getcwd(), "db.sqlite3")
-shutil.copy(file_path, destination_path)
-
-print(f"db.sqlite3 Copied to current directory: {destination_path}")
+
+
+if os.environ.get('INSIDEDOCKER'):
+    from huggingface_hub import hf_hub_download
+
+    custom_cache_dir = os.path.expanduser("/app/.cache/huggingface")
+    file_path = hf_hub_download(
+        repo_id="SushantGautam/BridgeMentor",
+        filename="db.sqlite3",
+        repo_type="dataset",
+        cache_dir=custom_cache_dir
+    )
+    print(f"Downloaded to: {file_path}")
+    destination_path = os.path.join(os.getcwd(), "db.sqlite3")
+    shutil.copy(file_path, destination_path)
+
+    print(f"db.sqlite3 Copied to current directory: {destination_path}")
 
 
 os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'BridgeMentor.settings')

Dockerfile

@@ -1,6 +1,7 @@
 # Use official Python image
 FROM python:3.11-slim
 
+ENV INSIDEDOCKER 1
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     default-libmysqlclient-dev \

manage.py

@@ -3,6 +3,24 @@
 import os
 import sys
 
+import shutil
+
+if os.environ.get('INSIDEDOCKER'):
+    from huggingface_hub import hf_hub_download
+    custom_cache_dir = os.path.expanduser("/app/.cache/huggingface")
+
+    file_path = hf_hub_download(
+        repo_id="SushantGautam/BridgeMentor",
+        filename="db.sqlite3",
+        repo_type="dataset",
+        cache_dir=custom_cache_dir
+    )
+    print(f"Downloaded to: {file_path}")
+    destination_path = os.path.join(os.getcwd(), "db.sqlite3")
+    shutil.copy(file_path, destination_path)
+    print(f"db.sqlite3 Copied to current directory: {destination_path}")
+
+
 def main():
     """Run administrative tasks."""
     os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'BridgeMentor.settings')