aiogram中的处理程序在子进程日志记录后停止工作。
创始人
2024-07-31 22:01:43
0

在aiogram的代码中添加一个asyncio的Event Loop,然后在关键部分使用run_in_executor方法来确保日志记录不会阻塞io运行。

以下是示例代码:

import asyncio
import logging
import subprocess

from aiogram import Bot, Dispatcher, types

# 配置日志
logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)


async def on_startup(dp):
    asyncio.ensure_future(subprocess_logging())


async def subprocess_logging():
    # 子进程运行一个脚本并记录输出
    proc = await asyncio.create_subprocess_shell(
        "python my_script.py", stdout=subprocess.PIPE, stderr=subprocess.PIPE
    )

    # 日志输出处理程序
    async def logger_output(fd, level):
        while True:
            data = await fd.readline()
            if not data:
                break
            logger.log(level, data.decode())

    # 记录标准输出和标准错误
    asyncio.ensure_future(logger_output(proc.stdout, logging.INFO))
    asyncio.ensure_future(logger_output(proc.stderr, logging.ERROR))

    # 等待子进程结束并记录退出码
    returncode = await proc.wait()
    logger.info(f"子进程已退出, 返回码: {returncode}")


async def start_bot():
    bot = Bot(token="BOT_TOKEN_HERE")
    dp = Dispatcher(bot=bot)
    dp.register_message_handler(echo, content_types=types.ContentType.TEXT)
    dp.register_message_handler(photo_handler, content_types=types.ContentType.PHOTO)
    dp.register_message_handler(video_handler, content_types=types.ContentType.VIDEO)

    # 这里添加一个asyncio的Event Loop
    loop = asyncio.get_event_loop()
    try:
        asyncio.ensure_future(on_startup(dp))
        await dp.start_polling()
    finally:
        await bot.session.close()
        loop.close()


async def echo(message):
    await message.answer(message.text)


async def photo_handler(message):
    photo_file_id = message.photo[-1].file_id
    await bot.send_photo(chat_id=message.chat.id, file_id=photo_file_id)


async def video_handler(message):
    video_file_id = message.video.file_id
    await bot.send_video(chat_id=message.chat.id, file_id=video_file_id)


if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    loop.run_until_complete(start_bot())

相关内容

热门资讯

安装apache-beam==... 出现此错误可能是因为用户的Python版本太低,而apache-beam==2.34.0需要更高的P...
避免在粘贴双引号时向VS 20... 在粘贴双引号时向VS 2022添加反斜杠的问题通常是由于编辑器的自动转义功能引起的。为了避免这个问题...
Android Recycle... 要在Android RecyclerView中实现滑动卡片效果,可以按照以下步骤进行操作:首先,在项...
omi系统和安卓系统哪个好,揭... OMI系统和安卓系统哪个好?这个问题就像是在问“苹果和橘子哪个更甜”,每个人都有自己的答案。今天,我...
原生ios和安卓系统,原生对比... 亲爱的读者们,你是否曾好奇过,为什么你的iPhone和安卓手机在操作体验上有着天壤之别?今天,就让我...
Android - 无法确定任... 这个错误通常发生在Android项目中,表示编译Debug版本的Java代码时出现了依赖关系问题。下...
Android - NDK 预... 在Android NDK的构建过程中,LOCAL_SRC_FILES只能包含一个项目。如果需要在ND...
Akka生成Actor问题 在Akka框架中,可以使用ActorSystem对象生成Actor。但是,当我们在Actor类中尝试...
Agora-RTC-React... 出现这个错误原因是因为在 React 组件中使用,import AgoraRTC from “ago...
安装了Anaconda之后找不... 在安装Anaconda后,如果找不到Jupyter Notebook,可以尝试以下解决方法:检查环境...