Current File : //opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrations/109_dos_detector.py
import logging
from defence360agent.contracts.config import (
_DOS_DETECTOR_MIN_LIMIT,
ConfigFile,
)
logger = logging.getLogger(__name__)
def migrate(migrator, database, fake=False, **kwargs):
if fake:
return
try:
config_file = ConfigFile()
config = config_file.config_to_dict(False)
if "DOS" not in config:
return
if "max_connections" in config["DOS"] and isinstance(
config["DOS"]["max_connections"], int
):
config["DOS"]["default_limit"] = max(
config["DOS"]["max_connections"], _DOS_DETECTOR_MIN_LIMIT
)
del config["DOS"]["max_connections"]
if "timeout" in config["DOS"]:
config["DOS"]["interval"] = config["DOS"]["timeout"]
del config["DOS"]["timeout"]
config_file.dict_to_config(config, overwrite=True)
except Exception:
logger.exception("Failed to replace DOS settings")
def rollback(migrator, database, fake=False, **kwargs):
pass
Mini Shell