From 58ac4c9d25e7d1ecb834a49f184b4543ebb4eff9 Mon Sep 17 00:00:00 2001
From: Caroline Kery <ckery@rti.org>
Date: Fri, 10 May 2024 13:58:12 -0400
Subject: [PATCH] fix project import/export

---
 .../commands/export_database_data.py          | 17 ++++++++++-------
 .../commands/ingest_database_data.py          | 19 +++++++++++--------
 2 files changed, 21 insertions(+), 15 deletions(-)

diff --git a/backend/django/core/management/commands/export_database_data.py b/backend/django/core/management/commands/export_database_data.py
index 0754c9a9..dc8964ae 100644
--- 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
index 0f6491c9..c918e56b 100644
--- 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)