Fix possible KeyError in ZZZNotes

thesadru · Nov 22, 2024 · 669bf1f · 669bf1f
1 parent 28ce2e9
commit 669bf1f
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/genshin/models/zzz/chronicle/notes.py b/genshin/models/zzz/chronicle/notes.py
@@ -85,9 +85,11 @@ def __transform_value(cls, v: typing.Literal["CardSignDone", "CardSignNotDone"])
 
     @pydantic.model_validator(mode="before")
     def __unnest_value(cls, values: dict[str, typing.Any]) -> dict[str, typing.Any]:
-        values["video_store_state"] = values["vhs_sale"]["sale_state"]
-        values["hollow_zero"] = {
-            "bounty_commission": values["bounty_commission"],
-            "survey_points": values["survey_points"],
-        }
+        if "video_store_state" not in values:
+            values["video_store_state"] = values["vhs_sale"]["sale_state"]
+        if "hollow_zero" not in values:
+            values["hollow_zero"] = {
+                "bounty_commission": values["bounty_commission"],
+                "survey_points": values["survey_points"],
+            }
         return values