bcsfe 3.1.0__tar.gz → 3.2.0b2__tar.gz

{bcsfe-3.1.0/src/bcsfe.egg-info → bcsfe-3.2.0b2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bcsfe
-Version: 3.1.0
+Version: 3.2.0b2
 Summary: A save file editor for The Battle Cats
 Author: fieryhenry
 License-Expression: GPL-3.0-or-later
@@ -368,10 +368,12 @@ Alternatively you can use pip directly, although it won't auto-update with the l
 git commits.
 
 ```sh
-py -m pip install git+https://codeberg.org/fieryhenry/BCSFE-Python.git
+py -m pip install -U git+https://codeberg.org/fieryhenry/BCSFE-Python.git
 py -m bcsfe
 ```
 
+Again, you might need change `py` for `python` or `python3`
+
 If you want to use the editor again all you need to do is run the `py -m bcsfe` command
 
 ## Documentation

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/README.md

@@ -339,10 +339,12 @@ Alternatively you can use pip directly, although it won't auto-update with the l
 git commits.
 
 ```sh
-py -m pip install git+https://codeberg.org/fieryhenry/BCSFE-Python.git
+py -m pip install -U git+https://codeberg.org/fieryhenry/BCSFE-Python.git
 py -m bcsfe
 ```
 
+Again, you might need change `py` for `python` or `python3`
+
 If you want to use the editor again all you need to do is run the `py -m bcsfe` command
 
 ## Documentation

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "3.1.0"
+__version__ = "3.2.0b2"
 
 from bcsfe import core, cli
 

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/__main__.py

@@ -52,7 +52,9 @@ def main():
         cli.main.Main.leave()
     except Exception as e:
         tb = traceback.format_exc()
-        cli.color.ColoredText.localize("error", error=e, traceback=tb)
+        cli.color.ColoredText.localize(
+            "error", error=e, version=bcsfe.__version__, traceback=tb
+        )
         try:
             cli.main.Main.exit_editor()
         except Exception:

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/cli/color.py

@@ -113,13 +113,9 @@ class ColoredText:
                 print(colored.stylize(text, fg), end="")  # type: ignore
 
     @staticmethod
-    def localize(
-        string: str, escape: bool = True, **kwargs: Any
-    ) -> ColoredText:
+    def localize(string: str, escape: bool = True, **kwargs: Any) -> ColoredText:
         return ColoredText(
-            core.core_data.local_manager.get_key(
-                string, escape=escape, **kwargs
-            )
+            core.core_data.local_manager.get_key(string, escape=escape, **kwargs)
         )
 
     def parse(self, txt: str) -> list[tuple[str, str]]:
@@ -134,11 +130,7 @@ class ColoredText:
         special_chars = core.LocalManager.get_special_chars()
         while i < len(txt):
             char = txt[i]
-            if (
-                char == "\\"
-                and i + 1 < len(txt)
-                and txt[i + 1] in special_chars
-            ):
+            if char == "\\" and i + 1 < len(txt) and txt[i + 1] in special_chars:
                 i += 1
                 char = txt[i]
                 text += char
@@ -183,7 +175,5 @@ class ColoredInput:
         return input()
 
     def localize(self, string: str, escape: bool = True, **kwargs: Any) -> str:
-        text = core.core_data.local_manager.get_key(
-            string, escape=escape, **kwargs
-        )
+        text = core.core_data.local_manager.get_key(string, escape=escape, **kwargs)
         return self.get(text)

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/cli/dialog_creator.py

@@ -327,7 +327,7 @@ class ChoiceInput:
         dialog = core.core_data.local_manager.get_key(key).format(
             min=self.get_min_value(), max=self.get_max_value()
         )
-        usr_input = color.ColoredInput().get(dialog).split(" ")
+        usr_input = color.ColoredInput().get(dialog).strip().split(" ")
         int_vals: list[int] = []
         for inp in usr_input:
             try:
@@ -689,7 +689,11 @@ class YesNoInput:
 
         if usr_input == core.core_data.local_manager.get_key("quit_key"):
             return None
-        return usr_input == core.core_data.local_manager.get_key("yes_key")
+        return (
+            usr_input == core.core_data.local_manager.get_key("yes_key")
+            or usr_input.lower().strip()
+            == core.core_data.local_manager.get_key("yes").lower().strip()
+        )
 
 
 class DialogBuilder:

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/cli/edits/basic_items.py

@@ -12,6 +12,12 @@ class BasicItems:
             return core.core_data.local_manager.get_key(key)
         return name.strip()
 
+    @staticmethod
+    def reset_golden_cat_cpus(save_file: core.SaveFile):
+        save_file.golden_cpu_count = 0
+
+        color.ColoredText.localize("reset_golden_cat_cpus_success")
+
     @staticmethod
     def edit_catfood(save_file: core.SaveFile):
         should_exit = not dialog_creator.YesNoInput().get_input_once("catfood_warning")

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/cli/edits/cat_editor.py

@@ -48,7 +48,6 @@ class CatEditor:
         if len(current_cats) > 50:
             color.ColoredText.localize("total_selected_cats", total=len(current_cats))
         else:
-            self.save_file.cats.bulk_download_names(self.save_file, current_cats)
             for cat in current_cats:
                 names = cat.get_names_cls(self.save_file)
                 if not names:

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/cli/edits/event_tickets.py

@@ -20,7 +20,8 @@ class EventTickets:
         )
 
         cli.color.ColoredText.localize("downloading_gatya_data")
