Added missing timezone in responses

Fixes #114
This commit is contained in:
Donkie
2023-09-20 20:12:55 +02:00
parent 5f5473481e
commit b795680226
4 changed files with 28 additions and 8 deletions

View File

@@ -1,14 +1,31 @@
"""Pydantic data models for typing the FastAPI request/responses."""
from datetime import datetime
from datetime import datetime, timezone
from typing import Optional
from pydantic import BaseModel, Field
from pydantic import BaseModel as PydanticBaseModel
from pydantic import Field
from spoolman.database import models
from spoolman.math import length_from_weight
def datetime_to_str(dt: datetime) -> str:
"""Convert a datetime object to a string."""
if dt.tzinfo is None:
dt = dt.astimezone(tz=timezone.utc)
return dt.isoformat().replace("+00:00", "Z")
class BaseModel(PydanticBaseModel):
class Config:
"""Pydantic configuration."""
json_encoders = { # noqa: RUF012
datetime: datetime_to_str,
}
class Message(BaseModel):
message: str = Field()