Mini Shell

import glob
import logging
from typing import List

from im360.model.global_whitelist import GlobalWhitelist

CUSTOM_WHITELIST_MASK = "/etc/imunify360/whitelist/*.txt"
CUSTOM_BLACKLIST_MASK = "/etc/imunify360/blacklist/*.txt"

logger = logging.getLogger(__name__)


class CustomWhitelist(GlobalWhitelist):
    _LIST_PATH = CUSTOM_WHITELIST_MASK

    @classmethod
    async def load(cls, group=None) -> List[str]:
        result = []  # type: List[str]

        for f in glob.glob(cls._LIST_PATH):
            logger.info("Loading %s to %s", f, cls.__name__)
            result.extend(cls._load_file(f, log_error=logger.warning))

        return result


class CustomBlacklist(CustomWhitelist):
    _LIST_PATH = CUSTOM_BLACKLIST_MASK