diff --git a/mini_bdx_runtime/mini_bdx_runtime/duck_config.py b/mini_bdx_runtime/mini_bdx_runtime/duck_config.py index f32b87e..c570c15 100644 --- a/mini_bdx_runtime/mini_bdx_runtime/duck_config.py +++ b/mini_bdx_runtime/mini_bdx_runtime/duck_config.py @@ -6,13 +6,17 @@ HOME_DIR = os.path.expanduser("~") class DuckConfig: + def __init__( - self, config_json_path: Optional[str] = f"{HOME_DIR}/duck_config.json" + self, + config_json_path: Optional[str] = f"{HOME_DIR}/duck_config.json", + ignore_default: bool = False, ): """ Looks for duck_config.json in the home directory by default. If not found, uses default values. """ + self.default = False try: self.json_config = ( json.load(open(config_json_path, "r")) if config_json_path else {} @@ -22,9 +26,25 @@ class DuckConfig: f"Warning : didn't find the config json file at {config_json_path}, using default values" ) self.json_config = {} + self.default = True if config_json_path is None: print("Warning : didn't provide a config json path, using default values") + self.default = True + + if self.default and not ignore_default: + print("") + print("") + print("") + print("") + print("======") + print( + "WARNING : Running with default values probably won't work well. Please make a duck_config.json file and set the parameters." + ) + res = input("Do you still want to run ? (y/N)") + if res.lower() != "y": + print("Exiting...") + exit(1) self.start_paused = self.json_config.get("start_paused", False) self.imu_upside_down = self.json_config.get("imu_upside_down", False) diff --git a/scripts/find_soft_offsets.py b/scripts/find_soft_offsets.py index 2212389..0c945b5 100644 --- a/scripts/find_soft_offsets.py +++ b/scripts/find_soft_offsets.py @@ -6,7 +6,7 @@ from mini_bdx_runtime.rustypot_position_hwi import HWI from mini_bdx_runtime.duck_config import DuckConfig import time -dummy_config = DuckConfig(config_json_path=None) +dummy_config = DuckConfig(config_json_path=None, ignore_default=True) print("======") print(