mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-10-03 16:53:32 +00:00
Wu Clan
c743182211
* Optimize serialization and algorithm performance * Fix get tree data to async * Removed CPU-intensive multi-threaded execution * Update pdm pre commit to 2.18.0 * Add jwt user info cache * Update ci * Fix ci * Fix redis delete prefix * Adapt to JWT user instance * Update jwt middleware token parse
55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Body, Depends, Path
|
|
|
|
from backend.app.task.service.task_service import task_service
|
|
from backend.common.response.response_schema import ResponseModel, response_base
|
|
from backend.common.security.jwt import DependsJwtAuth
|
|
from backend.common.security.permission import RequestPermission
|
|
from backend.common.security.rbac import DependsRBAC
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get('', summary='获取所有可执行任务模块', dependencies=[DependsJwtAuth])
|
|
async def get_all_tasks() -> ResponseModel:
|
|
tasks = task_service.get_list()
|
|
return response_base.success(data=tasks)
|
|
|
|
|
|
@router.get('/current', summary='获取当前正在执行的任务', dependencies=[DependsJwtAuth])
|
|
async def get_current_task() -> ResponseModel:
|
|
task = task_service.get()
|
|
return response_base.success(data=task)
|
|
|
|
|
|
@router.get('/{uid}/status', summary='获取任务状态', dependencies=[DependsJwtAuth])
|
|
async def get_task_status(uid: Annotated[str, Path(description='任务ID')]) -> ResponseModel:
|
|
status = task_service.get_status(uid)
|
|
return response_base.success(data=status)
|
|
|
|
|
|
@router.get('/{uid}', summary='获取任务结果', dependencies=[DependsJwtAuth])
|
|
async def get_task_result(uid: Annotated[str, Path(description='任务ID')]) -> ResponseModel:
|
|
task = task_service.get_result(uid)
|
|
return response_base.success(data=task)
|
|
|
|
|
|
@router.post(
|
|
'/{name}',
|
|
summary='执行任务',
|
|
dependencies=[
|
|
Depends(RequestPermission('sys:task:run')),
|
|
DependsRBAC,
|
|
],
|
|
)
|
|
async def run_task(
|
|
name: Annotated[str, Path(description='任务名称')],
|
|
args: Annotated[list | None, Body(description='任务函数位置参数')] = None,
|
|
kwargs: Annotated[dict | None, Body(description='任务函数关键字参数')] = None,
|
|
) -> ResponseModel:
|
|
task = task_service.run(name=name, args=args, kwargs=kwargs)
|
|
return response_base.success(data=task)
|