doomcc/dcc/fabricate.py

import contextlib
import dcc.config
import dcc.doom_base
import shutil
import subprocess
import tempfile


class Fabricate(dcc.doom_base.WadMap):
    def get_parser(self, prog_name):
        parser = super().get_parser(prog_name)
        parser.add_argument("--fg", action="store_true")
        return parser

    def take_action(self, parsed_args):
        with tempfile.TemporaryDirectory() as td:
            with contextlib.chdir(td):
                command = [self.dsda]
                if not parsed_args.fg and shutil.which("xvfb-run") is not None:
                    command = ["xvfb-run"] + command
                subprocess.run(
                    command + self.dsda_preamble()
                    + ["-timedemo", self.demo_in_path()]
                    + ["-viddump", self.video_path()]
                )

    def options_dict(self):
        opt_dict = super().options_dict()
        for k, v in self._config.get("fabricate_options", {}).items():
            opt_dict[k] = v
        return opt_dict