from enum import Enum, unique

from pydantic import BaseModel


@unique
class WorkStatusEnum(Enum):
    enum01 = "正常"
    enum02 = "出差"
    enum03 = "休息"


@unique
class ProjectStatusEnum(Enum):
    enum01 = "进行"
    enum02 = "完成"
    enum03 = "终止"


class EditProjectInfo(BaseModel):
    project: str
    client: str
    client_labels: str
    master: str
    participants: str
    departments: str
    project_type: str
    project_labels: str
    project_status: str
    start_date: str
    finish_date: str
    amount: str
    remarks: str


class QueryProjectByDepartment(BaseModel):
    department: str
    page: int = 1
    pagesize: int = 10


class QueryProjectByType(BaseModel):
    type: str
    year: str