Python中Firestore文档变更监听问题求助及代码排查

我明白你之前用轮询的方式踩了请求超限的坑，换成官方的快照监听后又碰到了程序直接终止的情况——这其实是Python脚本里的一个常见小坑，我来帮你捋清楚原因并解决它。

问题根源 #

Firestore的on_snapshot监听器是在后台线程异步工作的，它会默默等待Firestore的变更推送。但你的脚本在设置完监听器后，主线程就执行完所有代码了，直接退出，后台的监听线程也跟着被强制终止，所以你看不到任何回调输出，程序就悄无声息结束了。

解决方案：让主线程保持运行 #

你需要让主线程处于等待状态，这样监听器才能持续工作。这里有两种简单实用的实现方式：

方式1：无限循环+睡眠（最直接） #

这是入门级的解决方案，让主线程一直挂起，避免退出：

import firebase_admin
from firebase_admin import firestore, credentials
import time

cred = credentials.Certificate("AdminSDK.json")
firebase_admin.initialize_app(cred)
db = firestore.client()

def on_snapshot(col_snapshot, changes, read_time):
    print(u'收到快照回调')
    print(u'当前NeedClassification集合的文档变化：')
    for change in changes:
        if change.type.name == 'ADDED':
            print(u'新增文档：{}'.format(change.document.id))
        elif change.type.name == 'MODIFIED':
            print(u'修改文档：{}'.format(change.document.id))
        elif change.type.name == 'REMOVED':
            print(u'删除文档：{}'.format(change.document.id))

col_query = db.collection(u'NeedClassification')
query_watch = col_query.on_snapshot(on_snapshot)

# 让主线程持续运行
try:
    while True:
        time.sleep(3600)  # 每小时唤醒一次，避免占用过多系统资源
except KeyboardInterrupt:
    # 捕获Ctrl+C，优雅关闭监听器
    query_watch.unsubscribe()
    print("监听器已关闭")

方式2：信号处理（更优雅的退出方式） #

如果你希望能通过系统信号（比如Ctrl+C）优雅退出程序，可以用signal模块实现：

import firebase_admin
from firebase_admin import firestore, credentials
import signal
import sys

cred = credentials.Certificate("AdminSDK.json")
firebase_admin.initialize_app(cred)
db = firestore.client()

def on_snapshot(col_snapshot, changes, read_time):
    print(u'收到快照回调')
    print(u'当前NeedClassification集合的文档变化：')
    for change in changes:
        if change.type.name == 'ADDED':
            print(u'新增文档：{}'.format(change.document.id))
        elif change.type.name == 'MODIFIED':
            print(u'修改文档：{}'.format(change.document.id))
        elif change.type.name == 'REMOVED':
            print(u'删除文档：{}'.format(change.document.id))

def graceful_exit(signal, frame):
    print('\n正在关闭监听器...')
    query_watch.unsubscribe()
    sys.exit(0)

col_query = db.collection(u'NeedClassification')
query_watch = col_query.on_snapshot(on_snapshot)

# 注册信号处理，支持Ctrl+C和系统终止信号
signal.signal(signal.SIGINT, graceful_exit)
signal.signal(signal.SIGTERM, graceful_exit)

print("监听器已启动，按Ctrl+C可退出")
# 让主线程挂起等待信号
signal.pause()

额外小贴士 #

• 一定要记得在退出时调用query_watch.unsubscribe()，这样能正确释放资源，避免不必要的连接泄漏。
• 快照监听器是Firestore主动推送变更，相比你之前的轮询方式，不仅不会触发请求超限，效率也高得多。
• 如果你的脚本需要长期在服务器后台运行，建议用进程管理工具（比如systemd）托管，防止脚本意外退出。

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