fix project import/export

RTIInternational · May 10, 2024 · 58ac4c9 · 58ac4c9
1 parent b61b79c
commit 58ac4c9
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 15 deletions.
diff --git a/backend/django/core/management/commands/export_database_data.py b/backend/django/core/management/commands/export_database_data.py
@@ -9,15 +9,18 @@ class Command(BaseCommand):
 
     def handle(self, *args, **kwargs):
         export_projects = ExternalDatabase.objects.filter(cron_export=True).values_list(
-            flat=True
+            "project_id", flat=True
         )
         if len(export_projects) == 0:
             print("No projects have cron_export set to True.")
         for pk in export_projects:
             print("Exporting project", pk)
-            response = {}
-            export_table(pk, response)
-            if "success_message" in response:
-                print(response["success_message"])
-            if "error" in response:
-                print(response["error"])
+            try:
+                response = {}
+                export_table(pk, response)
+                if "success_message" in response:
+                    print(response["success_message"])
+                if "error" in response:
+                    print(response["error"])
+            except Exception as e:
+                print("ERROR:", e)
diff --git a/backend/django/core/management/commands/ingest_database_data.py b/backend/django/core/management/commands/ingest_database_data.py
@@ -9,16 +9,19 @@ class Command(BaseCommand):
 
     def handle(self, *args, **kwargs):
         ingest_projects = ExternalDatabase.objects.filter(cron_ingest=True).values_list(
-            flat=True
+            "project_id", flat=True
         )
         if len(ingest_projects) == 0:
             print("No projects have cron_ingest set to True.")
         for pk in ingest_projects:
             print("Ingesting project", pk)
-            project = Project.objects.get(pk=pk)
-            response = {}
-            response = load_ingest_table(project, response)
-            if "num_added" in response:
-                print("Imported", response["num_added"], "new items")
-            if "error" in response:
-                print(response["error"])
+            try:
+                project = Project.objects.get(pk=pk)
+                response = {}
+                response = load_ingest_table(project, response)
+                if "num_added" in response:
+                    print("Imported", response["num_added"], "new items")
+                if "error" in response:
+                    print(response["error"])
+            except Exception as e:
+                print("ERROR:", e)