🔨 feat: from pytz to zoneinfo

Author: Zerohertz

app/schemas/base.py

@@ -1,7 +1,7 @@
 import abc
 from datetime import datetime
+from zoneinfo import ZoneInfo
 
-import pytz
 from pydantic import BaseModel, ConfigDict, model_validator
 from typing_extensions import Self
 
@@ -20,6 +20,6 @@ class BaseResponse(BaseModel, abc.ABC):
 
     @model_validator(mode="after")
     def set_timezone(self) -> Self:
-        self.created_at = self.created_at.astimezone(pytz.timezone(configs.TZ))
-        self.updated_at = self.updated_at.astimezone(pytz.timezone(configs.TZ))
+        self.created_at = self.created_at.astimezone(ZoneInfo(configs.TZ))
+        self.updated_at = self.updated_at.astimezone(ZoneInfo(configs.TZ))
         return self

app/schemas/responses.py

@@ -1,7 +1,7 @@
 from datetime import datetime
 from typing import Generic, TypeVar
+from zoneinfo import ZoneInfo
 
-import pytz
 from pydantic import BaseModel
 
 from app.core.configs import configs
@@ -40,7 +40,7 @@ def success(cls, *, status: int, data: T) -> "APIResponse[T]":
             status=status,
             message="The request has been successfully processed.",
             data=data,
-            timestamp=datetime.now().astimezone(pytz.timezone(configs.TZ)),
+            timestamp=datetime.now().astimezone(ZoneInfo(configs.TZ)),
         )
 
     @classmethod
@@ -49,5 +49,5 @@ def error(cls, *, status: int, message: str) -> "APIResponse[T]":
             status=status,
             message=message,
             data=None,
-            timestamp=datetime.now().astimezone(pytz.timezone(configs.TZ)),
+            timestamp=datetime.now().astimezone(ZoneInfo(configs.TZ)),
         )

app/services/auth.py

@@ -1,8 +1,8 @@
 from datetime import datetime, timedelta
 from typing import overload
+from zoneinfo import ZoneInfo
 
 import httpx
-import pytz
 from jose import jwt
 from jose.exceptions import ExpiredSignatureError, JWTError
 from loguru import logger
@@ -64,8 +64,8 @@ def __init__(self) -> None:
     def _encode(self, *, sub: str, exp: timedelta) -> str:
         payload = JwtPayload(
             sub=sub,
-            iat=datetime.now().astimezone(pytz.timezone(configs.TZ)),
-            exp=datetime.now().astimezone(pytz.timezone(configs.TZ)) + exp,
+            iat=datetime.now().astimezone(ZoneInfo(configs.TZ)),
+            exp=datetime.now().astimezone(ZoneInfo(configs.TZ)) + exp,
         )
         return jwt.encode(
             claims=payload.model_dump(), key=self.secret, algorithm=self.algorithm

pyproject.toml

@@ -17,7 +17,6 @@ dependencies = [
     "pydantic[email]>=2.10.5",
     "python-jose>=3.3.0",
     "python-multipart>=0.0.20",
-    "pytz>=2024.2",
     "sqlalchemy>=2.0.38",
     "uv>=0.5.21",
     "uvicorn[standard]>=0.34.0",
@@ -33,7 +32,6 @@ lint = [
     "sqlalchemy[mypy]>=2.0.38",
     "types-passlib>=1.7.7.20241221",
     "types-python-jose>=3.3.4.20240106",
-    "types-pytz>=2024.2.0.20241221",
 ]
 test = [
     "pytest>=8.3.4",