Skip to content

Commit 83b29be

Browse files
author
Mithun Thomas
committed
Refactor datetime handling to include timezone
awareness Updated the datetime usage in multiple models to include timezone awareness by importing 'UTC' and applying it to the default_factory.
1 parent de90dae commit 83b29be

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

src/app/models/post.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import uuid as uuid_pkg
2-
from datetime import datetime
2+
from datetime import datetime, UTC
33

44
from sqlalchemy import String, DateTime, ForeignKey
55
from sqlalchemy.orm import Mapped, mapped_column
@@ -21,7 +21,7 @@ class Post(Base):
2121
media_url: Mapped[str | None] = mapped_column(String, default=None)
2222

2323
created_at: Mapped[datetime] = mapped_column(
24-
DateTime, default_factory=datetime.utcnow
24+
DateTime(timezone=True), default_factory=lambda: datetime.now(UTC)
2525
)
2626
updated_at: Mapped[datetime | None] = mapped_column(default=None)
2727
deleted_at: Mapped[datetime | None] = mapped_column(default=None)

src/app/models/rate_limit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from typing import Optional
2-
from datetime import datetime
2+
from datetime import datetime, UTC
33

44
from sqlalchemy import String, DateTime, ForeignKey, Integer
55
from sqlalchemy.orm import Mapped, mapped_column
@@ -19,6 +19,6 @@ class RateLimit(Base):
1919
period: Mapped[int] = mapped_column(Integer, nullable=False)
2020

2121
created_at: Mapped[datetime] = mapped_column(
22-
DateTime, default_factory=datetime.utcnow
22+
DateTime(timezone=True), default_factory=lambda: datetime.now(UTC)
2323
)
2424
updated_at: Mapped[Optional[datetime]] = mapped_column(default=None)

src/app/models/tier.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,6 @@ class Tier(Base):
1515
name: Mapped[str] = mapped_column(String, nullable=False, unique=True)
1616

1717
created_at: Mapped[datetime] = mapped_column(
18-
DateTime, default_factory=lambda: datetime.now(UTC).replace(tzinfo=None)
18+
DateTime(timezone=True), default_factory=lambda: datetime.now(UTC)
1919
)
2020
updated_at: Mapped[Optional[datetime]] = mapped_column(default=None)

src/app/models/user.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ class User(Base):
2424
default_factory=uuid_pkg.uuid4, primary_key=True, unique=True
2525
)
2626
created_at: Mapped[datetime] = mapped_column(
27-
DateTime, default_factory=lambda: datetime.now(UTC).replace(tzinfo=None)
27+
DateTime(timezone=True), default_factory=lambda: datetime.now(UTC)
2828
)
2929
updated_at: Mapped[Optional[datetime]] = mapped_column(default=None)
3030
deleted_at: Mapped[Optional[datetime]] = mapped_column(default=None)

0 commit comments

Comments
 (0)