annif.config

Complexity

Total Complexity

10

Size/Duplication

Total Lines	54
Duplicated Lines	72.22 %

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	AnnifConfigTOML.__getitem__()	2	2	1
A	AnnifConfigTOML.__init__()	8	8	3
A	AnnifConfigTOML.project_ids()	3	3	1
A	AnnifConfigCFG.project_ids()	3	3	1
A	AnnifConfigCFG.__getitem__()	2	2	1
A	AnnifConfigCFG.__init__()	11	11	3

"""Configuration file handling"""


import configparser
import tomli
import annif
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]