max sure it works also if no groups #310

Author: Hagellach37

mapswipe_workers/mapswipe_workers/firebase_to_postgres/archive_project.py

@@ -21,24 +21,28 @@ def archive_project(project_ids: list) -> None:
         fb_db.reference("v2/results/{0}".format(project_id)).set({})
 
         # get group keys for this project to estimate size in firebase
-        group_keys = list(
-            fb_db.reference("v2/tasks/{0}/".format(project_id)).get(shallow=True).keys()
-        )
-        chunk_size = 250
-        chunks = int(len(group_keys) / chunk_size) + 1
+        groups = fb_db.reference("v2/groups/{0}/".format(project_id)).get(shallow=True)
+
+        if not groups:
+            logger.info("no groups to delete in firebase")
+        else:
+            group_keys = list(groups.keys())
+            chunk_size = 250
+            chunks = int(len(group_keys) / chunk_size) + 1
 
-        # delete groups and tasks in firebase for each chunk using the update function
-        for i in range(1, chunks):
-            logger.info(
-                "Delete max {1} groups and tasks of project with the id {0}".format(
-                    project_id, chunk_size
+            # delete groups, tasks in firebase for each chunk using the update function
+            for i in range(0, chunks):
+                logger.info(
+                    "Delete max {1} groups and tasks of project with the id {0}".format(
+                        project_id, chunk_size
+                    )
                 )
-            )
-            update_dict = {}
-            for group_id in group_keys[:250]:
-                update_dict[group_id] = None
-            fb_db.reference("v2/groups/{0}".format(project_id)).update(update_dict)
-            fb_db.reference("v2/tasks/{0}".format(project_id)).update(update_dict)
+                update_dict = {}
+                for group_id in group_keys[:chunk_size]:
+                    update_dict[group_id] = None
+                fb_db.reference("v2/groups/{0}".format(project_id)).update(update_dict)
+                fb_db.reference("v2/tasks/{0}".format(project_id)).update(update_dict)
+                group_keys = group_keys[chunk_size:]
 
         logger.info(
             "Set status=archived in Firebase for project with the id {0}".format(