Fix timezone issue in ClaimedDailyReward

thesadru · Sep 19, 2024 · 99d99fb · 99d99fb
1 parent 9099c65
commit 99d99fb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/genshin/models/genshin/daily.py b/genshin/models/genshin/daily.py
@@ -3,7 +3,7 @@
 import datetime
 import typing
 
-import pydantic
+import pydantic.v1 as pydantic
 
 from genshin.constants import CN_TIMEZONE
 from genshin.models.model import Aliased, APIModel, Unique
@@ -40,7 +40,6 @@ class ClaimedDailyReward(APIModel, Unique):
     icon: str = Aliased("img")
     time: datetime.datetime = Aliased("created_at")
 
-    @pydantic.field_validator("time")
-    @classmethod
+    @pydantic.validator("time")
     def __add_timezone(cls, value: datetime.datetime) -> datetime.datetime:
         return value.replace(tzinfo=CN_TIMEZONE)