-        gatya_event_data = core.ServerHandler(self.save_file).download_gatya_data()
+        temp_save_file = core.SaveFile(cc=save_file.cc, gv=save_file.game_version)
+        gatya_event_data = core.ServerHandler(temp_save_file).download_gatya_data()
 
         if gatya_event_data is None:
             cli.color.ColoredText.localize("download_gatya_data_fail")

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/cli/edits/map.py

@@ -92,8 +92,12 @@ def edit_chapters(
     chapters: ChaptersType,
     letter_code: str,
     type: int | None = None,
+    no_r_prefix: bool = False,
+    base_index: int | None = None,
 ) -> dict[int, bool] | None:
-    map_names = core.MapNames(save_file, letter_code)
+    map_names = core.MapNames(
+        save_file, letter_code, no_r_prefix=no_r_prefix, base_index=base_index
+    )
     names = map_names.map_names
 
     choice = dialog_creator.ChoiceInput.from_reduced(

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/cli/feature_handler.py

@@ -36,9 +36,11 @@ class FeatureHandler:
                 "waydroid_push_rerun": save_management.SaveManagement.waydroid_push_rerun,
                 "adb_push": save_management.SaveManagement.adb_push,
                 "adb_push_rerun": save_management.SaveManagement.adb_push_rerun,
+                "root_push": save_management.SaveManagement.root_push,
+                "root_push_rerun": save_management.SaveManagement.root_push_rerun,
                 "export_save": save_management.SaveManagement.export_save,
                 "load_save": save_management.SaveManagement.load_save,
-                "init_save": save_management.SaveManagement.init_save,
+                # "init_save": save_management.SaveManagement.init_save,
                 "convert_region": save_management.SaveManagement.convert_save_cc,
                 "convert_version": save_management.SaveManagement.convert_save_gv,
             },
@@ -63,6 +65,7 @@ class FeatureHandler:
                 "100_million_tickets": edits.basic_items.BasicItems.edit_100_million_ticket,
                 "event_tickets": edits.event_tickets.EventTickets.edit,
                 "treasure_chests": edits.basic_items.BasicItems.edit_treasure_chests,
