49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
from fastapi import APIRouter, BackgroundTasks, HTTPException
|
|
from wechat_auto.models.activity import ActivityModel, TaskStatus
|
|
from wechat_auto.core.task_scheduler import task_scheduler
|
|
from wechat_auto.utils.logger import logger
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/api/publish", response_model=dict)
|
|
async def publish_activity(activity: ActivityModel, background_tasks: BackgroundTasks):
|
|
logger.info(f"收到发布活动请求: {activity.title}")
|
|
|
|
result = await task_scheduler.publish_activity(activity)
|
|
|
|
return {
|
|
"code": 200 if result["status"] == "success" else 500,
|
|
"message": "任务已提交" if result["status"] == "success" else "任务失败",
|
|
"data": result
|
|
}
|
|
|
|
|
|
@router.get("/api/task/{task_id}", response_model=dict)
|
|
async def get_task_status(task_id: str):
|
|
task = task_scheduler.get_task_status(task_id)
|
|
if not task:
|
|
raise HTTPException(status_code=404, detail="任务不存在")
|
|
|
|
return {
|
|
"code": 200,
|
|
"data": task
|
|
}
|
|
|
|
|
|
@router.get("/api/tasks", response_model=dict)
|
|
async def list_tasks():
|
|
tasks = task_scheduler.list_tasks()
|
|
return {
|
|
"code": 200,
|
|
"data": tasks
|
|
}
|
|
|
|
|
|
@router.get("/api/health")
|
|
async def health_check():
|
|
return {
|
|
"status": "ok",
|
|
"service": "wechat_auto"
|
|
}
|