ocrd.config

Complexity

Total Complexity

7

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
B	load_config_file()	0	20	7

from pathlib import Path
from datetime import datetime

from yaml import safe_load, safe_dump

from ocrd_models import OcrdConfig
from ocrd_utils import VERSION
import ocrd_utils
from ocrd_validators import OcrdConfigValidator
from ocrd_models.ocrd_config import DEFAULT_CONFIG

def load_config_file(basedir=None):
    """
    Load the configuration file
    """
    if not basedir:
        basedir = ocrd_utils.XDG_CONFIG_HOME
    fpath = Path(basedir, 'ocrd', 'config.yml')
    if not fpath.parent.exists():
        fpath.parent.mkdir(parents=True)
    obj = DEFAULT_CONFIG
    if not fpath.exists():
        with open(str(fpath), 'w', encoding='utf-8') as f_out:
            f_out.write("# Generated by OCR-D/core %s on %s\n" % (VERSION, datetime.now()))
            f_out.write(safe_dump(obj))
    with open(str(fpath), 'r', encoding='utf-8') as f_in:
        obj = {**obj, **safe_load(f_in.read())}
        report = OcrdConfigValidator.validate(obj)
        if not report.is_valid:
            raise ValueError("The configuration is invalid: %s" % report.errors)
        return OcrdConfig(obj)