|
| 1 | +import sys |
| 2 | +from mapswipe_workers.auth import firebaseDB |
| 3 | +from mapswipe_workers.definitions import logger |
| 4 | + |
| 5 | + |
| 6 | +def get_all_projects(project_type: int): |
| 7 | + """Get the user ids from all users in Firebase DB.""" |
| 8 | + |
| 9 | + project_id_list = [] |
| 10 | + fb_db = firebaseDB() |
| 11 | + |
| 12 | + status_list = ["active", "inactive"] |
| 13 | + |
| 14 | + for status in status_list: |
| 15 | + logger.info(f"query {status} projects") |
| 16 | + projects = ( |
| 17 | + fb_db.reference(f"v2/projects/") |
| 18 | + .order_by_child("status") |
| 19 | + .equal_to(status) |
| 20 | + .get() |
| 21 | + ) |
| 22 | + for project_id, data in projects.items(): |
| 23 | + if (data.get("projectType", 1) == project_type) & ( |
| 24 | + data.get("tutorialId", None) is None |
| 25 | + ): |
| 26 | + project_id_list.append(project_id) |
| 27 | + |
| 28 | + logger.info(f"got {len(project_id_list)} project from firebase.") |
| 29 | + return project_id_list |
| 30 | + |
| 31 | + |
| 32 | +def add_tutorial_id_to_projects(project_id_list, tutorial_id): |
| 33 | + fb_db = firebaseDB() |
| 34 | + for project_id in project_id_list: |
| 35 | + fb_db.reference(f"v2/projects/{project_id}/tutorialId").set(tutorial_id) |
| 36 | + logger.info(f"added tutorial id '{tutorial_id}' to project '{project_id}'") |
| 37 | + |
| 38 | + |
| 39 | +if __name__ == "__main__": |
| 40 | + """python add_tutorial_id_to_projects.py project_type tutorial_id""" |
| 41 | + project_type = sys.argv[1] |
| 42 | + tutorial_id = sys.argv[2] |
| 43 | + project_id_list = get_all_projects(1) |
| 44 | + add_tutorial_id_to_projects(project_id_list, tutorial_id) |
0 commit comments