+                "reset_golden_cat_cpus": edits.basic_items.BasicItems.reset_golden_cat_cpus,
             },
             "cats_special_skills": cat_features,
             "levels": {
@@ -70,7 +73,8 @@ class FeatureHandler:
                 "clear_story": core.game.map.story.StoryChapters.clear_story,
                 "challenge_score": core.game.map.challenge.edit_challenge_score,
                 "dojo_score": core.game.map.dojo.edit_dojo_score,
-                "enigma": core.game.map.enigma.edit_enigma,
+                "add_enigma_stages": core.game.map.enigma.edit_enigma,
+                "clear_enigma_stages": core.game.map.gauntlets.GauntletChapters.edit_enigma_stages,
                 "unlock_aku_realm": edits.aku_realm.unlock_aku_realm,
                 "story_treasures": core.game.map.story.StoryChapters.edit_treasures,
                 "outbreaks": core.game.map.outbreaks.Outbreaks.edit_outbreaks,
@@ -88,6 +92,7 @@ class FeatureHandler:
                 "legend_quest": core.game.map.legend_quest.LegendQuestChapters.edit_legend_quest,
                 "towers": core.game.map.tower.TowerChapters.edit_towers,
                 "zero_legends": core.game.map.zero_legends.ZeroLegendsChapters.edit_zero_legends,
+                "dojo_catclaw_championships": core.game.map.zero_legends.ZeroLegendsChapters.edit_catclaw_championships,
             },
             "gamototo": {
                 "engineers": edits.basic_items.BasicItems.edit_engineers,
@@ -117,6 +122,7 @@ class FeatureHandler:
             },
             "other": {
                 "unlocked_slots": edits.basic_items.BasicItems.edit_unlocked_slots,
+                "reset_gambling_events": core.GamblingEvent.reset_events,
                 "restart_pack": edits.basic_items.BasicItems.set_restart_pack,
                 "special_skills": edits.basic_items.BasicItems.edit_special_skills,
                 "playtime": core.game.catbase.playtime.edit,
@@ -229,6 +235,8 @@ class FeatureHandler:
 
         else:
             feats = self.search_features(usr_input)
+            if not feats:
+                color.ColoredText.localize("no_feature_with_name", name=usr_input)
             kv_map = list(feats.items())
             kv_map.sort(key=lambda v: v[1], reverse=True)
             selected_features = [v[0] for v in kv_map]

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/cli/save_management.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 from bcsfe import core
+import bcsfe
 from bcsfe.core import io
 from bcsfe.cli import main, color, dialog_creator, server_cli
 from bcsfe.core.country_code import CountryCode
@@ -196,6 +197,43 @@ class SaveManagement:
 
         return adb_handler
 
+    @staticmethod
+    def root_push(save_file: core.SaveFile) -> core.RootHandler | None:
+        """Push the save file to the device.
+
+        Args:
+            save_file (core.SaveFile): The save file to push.
+
+        Returns:
+            core.AdbHandler: The AdbHandler instance.
+        """
+        SaveManagement.save_save(save_file)
+        root_handler = core.RootHandler()
+        if not root_handler.is_android():
+            color.ColoredText.localize("root_push_not_android_error")
+            return None
+        if not root_handler.is_rooted():
+            color.ColoredText.localize("not_rooted_error")
+            return None
+        if save_file.used_storage and save_file.package_name is not None:
+            root_handler.set_package_name(save_file.package_name)
+        else:
+            packages = root_handler.get_battlecats_packages()
+            package_name = SaveManagement.select_package_name(packages)
+            if package_name is None:
+                color.ColoredText.localize("no_package_name_error")
+                return root_handler
+            root_handler.set_package_name(package_name)
+        if save_file.save_path is None:
+            return root_handler
+        result = root_handler.load_battlecats_save(save_file.save_path)
+        if result.success:
+            color.ColoredText.localize("root_push_success")
+        else:
+            color.ColoredText.localize("root_push_fail", error=result.result)
+
+        return root_handler
+
     @staticmethod
     def adb_push_rerun(save_file: core.SaveFile):
         """Push the save file to the device and rerun the game.
@@ -214,6 +252,24 @@ class SaveManagement:
         else:
             color.ColoredText.localize("adb_rerun_fail", error=result.result)
 
+    @staticmethod
+    def root_push_rerun(save_file: core.SaveFile):
+        """Push the save file to the device and rerun the game.
+
+        Args:
+            save_file (core.SaveFile): The save file to push.
+        """
+        root_handler = SaveManagement.root_push(save_file)
+        if not root_handler:
+            return
+        if root_handler.package_name is None:
+            return
+        result = root_handler.rerun_game()
+        if result.success:
+            color.ColoredText.localize("root_rerun_success")
+        else:
+            color.ColoredText.localize("root_rerun_fail", error=result.result)
+
     @staticmethod
     def export_save(save_file: core.SaveFile):
         """Export the save file to a json file.
@@ -302,7 +358,7 @@ class SaveManagement:
             "load_from_documents",
             "adb_pull_save",
             "load_save_data_json",
-            "create_new_save",
+            # "create_new_save",
         ]
         if starting_options:
             options.append("edit_config")
