from defence360agent.contracts.config import IConfig, LocalConfig def migrate( migrator, database, fake=False, config_file: IConfig = LocalConfig(), **kwargs ): if fake: return conf = config_file.config_to_dict() if not conf: return mod_sec_settings = conf.setdefault("MOD_SEC", {}) mod_sec_settings.pop("was_installed", None) mod_sec_settings.pop("OWASP_deleted", None) config_file.dict_to_config(conf, validate=False, overwrite=True) def rollback(migrator, database, fake=False, **kwargs): pass