23 lines
514 B
Python
23 lines
514 B
Python
from __future__ import annotations
|
|
|
|
from datetime import UTC, datetime
|
|
|
|
from pydantic import BaseModel, field_serializer
|
|
|
|
|
|
class UTCBaseModel(BaseModel):
|
|
@field_serializer("*", when_used="json")
|
|
def serialize_datetime(self, v, _info):
|
|
if isinstance(v, datetime):
|
|
if v.tzinfo is None:
|
|
v = v.replace(tzinfo=UTC)
|
|
return v.astimezone(UTC).isoformat()
|
|
return v
|
|
|
|
|
|
Cursor = dict[str, int]
|
|
|
|
|
|
class RespWithCursor(BaseModel):
|
|
cursor: Cursor | None = None
|