Add UTC timezone for decoded naive timestamp

mosquito · Dec 28, 2023 · e966a96 · e966a96
1 parent 8225acf
commit e966a96
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/aio_pika/message.py b/aio_pika/message.py
@@ -106,6 +106,8 @@ def decode_timestamp(value: Any) -> Optional[datetime]:
 
 @decode_timestamp.register(datetime)
 def decode_timestamp_datetime(value: datetime) -> datetime:
+    if value.tzinfo is None or value.tzinfo.utcoffset(value) is None:
+        value = value.replace(tzinfo=timezone.utc)
     return value