diff --git a/applications/common/tasks/events.py b/applications/common/tasks/events.py new file mode 100644 index 0000000..e49411e --- /dev/null +++ b/applications/common/tasks/events.py @@ -0,0 +1,54 @@ +from apscheduler.events import ( + EVENT_JOB_ADDED, + EVENT_JOB_ERROR, + EVENT_JOB_EXECUTED, + EVENT_JOB_MISSED, + EVENT_JOB_REMOVED, + EVENT_JOB_SUBMITTED, +) + +from applications.extensions.init_apscheduler import scheduler + + +def job_missed(event): + """Job missed event.""" + with scheduler.app.app_context(): + print(event) # noqa: T001 + + +def job_error(event): + """Job error event.""" + with scheduler.app.app_context(): + print(event) # noqa: T001 + + +def job_executed(event): + """Job executed event.""" + with scheduler.app.app_context(): + print(event) # noqa: T001 + + +def job_added(event): + """Job added event.""" + with scheduler.app.app_context(): + print(event) # noqa: T001 + + +def job_removed(event): + """Job removed event.""" + with scheduler.app.app_context(): + print(event) # noqa: T001 + + +def job_submitted(event): + """Job scheduled to run event.""" + with scheduler.app.app_context(): + print(event) # noqa: T001 + + +scheduler.add_listener(job_missed, EVENT_JOB_MISSED) +scheduler.add_listener(job_error, EVENT_JOB_ERROR) +scheduler.add_listener(job_executed, EVENT_JOB_EXECUTED) +scheduler.add_listener(job_added, EVENT_JOB_ADDED) +scheduler.add_listener(job_removed, EVENT_JOB_REMOVED) +scheduler.add_listener(job_submitted, EVENT_JOB_SUBMITTED) diff --git a/applications/common/tasks/tasks.py b/applications/common/tasks/tasks.py new file mode 100644 index 0000000..cac622a --- /dev/null +++ b/applications/common/tasks/tasks.py @@ -0,0 +1,51 @@ +import datetime +from flask import current_app + +from applications.extensions import db +from applications.extensions.init_apscheduler import scheduler +from applications.models import Dept + + +# @scheduler.task( +# "interval", +# id="job_sync", +# seconds=10, +# max_instances=1, +# start_date="2000-01-01 12:19:00", +# ) +# def task1(): +# # oh, do you need something from config? +# with scheduler.app.app_context(): +# print(scheduler.app.config) # noqa: T001 +# address = "ces" +# deptName = "测试" +# email = "123@qq.com" +# leader = "测试的" +# parentId = "1" +# phone = "123456" +# sort = "5" +# status = 1 +# dept = Dept( +# parent_id=parentId, +# dept_name=deptName, +# sort=sort, +# leader=leader, +# phone=phone, +# email=email, +# status=status, +# address=address +# ) +# r = db.session.add(dept) +# db.session.commit() + + +def task2(a, b): + print(f'定时任务_1_{a},{b},{datetime.datetime.now()}') + + +def task3(a, b): + print(f'定时任务_2_{a}{b}{datetime.datetime.now()}') + + +def task4(a, b): + print(f'定时任务_4_{a}{b}{datetime.datetime.now()}') diff --git a/applications/extensions/__init__.py b/applications/extensions/__init__.py index 87ac09c..75e800d 100644 --- a/applications/extensions/__init__.py +++ b/applications/extensions/__init__.py @@ -6,6 +6,7 @@ from .init_debug_tool import init_debug_tool from .init_template_directives import init_template_directives from .init_error_views import init_error_views from .init_mail import init_mail +from .init_apscheduler import init_scheduler def init_plugs(app: Flask) -> None: @@ -15,3 +16,4 @@ def init_plugs(app: Flask) -> None: init_template_directives(app) init_error_views(app) init_mail(app) + init_scheduler(app) diff --git a/applications/extensions/init_apscheduler.py b/applications/extensions/init_apscheduler.py new file mode 100644 index 0000000..18ae38d --- /dev/null +++ b/applications/extensions/init_apscheduler.py @@ -0,0 +1,12 @@ +from flask import Flask +from flask_apscheduler import APScheduler + +scheduler = APScheduler() + + +def init_scheduler(app: Flask): + scheduler.init_app(app) + with app.app_context(): + from applications.common.tasks import tasks + scheduler.start() + from applications.common.tasks import events diff --git a/applications/view/admin/__init__.py b/applications/view/admin/__init__.py index ebb9b47..9b58fcc 100644 --- a/applications/view/admin/__init__.py +++ b/applications/view/admin/__init__.py @@ -8,6 +8,7 @@ from applications.view.admin.power import admin_power from applications.view.admin.role import admin_role from applications.view.admin.user import admin_user from applications.view.admin.monitor import admin_monitor_bp +from applications.view.admin.task import admin_task def register_admin_views(app: Flask): @@ -19,3 +20,4 @@ def register_admin_views(app: Flask): app.register_blueprint(admin_power) app.register_blueprint(admin_role) app.register_blueprint(admin_dict) + app.register_blueprint(admin_task) diff --git a/applications/view/admin/admin_log.py b/applications/view/admin/admin_log.py index cc68c6a..4b582f1 100644 --- a/applications/view/admin/admin_log.py +++ b/applications/view/admin/admin_log.py @@ -1,6 +1,5 @@ from flask import Blueprint, request, render_template from sqlalchemy import desc - from applications.common.utils.http import table_api from applications.common.utils.rights import authorize from applications.models import AdminLog diff --git a/applications/view/admin/task.py b/applications/view/admin/task.py new file mode 100644 index 0000000..47122db --- /dev/null +++ b/applications/view/admin/task.py @@ -0,0 +1,61 @@ +from flask import Blueprint, request, jsonify +from applications.common.tasks.tasks import task4 +from applications.extensions.init_apscheduler import scheduler +from flask_apscheduler.utils import job_to_dict + +admin_task = Blueprint('adminTask', __name__, url_prefix='/admin/task') + + +# 暂停任务 +# scheduler.pause_job('third') +# 恢复任务 +# time.sleep(10) +# scheduler.resume_job('third') +# 删除任务 +# scheduler.remove_job('first') + +@admin_task.route('/pause', methods=['GET']) +def pause_job(): # 暂停 + job_id = request.args.get('id') + scheduler.pause_job(str(job_id)) + return "pause success!" + + +@admin_task.route('/resume', methods=['GET']) +def resume_job(): # 恢复 + job_id = request.args.get('id') + scheduler.resume_job(str(job_id)) + return "Success!" + + +@admin_task.route('/get_jobs', methods=['GET']) +def get_task(): # 获取 + jobs = scheduler.get_jobs() + jobs_list = [] + for job in jobs: + jobs_list.append(job_to_dict(job)) + return jsonify(jobs_list) + + +@admin_task.route('/remove_job', methods=['GET']) +def remove_job(): # 移除 + job_id = request.args.get('id') + scheduler.remove_job(str(job_id)) + return 'remove success' + + +@admin_task.route('/add_job', methods=['GET']) +def add_task(): + scheduler.add_job(func=task4, id='4', args=(1, 1), trigger='interval', seconds=3, + replace_existing=True) + return '6' + + # scheduler.add_job(func=task1, id='2', args=(1, 1), trigger='cron', day_of_week='0-6', hour=18, minute=24, + # second=10, replace_existing=True) + + # scheduler.add_job(func=task4, id='4', args=(2, 2), trigger='interval', seconds=3, + # replace_existing=True, misfire_grace_time=3) + + # # trigger='interval' 表示是一个循环任务，每隔多久执行一次 + # scheduler.add_job(func=task2, id='3', args=(2, 2), trigger='interval', seconds=3, + # replace_existing=True)