Update notes.py

thesadru · Sep 30, 2023 · f1965bf · f1965bf
1 parent a4e4fe5
commit f1965bf
Showing 1 changed file with 13 additions and 21 deletions.
diff --git a/genshin/models/genshin/chronicle/notes.py b/genshin/models/genshin/chronicle/notes.py
@@ -74,31 +74,23 @@ def seconds(self) -> int:
 
 
 class TaskReward(APIModel):
-    """Status of the Commission/Task.
+    """Status of the Commission/Task."""
 
-    status:
-        TaskRewardStatusTakenAward
-        TaskRewardStatusUnfinished
-    """
-
-    status: str
+    status: typing.Literal[
+        "TaskRewardStatusTakenAward",
+        "TaskRewardStatusUnfinished"
+    ]
 
 
 class AttendanceReward(APIModel):
-    """Status of the Encounter Point.
-
-    status:
-        AttendanceRewardStatusForbid
-        AttendanceRewardStatusTakenAward
-        AttendanceRewardStatusWaitTaken
-        AttendanceRewardStatusUnfinished
-
-    progress:
-        2000 == 100%
-        1000 == 50%
-    """
-
-    status: str
+    """Status of the Encounter Point."""
+
+    status: typing.Literal[
+        "TaskRewardStatusTakenAward",
+        "AttendanceRewardStatusTakenAward",
+        "AttendanceRewardStatusWaitTaken",
+        "AttendanceRewardStatusUnfinished"
+    ]
     progress: int