Source code for dffml.cli.config

import sys
import pathlib

from ..configloader.configloader import BaseConfigLoader
from ..configloader.json import JSONConfigLoader
from ..util.cli.cmd import CMD
from ..base import config, field


@config
class ConvertConfig:
    original: str = field("Config to convert")
    config_in: BaseConfigLoader = field(
        "ConfigLoader to use for importing", default=None,
    )
    config_out: BaseConfigLoader = field(
        "ConfigLoader to use for exporting", default=JSONConfigLoader,
    )


class Convert(CMD):

    CONFIG = ConvertConfig

    async def run(self):
        original_path = pathlib.Path(self.original)
        config_out = self.config_out.withconfig(self.extra_config)
        # Load input configloader
        config_in = self.config_in
        if config_in is None:
            config_type = original_path.suffix.replace(".", "")
            config_in = BaseConfigLoader.load(config_type)
        config_in = config_in.withconfig(self.extra_config)
        async with config_in as cl_in, config_out as cl_out:
            async with cl_in() as loader_in, cl_out() as loader_out:
                imported = await loader_in.loadb(original_path.read_bytes())
                sys.stdout.buffer.write(await loader_out.dumpb(imported))


class Config(CMD):

    convert = Convert