crawler_task_management/model/model.py

# -*- coding: utf-8 -*-
from pydantic import BaseModel, Field
from typing import Optional, Dict, Any


# 定义数据模型
class AccountStatus(BaseModel):
    account_id: str = Field(..., description="账号ID")
    app_name: str = Field(..., description="应用名称")


class AccountObtain(BaseModel):
    app_name: str = Field(..., description="应用名称")
    region: str = Field(..., description="地区")


class AccountUpdate(BaseModel):
    account_id: str = Field(..., description="账号ID")
    app_name: str = Field(..., description="应用名称")
    status: int = Field(..., description="状态：1-空闲，2-使用中，3-暂停使用(后续还能使用)，4-账号已无法使用")


class GoodsInfo(BaseModel):
    host: str = Field(..., description="客户所在地区")
    item_id: str = Field(..., description="商品ID")
    shop_id: str = Field(..., description="店铺ID")


class DataReceive(BaseModel):
    task_id: str = Field(..., description="任务ID")
    app_name: str = Field(..., description="应用名称")
    account_id: str = Field(..., description="使用账号")
    pad_code: str = Field(..., description="设备编号")
    shop_id: str = Field(..., description="店铺ID")
    item_id: str = Field(..., description="商品ID")
    region: str = Field(..., description="地区")
    goods_info: Dict[str, Any] = Field(..., description="商品信息")


class TokenItem(BaseModel):
    token: str = Field(..., description="token")


class CrawlerItem(BaseModel):
    region: str = Field(..., description="账号所在地区")
    app_name: str = Field(..., description="应用名称")


class ResetTask(BaseModel):
    task_id: str = Field(..., description="任务ID")


class AlterStatus(BaseModel):
    pad_code: str = Field(..., description="设备编号")


class BackupItem(BaseModel):
    pad_code: str = Field(..., description="设备编号")
    account_id: str = Field(..., description="账号ID")
    script_name: str = Field(..., description="脚本名称")
    is_restore: bool = Field(..., description="是否还原")