Skip to content

Commit b5bca5c

Browse files
committed
check if there are projects before updating data #175
1 parent 9feb195 commit b5bca5c

File tree

1 file changed

+12
-11
lines changed

1 file changed

+12
-11
lines changed

mapswipe_workers/mapswipe_workers/firebase_to_postgres/update_data.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -115,17 +115,18 @@ def update_project_data(project_ids=None):
115115
projects_ref = fb_db.reference('v2/projects/')
116116
projects = projects_ref.get()
117117

118-
for project_id, project in projects.items():
119-
query_update_project = '''
120-
UPDATE projects
121-
SET status=%s
122-
WHERE project_id=%s;
123-
'''
124-
# TODO: Is there need for fallback to ''
125-
# if project.status is not existent
126-
data_update_project = [project.get('status', ''), project_id]
127-
pg_db.query(query_update_project, data_update_project)
128-
logger.info(f"updated status for project {project_id} in postgres")
118+
if projects:
119+
for project_id, project in projects.items():
120+
query_update_project = '''
121+
UPDATE projects
122+
SET status=%s
123+
WHERE project_id=%s;
124+
'''
125+
# TODO: Is there need for fallback to ''
126+
# if project.status is not existent
127+
data_update_project = [project.get('status', ''), project_id]
128+
pg_db.query(query_update_project, data_update_project)
129+
logger.info(f"updated status for project {project_id} in postgres")
129130

130131
del(pg_db)
131132

0 commit comments

Comments
 (0)