doomcc/dcc/config.py

import io
import pathlib
import os
import re
import tomlkit.toml_file

from cliff.command import Command


class Base(Command):
    def get_parser(self, prog_name):
        parser = super().get_parser(prog_name)
        parser.add_argument(
            "--doom", default=pathlib.Path.home().joinpath("doom"))
        parser.add_argument("--config-name", default="config.toml")
        return parser

    def init_base(self, parsed_args):
        self._doom = pathlib.Path(parsed_args.doom)
        self._config_name = parsed_args.config_name
        self._config = tomlkit.toml_file.TOMLFile(
            self.doom.joinpath(self.config_name)).read()

        self._dsda = self._config.get("dsda")
        if self.dsda is None:
            raise Exception(
                "required key 'dsda' not set in config "
                + f"{self.doom.joinpath(self.config_name)}.")
        for d in ("iwads", "pwads", "demos", "fabricate"):
            self._init_attr([d], d, fn=self.doom.joinpath)

    def finalize(self):
        self._init_attr(["thumbnail", "height"], 720)
        self._init_attr(["thumbnail", "width"], 1280)
        self._init_attr(["thumbnail", "font"], None)
        self._init_attr(["thumbnail", "text_fill"], "white")
        self._init_attr(["thumbnail", "text_stroke"], "red")
        self._init_attr(
            ["fetch", "mirror"],
            "https://youfailit.net/pub/idgames"
        )

    def run(self, parsed_args):
        self.init_base(parsed_args)
        self.finalize()
        self.take_action(parsed_args)

    def _init_attr(self, what, default, fn=lambda x: x):
        propname = "_".join(what)
        val = self._config
        for w in what:
            val = val[w]
            if val is None:
                val = default
                break

        setattr(self, f"_{propname}", fn(val))
        setattr(
            type(self), propname,
            property(lambda self: getattr(self, f"_{propname}"))
        )

    @property
    def doom(self):
        return self._doom

    @property
    def config_name(self):
        return self._config_name

    @property
    def dsda(self):
        return self._doom.joinpath(self._dsda)