final adding

This commit is contained in:
2025-04-10 00:23:03 +03:00
parent 8b496aa782
commit d1d644d422
10 changed files with 42 additions and 93 deletions
+6 -6
View File
@@ -5,17 +5,17 @@ from argenta.command.flag.models import InputFlags
class CommandHandler:
def __init__(self, handler: Callable[[], None] | Callable[[InputFlags], None], handled_command: Command):
self.handler = handler
self.handled_command = handled_command
self._handler = handler
self._handled_command = handled_command
def handling(self, input_flags: InputFlags = None):
if input_flags is not None:
self.handler(input_flags)
self._handler(input_flags)
else:
self.handler()
self._handler()
def get_handler(self):
return self.handler
return self._handler
def get_handled_command(self):
return self.handled_command
return self._handled_command
+7
View File
@@ -110,6 +110,13 @@ class Router:
self._ignore_command_register = ignore_command_register
def get_triggers(self):
all_triggers: list[str] = []
for command_handler in self._command_handlers:
all_triggers.append(command_handler.get_handled_command().get_trigger())
return all_triggers
def get_command_handlers(self) -> CommandHandlers:
return self._command_handlers