如何在FastAPI全局配置Firestore后跨文件复用数据库实例

方法一：创建独立的数据库配置模块（推荐） #

将Firestore初始化逻辑抽离到单独模块，避免循环导入问题，让所有需要数据库的文件复用同一个实例。

1. 创建app/server/db.py文件，统一处理数据库初始化：

import os
import firebase_admin
from firebase_admin import credentials, firestore

# 设置模拟器环境变量（本地调试用）
os.environ["FIREBASE_AUTH_EMULATOR_HOST"] = "127.0.0.1:9099"
os.environ["FIRESTORE_EMULATOR_HOST"] = "127.0.0.1:8080"

# 初始化Firebase应用与Firestore客户端
cred = credentials.Certificate("path_to_file.json")
firebase_admin.initialize_app(cred)
db = firestore.client()

2. 修改app/server/app.py，从db模块导入实例：

from fastapi import FastAPI
from app.server.routes.users import router as User, get_data
from app.server.db import db  # 导入已初始化的数据库实例

app = FastAPI()
app.include_router(User, tags=["User"], prefix="/a1")

@app.on_event("startup")
async def start_up():
    await get_data()

3. 修改app/server/routes/users.py，直接导入db实例操作：

from fastapi import APIRouter, HTTPException, Request
from fastapi_utils.tasks import repeat_every
from app.server.db import db  # 导入数据库实例

router = APIRouter()

@repeat_every(seconds=5, max_repetitions=1)
async def get_data():
    # 直接使用db实例执行数据库操作
    db.collection("collection1").document("doc1").set({"a": 1})

方法二：使用FastAPI的Depends依赖注入（适合路由函数） #

如果是在路由处理函数中需要数据库实例，可通过依赖注入获取：

1. 在app/server/db.py中添加依赖函数：

# 新增依赖函数，供路由调用
def get_db():
    return db

2. 在路由函数中通过Depends获取实例：

from fastapi import APIRouter, Depends
from app.server.db import get_db
from firebase_admin import firestore

router = APIRouter()

@router.get("/get-data")
async def get_example_data(db: firestore.client = Depends(get_db)):
    doc = db.collection("collection1").document("doc1").get()
    return doc.to_dict() if doc.exists else {"message": "文档不存在"}

关键注意事项 #

• 确保firebase_admin.initialize_app(cred)只执行一次，多次初始化会抛出异常，独立模块的方式能避免这个问题。
• 使用本地模拟器时，需提前启动对应服务，且环境变量配置要与模拟器地址一致。

内容的提问来源于stack exchange，提问作者apod