Changed battleBg type to enum

atlasacademy · Jan 8, 2024 · c0a04fa · c0a04fa
1 parent 2f69cec
commit c0a04fa
Show file tree

Hide file tree

Showing 5 changed files with 38 additions and 2 deletions.
diff --git a/app/core/nice/quest.py b/app/core/nice/quest.py
@@ -25,6 +25,7 @@
 from ...schemas.common import Language, Region, ScriptLink
 from ...schemas.enums import STAGE_LIMIT_ACT_TYPE_NAME, get_class_name
 from ...schemas.gameenums import (
+    BATTLE_ENVIRONMENT_GRANT_TYPE_NAME,
     COND_TYPE_NAME,
     FREQUENCY_TYPE_NAME,
     QUEST_AFTER_CLEAR_NAME,
@@ -173,7 +174,7 @@ def get_nice_quest_restriction(
 def get_nice_battle_bg(battle_bg: MstBattleBg) -> NiceBattleBg:
     return NiceBattleBg(
         id=battle_bg.id,
-        type=battle_bg.type,
+        type=BATTLE_ENVIRONMENT_GRANT_TYPE_NAME[battle_bg.type],
         priority=battle_bg.priority,
         individuality=get_traits_list(battle_bg.individuality),
         imageId=battle_bg.imageId,

diff --git a/app/schemas/gameenums.py b/app/schemas/gameenums.py
@@ -5331,3 +5331,24 @@ class NiceWarBoardTreasureRarity(StrEnum):
     10: NiceWarBoardTreasureRarity.itemIconPlus,
     11: NiceWarBoardTreasureRarity.itemIconPlus2,
 }
+
+
+class BattleFieldEnvironmentDataGrantType(IntEnum):
+    None = 0
+    Stage = 1
+    Function = 2
+
+
+class NiceBattleFieldEnvironmentGrantType(StrEnum):
+    """Battle Field Environment Grant Type"""
+
+    none = "none"
+    stage = "stage"
+    function = "function"
+
+
+BATTLE_ENVIRONMENT_GRANT_TYPE_NAME: dict[int, NiceBattleFieldEnvironmentGrantType] = {
+    0: NiceBattleFieldEnvironmentGrantType.none,
+    1: NiceBattleFieldEnvironmentGrantType.stage,
+    2: NiceBattleFieldEnvironmentGrantType.function,
+}
diff --git a/app/schemas/nice.py b/app/schemas/nice.py
@@ -45,6 +45,7 @@
     NiceAiActTarget,
     NiceAiActType,
     NiceAiCond,
+    NiceBattleFieldEnvironmentGrantType,
     NiceBuffType,
     NiceCardType,
     NiceClassBoardSkillType,
@@ -2370,7 +2371,7 @@ class NiceStageCutIn(BaseModelORJson):
 
 class NiceBattleBg(BaseModelORJson):
     id: int
-    type: int
+    type: NiceBattleFieldEnvironmentGrantType
     priority: int
     individuality: list[NiceTrait]
     imageId: int

diff --git a/scripts/enum.ts b/scripts/enum.ts
@@ -1811,3 +1811,9 @@ export enum NiceWarBoardTreasureRarity {
     ITEM_ICON_PLUS = "itemIconPlus",
     ITEM_ICON_PLUS2 = "itemIconPlus2",
 }
+
+export enum NiceBattleFieldEnvironmentGrantType {
+    None = "none",
+    Stage = "stage",
+    Function = "function",
+}
diff --git a/scripts/extract_enums.py b/scripts/extract_enums.py
@@ -578,6 +578,13 @@ def cs_enum_to_ts(cs_enums: list[str], raw_class: str, nice_class: str) -> list[
         "War Board Treasure Rarity",
         "WAR_BOARD_TREASURE_RARITY_NAME",
     ),
+    (
+        "BattleFieldEnvironmentData.GrantType",
+        "BattleFieldEnvironmentDataGrantType",
+        "NiceBattleFieldEnvironmentGrantType",
+        "Battle Field Environment Grant Type",
+        "BATTLE_ENVIRONMENT_GRANT_TYPE_NAME",
+    ),
 ]