mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 10:05:28 +03:00
135 lines
4.6 KiB
Python
135 lines
4.6 KiB
Python
__all__ = [
|
|
"ArgSpace",
|
|
"ArgParser",
|
|
]
|
|
|
|
import sys
|
|
from argparse import ArgumentParser, Namespace
|
|
from typing import Never, Self
|
|
|
|
from argenta.orchestrator.argparser.arguments.models import (
|
|
BaseArgument,
|
|
BooleanArgument,
|
|
InputArgument,
|
|
ValueArgument,
|
|
)
|
|
|
|
|
|
class ArgSpace:
|
|
def __init__(self, all_arguments: list[InputArgument]) -> None:
|
|
self.all_arguments = all_arguments
|
|
|
|
self._name_object_paired_args: dict[str, InputArgument] = {}
|
|
self._type_object_paired_args: dict[type[BaseArgument], list[InputArgument]] = {
|
|
BooleanArgument: [],
|
|
ValueArgument: []
|
|
}
|
|
|
|
self._setup_getters()
|
|
|
|
@classmethod
|
|
def from_namespace(
|
|
cls,
|
|
namespace: Namespace,
|
|
processed_args: list[ValueArgument | BooleanArgument]
|
|
) -> Self:
|
|
name_type_paired_processed_args: dict[str, type[BaseArgument]] = {
|
|
arg.name: type(arg) for arg in processed_args
|
|
}
|
|
parsed_arguments: list[InputArgument] = []
|
|
|
|
for name, value in vars(namespace).items():
|
|
parsed_arguments.append(
|
|
InputArgument(
|
|
name=name,
|
|
value=value,
|
|
founder_class=name_type_paired_processed_args[name]
|
|
)
|
|
)
|
|
|
|
return cls(parsed_arguments)
|
|
|
|
def _setup_getters(self) -> None:
|
|
if not self.all_arguments:
|
|
return
|
|
for input_arg in self.all_arguments:
|
|
self._name_object_paired_args[input_arg.name] = input_arg
|
|
self._type_object_paired_args[input_arg.founder_class].append(input_arg)
|
|
|
|
def get_by_name(self, name: str) -> InputArgument | None:
|
|
return self._name_object_paired_args.get(name)
|
|
|
|
def get_by_type(self, arg_type: type[BaseArgument]) -> list[InputArgument] | list[Never]:
|
|
return self._type_object_paired_args.get(arg_type, [])
|
|
|
|
|
|
class ArgParser:
|
|
def __init__(
|
|
self,
|
|
processed_args: list[ValueArgument | BooleanArgument],
|
|
*,
|
|
name: str = "Argenta",
|
|
description: str = "Argenta available arguments",
|
|
epilog: str = "github.com/koloideal/Argenta | made by kolo",
|
|
) -> None:
|
|
"""
|
|
Public. Cmd argument parser and configurator at startup
|
|
:param name: the name of the ArgParse instance
|
|
:param description: the description of the ArgParse instance
|
|
:param epilog: the epilog of the ArgParse instance
|
|
:param processed_args: registered and processed arguments
|
|
"""
|
|
self.name: str = name
|
|
self.description: str = description
|
|
self.epilog: str = epilog
|
|
self.processed_args: list[ValueArgument | BooleanArgument] = processed_args
|
|
|
|
self.parsed_argspace: ArgSpace = ArgSpace([])
|
|
|
|
self._core: ArgumentParser = ArgumentParser(prog=name, description=description, epilog=epilog)
|
|
self._register_args(processed_args)
|
|
|
|
def _parse_args(self) -> None:
|
|
self.parsed_argspace = ArgSpace.from_namespace(
|
|
namespace=self._core.parse_args(), processed_args=self.processed_args
|
|
)
|
|
|
|
def _register_args(self, processed_args: list[ValueArgument | BooleanArgument]) -> None:
|
|
if sys.version_info >= (3, 13):
|
|
for arg in processed_args:
|
|
if isinstance(arg, BooleanArgument):
|
|
_ = self._core.add_argument(
|
|
arg.string_entity,
|
|
action=arg.action,
|
|
help=arg.help,
|
|
deprecated=arg.is_deprecated
|
|
)
|
|
else:
|
|
_ = self._core.add_argument(
|
|
arg.string_entity,
|
|
action=arg.action,
|
|
help=arg.help,
|
|
default=arg.default,
|
|
choices=arg.possible_values,
|
|
required=arg.is_required,
|
|
deprecated=arg.is_deprecated,
|
|
)
|
|
else:
|
|
for arg in processed_args:
|
|
if isinstance(arg, BooleanArgument):
|
|
_ = self._core.add_argument(
|
|
arg.string_entity,
|
|
action=arg.action,
|
|
help=arg.help,
|
|
)
|
|
else:
|
|
_ = self._core.add_argument(
|
|
arg.string_entity,
|
|
action=arg.action,
|
|
help=arg.help,
|
|
default=arg.default,
|
|
choices=arg.possible_values,
|
|
required=arg.is_required
|
|
)
|
|
|