loy

app/api/test.py

@@ -0,0 +1,55 @@
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+from ..core.database import get_db
+from ..core.response_code import ResponseCode
+from ..schemas.comprehensive_data import DataResponse
+import logging
+from ..services.daily import DailyDataService
+from ..services.checkpoint import CheckpointService
+from ..services.settlement_data import SettlementDataService
+from ..utils.scheduler import scheduled_get_max_nyid_by_point_id
+# 导入全局定时任务调度器实例和目标任务函数
+from ..utils.scheduler import task_scheduler, scheduled_get_max_nyid_by_point_id
+
+router = APIRouter(prefix="/test", tags=["测试"])
+logger = logging.getLogger(__name__)
+
+@router.get("/trigger_max_nyid_task", response_model=DataResponse)
+def trigger_max_nyid_task(db: Session = Depends(get_db)):
+    """接口：直接触发调度器中的 scheduled_get_max_nyid_by_point_id 定时任务"""
+    try:
+        # 触发任务执行
+        # task_scheduler.run_job(scheduled_get_max_nyid_by_point_id.__name__)
+        # settlement = SettlementDataService()
+        # # 获取模型实例列表
+        # checkpoint_instances = settlement.get_by_nyid(db, nyid="4993546")
+        # # 转为字典列表（核心修正）
+        # checkpoint_data = [instance.__dict__ for instance in checkpoint_instances]
+        # # 清理 SQLAlchemy 内部属性（可选，避免多余字段）
+        # checkpoint_data = [{k: v for k, v in item.items() if not k.startswith('_sa_')} for item in checkpoint_data]
+
+        # return DataResponse(
+        #     code=ResponseCode.SUCCESS,
+        #     message="定时任务触发执行成功！任务已开始处理（具体结果查看系统日志）",
+        #     total=len(checkpoint_data),  # 修正为实际数据长度
+        #     data=checkpoint_data
+        # )
+
+        # scheduled_get_max_nyid_by_point_id()
+        daily_service = DailyDataService()
+        daily_data = daily_service.get_daily_data_by_account(db,account_id=1)
+        
+        return DataResponse(
+            code=ResponseCode.SUCCESS,
+            message="定时任务触发执行成功！任务已开始处理（具体结果查看系统日志）",
+            total=1,
+            data=daily_data
+        )
+    except Exception as e:
+        logger.error(f"接口触发定时任务失败：{str(e)}", exc_info=True)
+        return DataResponse(
+            code=ResponseCode.QUERY_FAILED,
+            message=f"定时任务触发失败：{str(e)}",
+            total=0,
+            data={}
+        )