Python中访问Firestore命名数据库的实用指南

pip install --upgrade google-cloud-firestore

from google.cloud import firestore
import os

# 确保已设置GOOGLE_APPLICATION_CREDENTIALS环境变量
# 或在代码中直接指定凭证文件路径
# os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "/path/to/your/service-account-key.json"

# 初始化Firestore客户端，指定命名数据库
# 将 "your-named-database" 替换为您的实际数据库名称
# 注意：数据库名称通常是您在Google Cloud控制台创建时指定的ID
try:
    firestore_client_named = firestore.Client(database="my-custom-db")

    # 示例：在命名数据库中写入数据
    doc_ref_write = firestore_client_named.collection("users").document("alovelace")
    doc_ref_write.set({
        "first": "Ada",
        "last": "Lovelace",
        "born": 1815
    })
    print(f"数据已写入命名数据库 '{firestore_client_named.database_id}' 中的 'users/alovelace'。")

    # 示例：从命名数据库中读取数据
    doc = doc_ref_write.get()
    if doc.exists:
        print(f"从命名数据库 '{firestore_client_named.database_id}' 读取到数据: {doc.to_dict()}")
    else:
        print(f"命名数据库 '{firestore_client_named.database_id}' 中的文档不存在。")

except Exception as e:
    print(f"访问命名数据库时发生错误: {e}")

# 您也可以同时初始化默认数据库的客户端
try:
    firestore_client_default = firestore.Client()
    doc_ref_default = firestore_client_default.collection("default_collection").document("data")
    # 尝试写入或读取默认数据库
    doc_ref_default.set({"message": "Hello from default database"})
    doc_default = doc_ref_default.get()
    if doc_default.exists:
        print(f"从默认数据库 '{firestore_client_default.database_id}' 读取到数据: {doc_default.to_dict()}")
    else:
        print(f"默认数据库 '{firestore_client_default.database_id}' 中的文档不存在。")
except Exception as e:
    print(f"访问默认数据库时发生错误: {e}")

import firebase_admin
from firebase_admin import credentials
from google.cloud import firestore # 注意：这里是google.cloud.firestore

# 初始化Firebase Admin SDK，用于凭证管理和项目ID
# 如果您已经设置了GOOGLE_APPLICATION_CREDENTIALS环境变量，ApplicationDefault()会自动找到
# 否则，您可能需要提供 serviceAccountKey.json 路径
cred = credentials.ApplicationDefault()
# 确保projectId是您的Firebase/GCP项目ID
firebase_admin.initialize_app(cred, options={"projectId": "your-gcp-project-id"})

# 现在，使用google.cloud.firestore.Client来创建针对命名数据库的客户端
# 这里的projectId可以省略，因为firebase_admin.initialize_app已经设置了默认项目
try:
    firestore_client_named_via_admin = firestore.Client(database="my-another-db")

    # 示例：在命名数据库中查询数据
    docs = firestore_client_named_via_admin.collection("cities").stream()
    print(f"从命名数据库 '{firestore_client_named_via_admin.database_id}' 读取 'cities' 集合中的文档:")
    for doc in docs:
        print(f"  {doc.id} => {doc.to_dict()}")

except Exception as e:
    print(f"通过Firebase Admin SDK访问命名数据库时发生错误: {e}")

# 您仍然可以通过firebase_admin.firestore来获取默认数据库客户端
from firebase_admin import firestore as admin_firestore
default_db_client = admin_firestore.client()
print(f"\n默认数据库客户端的数据库ID: {default_db_client.database_id}")