Fully ensure date_added is in UTC

le717 · Jun 8, 2024 · 304091a · 304091a
1 parent c02872c
commit 304091a
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/core/database/schema.py b/src/core/database/schema.py
@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import UTC, datetime
 from typing import Any
 
 from flask_sqlalchemy import SQLAlchemy
@@ -16,6 +16,11 @@ class Base(DeclarativeBase): ...
 db = SQLAlchemy(model_class=Base)
 
 
+def now_in_utc() -> datetime:
+    """Get the current datetime, expressed in UTC."""
+    return datetime.now(tz=UTC)
+
+
 class HelperMethods:
     def as_dict(self) -> dict[str, Any]:
         """Return a model as a dictionary."""
@@ -40,8 +45,8 @@ class WebLink(HelperMethods, Base):
     )
     date_added: Mapped[datetime] = mapped_column(
         nullable=False,
-        default=datetime.now,
-        onupdate=datetime.now,
+        default=now_in_utc,
+        onupdate=now_in_utc,
     )
     is_dead: Mapped[bool] = mapped_column(default=False)
     is_web_archive: Mapped[bool] = mapped_column(default=False)