Move django.setup before import

cisagov · Nov 20, 2024 · b101bb1 · b101bb1
1 parent f46b571
commit b101bb1
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/backend/src/xfd_django/xfd_api/tasks/run_syncdb.py b/backend/src/xfd_django/xfd_api/tasks/run_syncdb.py
@@ -3,6 +3,14 @@
 
 # Third-Party Libraries
 import django
+
+# Set the Django settings module
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "xfd_django.settings")
+os.environ["DJANGO_ALLOW_ASYNC_UNSAFE"] = "true"
+
+# Initialize Django
+django.setup()
+
 from django.core.management import call_command
 from xfd_api.management.commands.syncdb import (
     populate_sample_data,
@@ -14,12 +22,6 @@ def handler(event, context):
     """
     Lambda handler to trigger syncdb.
     """
-    # Set the Django settings module
-    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "xfd_django.settings")
-    os.environ["DJANGO_ALLOW_ASYNC_UNSAFE"] = "true"
-
-    # Initialize Django
-    django.setup()
 
     # Parse arguments from the event
     dangerouslyforce = event.get("dangerouslyforce", False)