From 339ea544ac7394b832a55e425040a129671c69c4 Mon Sep 17 00:00:00 2001 From: yagudin Date: Tue, 1 Jun 2021 18:59:04 +0300 Subject: [PATCH] Add support of different platforms --- firebase_requests.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/firebase_requests.py b/firebase_requests.py index 150a536..83cf5c0 100644 --- a/firebase_requests.py +++ b/firebase_requests.py @@ -12,7 +12,9 @@ db = firestore.Client(credentials=credentials, project="gjo-calibration") def get_forecasts(uid, questions, platform_url, headers, cookies): - db_forecasts = db.collection("users").document(uid).get().to_dict() + platform = "gjo" if platform_url == "https://www.gjopen.com" else "cset" + + db_forecasts = db.collection(f"users_{platform}").document(uid).get().to_dict() db_forecasts = dict() if db_forecasts is None else db_forecasts missing_forecasts_qs = list(set(questions) - set(db_forecasts)) @@ -22,14 +24,16 @@ def get_forecasts(uid, questions, platform_url, headers, cookies): if missing_forecasts: if not db_forecasts: - db.collection("users").add({}, uid) - db.collection("users").document(uid).update(missing_forecasts) + db.collection(f"users_{platform}").add({}, uid) + db.collection(f"users_{platform}").document(uid).update(missing_forecasts) return {key: value for key, value in {**db_forecasts, **missing_forecasts}.items() if key in questions} def get_resolutions(questions, platform_url, headers, cookies): - db_resolutions = db.collection("questions").document("resolutions").get().to_dict() + platform = "gjo" if platform_url == "https://www.gjopen.com" else "cset" + + db_resolutions = db.collection(f"questions_{platform}").document("resolutions").get().to_dict() db_resolutions = dict() if db_resolutions is None else db_resolutions relevant_resolutions = { key: value for key, value in db_resolutions.items() if key in set(questions) @@ -41,6 +45,6 @@ def get_resolutions(questions, platform_url, headers, cookies): ) if missing_resolutions: - db.collection("questions").document("resolutions").update(missing_resolutions) + db.collection(f"questions_{platform}").document("resolutions").update(missing_resolutions) return {**relevant_resolutions, **missing_resolutions}