python之APScheduler
介绍
APScheduler是Python中一个强大的第三方库,用于在后台执行定时任务。它允许我们根据设定的时间间隔、日期规则或特定时间来执行任务,适用于定时执行脚本、定时发送邮件、定时处理数据等场景。
APScheduler的功能使得在Python中实现定时任务变得非常简单和高效。
1. 安装和导入
首先,我们需要安装APScheduler库。可以使用pip命令进行安装:
pip install apscheduler
安装完成后,我们可以在Python代码中导入APScheduler:
from apscheduler.schedulers.background import BackgroundScheduler
2. 创建定时任务
APScheduler提供了BackgroundScheduler和BlockingScheduler两种类型的调度器,用于创建定时任务。
BackgroundScheduler在后台运行,不会阻塞主线程;
BlockingScheduler会阻塞主线程直到所有任务完成。
from apscheduler.schedulers.background import BackgroundScheduler
import time
# 创建后台调度器
scheduler = BackgroundScheduler()
# 定义任务函数
def job():
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,每隔5秒执行一次
scheduler.add_job(job, 'interval', seconds=5)
# 启动调度器
scheduler.start()
# 主线程等待一段时间后结束
time.sleep(20)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们首先创建了一个后台调度器scheduler,然后定义了一个名为job的任务函数,在其中打印当前时间。使用scheduler.add_job()添加了一个定时任务,设置为每隔5秒执行一次。
然后,我们启动了调度器scheduler,让定时任务在后台执行。主线程等待20秒后结束,并调用scheduler.shutdown()关闭调度器。
3. 定时任务触发器
APScheduler提供了多种触发器类型,用于设置定时任务的触发条件。
interval触发器: 按照设定的时间间隔来触发任务。
from apscheduler.schedulers.background import BackgroundScheduler
import time
# 创建后台调度器
scheduler = BackgroundScheduler()
# 定义任务函数
def job():
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,每隔5秒执行一次
scheduler.add_job(job, 'interval', seconds=5)
# 启动调度器
scheduler.start()
# 主线程等待一段时间后结束
time.sleep(20)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们使用'interval'触发器,设置任务每隔5秒执行一次。
cron触发器: 使用类似于Linux中cron表达式的规则来触发任务,可以精确到秒。
from apscheduler.schedulers.background import BackgroundScheduler
import time
# 创建后台调度器
scheduler = BackgroundScheduler()
# 定义任务函数
def job():
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,每天的13点30分触发任务
scheduler.add_job(job, 'cron', hour=13, minute=30)
# 启动调度器
scheduler.start()
# 主线程等待一段时间后结束
time.sleep(60)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们使用'cron'触发器,设置任务每天的13点30分触发。
date触发器: 在指定的时间点触发任务。
from apscheduler.schedulers.background import BackgroundScheduler
import time
# 创建后台调度器
scheduler = BackgroundScheduler()
# 定义任务函数
def job():
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,设置任务在2023年7月31日10点30分触发
scheduler.add_job(job, 'date', run_date='2023-07-31 10:30:00')
# 启动调度器
scheduler.start()
# 主线程等待一段时间后结束
time.sleep(60)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们使用'date'触发器,设置任务在2023年7月31日10点30分触发。
4. 任务存储
APScheduler支持将任务存储在不同的后端存储中,如内存、数据库等。默认情况下,任务是存储在内存中的。
from apscheduler.schedulers.background import BackgroundScheduler
import time
# 创建后台调度器
scheduler = BackgroundScheduler()
# 定义任务函数
def job():
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,每隔5秒执行一次
scheduler.add_job(job, 'interval', seconds=5)
# 启动调度器
scheduler.start()
# 主线程等待一段时间后结束
time.sleep(20)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们使用默认的内存存储来存储任务。
如果需要将任务存储在数据库中,可以使用jobstores参数来设置。
在上述代码中,我们使用了SQLAlchemyJobStore来将任务存储在SQLite数据库中。
5. 并发执行
默认情况下,APScheduler会将任务串行执行,也就是说一个任务结束后才会执行下一个任务。如果希望并发执行多个任务,可以使用max_instances参数来设置。
from apscheduler.schedulers.background import BackgroundScheduler
import time
# 创建后台调度器
scheduler = BackgroundScheduler()
# 定义任务函数
def job(index):
print(f"定时任务{index}执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,每隔5秒执行一次,最多并发3个任务
scheduler.add_job(job, 'interval', seconds=5, args=[1], max_instances=3)
scheduler.add_job(job, 'interval', seconds=5, args=[2], max_instances=3)
scheduler.add_job(job, 'interval', seconds=5, args=[3], max_instances=3)
# 启动调度器
scheduler.start()
# 主线程等待一段时间后结束
time.sleep(20)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们使用了args参数传递参数给任务函数,并使用max_instances参数设置最多并发3个任务。
6. 阻塞和非阻塞
APScheduler提供了阻塞和非阻塞两种调度器类型。
阻塞调度器: 在调度器启动后,会阻塞主线程直到所有任务完成。
from apscheduler.schedulers.blocking import BlockingScheduler
import time
# 创建阻塞调度器
scheduler = BlockingScheduler()
# 定义任务函数
def job():
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,每隔5秒执行一次
scheduler.add_job(job, 'interval', seconds=5)
# 启动调度器
scheduler.start()
print("主线程结束")
非阻塞调度器: 在调度器启动后,不会阻塞主线程。
from apscheduler.schedulers.background import BackgroundScheduler
import time
# 创建后台调度器
scheduler = BackgroundScheduler()
# 定义任务函数
def job():
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,每隔5秒执行一次
scheduler.add_job(job, 'interval', seconds=5)
# 启动调度器
scheduler.start()
# 主线程等待一段时间后结束
time.sleep(20)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们分别使用BlockingScheduler和BackgroundScheduler创建了阻塞和非阻塞调度器。
7. 错误处理
在任务执行过程中,可能会出现异常。APScheduler提供了异常处理机制,我们可以通过try...except...捕获任务函数中的异常,并进行相应的处理。
from apscheduler.schedulers.background import BackgroundScheduler
import time
# 创建后台调度器
scheduler = BackgroundScheduler()
# 定义任务函数
def job():
try:
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 抛出一个异常
raise ValueError("任务出现异常")
except Exception as e:
print("任务执行过程中发生异常:", str(e))
# 添加定时任务,每隔5秒执行一次
scheduler.add_job(job, 'interval', seconds=5)
# 启动调度器
scheduler.start()
# 主线程等待一段时间后结束
time.sleep(20)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们在任务函数中抛出了一个ValueError异常,并通过try...except...捕获并输出了异常信息。
8. 立即执行任务
有时候我们可能需要立即执行一个任务,而不是等到下次触发时间。APScheduler提供了run_job方法来立即执行任务。
from apscheduler.schedulers.background import BackgroundScheduler
import time
# 创建后台调度器
scheduler = BackgroundScheduler()
# 定义任务函数
def job():
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,每隔5秒执行一次
scheduler.add_job(job, 'interval', seconds=5)
# 启动调度器
scheduler.start()
# 立即执行任务
scheduler.run_job(job)
# 主线程等待一段时间后结束
time.sleep(20)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们使用scheduler.run_job(job)方法立即执行了任务。
9. 调度器持久化
在实际应用中,我们可能需要将调度器的配置保存到文件中,以便在下次启动时恢复。
from apscheduler.schedulers.background import BackgroundScheduler
from apscheduler.jobstores.sqlalchemy import SQLAlchemyJobStore
import time
# 创建数据库存储
jobstores = {
'default': SQLAlchemyJobStore(url='sqlite:///jobs.sqlite')
}
# 创建后台调度器,并指定jobstores参数
scheduler = BackgroundScheduler(jobstores=jobstores)
# 定义任务函数
def job():
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,每隔5秒执行一次
scheduler.add_job(job, 'interval', seconds=5)
# 启动调度器
scheduler.start()
# 主线程等待一段时间后结束
time.sleep(20)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们创建了一个数据库存储jobstores,并在创建后台调度器时指定了jobstores参数。这样,在调度器运行过程中,任务的配置将会被持久化到数据库中。
10. 任务监听器
APScheduler提供了任务监听器,用于监听任务的状态变化。我们可以通过add_listener方法添加监听器,并在任务状态发生变化时进行相应的处理。
from apscheduler.schedulers.background import BackgroundScheduler
import time
# 创建后台调度器
scheduler = BackgroundScheduler()
# 定义任务函数
def job():
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,每隔5秒执行一次
scheduler.add_job(job, 'interval', seconds=5)
# 定义任务监听器
def my_listener(event):
if event.exception:
print("任务执行过程中发生异常:", str(event.exception))
else:
print("任务执行成功")
# 添加任务监听器
scheduler.add_listener(my_listener, mask='all')
# 启动调度器
scheduler.start()
# 主线程等待一段时间后结束
time.sleep(20)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们创建了一个任务监听器my_listener,并在任务执行过程中通过if...else...判断是否出现异常。然后通过scheduler.add_listener(my_listener, mask='all')方法添加了监听器。
11. 移除定时任务
如果我们希望在调度器运行过程中移除某个定时任务,可以使用scheduler.remove_job(job_id)方法。
from apscheduler.schedulers.background import BackgroundScheduler
import time
# 创建后台调度器
scheduler = BackgroundScheduler()
# 定义任务函数
def job():
print("定时任务执行:", time.strftime("%Y-%m-%d %H:%M:%S"))
# 添加定时任务,每隔5秒执行一次,并获取任务ID
job_id = scheduler.add_job(job, 'interval', seconds=5).id
# 启动调度器
scheduler.start()
# 主线程等待一段时间后移除定时任务
time.sleep(10)
scheduler.remove_job(job_id)
# 主线程等待一段时间后结束
time.sleep(10)
# 关闭调度器
scheduler.shutdown()
print("主线程结束")
在上述代码中,我们通过scheduler.add_job(job, 'interval', seconds=5).id获取了定时任务的ID,并使用scheduler.remove_job(job_id)移除了定时任务。