Check if aggregated table exists

Author: thenav56

mapswipe_workers/mapswipe_workers/auth.py

@@ -80,6 +80,12 @@ def copy_expert(self, sql, file):
         self._db_connection.commit()
         self._db_cur.close()
 
+    def table_exists(self, table_name):
+        resp = self.retr_query(
+            "SELECT to_regclass(%(table_name)s);", {"table_name": table_name}
+        )
+        return resp[0][0] is not None
+
     def retr_query(self, query, data=None):
         self._db_cur = self._db_connection.cursor()
         self._db_cur.execute(query, data)

mapswipe_workers/mapswipe_workers/firebase_to_postgres/delete_project.py

@@ -114,10 +114,13 @@ def delete_project(project_ids: list) -> bool:
         # -- Table from django/apps/aggregated/models.py. Used to cache stats data
         # NOTE: Django doesn't support database-level CASCADE delete
         #  https://docs.djangoproject.com/en/4.1/ref/models/fields/#django.db.models.ForeignKey.on_delete
-        sql_query = "DELETE FROM aggregated_aggregateduserstatdata WHERE project_id = %(project_id)s;"
-        pg_db.query(sql_query, {"project_id": project_id})
-        sql_query = "DELETE FROM aggregated_aggregatedusergroupstatdata WHERE project_id = %(project_id)s;"
-        pg_db.query(sql_query, {"project_id": project_id})
+        for aggregated_table_name in [
+            "aggregated_aggregateduserstatdata",
+            "aggregated_aggregatedusergroupstatdata",
+        ]:
+            if pg_db.table_exists(aggregated_table_name):
+                sql_query = f"DELETE FROM {aggregated_table_name} WHERE project_id = %(project_id)s;"
+                pg_db.query(sql_query, {"project_id": project_id})
         # Finally delete the project
         sql_query = "DELETE FROM projects WHERE project_id = %(project_id)s;"
         pg_db.query(sql_query, {"project_id": project_id})