from datetime import datetime

from pydantic import BaseModel, Field


class LogAlertTrigger(BaseModel):
    alter_name: str
    timestamp: int


class LogAlertResponse(BaseModel):
    alterName: str
    alterTime: datetime
    logTime: datetime
    logTopic: str
    logMessage: str
    logStatus: str

    class Config:
        from_attributes = True
        json_encoders = {
            # 自定义 datetime 类型的序列化格式
            datetime: lambda dt: dt.strftime('%Y-%m-%d %H:%M:%S')
        }

class OpenobserveQuery(BaseModel):
    start_time: int
    end_time: int
    from_: int = Field(default=0, alias="from")
    size: int = 10
    sql: str = "select * from default"