如何让 FastAPI 和 Telegram 机器人在一个代码中和谐共处？

python

2024-03-25 04:17:51

FastAPI 与 Telegram 机器人的和谐共存：在一个代码中实现

前言

在现代应用开发中，整合不同的技术栈以满足特定的业务需求已变得至关重要。本文探讨如何在同一代码中同时运行 FastAPI 应用程序和 Telegram 机器人，以实现二者的无缝协作。

问题概述

FastAPI 是一个流行的 Python 异步框架，用于构建 RESTful API，而 Telegram 机器人是用于创建基于消息的聊天机器人的一个平台。将这两项技术结合起来可以带来强大的可能性，但同时也会带来一些独特的挑战。

挑战：独立的事件循环

FastAPI 和 Telegram 机器人通常通过独立的事件循环运行。这种分离导致两个应用程序无法共享数据或同时响应事件。

解决办法：统一事件循环

为了解决这一挑战，我们引入了一个异步函数，将 FastAPI 应用程序和 Telegram 机器人轮询函数包装在一起。通过使用 asyncio.run() 在主线程中运行此函数，我们将这两个事件循环统一到了一个单一的事件循环中。

实现细节

以下是一个示例代码，展示了如何同时运行 FastAPI 应用程序和 Telegram 机器人：

import asyncio
from fastapi import FastAPI
from telegram.ext import Application, PicklePersistence, ApplicationBuilder

# 创建 FastAPI 应用程序
f_app = FastAPI()

# 创建 Telegram 机器人应用程序
pers = PicklePersistence("storage")
bot_app = ApplicationBuilder().token("YOUR_TELEGRAM_BOT_TOKEN").persistence(pers).build()

# 异步函数，同时运行 FastAPI 应用程序和 Telegram 机器人轮询
async def run_all():
    # 启动 FastAPI 应用程序
    uvicorn.run(f_app, port=9001)

    # 启动 Telegram 机器人轮询
    await bot_app.run_polling()

# 在主线程中运行异步函数
asyncio.run(run_all())

注意事项

确保 run_all() 函数中使用的事件循环与 FastAPI 应用程序和 Telegram 机器人轮询使用的事件循环相同。
如果需要在 FastAPI 应用程序中访问 Telegram 机器人数据，可以使用 f_app.state.bot_app 访问 Telegram 机器人应用程序对象。