如何解决firebase_admin调用get()查询Firestore文档的上下文错误

你之前对报错原因的判断方向错了，这个RuntimeError: cannot exit context: thread state references a different context object和SDK异步调用等待逻辑无关，触发原因基本集中在两类：

• 跨线程/跨协程上下文复用了Firestore客户端实例：Firestore底层依赖gRPC的上下文绑定机制，在A线程/协程初始化的客户端，拿到B线程/协程调用就会抛出这个上下文不匹配错误
• 依赖版本兼容问题：Python 3.10+版本搭配6.2.0以下的firebase_admin、或者grpcio版本高于1.66.x时，gRPC内部的线程状态管理存在已知bug，会稳定复现这个报错

1. 先对齐依赖版本，排除兼容问题 #

先卸载存在冲突的依赖包，安装经过验证的兼容版本，执行以下命令：

pip uninstall -y firebase_admin google-cloud-firestore grpcio grpcio-status
pip install firebase_admin==6.5.0 google-cloud-firestore==2.16.0 grpcio==1.64.1

注意：不要单独将grpcio升级到最新版，目前1.65以上版本的grpcio和稳定版firebase_admin存在已知的上下文管理冲突

2. 修正客户端初始化逻辑，避免跨上下文复用 #

你之前的写法把客户端初始化和调用放在不同上下文的概率很高，注意：firebase_admin默认提供的Firestore客户端是同步阻塞版本，本身调用.get()就会等待请求返回，不需要额外引入asyncio包裹。正确写法要保证客户端初始化和数据请求在同一个线程/协程上下文内，不要在全局作用域初始化db实例后跨场景复用：

import firebase_admin
from firebase_admin import credentials, firestore

def query_subscription(purchase_token: str):
    # 避免重复初始化app
    if not firebase_admin._apps:
        cred = credentials.Certificate("你的服务账号证书路径.json")
        firebase_admin.initialize_app(cred)
    # 每次调用在当前上下文初始化客户端，不要复用全局db实例
    db = firestore.client()
    doc = db.collection("subscriptions").document(purchase_token).get()
    return doc.to_dict() if doc.exists else None

补充说明：如果要在异步场景使用Firestore，不要给同步客户端的方法加await，需要单独引入官方异步客户端实现，混用同步客户端和asyncio事件循环也会触发上下文错误，异步场景正确写法参考：

import asyncio
import firebase_admin
from firebase_admin import credentials
from google.cloud.firestore_v1.async_client import AsyncClient

async def async_query_subscription(purchase_token: str):
    if not firebase_admin._apps:
        cred = credentials.Certificate("你的服务账号证书路径.json")
        app = firebase_admin.initialize_app(cred)
    db = AsyncClient(credentials=app.credential.get_credential(), project=app.project_id)
    doc = await db.collection("subscriptions").document(purchase_token).get()
    return doc.to_dict() if doc.exists else None

• 在FastAPI、Flask等Web框架中使用时，不要在启动全局初始化db实例后注入到路由请求中处理，要在每个请求的上下文内初始化客户端
• 多线程任务场景下，不要复用主线程创建的db实例，每个子线程需要独立初始化客户端
• 不要给同步客户端的.get()、.where()等方法加await关键字，会直接触发协程上下文和gRPC线程状态的冲突

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