@@ -407,32 +463,32 @@ class SaveManagement:
                 save_path, cc = data
             else:
                 save_path = None
-        elif choice == 5:
-            color.ColoredText.localize("create_new_save_warning")
-            cc = core.CountryCode.select()
-            if cc is None:
-                return None, False
-            try:
-                gv = core.GameVersion.from_string(
-                    color.ColoredInput().localize(
-                        "game_version_dialog",
-                    )
-                )
-            except ValueError:
-                color.ColoredText.localize("invalid_game_version")
-                return None, False
-            save = core.SaveFile(cc=cc, gv=gv, load=False)
-            save_path = main.Main.save_save_dialog(save)
-            if save_path is None:
-                return None, False
-            save.to_file(save_path)
-            color.ColoredText.localize("create_new_save_success")
-
-        elif choice == 6 and starting_options:
+        # elif choice == 5:
+        #     color.ColoredText.localize("create_new_save_warning")
+        #     cc = core.CountryCode.select()
+        #     if cc is None:
+        #         return None, False
+        #     try:
+        #         gv = core.GameVersion.from_string(
+        #             color.ColoredInput().localize(
+        #                 "game_version_dialog",
+        #             )
+        #         )
+        #     except ValueError:
+        #         color.ColoredText.localize("invalid_game_version")
+        #         return None, False
+        #     save = core.SaveFile(cc=cc, gv=gv, load=False)
+        #     save_path = main.Main.save_save_dialog(save)
+        #     if save_path is None:
+        #         return None, False
+        #     save.to_file(save_path)
+        #     color.ColoredText.localize("create_new_save_success")
+
+        elif choice == 5 and starting_options:
             core.core_data.config.edit_config()
-        elif choice == 7 and starting_options:
+        elif choice == 6 and starting_options:
             core.update_external_content()
-        elif choice == 8 and starting_options:
+        elif choice == 7 and starting_options:
             main.Main.exit_editor(check_temp=False)
 
         if save_path is None or not save_path.exists():
@@ -465,12 +521,16 @@ class SaveManagement:
                 save_file = core.SaveFile(save_path.read(), cc)
             except Exception:
                 tb = core.core_data.logger.get_traceback()
-                color.ColoredText.localize("parse_save_error", error=tb)
+                color.ColoredText.localize(
+                    "parse_save_error", error=tb, version=bcsfe.__version__
+                )
                 return None
 
         except Exception:
             tb = core.core_data.logger.get_traceback()
-            color.ColoredText.localize("parse_save_error", error=tb)
+            color.ColoredText.localize(
+                "parse_save_error", error=tb, version=bcsfe.__version__
+            )
             return None
 
         save_file.save_path = save_path
@@ -503,6 +563,7 @@ class SaveManagement:
         if new_save_file is None:
             return stop
         save_file.load_save_file(new_save_file)
+        core.core_data.init_data()
         color.ColoredText.localize("load_save_success")
         return False
 
@@ -515,6 +576,7 @@ class SaveManagement:
             return
         save_file.set_cc(cc)
         core.ServerHandler(save_file).create_new_account()
+        core.core_data.init_data()
         color.ColoredText.localize("country_code_set", cc=cc)
 
     @staticmethod
@@ -530,4 +592,5 @@ class SaveManagement:
             color.ColoredText.localize("invalid_game_version")
             return
         save_file.set_gv(gv)
+        core.core_data.init_data()
         color.ColoredText.localize("game_version_set", version=gv.to_string())

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/core/__init__.py

@@ -4,7 +4,7 @@ from typing import Any
 from requests.exceptions import ConnectionError
 from requests import Response
 from json.decoder import JSONDecodeError
