我有一个用非重写版本的discord.py写的Discord bot,它发送一个类似心跳的消息(除其他外) . 我不知道我是否理解正确但是从测试中我发现我需要在 main.py
文件中使用 async def heartbeat()
函数 .
摘自 main.py
(心跳按预期工作):
[...]
import asyncio
import datetime
from configparser import ConfigParser
startup_time = datetime.datetime.utcnow()
[...]
async def heartbeat():
await bot.wait_until_ready()
heartbeat_config = ConfigParser()
heartbeat_config.read('./config/config.ini')
hb_freq = int(heartbeat_config.get('Heartbeat', 'hb_freq')) # frequency of heartbeat message
hb_channel = heartbeat_config.get('Heartbeat', 'hb_channel') # target channel of heartbeat message
hb_channel = bot.get_channel(hb_channel) # get channel from bot's channels
await bot.send_message(hb_channel, "Starting up at: `" + str(startup_time) + "`")
await asyncio.sleep(hb_freq) # sleep for hb_freq seconds before entering loop
while not bot.is_closed:
now = datetime.datetime.utcnow() # time right now
tdelta = now - startup_time # time since startup
tdelta = tdelta - datetime.timedelta(microseconds=tdelta.microseconds) # remove microseconds from tdelta
beat = await bot.send_message(hb_channel, "Still running\nSince: `" + str(startup_time) + "`.\nCurrent uptime: `" + str(tdelta))
await asyncio.sleep(hb_freq) # sleep for hb_freq seconds before initialising next beat
await bot.delete_message(beat) # delete old beat so it can be replaced
[...]
if __name__ == "__main__":
global heartbeat_task
heartbeat_task = bot.loop.create_task(heartbeat()) # creates heartbeat task in the background
bot.run(token) # run bot
我有一些应该与创建的 heartbeat_task
交互的命令,但是它们位于不同的模块中,称为 dev.py
(与 main.py
驻留在同一目录中) .
摘自 dev.py
:
[...]
from main import heartbeat_task, heartbeat
[...]
@commands.group(pass_context=True)
async def heart(self, ctx):
if ctx.invoked_subcommand is None:
return
@heart.command(pass_context=True)
async def stop(self, ctx):
# should cancel the task from main.py
heartbeat_task.cancel()
await self.bot.say('Heartbeat stopped by user {}'.format(ctx.message.author.name))
@heart.command(pass_context=True)
async def start(self, ctx):
# start the heartbeat if it is not running
global heartbeat_task
if heartbeat_task.cancelled():
heartbeat_task = self.bot.loop.create_task(heartbeat())
await self.bot.say('Heartbeat started by user {}'.format(ctx.message.author.name))
else:
return
[...]
当这些命令属于 main.py
(当然需要进行必要的调整,比如删除 self
,导入等)时,这些命令工作得非常好,但是因为我希望所有与开发人员相关的命令都进入他们自己的模块,所以我试着移动他们 .
我尝试加载模块时出现以下错误:
ImportError: cannot import name 'heartbeat_task'.
从 dev.py
删除该导入会导致模块成功加载,但在使用任一命令时,控制台会抛出错误:
NameError: name 'heartbeat_task' is not defined
哪条回溯到 heartbeat_task.cancel()
行(在 heart stop
// if heartbeat_task.cancelled():
的情况下(在 heart start
的情况下) .
现在我的问题 . 如何在 main.py
中使用async heartbeat()
但是使用 dev.py
模块中的命令影响任务?
如果我不能,那么在 dev.py
中保存命令的可行替代方法是什么(函数本身不需要保留在 main.py
中,但最好留在那里)?
(我已经搜索了很长时间,找不到像我这样的问题或者碰巧也适合我的解决方案)
1 回答
在cog中执行后台任务的最简单方法是向将启动后台任务的cog添加
on_ready
协程,而不是手动启动它:请注意,您不需要使用cog中的任何内容来装饰
on_ready
,add_cog
机器将根据其名称进行拾取 .