관리-도구

편집 파일: 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