daily/Schemas/DailySchemas.py

from enum import Enum, unique
from typing import Optional, List, Union
from pydantic import BaseModel
from datetime import datetime

from Schemas.UserSchemas import UserInfo


@unique
class DailyTypeEnum(Enum):
    enum01 = "董监高日报"
    enum02 = "部门子公司日报"
    enum03 = "监管和同业动态"
    enum04 = "行业信息专题分析"


class DailyInfo(BaseModel):
    id: int
    type: DailyTypeEnum
    fill_user: str
    department: int
    user: UserInfo
    post: int
    content: str
    daily_time: datetime
    create_time: datetime
    update_time: datetime


class DailyGetReq(BaseModel):
    id: int
    pass


class DailyGetRes(DailyInfo):
    pass


class DailyChangeReq(BaseModel):
    id: int
    type: Optional[DailyTypeEnum]
    fill_user: Optional[str]
    name: Optional[str]
    department: Optional[int]
    post: Optional[int]
    content: Optional[str]
    daily_time: Optional[datetime]
    create_time: Optional[datetime]
    update_time: Optional[datetime]


class DailyChangeRes(DailyInfo):
    pass


class DailyQuery(BaseModel):
    id: Optional[int]
    type: Optional[DailyTypeEnum]
    fill_user: Optional[str]
    name: Optional[str]
    department: Optional[List[int]]
    post: Optional[List[int]]
    content: Optional[str]
    daily_time: Optional[datetime]
    create_time: Optional[datetime]
    update_time: Optional[datetime]
    page_size: Optional[int]
    page: Optional[int]


class DailyQueryRes(BaseModel):
    items: List[DailyInfo]
    count: int
    pass


class DailyAddReq(BaseModel):
    type: DailyTypeEnum
    fill_user: Optional[str]
    department: int
    content: str
    post:int
    daily_time: Optional[datetime]
    title: Optional[str]


class DailyAddRes(DailyInfo):
    pass


class DailyDeleteReq(BaseModel):
    id: int


class DailyDeleteRes(BaseModel):
    msg: str
    state: int