from sqlmodel import SQLModel, Field
from typing import Optional
from datetime import datetime


class TimestampMixin(SQLModel):
    create_time: Optional[datetime] = Field(default_factory=datetime.now)
    update_time: Optional[datetime] = Field(default_factory=datetime.now)

    class Config:
        json_encoders = {
            datetime: lambda v: v.strftime('%Y-%m-%d %H:%M:%S')
        }