-from bcsfe.cli import color
+from bcsfe.cli import color, dialog_creator
 
 from bcsfe.core import (
     country_code,
@@ -30,7 +30,14 @@ from bcsfe.core.game.battle.enemy import (
     EnemyDictionary,
 )
 from bcsfe.core.game.catbase.beacon_base import BeaconEventListScene
-from bcsfe.core.game.catbase.cat import Cat, Cats, UnitBuy, TalentData
+from bcsfe.core.game.catbase.cat import (
+    Cat,
+    Cats,
+    UnitBuy,
+    TalentData,
+    NyankoPictureBook,
+)
+from bcsfe.core.game.catbase.gambling import GamblingEvent
 from bcsfe.core.game.catbase.gatya import (
     Gatya,
     GatyaInfos,
@@ -307,6 +314,14 @@ def update_external_content(_: Any = None):
     ExternalLocaleManager.update_all_external_locales()
     core_data.init_data()
 
+    clear_game_data = dialog_creator.YesNoInput().get_input_once("clear_game_data_q")
+    if clear_game_data is None:
+        return
+
+    if clear_game_data:
+        GameDataGetter.delete_old_versions(0)
+        color.ColoredText.localize("cleared_game_data")
+
 
 def print_no_internet():
     color.ColoredText.localize("no_internet")
@@ -368,4 +383,7 @@ __all__ = [
     "GatyaDataOptionSet",
     "GatyaDataOption",
     "MaxValueType",
+    "GamblingEvent",
+    "UnitBuy",
+    "NyankoPictureBook",
 ]

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/core/game/catbase/__init__.py

@@ -21,6 +21,7 @@ from bcsfe.core.game.catbase import (
     powerup,
     drop_chara,
     playtime,
+    gambling,
 )
 
 __all__ = [
@@ -46,4 +47,5 @@ __all__ = [
     "powerup",
     "drop_chara",
     "playtime",
+    "gambling",
 ]

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/core/game/catbase/cat.py

@@ -227,7 +227,7 @@ class TalentData:
         talent_data_cat = self.get_cat_skill(cat.id)
         if talent_data_cat is None or cat.talents is None:
             return None
-        save_talent_data = cat.talents
+        # save_talent_data = cat.talents
         talent_names: list[str] = []
         max_levels: list[int] = []
         current_levels: list[int] = []
@@ -836,7 +836,6 @@ class Cats:
         self.unit_limit: UnitLimit | None = None
         self.nyanko_picture_book: NyankoPictureBook | None = None
         self.talent_data: TalentData | None = None
-        self.bulk_downloaded = False
 
     def get_all_cats(self) -> list[Cat]:
         return self.cats
@@ -882,7 +881,7 @@ class Cats:
 
     def get_non_gacha_cats(self, save_file: core.SaveFile) -> list[Cat]:
         unitbuy = self.read_unitbuy(save_file)
-        cats = []
+        cats: list[Cat] = []
         for cat in self.cats:
             unit_buy_data = unitbuy.get_unit_buy(cat.id)
             if unit_buy_data is None:
@@ -922,7 +921,6 @@ class Cats:
         save_file: core.SaveFile,
         search_name: str,
     ) -> list[Cat]:
-        self.bulk_download_names(save_file)
         cats: list[Cat] = []
         for cat in self.cats:
             names = cat.get_names_cls(save_file)
@@ -934,32 +932,6 @@ class Cats:
                     break
         return cats
 
-    def bulk_download_names(
-        self, save_file: core.SaveFile, current_cats: list[core.Cat] | None = None
-    ):
-        if self.bulk_downloaded and current_cats is None:
-            return
-        file_names: list[str] = []
-        gdg = core.core_data.get_game_data_getter(save_file)
-        if current_cats is None:
-            cats = self.cats
-        else:
-            cats = current_cats
-        for cat in cats:
-            if cat.names is None:
-                file_name = f"Unit_Explanation{cat.id + 1}_{core.core_data.get_lang(save_file)}.csv"
-                if gdg.is_downloaded("resLocal", file_name):
-                    continue
-                file_names.append(file_name)
-
-        if len(file_names) > 50:
-            gdg.save_all_cat_names_fast()
-
-        core.core_data.get_game_data_getter(save_file).download_all(
-            "resLocal", file_names
-        )
-        self.bulk_downloaded = current_cats is None
-
     def get_cats_obtainable(self, save_file: core.SaveFile) -> list[Cat] | None:
         nyanko_picture_book = self.read_nyanko_picture_book(save_file)
         obtainable_cats = nyanko_picture_book.get_obtainable_cats()

bcsfe-3.2.0b2/src/bcsfe/core/game/catbase/gambling.py

@@ -0,0 +1,111 @@
+from __future__ import annotations
+from bcsfe import core
+from typing import Any
+
+from bcsfe.cli import color
+
+
+class GamblingEvent:
+    def __init__(
+        self,
+        completed: dict[int, bool],
+        values: dict[int, dict[int, int]],
+        start_times: dict[int, int | float],
+    ):
+        self.completed = completed
+        self.values = values
+        self.start_times = start_times
+
+    @staticmethod
+    def init() -> GamblingEvent:
+        return GamblingEvent({}, {}, {})
+
+    @staticmethod
+    def read(data: core.Data, game_version: core.GameVersion) -> GamblingEvent:
+        total = data.read_short()
+        completed: dict[int, bool] = {}
+
+        for _ in range(total):
+            key = data.read_short()
+            completed[key] = data.read_bool()
+
+        total = data.read_short()
+        values: dict[int, dict[int, int]] = {}
+
+        for _ in range(total):
+            key = data.read_short()
+            if key not in values:
+                values[key] = {}
+
+            total2 = data.read_short()
+            for _ in range(total2):
+                key2 = data.read_short()
+
+                values[key][key2] = data.read_short()
+
+        total = data.read_short()
+        start_times: dict[int, int | float] = {}
+
+        for _ in range(total):
+            key = data.read_short()
+
+            if game_version < 90100:
+                value = data.read_double()
+            else:
+                value = data.read_int()
+
+            start_times[key] = value
+
+        return GamblingEvent(completed, values, start_times)
+
+    def write(self, data: core.Data, game_version: core.GameVersion):
+        data.write_short(len(self.completed))
+        data.write_short_bool_dict(self.completed, write_length=False)
+
+        data.write_short(len(self.values))
+
+        for key, value in self.values.items():
+            data.write_short(key)
+            data.write_short(len(value))
+
+            for key2, value2 in value.items():
+                data.write_short(key2)
+                data.write_short(value2)
+
+        data.write_short(len(self.start_times))
+        for key, value in self.start_times.items():
+            data.write_short(key)
+
+            # this is a bad conversion, since float is timestamp i assume and int as the date as YYYMMDD. FIXME
+            if game_version < 90100:
+                data.write_double(float(value))
+            else:
+                data.write_int(int(value))
+
+    def serialize(self) -> dict[str, Any]:
+        return {
+            "completed": self.completed,
+            "values": self.values,
+            "start_times": self.start_times,
+        }
+
+    @staticmethod
+    def deserialize(data: dict[str, Any]) -> GamblingEvent:
+        return GamblingEvent(
+            data.get("completed", {}),
+            data.get("values", {}),
+            data.get("start_times", {}),
+        )
+
+    def reset(self):
+        self.completed = {}
+        self.values = {}
+        # TODO: check start times
+        self.start_times = {}
+
+    @staticmethod
+    def reset_events(save_file: core.SaveFile):
+        save_file.wildcat_slots.reset()
+        color.ColoredText.localize("reset_wildcat_slots")
+        save_file.cat_scratcher.reset()
+        color.ColoredText.localize("reset_cat_scratcher")

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/core/game/map/enigma.py

@@ -195,6 +195,9 @@ class Enigma:
             if wipe is None:
                 return
             if wipe:
+                for stage in self.stages:
+                    id = stage.stage_id
+                    save_file.event_stages.chapter_completion_count[id] = 0
                 self.stages = []
 
         ids, _ = dialog_creator.ChoiceInput(
@@ -209,6 +212,8 @@ class Enigma:
 
         for enigma_id in ids:
             abs_id = enigma_id + base_level
+            save_file.event_stages.chapter_completion_count[abs_id] = 0
+            # TODO: level? they can go much higher than 3... not sure it really matters though
             stage = Stage(3, abs_id, 2, int(time.time()))
             self.stages.append(stage)
 

{bcsfe-3.1.0 → bcsfe-3.2.0b2}/src/bcsfe/core/game/map/gauntlets.py

@@ -334,6 +334,10 @@ class GauntletChapters:
         gauntlets = save_file.behemoth_culling
         gauntlets.edit_chapters(save_file, "Q")
 
+    @staticmethod
+    def edit_enigma_stages(save_file: core.SaveFile):
+        save_file.enigma_clears.edit_chapters(save_file, "H")
+
     def edit_chapters(self, save_file: core.SaveFile, letter_code: str):
         edits.map.edit_chapters(save_file, self, letter_code)