Use FIREBASE_PROJECT_ID for Firestore database (#26)

* Use FIREBASE_PROJECT_ID for Firestore database * Use project keyword for AsyncClient
hcholab · Jan 2, 2024 · 87c27ef · 87c27ef
1 parent 18f54da
commit 87c27ef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/__init__.py b/src/__init__.py
@@ -27,7 +27,10 @@ def create_app() -> Quart:
     app = cors(app, allow_origin=list(origins))
 
     app.config.from_mapping(
-        SECRET_KEY=secrets.token_hex(16), DATABASE=firestore.AsyncClient()
+        SECRET_KEY=secrets.token_hex(16),
+        DATABASE=firestore.AsyncClient(
+            project=os.getenv("FIREBASE_PROJECT_ID", "broad-cho-priv1"),
+        ),
     )
 
     app.register_blueprint(status.bp)

diff --git a/src/web/web.py b/src/web/web.py
@@ -42,7 +42,7 @@ async def create_custom_token() -> Response:
                 {
                     "customToken": custom_token.decode("utf-8"),
                     "firebaseApiKey": firebase_api_key,
-                    "firebaseProjectId": os.getenv("FIREBASE_PROJECT_ID", "broad-cho-priv1"),
+                    "firebaseProjectId": current_app.config["DATABASE"].project,
                 }
             ),
             200,