mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 18:15:28 +03:00
final adding
This commit is contained in:
@@ -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
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user