Source code for neuralib.util.cli_args

from __future__ import annotations

import attrs

from neuralib.util.deprecation import deprecated_func

__all__ = ['CliArgs']




@deprecated_func(remarks='over engineering', removal_version="0.4.0")
@attrs.define
class CliArgs:
    flag: str | None = attrs.field(default=None)
    args: str | None = attrs.field(default=None)

    # noinspection PyUnresolvedReferences
    @flag.validator
    def _check_flag(self, attribute, value: str | None) -> None:
        if value is None:
            return

        if not isinstance(value, str):
            raise TypeError(f'flag not a str: "{value}"')

        valid1 = value.startswith('-') and len(value) > 1
        valid2 = value.startswith('--') and len(value) > 2
        if not (valid1 or valid2):
            raise ValueError(f'{self} {attribute.name} not valid')

    # noinspection PyUnresolvedReferences
    @args.validator
    def _check_args(self, attribute, value: str) -> None:
        if value is None:
            return

        if not isinstance(value, str):
            self.args = str(value)

    def as_command(self) -> list[str]:

        # position arg
        if self.flag is None and self.args is not None:
            return [self.args]

        # flag
        elif self.flag is not None and self.args is None:
            return [self.flag]

        # flag + arg
        elif self.flag is not None and self.args is not None:
            return [self.flag, self.args]
        else:
            raise RuntimeError('')

    @classmethod
    def concat_command(cls, args: list[CliArgs]) -> list[str]:
        """concat multiple ``CliArgs`` as list of subprocess command line"""
        ret = []
        for arg in args:
            ret.extend(arg.as_command())

        return ret