1#@!#!123s

: / opt / cloudlinux / venv / lib / python3.11 / site-packages / setoptconf / source /
Filename : filebased.py
back
import os.path

from copy import deepcopy

from ..config import Configuration
from .base import Source


__all__ = ("HomeDirectory", "ConfigDirectory", "FileBasedSource")


class DirectoryModifier(object):
    def __init__(self, target_file):
        self.target_file = target_file

    def __call__(self):
        raise NotImplementedError()


class HomeDirectory(DirectoryModifier):
    def __call__(self):
        return os.path.expanduser(os.path.join("~", self.target_file))


class ConfigDirectory(DirectoryModifier):
    def __call__(self):
        config_dir = os.getenv("XDG_CONFIG_HOME") or os.path.expanduser(
            os.path.join("~", ".config")
        )

        return os.path.join(config_dir, self.target_file)


class FileBasedSource(Source):
    def __init__(self, files, base_path=None, combine=False):
        super(FileBasedSource, self).__init__()

        if isinstance(files, (str, DirectoryModifier)):
            files = [files]
        elif not isinstance(files, (tuple, list)):
            raise TypeError("files must be a string or list of strings")

        self.files = []
        for target in files:
            if isinstance(target, str):
                self.files.append(target)
            elif isinstance(target, DirectoryModifier):
                self.files.append(target())
            else:
                raise TypeError("files must be a string or list of strings")

        self.base_path = base_path or os.getcwd()
        self.combine = combine

    def get_config(self, settings, manager=None, parent=None):
        parsed_settings = []

        for file_source in self.files:
            if os.path.isabs(file_source):
                file_path = file_source
            else:
                file_path = os.path.join(self.base_path, file_source)

            if os.path.exists(file_path):
                file_settings = self.get_settings_from_file(
                    file_path, deepcopy(settings), manager=manager
                )

                if file_settings:
                    parsed_settings.append(file_settings)

                    if not self.combine:
                        # No need to gather any more, we only want one.
                        break

        if parsed_settings:
            config = parent
            for parsed_setting in reversed(parsed_settings):
                config = Configuration(settings=parsed_setting, parent=config)

        else:
            config = Configuration(settings=settings, parent=parent)

        return config

    def get_settings_from_file(self, file_path, settings, manager=None):
        raise NotImplementedError()