fix: use plain datetime for model fields (server returns naive datetimes)

Author: jackharrhy

justfile

@@ -6,7 +6,8 @@ generate:
     --input-file-type openapi \
     --output-model-type pydantic_v2.BaseModel \
     --output stackcoin/stackcoin/models.py \
-    --target-python-version 3.13
+    --target-python-version 3.13 \
+    --output-datetime-class datetime
   uvx ruff format stackcoin/
 
 dev:

stackcoin/stackcoin/gateway.py

@@ -12,6 +12,7 @@
 
 class Event(BaseModel):
     """A StackCoin event received via the gateway."""
+
     id: int
     type: str
     data: dict[str, Any]
@@ -20,21 +21,21 @@ class Event(BaseModel):
 
 class Gateway:
     """WebSocket gateway for receiving real-time StackCoin events.
-    
+
     Usage::
-    
+
         gateway = stackcoin.Gateway(
             ws_url="ws://localhost:4000/bot/websocket",
             token="...",
         )
-        
+
         @gateway.on("request.accepted")
         async def handle_accepted(event: stackcoin.Event):
             print(event.data["request_id"])
-        
+
         await gateway.connect()
     """
-    
+
     def __init__(
         self,
         ws_url: str,
@@ -58,9 +59,11 @@ def last_event_id(self) -> int:
 
     def on(self, event_type: str) -> Callable[[EventHandler], EventHandler]:
         """Decorator to register an event handler."""
+
         def decorator(func: EventHandler) -> EventHandler:
             self.register_handler(event_type, func)
             return func
+
         return decorator
 
     def register_handler(self, event_type: str, handler: EventHandler) -> None:
@@ -72,41 +75,43 @@ def register_handler(self, event_type: str, handler: EventHandler) -> None:
     async def connect(self) -> None:
         """Connect and listen for events. Reconnects automatically on failure."""
         import websockets
-        
+
         self._running = True
-        
+
         while self._running:
             try:
                 url = f"{self._ws_url}?token={self._token}&vsn=2.0.0"
-                
+
                 async with websockets.connect(url) as ws:
                     self._ws = ws
                     await self._join_channel(ws)
-                    
+
                     heartbeat_task = asyncio.create_task(self._heartbeat(ws))
                     try:
                         async for raw_msg in ws:
                             msg = json.loads(raw_msg)
                             await self._handle_message(msg)
                     finally:
                         heartbeat_task.cancel()
-            
+
             except Exception:
                 if self._running:
                     await asyncio.sleep(5)
 
     async def _join_channel(self, ws: Any) -> None:
         """Join the user:self channel with event replay."""
         self._ref_counter += 1
-        join_msg = json.dumps([
-            None,
-            str(self._ref_counter),
-            "user:self",
-            "phx_join",
-            {"last_event_id": self._last_event_id},
-        ])
+        join_msg = json.dumps(
+            [
+                None,
+                str(self._ref_counter),
+                "user:self",
+                "phx_join",
+                {"last_event_id": self._last_event_id},
+            ]
+        )
         await ws.send(join_msg)
-        
+
         reply = json.loads(await asyncio.wait_for(ws.recv(), timeout=10))
         if not (reply[3] == "phx_reply" and reply[4].get("status") == "ok"):
             raise ConnectionError(f"Failed to join channel: {reply}")

stackcoin/stackcoin/models.py

@@ -1,10 +1,12 @@
 # generated by datamodel-codegen:
 #   filename:  openapi.json
-#   timestamp: 2026-03-05T03:46:00+00:00
+#   timestamp: 2026-03-05T04:04:56+00:00
 
 from __future__ import annotations
 
-from pydantic import AwareDatetime, BaseModel, Field
+from datetime import datetime
+
+from pydantic import BaseModel, Field
 
 
 class CreateRequestParams(BaseModel):
@@ -25,7 +27,7 @@ class Responder(BaseModel):
 class CreateRequestResponse(BaseModel):
     amount: int = Field(..., description="Requested amount")
     request_id: int = Field(..., description="Created request ID")
-    requested_at: AwareDatetime = Field(..., description="Request timestamp")
+    requested_at: datetime = Field(..., description="Request timestamp")
     requester: Requester
     responder: Responder
     status: str = Field(..., description="Request status")
@@ -37,7 +39,7 @@ class DiscordGuild(BaseModel):
         None, description="Designated channel snowflake ID"
     )
     id: int = Field(..., description="Guild ID")
