import os import yaml from defence360agent.contracts.config import IConfigFile, LocalConfig from defence360agent.utils import log_error_and_ignore @log_error_and_ignore() def migrate( migrator, database, fake=False, config_file: IConfigFile = LocalConfig(), **kwargs ): if fake: return if not os.path.exists(config_file.path): return with open(config_file.path) as f: conf = yaml.safe_load(f) malware_settings = conf.setdefault("MALWARE_SCANNING", {}) value = malware_settings.pop("scan_modified_files", None) malware_settings["scan_modified_files"] = value config_file.dict_to_config(conf, validate=False) def rollback(migrator, database, fake=False, **kwargs): pass