refactor, new model e.t.c.

argenta/command/entity.py

@@ -1,4 +1,5 @@
-from argenta.command.flag.entity import Flag
+from argenta.command.flag.registered_flag.entity import Flag
+from argenta.command.flag.input_flag.entity import InputFlag
 from argenta.command.flag.flags_group import FlagsGroup
 from .exceptions import (UnprocessedInputFlagException,
                          RepeatedInputFlagsException,
@@ -32,7 +33,7 @@ class Command(Generic[CommandType]):
         return self._registered_flags
 
 
-    def validate_input_flag(self, flag: Flag):
+    def validate_input_flag(self, flag: InputFlag):
         registered_flags: FlagsGroup | None = self.get_registered_flags()
         if registered_flags:
             if isinstance(registered_flags, Flag):
@@ -85,8 +86,8 @@ class Command(Generic[CommandType]):
                 flag_prefix_last_symbol_index = current_flag_name.rfind('-')
                 flag_prefix = current_flag_name[:flag_prefix_last_symbol_index+1]
                 flag_name = current_flag_name[flag_prefix_last_symbol_index+1:]
-                input_flag = Flag(flag_name=flag_name,
-                                  flag_prefix=cast(Literal['-', '--', '---'], flag_prefix))
+                input_flag = InputFlag(flag_name=flag_name,
+                                       flag_prefix=cast(Literal['-', '--', '---'], flag_prefix))
                 input_flag.set_value(current_flag_value)
 
                 all_flags = [x.get_string_entity() for x in flags.get_flags()]

argenta/command/exceptions.py

@@ -1,4 +1,5 @@
-from argenta.command.flag.entity import Flag
+from argenta.command.flag.input_flag.entity import InputFlag
+from argenta.command.flag.registered_flag.entity import Flag
 
 
 class UnprocessedInputFlagException(Exception):
@@ -7,7 +8,7 @@ class UnprocessedInputFlagException(Exception):
 
 
 class RepeatedInputFlagsException(Exception):
-    def __init__(self, flag: Flag):
+    def __init__(self, flag: Flag | InputFlag):
         self.flag = flag
     def __str__(self):
         return ("Repeated Input Flags\n"

argenta/command/flag/__init__.py

@@ -1,5 +0,0 @@
-__all__ = ["Flag", "FlagsGroup"]
-
-
-from .entity import Flag
-from .flags_group.entity import FlagsGroup

argenta/command/flag/base_flag/entity.py

@@ -1,7 +1,7 @@
 from typing import Literal, Pattern
 
 
-class Flag:
+class BaseFlag:
     def __init__(self, flag_name: str,
                  flag_prefix: Literal['-', '--', '---'] = '--',
                  possible_flag_values: list[str] | Pattern[str] | False = True):
@@ -9,8 +9,6 @@ class Flag:
         self._flag_prefix = flag_prefix
         self.possible_flag_values = possible_flag_values
 
-        self._flag_value = None
-
     def get_string_entity(self):
         string_entity: str = self._flag_prefix + self._flag_name
         return string_entity
@@ -21,12 +19,6 @@ class Flag:
     def get_flag_prefix(self):
         return self._flag_prefix
 
-    def get_value(self):
-        return self._flag_value
-
-    def set_value(self, value):
-        self._flag_value = value
-
     def validate_input_flag_value(self, input_flag_value: str | None):
         if self.possible_flag_values is False:
             if input_flag_value is None:

argenta/command/flag/flags_group/entity.py

@@ -1,22 +1,23 @@
-from argenta.command.flag import Flag
+from argenta.command.flag.input_flag.entity import InputFlag
+from argenta.command.flag.registered_flag import Flag
 
 
 class FlagsGroup:
-    def __init__(self, *flags: Flag):
-        self._flags: list[Flag] = [] if not flags else flags
+    def __init__(self, *flags: Flag | InputFlag):
+        self._flags: list[Flag | InputFlag] = [] if not flags else flags
 
-    def get_flags(self) -> list[Flag]:
+    def get_flags(self) -> list[Flag | InputFlag]:
         return self._flags
 
-    def add_flag(self, flag: Flag):
+    def add_flag(self, flag: Flag | InputFlag):
         self._flags.append(flag)
 
-    def add_flags(self, flags: list[Flag]):
+    def add_flags(self, flags: list[Flag | InputFlag]):
         self._flags.extend(flags)
 
     def unparse_to_dict(self):
         result_dict: dict[str, dict] = {}
-        for flag in self._flags:
+        for flag in self.get_flags():
             result_dict[flag.get_flag_name()] = {
                 'name': flag.get_flag_name(),
                 'string_entity': flag.get_string_entity(),

argenta/command/flag/input_flag/entity.py

@@ -0,0 +1,18 @@
+from re import Pattern
+from typing import Literal
+
+from argenta.command.flag.base_flag.entity import BaseFlag
+
+
+class InputFlag(BaseFlag):
+    def __init__(self, flag_name: str,
+                 flag_prefix: Literal['-', '--', '---'] = '--',
+                 possible_flag_values: list[str] | Pattern[str] | False = True):
+        super().__init__(flag_name, flag_prefix, possible_flag_values)
+        self._flag_value = None
+
+    def get_value(self):
+        return self._flag_value
+
+    def set_value(self, value):
+        self._flag_value = value

argenta/command/flag/registered_flag/__init__.py

@@ -0,0 +1,5 @@
+__all__ = ["Flag", "FlagsGroup"]
+
+
+from .entity import Flag
+from argenta.command.flag.flags_group import FlagsGroup

argenta/command/flag/registered_flag/defaults.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from argenta.command.flag import Flag
+from argenta.command.flag.registered_flag import Flag
 import re
 
 

argenta/command/flag/registered_flag/entity.py

@@ -0,0 +1,5 @@
+from argenta.command.flag.base_flag.entity import BaseFlag
+
+
+class Flag(BaseFlag):
+    pass