annif.config

Complexity

Total Complexity

16

Size/Duplication

Total Lines	86
Duplicated Lines	77.91 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
B	parse_config()	28	28	6

"""Configuration file handling"""


import os.path
import configparser
import tomli
import annif
import annif.util
from annif.exception import ConfigurationException


logger = annif.logger


class AnnifConfigCFG:

This code seems to be duplicated in your project.

    """Class for reading configuration in CFG/INI format"""

    def __init__(self, filename):
        self._config = configparser.ConfigParser()
        self._config.optionxform = annif.util.identity
        with open(filename, encoding='utf-8-sig') as projf:
            try:
                logger.debug(
                    f"Reading configuration file {filename} in CFG format")
                self._config.read_file(projf)
            except (configparser.DuplicateOptionError,
                    configparser.DuplicateSectionError) as err:
                raise ConfigurationException(err)

    @property
    def project_ids(self):
        return self._config.sections()

    def __getitem__(self, key):
        return self._config[key]


class AnnifConfigTOML:

This code seems to be duplicated in your project.

    """Class for reading configuration in TOML format"""

    def __init__(self, filename):
        with open(filename, "rb") as projf:
            try:
                logger.debug(
                    f"Reading configuration file {filename} in TOML format")
                self._config = tomli.load(projf)
            except tomli.TOMLDecodeError as err:
                raise ConfigurationException(err)

    @property
    def project_ids(self):
        return self._config.keys()

    def __getitem__(self, key):
        return self._config[key]


def parse_config(projects_file):

This code seems to be duplicated in your project.

    if projects_file:
        if not os.path.exists(projects_file):
            logger.warning(
                f'Project configuration file "{projects_file}" is ' +
                'missing. Please provide one. ' +
                'You can set the path to the project configuration ' +
                'file using the ANNIF_PROJECTS environment ' +
                'variable or the command-line option "--projects".')
            return None
    else:
        if os.path.exists('projects.cfg'):
            projects_file = 'projects.cfg'
        elif os.path.exists('projects.toml'):
            projects_file = 'projects.toml'
        else:
            logger.warning(
                'Could not find project configuration file ' +
                '"projects.cfg" or "projects.toml". ' +
                'You can set the path to the project configuration ' +
                'file using the ANNIF_PROJECTS environment ' +
                'variable or the command-line option "--projects".')
            return None

    if projects_file.endswith('.toml'):  # TOML format
        return AnnifConfigTOML(projects_file)
    else:  # classic CFG/INI style format
        return AnnifConfigCFG(projects_file)