添加令牌验证和推荐系统接口
Change-Id: I2d318c167683a5106ced96047ab9f39b33b75bd1
diff --git a/Merge/back_rhj/app/__init__.py b/Merge/back_rhj/app/__init__.py
index c50a674..098898c 100644
--- a/Merge/back_rhj/app/__init__.py
+++ b/Merge/back_rhj/app/__init__.py
@@ -1,5 +1,7 @@
from flask import Flask
from flask_cors import CORS
+import atexit
+import logging
def create_app():
app = Flask(__name__)
@@ -17,6 +19,26 @@
# Register recommendation blueprint
from .blueprints.recommend import recommend_bp
app.register_blueprint(recommend_bp)
+
+ # Register scheduler blueprint
+ from .blueprints.scheduler import scheduler_bp
+ app.register_blueprint(scheduler_bp)
+
+ # 初始化定时任务管理器
+ from .utils.scheduler_manager import SchedulerManager
+
+ scheduler_manager = SchedulerManager()
+ scheduler_manager.init_scheduler(app)
+
+ # 检查是否启用定时任务
+ scheduler_enabled = getattr(app.config, 'SCHEDULER_ENABLED', True)
+ if scheduler_enabled:
+ # 从配置获取重建间隔
+ rebuild_interval = getattr(app.config, 'GRAPH_REBUILD_INTERVAL', 1)
+ scheduler_manager.start_graph_rebuild_task(interval_minutes=rebuild_interval)
+
+ # 注册关闭时的清理函数
+ atexit.register(lambda: scheduler_manager.shutdown())
return app