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 AccountUpdate(BaseModel):
    account_id: str = Field(..., description="账号ID")
    app_name: str = Field(..., description="应用名称")
    status: int = Field(..., ge=1, le=10, 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="应用名称")
    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")