83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
Pydantic 数据模型 —— 用于 REST API 请求 / 响应以及内部状态。
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import time
|
|||
|
|
import uuid
|
|||
|
|
from typing import Any, Dict, List, Optional
|
|||
|
|
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
|
|||
|
|
from common.protocol import TaskStatus, TaskType
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ────────────────────────── Worker ──────────────────────────
|
|||
|
|
|
|||
|
|
class BrowserProfile(BaseModel):
|
|||
|
|
"""比特浏览器窗口信息（Worker 上报）。"""
|
|||
|
|
id: str
|
|||
|
|
name: str = ""
|
|||
|
|
remark: str = ""
|
|||
|
|
|
|||
|
|
|
|||
|
|
class WorkerInfo(BaseModel):
|
|||
|
|
"""一台 Worker 的运行时信息（内存中保存）。"""
|
|||
|
|
worker_id: str
|
|||
|
|
worker_name: str = ""
|
|||
|
|
browsers: List[BrowserProfile] = []
|
|||
|
|
online: bool = True
|
|||
|
|
last_heartbeat: float = Field(default_factory=time.time)
|
|||
|
|
connected_at: float = Field(default_factory=time.time)
|
|||
|
|
current_task_id: Optional[str] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
class WorkerOut(BaseModel):
|
|||
|
|
"""返回给前端的 Worker 信息。"""
|
|||
|
|
worker_id: str
|
|||
|
|
worker_name: str
|
|||
|
|
browsers: List[BrowserProfile]
|
|||
|
|
online: bool
|
|||
|
|
current_task_id: Optional[str] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ────────────────────────── Task ──────────────────────────
|
|||
|
|
|
|||
|
|
class TaskCreate(BaseModel):
|
|||
|
|
"""前端提交任务的请求体。"""
|
|||
|
|
task_type: TaskType
|
|||
|
|
worker_id: Optional[str] = None # 手动指定机器
|
|||
|
|
account_name: Optional[str] = None # 按比特浏览器窗口名自动路由
|
|||
|
|
params: Dict[str, Any] = {} # 任务参数（如 job_title, max_greet 等）
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TaskInfo(BaseModel):
|
|||
|
|
"""任务完整信息（内存 / 返回前端）。"""
|
|||
|
|
task_id: str = Field(default_factory=lambda: uuid.uuid4().hex[:12])
|
|||
|
|
task_type: TaskType
|
|||
|
|
status: TaskStatus = TaskStatus.PENDING
|
|||
|
|
worker_id: Optional[str] = None
|
|||
|
|
account_name: Optional[str] = None
|
|||
|
|
params: Dict[str, Any] = {}
|
|||
|
|
progress: Optional[str] = None # 最新进度描述
|
|||
|
|
result: Any = None # 最终结果
|
|||
|
|
error: Optional[str] = None
|
|||
|
|
created_at: float = Field(default_factory=time.time)
|
|||
|
|
updated_at: float = Field(default_factory=time.time)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TaskOut(BaseModel):
|
|||
|
|
"""返回给前端的任务信息。"""
|
|||
|
|
task_id: str
|
|||
|
|
task_type: TaskType
|
|||
|
|
status: TaskStatus
|
|||
|
|
worker_id: Optional[str] = None
|
|||
|
|
account_name: Optional[str] = None
|
|||
|
|
params: Dict[str, Any] = {}
|
|||
|
|
progress: Optional[str] = None
|
|||
|
|
result: Any = None
|
|||
|
|
error: Optional[str] = None
|
|||
|
|
created_at: float
|
|||
|
|
updated_at: float
|