kolo/Argenta
1
0
Fork 0
mirror of https://github.com/koloideal/Argenta.git synced 2026-06-10 18:15:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

work

Browse Source
This commit is contained in:
kolo
2025-04-02 16:34:43 +03:00
parent 1c6f896b73
commit 253790fe2e
1 changed files with 12 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
argenta/app/entity.py
+12 -17
View File
@@ -1,5 +1,6 @@
from typing import Callable from typing import Callable
from inspect import getfullargspec from inspect import getfullargspec
from rich.console import Console
import re import re
from argenta.command.models import Command, InputCommand from argenta.command.models import Command, InputCommand
@@ -19,7 +20,7 @@ from argenta.app.registered_routers.entity import RegisteredRouters
class App: class App:
def __init__(self, def __init__(self,
prompt: str = 'Enter a command', prompt: str = 'Enter a command\n',
initial_message: str = '\nHello, I am Argenta\n', initial_message: str = '\nHello, I am Argenta\n',
farewell_message: str = '\nGoodBye\n', farewell_message: str = '\nGoodBye\n',
exit_command: str = 'Q', exit_command: str = 'Q',
@@ -30,7 +31,7 @@ class App:
line_separate: str = '', line_separate: str = '',
command_group_description_separate: str = '', command_group_description_separate: str = '',
repeat_command_groups: bool = True, repeat_command_groups: bool = True,
print_func: Callable[[str], None] = print) -> None: print_func: Callable[[str], None] = Console().print) -> None:
self._prompt = prompt self._prompt = prompt
self._print_func = print_func self._print_func = print_func
self._exit_command = exit_command self._exit_command = exit_command
@@ -66,14 +67,12 @@ class App:
if not self._repeat_command_groups_description: if not self._repeat_command_groups_description:
self._print_command_group_description() self._print_command_group_description()
self._print_func(self._prompt)
while True: while True:
if self._repeat_command_groups_description: if self._repeat_command_groups_description:
self._print_command_group_description() self._print_command_group_description()
self._print_func(self._prompt)
raw_command: str = input() raw_command: str = Console().input(self._prompt)
try: try:
input_command: InputCommand = InputCommand.parse(raw_command=raw_command) input_command: InputCommand = InputCommand.parse(raw_command=raw_command)
@@ -86,14 +85,12 @@ class App:
self._print_func(self._prompt) self._print_func(self._prompt)
continue continue
is_exit = self._is_exit_command(input_command) if self._is_exit_command(input_command):
if is_exit:
return return
self._print_func(self._line_separate) self._print_func(self._line_separate)
is_unknown_command: bool = self._check_is_command_unknown(input_command)
if is_unknown_command: if self._is_unknown_command(input_command):
self._print_func(self._line_separate) self._print_func(self._line_separate)
if not self._repeat_command_groups_description: if not self._repeat_command_groups_description:
self._print_func(self._prompt) self._print_func(self._prompt)
@@ -209,23 +206,21 @@ class App:
if self._ignore_exit_command_register: if self._ignore_exit_command_register:
system_router.input_command_handler(command) system_router.input_command_handler(command)
return True return True
else: elif command.get_trigger() == self._exit_command:
if command.get_trigger() == self._exit_command: system_router.input_command_handler(command)
system_router.input_command_handler(command) return True
return True
return False return False
def _check_is_command_unknown(self, command: InputCommand): def _is_unknown_command(self, command: InputCommand):
for router_entity in self._registered_routers: for router_entity in self._registered_routers:
for command_handler in router_entity.get_command_handlers(): for command_handler in router_entity.get_command_handlers():
handled_command_trigger = command_handler.get_handled_command().get_trigger() handled_command_trigger = command_handler.get_handled_command().get_trigger()
if handled_command_trigger.lower() == command.get_trigger().lower(): if handled_command_trigger.lower() == command.get_trigger().lower():
if self._ignore_command_register: if self._ignore_command_register:
return False return False
else: elif handled_command_trigger == command.get_trigger():
if handled_command_trigger == command.get_trigger(): return False
return False
self._unknown_command_handler(command) self._unknown_command_handler(command)
return True return True