add script to automatically update project status

Author: Hagellach37

mapswipe_workers/mapswipe_workers/definitions.py

@@ -113,6 +113,8 @@ class MessageType(Enum):
     FAIL = 2
     NOTIFICATION_90 = 3
     NOTIFICATION_100 = 4
+    PROJECT_STATUS_FINISHED = 5
+    PROJECT_STATUS_ACTIVE = 6
 
 
 class ProjectType(Enum):

mapswipe_workers/mapswipe_workers/utils/slack_helper.py

@@ -58,6 +58,24 @@ def send_slack_message(
             + "and activate another one."
         )
         slack_client.chat_postMessage(channel="mapswipe_managers", text=message)
+    elif message_type == MessageType.PROJECT_STATUS_FINISHED:
+        message = (
+            "### SET PROJECT STATUS TO FINISHED ###\n"
+            + f"Project Name: {project_name}\n"
+            + f"Project Id: {project_id}\n\n"
+            + "The status of the project has been set to 'finished' "
+            + "by MapSwipe's backend workers."
+        )
+        slack_client.chat_postMessage(channel=SLACK_CHANNEL, text=message)
+    elif message_type == MessageType.PROJECT_STATUS_ACTIVE:
+        message = (
+            "### SET PROJECT STATUS TO ACTIVE ###\n"
+            + f"Project Name: {project_name}\n"
+            + f"Project Id: {project_id}\n\n"
+            + "The status of the project has been set to 'active' "
+            + "by MapSwipe's backend workers."
+        )
+        slack_client.chat_postMessage(channel=SLACK_CHANNEL, text=message)
     else:
         # TODO: Raise an Exception
         pass

mapswipe_workers/python_scripts/update_project_status.py

@@ -0,0 +1,69 @@
+from mapswipe_workers.auth import firebaseDB
+from mapswipe_workers.definitions import MessageType, logger
+from mapswipe_workers.utils.slack_helper import send_slack_message
+
+
+def get_projects(status):
+    """Load 'active' projects from Firebase."""
+    fb_db = firebaseDB()
+    projects = (
+        fb_db.reference("v2/projects/").order_by_child("status").equal_to(status).get()
+    )
+    logger.info(f"got {len(projects)} {status} projects from firebase")
+    return projects
+
+
+def filter_projects_by_name_and_progress(projects, filter_string, progress_threshold):
+    """Load 'active' projects from Firebase."""
+    selected_project_ids = []
+    for project_id in projects.keys():
+        name = projects[project_id]["name"]
+        progress = projects[project_id]["progress"]
+        if filter_string in name and progress >= progress_threshold:
+            selected_project_ids.append(project_id)
+
+    logger.info(
+        f"selected {len(selected_project_ids)} project(s) which contain(s) "
+        f"'{filter_string}' in the project name and progress >= {progress_threshold}%."
+    )
+    return selected_project_ids
+
+
+def set_status_in_firebase(project_id, project_name, new_status):
+    """Change status of a project in Firebase."""
+    # change status in firebase
+    fb_db = firebaseDB()
+    fb_db.reference(f"v2/projects/{project_id}/status").set(new_status)
+    logger.info(f"set project status to '{new_status}' for project {project_id}")
+
+    # send slack message
+    if new_status == "finished":
+        send_slack_message(
+            MessageType.PROJECT_STATUS_FINISHED, project_name, project_id
+        )
+    elif new_status == "active":
+        send_slack_message(MessageType.PROJECT_STATUS_ACTIVE, project_name, project_id)
+
+
+if __name__ == "__main__":
+    """Check if project status should be updated and change value in Firebase."""
+    filter_string = "Test"
+
+    active_projects = get_projects(status="active")
+    finished_projects = filter_projects_by_name_and_progress(
+        active_projects, filter_string, progress_threshold=1,
+    )
+
+    inactive_projects = get_projects(status="inactive")
+    new_active_projects = filter_projects_by_name_and_progress(
+        inactive_projects, filter_string, progress_threshold=0,
+    )[0 : len(finished_projects)]
+
+    if len(new_active_projects) > 0:
+        for project_id in finished_projects:
+            project_name = active_projects[project_id]["name"]
+            set_status_in_firebase(project_id, project_name, new_status="finished")
+
+        for project_id in new_active_projects:
+            project_name = inactive_projects[project_id]["name"]
+            set_status_in_firebase(project_id, project_name, new_status="active")