from pydantic import BaseModel
from datetime import datetime


class RawData(BaseModel):
    raw_dt: str
    mg_equip_rtu_seq: str
    acc_pg: str
    pg: str


class RtuGenerator(BaseModel):
    month: float
    day: float
    hour: float
    rtu_seq: float
    icsr: float
    dc10tca: float


class RawDataHerit(BaseModel):
    tm: str
    content_type: str
    X_HIT_TransactionId: str
    content_Length: str
    body: dict


class CardExpenseCreate(BaseModel):
    jang: str
    yeo: str
    kim: str
    choi: str
    amount: float
    purpose: str


class CardExpense(BaseModel):
    id: int
    jang: str
    yeo: str
    kim: str
    choi: str
    amount: float
    purpose: str
    created_at: datetime

    class Config:
        orm_mode = True


class DatePayload(BaseModel):
    year: int
    month: int