From 1b00bbc890c1c6000af443249f852a7a5805ada6 Mon Sep 17 00:00:00 2001 From: brandonzhang Date: Tue, 8 Oct 2024 19:17:17 +0800 Subject: [PATCH 1/2] feat(endpoints): run scheduler through api by token --- app/api/endpoints/system.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/app/api/endpoints/system.py b/app/api/endpoints/system.py index 3b3e923d..9bfc6555 100644 --- a/app/api/endpoints/system.py +++ b/app/api/endpoints/system.py @@ -12,7 +12,7 @@ from app.chain.search import SearchChain from app.chain.system import SystemChain from app.core.config import settings, global_vars from app.core.module import ModuleManager -from app.core.security import verify_token, verify_uri_token +from app.core.security import verify_token, verify_uri_token, verify_apitoken from app.db.models import User from app.db.systemconfig_oper import SystemConfigOper from app.db.user_oper import get_current_active_superuser @@ -365,3 +365,15 @@ def run_scheduler(jobid: str, return schemas.Response(success=False, message="命令不能为空！") Scheduler().start(jobid) return schemas.Response(success=True) + +@router.get("/runscheduler2", summary="运行服务（API_TOKEN）", response_model=schemas.Response) +def run_scheduler2(jobid: str, + _: str = Depends(verify_apitoken)): + """ + 执行命令（API_TOKEN认证） + """ + if not jobid: + return schemas.Response(success=False, message="命令不能为空！") + + Scheduler().start(jobid) + return schemas.Response(success=True) From 78cbe1aaedb6e4a1290257e5c02ce5237d2296cd Mon Sep 17 00:00:00 2001 From: Cabbagec Date: Wed, 9 Oct 2024 11:54:49 +0800 Subject: [PATCH 2/2] fix: formatting app/api/endpoints/system.py --- app/api/endpoints/system.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/app/api/endpoints/system.py b/app/api/endpoints/system.py index 9bfc6555..e1eecf6c 100644 --- a/app/api/endpoints/system.py +++ b/app/api/endpoints/system.py @@ -366,9 +366,10 @@ def run_scheduler(jobid: str, Scheduler().start(jobid) return schemas.Response(success=True) + @router.get("/runscheduler2", summary="运行服务（API_TOKEN）", response_model=schemas.Response) def run_scheduler2(jobid: str, - _: str = Depends(verify_apitoken)): + _: str = Depends(verify_apitoken)): """ 执行命令（API_TOKEN认证） """