import os import yaml from defence360agent.contracts.config import LocalConfig def migrate(migrator, database, fake=False, **kwargs): if fake: return local_config = LocalConfig() if not os.path.exists(local_config.path): return with open(local_config.path) as f: conf = yaml.safe_load(f) malware_settings = conf.setdefault("MALWARE_SCANNING", {}) value = malware_settings.pop("enable_scan_uploaded_files", True) malware_settings["enable_scan_pure_ftpd"] = value malware_settings["enable_scan_modsec"] = value local_config.dict_to_config(conf, validate=False) def rollback(migrator, database, fake=False, **kwargs): pass