我正在使用 discord.py 制作一个 Discord 机器人,有时我用来运行它的脚本需要关闭机器人。当我在不使用信号处理程序的情况下关闭它时,有很多关于循环未关闭的错误,所以我添加了一个信号处理程序(使用下面的代码),在里面我需要调用 client.close()
和 client.logout()
,但问题是那些是异步函数,因此需要我等待它们,但我不能等待这些函数,因为信号处理程序不能是异步函数。
这是代码:
def handler():
print("Logging out of Discord Bot")
client.logout()
client.close()
sys.exit()
@client.event
async def on_ready():
print('We have logged in as {0.user}'.format(client))
for signame in ('SIGINT', 'SIGTERM'):
client.loop.add_signal_handler(getattr(signal, signame),
lambda: asyncio.ensure_future(handler()))
有没有办法使用信号处理程序正确注销,或者至少只是使代码内部的警告和错误静音,以便在控制台中不打印错误。
最佳答案
您的方法走在正确的轨道上 - 因为 add_signal_handler
需要一个普通函数而不是异步函数,你需要调用 ensure_future
(或其表亲 create_task
)提交异步函数以在事件循环中运行。下一步是实际制作handler
异步和 await
它调用的协程:
async def handler():
print("Logging out of Discord Bot")
await client.logout()
await client.close()
asyncio.get_event_loop().stop()
注意我改了 sys.exit()
显式停止事件循环,因为 asyncio 对 sys.exit()
react 不佳从回调中间调用(它捕获 SystemExit
异常并提示未检索到的异常)。由于我没有要测试的不和谐,我通过更改注销并关闭 sleep 来测试它:
import asyncio, signal
async def handler():
print("sleeping a bit...")
await asyncio.sleep(0.2)
print('exiting')
asyncio.get_event_loop().stop()
def setup():
loop = asyncio.get_event_loop()
for signame in ('SIGINT', 'SIGTERM'):
loop.add_signal_handler(getattr(signal, signame),
lambda: asyncio.create_task(handler()))
setup()
asyncio.get_event_loop().run_forever()
如果您使用的不是 run_forever
的东西启动事件循环,例如 asyncio.run(some_function())
,那么您将需要替换 loop.stop()
使用设置主协程等待的任何事件的代码。例如,如果它等待 server.serve_forever()
在服务器上,然后您将服务器对象传递给 handler
并调用 server.close()
, 等等。
关于python - 在 Signal Handler 中调用 Async 函数,Asyncio,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66510288/