mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-12-01 09:34:46 +08:00
162 lines
5.2 KiB
Markdown
162 lines
5.2 KiB
Markdown
# 计划任务
|
||
|
||
NoneBot 可选地内置了计划任务功能，在指南的 [添加计划任务](../guide/scheduler.md) 已经进行了简单的介绍。这里列出更多常见的用法。
|
||
|
||
## 固定的计划任务
|
||
|
||
可以利用固定的*触发器*（trigger）来触发某些任务。
|
||
|
||
### 一次性任务
|
||
|
||
[`date` 触发器](https://apscheduler.readthedocs.io/en/stable/modules/triggers/date.html#module-apscheduler.triggers.date) 固定时间触发，仅触发一次：
|
||
|
||
```python
|
||
from datetime import datetime
|
||
|
||
@nonebot.scheduler.scheduled_job(
|
||
'date',
|
||
run_date=datetime(2021, 1, 1, 0, 0),
|
||
# timezone=None,
|
||
)
|
||
async def _():
|
||
await bot.send_group_msg(group_id=123456,
|
||
message="2021，新年快乐！")
|
||
```
|
||
|
||
### 定期任务
|
||
|
||
[`cron` 触发器](https://apscheduler.readthedocs.io/en/stable/modules/triggers/cron.html#module-apscheduler.triggers.cron) 从 `start_date` 开始到 `end_date` 结束，根据类似 [Cron](https://wiki.archlinux.org/index.php/Cron_(%E7%AE%80%E4%BD%93%E4%B8%AD%E6%96%87)) 的规则触发任务：
|
||
|
||
```python
|
||
@nonebot.scheduler.scheduled_job(
|
||
'cron',
|
||
# year=None,
|
||
# month=None,
|
||
# day=None,
|
||
# week=None,
|
||
day_of_week="mon,tue,wed,thu,fri",
|
||
hour=7,
|
||
# minute=None,
|
||
# second=None,
|
||
# start_date=None,
|
||
# end_date=None,
|
||
# timezone=None,
|
||
)
|
||
async def _():
|
||
await bot.send_group_msg(group_id=123456,
|
||
message="起床啦！")
|
||
```
|
||
|
||
### 间隔任务
|
||
|
||
[`interval` 触发器](https://apscheduler.readthedocs.io/en/stable/modules/triggers/interval.html#module-apscheduler.triggers.interval) 从 `start_date` 开始，每间隔一段时间触发，到 `end_date` 结束：
|
||
|
||
```python
|
||
@nonebot.scheduler.scheduled_job(
|
||
'interval',
|
||
# weeks=0,
|
||
# days=0,
|
||
# hours=0,
|
||
minutes=5,
|
||
# seconds=0,
|
||
# start_date=time.now(),
|
||
# end_date=None,
|
||
)
|
||
async def _():
|
||
has_new_item = check_new_item()
|
||
if has_new_item:
|
||
await bot.send_group_msg(group_id=123456,
|
||
message="XX有更新啦！")
|
||
```
|
||
|
||
## 动态的计划任务
|
||
|
||
有时，我们需要机器人在运行的过程中，添加或删除计划任务，那么我们就需要 `scheduler.add_job` 来帮忙。这里，我们以*一次性任务*为例，其他类型的任务可以用相同的方法：
|
||
|
||
```python
|
||
import datetime
|
||
|
||
from apscheduler.triggers.date import DateTrigger # 一次性触发器
|
||
# from apscheduler.triggers.cron import CronTrigger # 定期触发器
|
||
# from apscheduler.triggers.interval import IntervalTrigger # 间隔触发器
|
||
from nonebot import on_command, scheduler
|
||
|
||
@on_command('赖床')
|
||
async def _(session: CommandSession):
|
||
await session.send('我会在5分钟后再喊你')
|
||
|
||
# 制作一个“5分钟后”触发器
|
||
delta = datetime.timedelta(minutes=5)
|
||
trigger = DateTrigger(
|
||
run_date=datetime.datetime.now() + delta
|
||
)
|
||
|
||
# 添加任务
|
||
scheduler.add_job(
|
||
func=session.send, # 要添加任务的函数，不要带参数
|
||
trigger=trigger, # 触发器
|
||
args=('不要再赖床啦！',), # 函数的参数列表，注意：只有一个值时，不能省略末尾的逗号
|
||
# kwargs=None,
|
||
misfire_grace_time=60, # 允许的误差时间，建议不要省略
|
||
# jobstore='default', # 任务储存库，在下一小节中说明
|
||
)
|
||
```
|
||
|
||
<!-- ## 持久化存储任务
|
||
|
||
有时，我们动态添加的一些计划任务需要持久化储存，否则任务会在重启后丢失，这时我们就需要 `jobstore` 来帮忙。
|
||
|
||
APScheduler 可以将任务存储在内存中或数据库中，默认 jobstore 将所有任务储存在内存中，关闭后即丢失。这里，我们以 SQLite 为例，将任务添加到数据库中。
|
||
|
||
我们先创建一个数据库：
|
||
|
||
```python
|
||
import asyncio
|
||
import os
|
||
|
||
from apscheduler.jobstores.sqlalchemy import SQLAlchemyJobStore
|
||
from nonebot import on_command, scheduler, get_bot
|
||
|
||
database_path = os.path.join(os.path.dirname(__file__), 'job_store.db')
|
||
|
||
store = SQLAlchemyJobStore(url='sqlite:///'+database_path)
|
||
|
||
scheduler.add_jobstore(store, alias='my_job_store')
|
||
```
|
||
|
||
之后，我们在添加新任务时，可以指定任务的储存库
|
||
|
||
```python
|
||
async def alarm(*args, **kwargs):
|
||
bot = get_bot()
|
||
await bot.send(*args, **kwargs)
|
||
|
||
@on_command('提醒收菜')
|
||
async def _(session: CommandSession):
|
||
await session.send('我会在一天后提醒你收菜')
|
||
delta = datetime.timedelta(days=1)
|
||
trigger = DateTrigger(
|
||
run_date=datetime.datetime.now() + delta
|
||
)
|
||
|
||
# 添加任务
|
||
scheduler.add_job(
|
||
func=alarm,
|
||
trigger=trigger,
|
||
kwargs = {
|
||
'context': session.ctx,
|
||
'message': '起床收菜啦！',
|
||
},
|
||
misfire_grace_time=60,
|
||
jobstore='my_job_store', # 任务储存库，指定为刚才创建的储存库
|
||
)
|
||
```
|
||
|
||
**踩坑预警：**
|
||
|
||
由于 `apscheduler` 自带的 jobstore 无法将协程任务储存进数据库，
|
||
所以必须将任务转化为同步任务再储存。
|
||
并且 `apscheduler` 中的 `AsyncIOScheduler` 在执行同步任务时，会新建一个执行器（executor），
|
||
导致这个任务里无法使用 `asyncio.get_running_loop()` 来获取事件循环，
|
||
只能使用 `asyncio.run(...)` 来运行异步函数。 -->
|