-    last_updated: AwareDatetime = Field(..., description="Last updated timestamp")
+    last_updated: datetime = Field(..., description="Last updated timestamp")
     name: str = Field(..., description="Guild name")
     snowflake: str = Field(..., description="Discord guild snowflake ID")
 
@@ -47,7 +49,7 @@ class DiscordGuildResponse(BaseModel):
         None, description="Designated channel snowflake ID"
     )
     id: int = Field(..., description="Guild ID")
-    last_updated: AwareDatetime = Field(..., description="Last updated timestamp")
+    last_updated: datetime = Field(..., description="Last updated timestamp")
     name: str = Field(..., description="Guild name")
     snowflake: str = Field(..., description="Discord guild snowflake ID")
 
@@ -72,17 +74,17 @@ class Request(BaseModel):
     amount: int = Field(..., description="Requested amount")
     id: int = Field(..., description="Request ID")
     label: str | None = Field(None, description="Request label")
-    requested_at: AwareDatetime = Field(..., description="Request timestamp")
+    requested_at: datetime = Field(..., description="Request timestamp")
     requester: Requester
-    resolved_at: AwareDatetime | None = Field(None, description="Resolution timestamp")
+    resolved_at: datetime | None = Field(None, description="Resolution timestamp")
     responder: Responder
     status: str = Field(..., description="Request status")
     transaction_id: int | None = Field(None, description="Associated transaction ID")
 
 
 class RequestActionResponse(BaseModel):
     request_id: int = Field(..., description="Request ID")
-    resolved_at: AwareDatetime = Field(..., description="Resolution timestamp")
+    resolved_at: datetime = Field(..., description="Resolution timestamp")
     status: str = Field(..., description="New request status")
     success: bool = Field(..., description="Whether the operation succeeded")
     transaction_id: int | None = Field(None, description="Associated transaction ID")
@@ -92,9 +94,9 @@ class RequestResponse(BaseModel):
     amount: int = Field(..., description="Requested amount")
     id: int = Field(..., description="Request ID")
     label: str | None = Field(None, description="Request label")
-    requested_at: AwareDatetime = Field(..., description="Request timestamp")
+    requested_at: datetime = Field(..., description="Request timestamp")
     requester: Requester
-    resolved_at: AwareDatetime | None = Field(None, description="Resolution timestamp")
+    resolved_at: datetime | None = Field(None, description="Resolution timestamp")
     responder: Responder
     status: str = Field(..., description="Request status")
     transaction_id: int | None = Field(None, description="Associated transaction ID")
@@ -133,7 +135,7 @@ class Transaction(BaseModel):
     from_: From = Field(..., alias="from")
     id: int = Field(..., description="Transaction ID")
     label: str | None = Field(None, description="Transaction label")
-    time: AwareDatetime = Field(..., description="Transaction timestamp")
+    time: datetime = Field(..., description="Transaction timestamp")
     to: To
 
 
@@ -142,7 +144,7 @@ class TransactionResponse(BaseModel):
     from_: From = Field(..., alias="from")
     id: int = Field(..., description="Transaction ID")
     label: str | None = Field(None, description="Transaction label")
-    time: AwareDatetime = Field(..., description="Transaction timestamp")
+    time: datetime = Field(..., description="Transaction timestamp")
     to: To
 
 
@@ -156,8 +158,8 @@ class User(BaseModel):
     balance: int = Field(..., description="User's STK balance")
     banned: bool = Field(..., description="Whether user is banned")
     id: int | None = Field(None, description="User ID")
-    inserted_at: AwareDatetime | None = Field(None, description="Creation timestamp")
-    updated_at: AwareDatetime | None = Field(None, description="Update timestamp")
+    inserted_at: datetime | None = Field(None, description="Creation timestamp")
+    updated_at: datetime | None = Field(None, description="Update timestamp")
     username: str = Field(..., description="Username")
 
 
@@ -166,8 +168,8 @@ class UserResponse(BaseModel):
     balance: int = Field(..., description="User's STK balance")
     banned: bool = Field(..., description="Whether user is banned")
     id: int | None = Field(None, description="User ID")
-    inserted_at: AwareDatetime | None = Field(None, description="Creation timestamp")
-    updated_at: AwareDatetime | None = Field(None, description="Update timestamp")
+    inserted_at: datetime | None = Field(None, description="Creation timestamp")
+    updated_at: datetime | None = Field(None, description="Update timestamp")
     username: str = Field(..., description="Username")