45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
Worker 查询 API（需要登录）。
|
||
"""
|
||
from typing import List
|
||
|
||
from fastapi import APIRouter, Depends, HTTPException
|
||
|
||
from server.models import WorkerOut
|
||
from server.core.worker_manager import worker_manager
|
||
from server.api.deps import require_auth
|
||
|
||
router = APIRouter(prefix="/api/workers", tags=["workers"], dependencies=[Depends(require_auth)])
|
||
|
||
|
||
@router.get("", response_model=List[WorkerOut])
|
||
async def list_workers():
|
||
"""获取所有已注册的 Worker（含在线状态与浏览器列表）。"""
|
||
workers = worker_manager.get_all_workers()
|
||
return [
|
||
WorkerOut(
|
||
worker_id=w.worker_id,
|
||
worker_name=w.worker_name,
|
||
browsers=w.browsers,
|
||
online=w.online,
|
||
current_task_id=w.current_task_id,
|
||
)
|
||
for w in workers
|
||
]
|
||
|
||
|
||
@router.get("/{worker_id}", response_model=WorkerOut)
|
||
async def get_worker(worker_id: str):
|
||
"""获取指定 Worker 的详情。"""
|
||
w = worker_manager.get_worker(worker_id)
|
||
if not w:
|
||
raise HTTPException(status_code=404, detail=f"Worker {worker_id} 不存在")
|
||
return WorkerOut(
|
||
worker_id=w.worker_id,
|
||
worker_name=w.worker_name,
|
||
browsers=w.browsers,
|
||
online=w.online,
|
||
current_task_id=w.current_task_id,
|
||
)
|