from enum import Enum, unique
from pydantic import BaseModel


@unique
class GenerationFlowStatusEnum(Enum):
    enum01 = "进行"
    enum02 = "完成"
    enum03 = "取消"


@unique
class ReportNameEnum(Enum):
    name01 = "主体信用评级报告模板"


class CreateReportFlowReqBody(BaseModel):
    rating_process_id = "评级流程ID"
    report = "主体信用评级报告模板"
    begin_date = "开始日期"
    end_date = "结束日期"


class SearchReportFlowReqBody(BaseModel):
    report_generation_id = "报告流程ID"


class SaveReportDataReqBody(BaseModel):
    report_generation_id = "报告流程ID"
    report_data = {}