add option to pass project ids in cli for transfer results #481

Author: Hagellach37

mapswipe_workers/mapswipe_workers/firebase_to_postgres/transfer_results.py

@@ -29,6 +29,7 @@ def transfer_results(project_id_list=None):
     # We will only transfer results for projects we in postgres.
     postgres_project_ids = get_projects_from_postgres()
 
+    project_id_list_transfered = []
     for project_id in project_id_list:
         if project_id not in postgres_project_ids:
             logger.info(
@@ -49,8 +50,9 @@ def transfer_results(project_id_list=None):
             results = results_ref.get()
             del fb_db
             transfer_results_for_project(project_id, results)
+            project_id_list_transfered.appen(project_id)
 
-    return project_id_list
+    return project_id_list_transfered
 
 
 def transfer_results_for_project(project_id, results):

mapswipe_workers/mapswipe_workers/mapswipe_workers.py

@@ -107,14 +107,28 @@ def run_create_projects():
 
 
 @cli.command("firebase-to-postgres")
-def run_firebase_to_postgres() -> list:
+@click.option(
+    "--project_ids",
+    cls=PythonLiteralOption,
+    default="[]",
+    help=(
+        "Project ids for which to generate stats as a list of strings: "
+        """ '["project_a", "project_b"]' """
+        "(You need the quotes.)"
+    ),
+)
+def run_firebase_to_postgres(project_ids: list) -> list:
     """Update users and transfer results from Firebase to Postgres."""
-    project_ids = transfer_results.transfer_results()
-    for project_id in project_ids:
-        update_data.set_progress_in_firebase(project_id)
-        update_data.set_contributor_count_in_firebase(project_id)
-        send_progress_notification(project_id)
+    project_ids = transfer_results.transfer_results(project_ids)
+
+    if len(project_ids) > 0:
+        for project_id in project_ids:
+            update_data.set_progress_in_firebase(project_id)
+            update_data.set_contributor_count_in_firebase(project_id)
+            send_progress_notification(project_id)
+
     update_data.update_project_data()
+
     return project_ids