This commit is contained in:
2025-03-01 16:44:33 +03:00
parent fd4f2e1570
commit a378163431
14 changed files with 138 additions and 154 deletions
+9 -3
View File
@@ -1,11 +1,11 @@
from typing import Literal
from typing import Literal, Pattern
class Flag:
def __init__(self, flag_name: str,
flag_prefix: Literal['-', '--', '---'] = '-',
ignore_flag_value_register: bool = False,
possible_flag_values: list[str] = False):
possible_flag_values: list[str] | Pattern[str] = False):
self._flag_name = flag_name
self._flag_prefix = flag_prefix
self.possible_flag_values = possible_flag_values
@@ -30,7 +30,13 @@ class Flag:
self._value = value
def validate_input_flag_value(self, input_flag_value: str):
if self.possible_flag_values:
if isinstance(self.possible_flag_values, Pattern):
is_valid = bool(self.possible_flag_values.match(input_flag_value))
if bool(is_valid):
return True
else:
return False
if isinstance(self.possible_flag_values, list):
if self.ignore_flag_value_register:
if input_flag_value.lower() in [x.lower() for x in self.possible_flag_values]:
return True
@@ -1,18 +1,17 @@
from argenta.command.params.flag.entity import Flag
from argenta.command.params.flag.input_flag.entity import InputFlag
class FlagsGroup:
def __init__(self, flags: list[Flag | InputFlag] = None):
self._flags: list[Flag | InputFlag] = [] if not flags else flags
def __init__(self, flags: list[Flag] = None):
self._flags: list[Flag] = [] if not flags else flags
def get_flags(self):
return self._flags
def add_flag(self, flag: Flag | InputFlag):
def add_flag(self, flag: Flag):
self._flags.append(flag)
def add_flags(self, flags: list[Flag | InputFlag]):
def add_flags(self, flags: list[Flag]):
self._flags.extend(flags)
def __iter__(self):
@@ -1,11 +0,0 @@
from ...flag.entity import Flag
class InputFlag(Flag):
def set_value(self, value: str):
self._value = value
def get_value(self) -> str:
return self._value