增加 APScheduler 模块

2 files changed, 6 insertions, 2 deletions

diff --git a/server/app.py b/server/app.py
index 43dca54..1870c91 100644
--- a/server/app.py
+++ b/server/app.py
@@ -8,7 +8,7 @@ from apps.sysinfo import bp as sysbp
 # from apps.sysmange import bp as mangbp
 from apps.target import bp as targetbp
 from apps.task import bp as taskbp
-from exts import db, login_manager, CORS, Migrate
+from exts import db, login_manager, CORS, Migrate, scheduler
 
 # 注册蓝图
 app = APIFlask(__name__, template_folder='./static/templates')
@@ -38,6 +38,8 @@ db.init_app(app)
 migrate = Migrate(app, db)
 
 login_manager.init_app(app)
+scheduler.init_app(app)
+scheduler.start()
 
 @app.get('/')
 @app.doc("获取测试页面")
diff --git a/server/exts.py b/server/exts.py
index 84a79cb..dccacdb 100644
--- a/server/exts.py
+++ b/server/exts.py
@@ -2,6 +2,8 @@ from flask_sqlalchemy import SQLAlchemy
 from flask_login import LoginManager
 from flask_cors import CORS
 from flask_migrate import Migrate
+from flask_apscheduler import APScheduler
 
 login_manager = LoginManager()
-db = SQLAlchemy()
\ No newline at end of file
+db = SQLAlchemy()
+scheduler = APScheduler()
\ No newline at end of file