mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 10:05:28 +03:00
Compare commits
14 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 07ac2af71e | |||
| c4b3aa7db8 | |||
| 61ef6a6466 | |||
| 477f3a7dec | |||
| adf3431388 | |||
| 83955aa046 | |||
| 5a17e916eb | |||
| 1159dda16e | |||
| 315508a36e | |||
| 9d6598c4e0 | |||
| eb43806da6 | |||
| e076dbf84f | |||
| 2f090b6b47 | |||
| c9dbf2bbae |
@@ -74,7 +74,7 @@
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||

|

|
||||||
An example of the TUI appearance
|
An example of the TUI appearance
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|||||||
Binary file not shown.
|
After Width: | Height: | Size: 50 KiB |
@@ -1,7 +1,26 @@
|
|||||||
|
from mock.mock_app.handlers.routers import work_router
|
||||||
|
|
||||||
from argenta.app import App
|
from argenta.app import App
|
||||||
|
from argenta.app.defaults import PredefinedMessages
|
||||||
|
from argenta.app.autocompleter import AutoCompleter
|
||||||
from argenta.orchestrator import Orchestrator
|
from argenta.orchestrator import Orchestrator
|
||||||
|
from argenta.orchestrator.argparser import ArgParser
|
||||||
|
from argenta.orchestrator.argparser.arguments import BooleanArgument
|
||||||
|
|
||||||
app = App(repeat_command_groups=True)
|
|
||||||
|
|
||||||
orchestrator = Orchestrator()
|
arg_parser = ArgParser(processed_args=[BooleanArgument('repeat')])
|
||||||
orchestrator.start_polling(app)
|
app: App = App(autocompleter=AutoCompleter('.hist'))
|
||||||
|
orchestrator: Orchestrator = Orchestrator()
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
app.include_router(work_router)
|
||||||
|
|
||||||
|
app.add_message_on_startup(PredefinedMessages.USAGE)
|
||||||
|
app.add_message_on_startup(PredefinedMessages.AUTOCOMPLETE)
|
||||||
|
app.add_message_on_startup(PredefinedMessages.HELP)
|
||||||
|
|
||||||
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
|
|||||||
+21
-4
@@ -1,6 +1,23 @@
|
|||||||
|
from argenta.response import Response, Status
|
||||||
from argenta.app import App
|
from argenta.app import App
|
||||||
from argenta.command.models import InputCommand
|
from argenta.app.dividing_line import StaticDividingLine, DynamicDividingLine
|
||||||
|
from argenta.app.autocompleter import AutoCompleter
|
||||||
|
from argenta.app.defaults import PredefinedMessages
|
||||||
|
from argenta.command import Command
|
||||||
|
from argenta.command.flags import Flags, InputFlags, InvalidValueInputFlags, UndefinedInputFlags, ValidInputFlags
|
||||||
|
from argenta.command.flag import Flag, InputFlag
|
||||||
|
from argenta.command.flag.defaults import PredefinedFlags
|
||||||
|
from argenta.router import Router
|
||||||
|
from argenta.orchestrator import Orchestrator
|
||||||
|
|
||||||
|
from argenta.command.models import InputCommand
|
||||||
|
import inspect
|
||||||
|
|
||||||
|
|
||||||
|
router = Router()
|
||||||
|
|
||||||
|
|
||||||
|
@router.command(Command('some'))
|
||||||
|
def handler(res: Response) -> Response:
|
||||||
|
pass
|
||||||
|
|
||||||
app = App()
|
|
||||||
app._all_registered_triggers_in_lower = ['fr', 'Tre', 'Pre']
|
|
||||||
print(app._is_unknown_command(InputCommand('fr')))
|
|
||||||
@@ -1,6 +1,9 @@
|
|||||||
from rich.console import Console
|
from rich.console import Console
|
||||||
|
|
||||||
from argenta.command import Command
|
from argenta.command import Command
|
||||||
|
from argenta.command.flag.defaults import PredefinedFlags
|
||||||
|
from argenta.command.flags import Flags
|
||||||
|
from argenta.response import Response
|
||||||
from argenta.router import Router
|
from argenta.router import Router
|
||||||
|
|
||||||
|
|
||||||
@@ -9,14 +12,20 @@ work_router: Router = Router(title='Work points:')
|
|||||||
console = Console()
|
console = Console()
|
||||||
|
|
||||||
|
|
||||||
@work_router.command(Command('get', 'Get Help', aliases=['help', 'Get_help']))
|
@work_router.command(Command('get', 'Get Help', aliases=['help', 'Get_help'], flags=Flags(PredefinedFlags.PORT, PredefinedFlags.HOST)))
|
||||||
def command_help():
|
def command_help(response: Response):
|
||||||
pass
|
print(response.status)
|
||||||
|
print(response.undefined_flags.get_flags())
|
||||||
|
print(response.valid_flags.get_flags())
|
||||||
|
print(response.invalid_value_flags.get_flags())
|
||||||
|
|
||||||
|
|
||||||
@work_router.command(Command('run', 'Run All'))
|
@work_router.command('run')
|
||||||
def command_start_solving():
|
def command_start_solving(response: Response):
|
||||||
pass
|
print(response.status)
|
||||||
|
print(response.undefined_flags.get_flags())
|
||||||
|
print(response.valid_flags.get_flags())
|
||||||
|
print(response.invalid_value_flags.get_flags())
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -11,7 +11,8 @@ from argenta.orchestrator.argparser.arguments import BooleanArgument
|
|||||||
|
|
||||||
arg_parser = ArgParser(processed_args=[BooleanArgument('repeat')])
|
arg_parser = ArgParser(processed_args=[BooleanArgument('repeat')])
|
||||||
app: App = App(dividing_line=DynamicDividingLine(),
|
app: App = App(dividing_line=DynamicDividingLine(),
|
||||||
autocompleter=AutoCompleter('./mock/.hist'))
|
autocompleter=AutoCompleter('./mock/.hist'),
|
||||||
|
repeat_command_groups=False,)
|
||||||
orchestrator: Orchestrator = Orchestrator(arg_parser)
|
orchestrator: Orchestrator = Orchestrator(arg_parser)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
+3
-3
@@ -1,7 +1,7 @@
|
|||||||
[project]
|
[project]
|
||||||
name = "argenta"
|
name = "argenta"
|
||||||
version = "1.0.0-alpha4"
|
version = "1.0.0-beta2"
|
||||||
description = "Python library for creating TUI"
|
description = "Python library for building modular CLI applications"
|
||||||
authors = [{ name = "kolo", email = "kolo.is.main@gmail.com" }]
|
authors = [{ name = "kolo", email = "kolo.is.main@gmail.com" }]
|
||||||
requires-python = ">=3.11, <4.0"
|
requires-python = ">=3.11, <4.0"
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
@@ -9,7 +9,7 @@ license = { text = "MIT" }
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"rich (>=14.0.0,<15.0.0)",
|
"rich (>=14.0.0,<15.0.0)",
|
||||||
"art (>=6.4,<7.0)",
|
"art (>=6.4,<7.0)",
|
||||||
"pyreadline3 (>=3.5.4,<4.0.0)",
|
"pyreadline3>=3.5.4",
|
||||||
]
|
]
|
||||||
|
|
||||||
[dependency-groups]
|
[dependency-groups]
|
||||||
|
|||||||
+42
-21
@@ -16,6 +16,7 @@ from argenta.command.exceptions import (UnprocessedInputFlagException,
|
|||||||
EmptyInputCommandException,
|
EmptyInputCommandException,
|
||||||
BaseInputCommandException)
|
BaseInputCommandException)
|
||||||
from argenta.app.registered_routers.entity import RegisteredRouters
|
from argenta.app.registered_routers.entity import RegisteredRouters
|
||||||
|
from argenta.response import Response
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@@ -58,7 +59,7 @@ class BaseApp:
|
|||||||
self._repeated_input_flags_handler: Callable[[str], None] = lambda raw_command: print_func(f'Repeated input flags: {raw_command}')
|
self._repeated_input_flags_handler: Callable[[str], None] = lambda raw_command: print_func(f'Repeated input flags: {raw_command}')
|
||||||
self._empty_input_command_handler: Callable[[], None] = lambda: print_func('Empty input command')
|
self._empty_input_command_handler: Callable[[], None] = lambda: print_func('Empty input command')
|
||||||
self._unknown_command_handler: Callable[[InputCommand], None] = lambda command: print_func(f"Unknown command: {command.get_trigger()}")
|
self._unknown_command_handler: Callable[[InputCommand], None] = lambda command: print_func(f"Unknown command: {command.get_trigger()}")
|
||||||
self._exit_command_handler: Callable[[], None] = lambda: print_func(self._farewell_message)
|
self._exit_command_handler: Callable[[Response], None] = lambda response: print_func(self._farewell_message)
|
||||||
|
|
||||||
|
|
||||||
def set_description_message_pattern(self, _: Callable[[str, str], str]) -> None:
|
def set_description_message_pattern(self, _: Callable[[str, str], str]) -> None:
|
||||||
@@ -138,7 +139,7 @@ class BaseApp:
|
|||||||
"""
|
"""
|
||||||
if isinstance(self._dividing_line, StaticDividingLine):
|
if isinstance(self._dividing_line, StaticDividingLine):
|
||||||
self._print_func(self._dividing_line.get_full_static_line(self._override_system_messages))
|
self._print_func(self._dividing_line.get_full_static_line(self._override_system_messages))
|
||||||
self._print_func(text)
|
print(text.strip('\n'))
|
||||||
self._print_func(self._dividing_line.get_full_static_line(self._override_system_messages))
|
self._print_func(self._dividing_line.get_full_static_line(self._override_system_messages))
|
||||||
|
|
||||||
elif isinstance(self._dividing_line, DynamicDividingLine):
|
elif isinstance(self._dividing_line, DynamicDividingLine):
|
||||||
@@ -210,30 +211,51 @@ class BaseApp:
|
|||||||
system_router.set_title(self._system_router_title)
|
system_router.set_title(self._system_router_title)
|
||||||
|
|
||||||
@system_router.command(self._exit_command)
|
@system_router.command(self._exit_command)
|
||||||
def exit_command():
|
def exit_command(response: Response) -> None:
|
||||||
self._exit_command_handler()
|
self._exit_command_handler(response)
|
||||||
|
|
||||||
if system_router not in self._registered_routers.get_registered_routers():
|
if system_router not in self._registered_routers.get_registered_routers():
|
||||||
system_router.set_command_register_ignore(self._ignore_command_register)
|
system_router.set_command_register_ignore(self._ignore_command_register)
|
||||||
self._registered_routers.add_registered_router(system_router)
|
self._registered_routers.add_registered_router(system_router)
|
||||||
|
|
||||||
|
|
||||||
|
def _most_similar_command(self, unknown_command: str) -> str | None:
|
||||||
|
all_commands = self._all_registered_triggers_in_lower if self._ignore_command_register else self._all_registered_triggers_in_default_case
|
||||||
|
matches: list[str] | list = sorted(cmd for cmd in all_commands if cmd.startswith(unknown_command))
|
||||||
|
if not matches:
|
||||||
|
matches: list[str] | list = sorted(cmd for cmd in all_commands if unknown_command.startswith(cmd))
|
||||||
|
if len(matches) == 1:
|
||||||
|
return matches[0]
|
||||||
|
elif len(matches) > 1:
|
||||||
|
return sorted(matches, key=lambda cmd: len(cmd))[0]
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
def _setup_default_view(self) -> None:
|
def _setup_default_view(self) -> None:
|
||||||
"""
|
"""
|
||||||
Private. Sets up default app view
|
Private. Sets up default app view
|
||||||
:return: None
|
:return: None
|
||||||
"""
|
"""
|
||||||
if not self._override_system_messages:
|
self._prompt = '[italic dim bold]What do you want to do?\n'
|
||||||
self._initial_message = f'\n[bold red]{text2art(self._initial_message, font="tarty1")}\n\n'
|
self._initial_message = f'\n[bold red]{text2art(self._initial_message, font="tarty1")}\n'
|
||||||
self._farewell_message = (f'[bold red]\n{text2art(f"\n{self._farewell_message}\n", font="chanky")}[/bold red]\n'
|
self._farewell_message = (f'[bold red]\n{text2art(f"\n{self._farewell_message}\n", font="chanky")}[/bold red]\n'
|
||||||
f'[red i]github.com/koloideal/Argenta[/red i] | [red bold i]made by kolo[/red bold i]\n')
|
f'[red i]github.com/koloideal/Argenta[/red i] | [red bold i]made by kolo[/red bold i]\n')
|
||||||
self._description_message_gen = lambda command, description: (f'[bold red]{escape("[" + command + "]")}[/bold red] '
|
self._description_message_gen = lambda command, description: (f'[bold red]{escape("[" + command + "]")}[/bold red] '
|
||||||
f'[blue dim]*=*=*[/blue dim] '
|
f'[blue dim]*=*=*[/blue dim] '
|
||||||
f'[bold yellow italic]{escape(description)}')
|
f'[bold yellow italic]{escape(description)}')
|
||||||
self._invalid_input_flags_handler = lambda raw_command: self._print_func(f'[red bold]Incorrect flag syntax: {escape(raw_command)}')
|
self._invalid_input_flags_handler = lambda raw_command: self._print_func(f'[red bold]Incorrect flag syntax: {escape(raw_command)}')
|
||||||
self._repeated_input_flags_handler = lambda raw_command: self._print_func(f'[red bold]Repeated input flags: {escape(raw_command)}')
|
self._repeated_input_flags_handler = lambda raw_command: self._print_func(f'[red bold]Repeated input flags: {escape(raw_command)}')
|
||||||
self._empty_input_command_handler = lambda: self._print_func('[red bold]Empty input command')
|
self._empty_input_command_handler = lambda: self._print_func('[red bold]Empty input command')
|
||||||
self._unknown_command_handler = lambda command: self._print_func(f"[red bold]Unknown command: {escape(command.get_trigger())}")
|
|
||||||
|
def unknown_command_handler(command: InputCommand) -> None:
|
||||||
|
cmd_trg: str = command.get_trigger()
|
||||||
|
mst_sim_cmd: str | None = self._most_similar_command(cmd_trg)
|
||||||
|
first_part_of_text = f"[red]Unknown command:[/red] [blue]{escape(cmd_trg)}[/blue]"
|
||||||
|
second_part_of_text = ('[red], most similar:[/red] ' + ('[blue]' + mst_sim_cmd + '[/blue]')) if mst_sim_cmd else ''
|
||||||
|
self._print_func(first_part_of_text + second_part_of_text)
|
||||||
|
|
||||||
|
self._unknown_command_handler = unknown_command_handler
|
||||||
|
|
||||||
|
|
||||||
def _pre_cycle_setup(self) -> None:
|
def _pre_cycle_setup(self) -> None:
|
||||||
@@ -241,7 +263,6 @@ class BaseApp:
|
|||||||
Private. Configures various aspects of the application before the start of the cycle
|
Private. Configures various aspects of the application before the start of the cycle
|
||||||
:return: None
|
:return: None
|
||||||
"""
|
"""
|
||||||
self._setup_default_view()
|
|
||||||
self._setup_system_router()
|
self._setup_system_router()
|
||||||
|
|
||||||
for router_entity in self._registered_routers:
|
for router_entity in self._registered_routers:
|
||||||
@@ -253,12 +274,15 @@ class BaseApp:
|
|||||||
|
|
||||||
self._autocompleter.initial_setup(self._all_registered_triggers_in_lower)
|
self._autocompleter.initial_setup(self._all_registered_triggers_in_lower)
|
||||||
|
|
||||||
|
if not self._override_system_messages:
|
||||||
|
self._setup_default_view()
|
||||||
|
|
||||||
self._print_func(self._initial_message)
|
self._print_func(self._initial_message)
|
||||||
|
|
||||||
for message in self._messages_on_startup:
|
for message in self._messages_on_startup:
|
||||||
self._print_func(message)
|
self._print_func(message)
|
||||||
if self._messages_on_startup:
|
if self._messages_on_startup:
|
||||||
print('\n\n')
|
print('\n')
|
||||||
|
|
||||||
if not self._repeat_command_groups_description:
|
if not self._repeat_command_groups_description:
|
||||||
self._print_command_group_description()
|
self._print_command_group_description()
|
||||||
@@ -267,7 +291,7 @@ class BaseApp:
|
|||||||
|
|
||||||
class App(BaseApp):
|
class App(BaseApp):
|
||||||
def __init__(self,
|
def __init__(self,
|
||||||
prompt: str = '[italic dim bold]What do you want to do?\n',
|
prompt: str = 'What do you want to do?\n',
|
||||||
initial_message: str = '\nArgenta\n',
|
initial_message: str = '\nArgenta\n',
|
||||||
farewell_message: str = '\nSee you\n',
|
farewell_message: str = '\nSee you\n',
|
||||||
exit_command: Command = Command('Q', 'Exit command'),
|
exit_command: Command = Command('Q', 'Exit command'),
|
||||||
@@ -346,9 +370,6 @@ class App(BaseApp):
|
|||||||
res: str = f.getvalue()
|
res: str = f.getvalue()
|
||||||
self._print_framed_text(res)
|
self._print_framed_text(res)
|
||||||
|
|
||||||
if not self._repeat_command_groups_description:
|
|
||||||
self._print_func(self._prompt)
|
|
||||||
|
|
||||||
|
|
||||||
def include_router(self, router: Router) -> None:
|
def include_router(self, router: Router) -> None:
|
||||||
"""
|
"""
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
from argenta.command.flag.models import InputFlag, Flag
|
from argenta.command.flag.models import Flag, InputFlag
|
||||||
|
|
||||||
|
|
||||||
class BaseInputCommandException(Exception):
|
class BaseInputCommandException(Exception):
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
__all__ = ('InputFlags', 'InputFlag', 'Flag', 'Flags')
|
__all__ = ["Flag", "InputFlag"]
|
||||||
|
|
||||||
|
|
||||||
from argenta.command.flag.models import InputFlags, InputFlag, Flags, Flag
|
from argenta.command.flag.models import Flag, InputFlag
|
||||||
|
|||||||
@@ -1,8 +1,8 @@
|
|||||||
from typing import Literal, Pattern
|
from typing import Literal, Pattern
|
||||||
from abc import ABC, abstractmethod
|
|
||||||
|
|
||||||
|
|
||||||
class BaseFlag(ABC):
|
|
||||||
|
class BaseFlag:
|
||||||
def __init__(self, name: str,
|
def __init__(self, name: str,
|
||||||
prefix: Literal['-', '--', '---'] = '--') -> None:
|
prefix: Literal['-', '--', '---'] = '--') -> None:
|
||||||
"""
|
"""
|
||||||
@@ -36,37 +36,8 @@ class BaseFlag(ABC):
|
|||||||
"""
|
"""
|
||||||
return self._prefix
|
return self._prefix
|
||||||
|
|
||||||
|
def __eq__(self, other) -> bool:
|
||||||
|
return self.get_string_entity() == other.get_string_entity()
|
||||||
class InputFlag(BaseFlag):
|
|
||||||
def __init__(self, name: str,
|
|
||||||
prefix: Literal['-', '--', '---'] = '--',
|
|
||||||
value: str = None):
|
|
||||||
"""
|
|
||||||
Public. The entity of the flag of the entered command
|
|
||||||
:param name: the name of the input flag
|
|
||||||
:param prefix: the prefix of the input flag
|
|
||||||
:param value: the value of the input flag
|
|
||||||
:return: None
|
|
||||||
"""
|
|
||||||
super().__init__(name, prefix)
|
|
||||||
self._flag_value = value
|
|
||||||
|
|
||||||
def get_value(self) -> str | None:
|
|
||||||
"""
|
|
||||||
Public. Returns the value of the flag
|
|
||||||
:return: the value of the flag as str
|
|
||||||
"""
|
|
||||||
return self._flag_value
|
|
||||||
|
|
||||||
def set_value(self, value):
|
|
||||||
"""
|
|
||||||
Private. Sets the value of the flag
|
|
||||||
:param value: the fag value to set
|
|
||||||
:return: None
|
|
||||||
"""
|
|
||||||
self._flag_value = value
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class Flag(BaseFlag):
|
class Flag(BaseFlag):
|
||||||
@@ -113,106 +84,35 @@ class Flag(BaseFlag):
|
|||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
class InputFlag(BaseFlag):
|
||||||
class BaseFlags(ABC):
|
def __init__(self, name: str,
|
||||||
"""
|
prefix: Literal['-', '--', '---'] = '--',
|
||||||
Private. Base class for groups of flags
|
value: str = None):
|
||||||
"""
|
|
||||||
__slots__ = ('_flags',)
|
|
||||||
|
|
||||||
@abstractmethod
|
|
||||||
def get_flags(self):
|
|
||||||
"""
|
"""
|
||||||
Public. Returns a list of flags
|
Public. The entity of the flag of the entered command
|
||||||
:return: list of flags
|
:param name: the name of the input flag
|
||||||
"""
|
:param prefix: the prefix of the input flag
|
||||||
pass
|
:param value: the value of the input flag
|
||||||
|
|
||||||
@abstractmethod
|
|
||||||
def add_flag(self, flag: Flag | InputFlag):
|
|
||||||
"""
|
|
||||||
Public. Adds a flag to the list of flags
|
|
||||||
:param flag: flag to add
|
|
||||||
:return: None
|
:return: None
|
||||||
"""
|
"""
|
||||||
pass
|
super().__init__(name, prefix)
|
||||||
|
self._flag_value = value
|
||||||
|
|
||||||
@abstractmethod
|
def get_value(self) -> str | None:
|
||||||
def add_flags(self, flags: list[Flag] | list[InputFlag]):
|
|
||||||
"""
|
"""
|
||||||
Public. Adds a list of flags to the list of flags
|
Public. Returns the value of the flag
|
||||||
:param flags: list of flags to add
|
:return: the value of the flag as str
|
||||||
|
"""
|
||||||
|
return self._flag_value
|
||||||
|
|
||||||
|
def set_value(self, value):
|
||||||
|
"""
|
||||||
|
Private. Sets the value of the flag
|
||||||
|
:param value: the fag value to set
|
||||||
:return: None
|
:return: None
|
||||||
"""
|
"""
|
||||||
pass
|
self._flag_value = value
|
||||||
|
|
||||||
@abstractmethod
|
def __eq__(self, other) -> bool:
|
||||||
def get_flag(self, name: str):
|
return self.get_string_entity() == other.get_string_entity() and self.get_value() == other.get_value()
|
||||||
"""
|
|
||||||
Public. Returns the flag entity by its name or None if not found
|
|
||||||
:param name: the name of the flag to get
|
|
||||||
:return: entity of the flag or None
|
|
||||||
"""
|
|
||||||
pass
|
|
||||||
|
|
||||||
def __iter__(self):
|
|
||||||
return iter(self._flags)
|
|
||||||
|
|
||||||
def __next__(self):
|
|
||||||
return next(iter(self))
|
|
||||||
|
|
||||||
def __getitem__(self, item):
|
|
||||||
return self._flags[item]
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class Flags(BaseFlags, ABC):
|
|
||||||
def __init__(self, *flags: Flag):
|
|
||||||
"""
|
|
||||||
Public. A model that combines the registered flags
|
|
||||||
:param flags: the flags that will be registered
|
|
||||||
:return: None
|
|
||||||
"""
|
|
||||||
self._flags = flags if flags else []
|
|
||||||
|
|
||||||
def get_flags(self) -> list[Flag]:
|
|
||||||
return self._flags
|
|
||||||
|
|
||||||
def add_flag(self, flag: Flag):
|
|
||||||
self._flags.append(flag)
|
|
||||||
|
|
||||||
def add_flags(self, flags: list[Flag]):
|
|
||||||
self._flags.extend(flags)
|
|
||||||
|
|
||||||
def get_flag(self, name: str) -> Flag | None:
|
|
||||||
if name in [flag.get_name() for flag in self._flags]:
|
|
||||||
return list(filter(lambda flag: flag.get_name() == name, self._flags))[0]
|
|
||||||
else:
|
|
||||||
return None
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class InputFlags(BaseFlags, ABC):
|
|
||||||
def __init__(self, *flags: InputFlag):
|
|
||||||
"""
|
|
||||||
Public. A model that combines the input flags of the input command
|
|
||||||
:param flags: all input flags
|
|
||||||
:return: None
|
|
||||||
"""
|
|
||||||
self._flags = flags if flags else []
|
|
||||||
|
|
||||||
def get_flags(self) -> list[InputFlag]:
|
|
||||||
return self._flags
|
|
||||||
|
|
||||||
def add_flag(self, flag: InputFlag):
|
|
||||||
self._flags.append(flag)
|
|
||||||
|
|
||||||
def add_flags(self, flags: list[InputFlag]):
|
|
||||||
self._flags.extend(flags)
|
|
||||||
|
|
||||||
def get_flag(self, name: str) -> InputFlag | None:
|
|
||||||
if name in [flag.get_name() for flag in self._flags]:
|
|
||||||
return list(filter(lambda flag: flag.get_name() == name, self._flags))[0]
|
|
||||||
else:
|
|
||||||
return None
|
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,10 @@
|
|||||||
|
__all__ = ["Flags", "InputFlags",
|
||||||
|
"UndefinedInputFlags",
|
||||||
|
"InvalidValueInputFlags",
|
||||||
|
"ValidInputFlags"]
|
||||||
|
|
||||||
|
|
||||||
|
from argenta.command.flags.models import (Flags, InputFlags,
|
||||||
|
UndefinedInputFlags,
|
||||||
|
InvalidValueInputFlags,
|
||||||
|
ValidInputFlags)
|
||||||
@@ -0,0 +1,87 @@
|
|||||||
|
from argenta.command.flag.models import InputFlag, Flag
|
||||||
|
from typing import Generic, TypeVar
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
FlagType = TypeVar('FlagType')
|
||||||
|
|
||||||
|
|
||||||
|
class BaseFlags(Generic[FlagType]):
|
||||||
|
def __init__(self, *flags: FlagType):
|
||||||
|
"""
|
||||||
|
Public. A model that combines the registered flags
|
||||||
|
:param flags: the flags that will be registered
|
||||||
|
:return: None
|
||||||
|
"""
|
||||||
|
self._flags = flags if flags else []
|
||||||
|
|
||||||
|
def get_flags(self) -> list[FlagType]:
|
||||||
|
"""
|
||||||
|
Public. Returns a list of flags
|
||||||
|
:return: list of flags as list[FlagType]
|
||||||
|
"""
|
||||||
|
return self._flags
|
||||||
|
|
||||||
|
def add_flag(self, flag: FlagType):
|
||||||
|
"""
|
||||||
|
Public. Adds a flag to the list of flags
|
||||||
|
:param flag: flag to add
|
||||||
|
:return: None
|
||||||
|
"""
|
||||||
|
self._flags.append(flag)
|
||||||
|
|
||||||
|
def add_flags(self, flags: list[FlagType]):
|
||||||
|
"""
|
||||||
|
Public. Adds a list of flags to the list of flags
|
||||||
|
:param flags: list of flags to add
|
||||||
|
:return: None
|
||||||
|
"""
|
||||||
|
self._flags.extend(flags)
|
||||||
|
|
||||||
|
def get_flag(self, name: str) -> FlagType | None:
|
||||||
|
"""
|
||||||
|
Public. Returns the flag entity by its name or None if not found
|
||||||
|
:param name: the name of the flag to get
|
||||||
|
:return: entity of the flag or None
|
||||||
|
"""
|
||||||
|
if name in [flag.get_name() for flag in self._flags]:
|
||||||
|
return list(filter(lambda flag: flag.get_name() == name, self._flags))[0]
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
|
def __iter__(self):
|
||||||
|
return iter(self._flags)
|
||||||
|
|
||||||
|
def __next__(self):
|
||||||
|
return next(iter(self))
|
||||||
|
|
||||||
|
def __getitem__(self, item):
|
||||||
|
return self._flags[item]
|
||||||
|
|
||||||
|
def __bool__(self):
|
||||||
|
return bool(self._flags)
|
||||||
|
|
||||||
|
def __eq__(self, other):
|
||||||
|
if len(self.get_flags()) != len(other.get_flags()):
|
||||||
|
return False
|
||||||
|
else:
|
||||||
|
for flag, other_flag in zip(self.get_flags(), other.get_flags()):
|
||||||
|
if not flag == other_flag:
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
class Flags(BaseFlags[Flag]): pass
|
||||||
|
|
||||||
|
|
||||||
|
class InputFlags(BaseFlags[InputFlag]): pass
|
||||||
|
|
||||||
|
|
||||||
|
class ValidInputFlags(InputFlags): pass
|
||||||
|
|
||||||
|
|
||||||
|
class UndefinedInputFlags(InputFlags): pass
|
||||||
|
|
||||||
|
|
||||||
|
class InvalidValueInputFlags(InputFlags): pass
|
||||||
|
|
||||||
@@ -1,4 +1,5 @@
|
|||||||
from argenta.command.flag.models import Flag, InputFlag, Flags, InputFlags
|
from argenta.command.flag.models import Flag, InputFlag
|
||||||
|
from argenta.command.flags.models import InputFlags, Flags
|
||||||
from argenta.command.exceptions import (UnprocessedInputFlagException,
|
from argenta.command.exceptions import (UnprocessedInputFlagException,
|
||||||
RepeatedInputFlagsException,
|
RepeatedInputFlagsException,
|
||||||
EmptyInputCommandException)
|
EmptyInputCommandException)
|
||||||
@@ -38,7 +39,7 @@ class Command(BaseCommand):
|
|||||||
"""
|
"""
|
||||||
super().__init__(trigger)
|
super().__init__(trigger)
|
||||||
self._registered_flags: Flags = flags if isinstance(flags, Flags) else Flags(flags) if isinstance(flags, Flag) else Flags()
|
self._registered_flags: Flags = flags if isinstance(flags, Flags) else Flags(flags) if isinstance(flags, Flag) else Flags()
|
||||||
self._description = f'Description for "{self._trigger}" command' if not description else description
|
self._description = f'Very useful command' if not description else description
|
||||||
self._aliases = aliases if isinstance(aliases, list) else []
|
self._aliases = aliases if isinstance(aliases, list) else []
|
||||||
|
|
||||||
def get_registered_flags(self) -> Flags:
|
def get_registered_flags(self) -> Flags:
|
||||||
@@ -55,7 +56,7 @@ class Command(BaseCommand):
|
|||||||
"""
|
"""
|
||||||
return self._aliases
|
return self._aliases
|
||||||
|
|
||||||
def validate_input_flag(self, flag: InputFlag) -> bool:
|
def validate_input_flag(self, flag: InputFlag) -> Literal['Undefined', 'Valid', 'Invalid']:
|
||||||
"""
|
"""
|
||||||
Private. Validates the input flag
|
Private. Validates the input flag
|
||||||
:param flag: input flag for validation
|
:param flag: input flag for validation
|
||||||
@@ -67,14 +68,21 @@ class Command(BaseCommand):
|
|||||||
if registered_flags.get_string_entity() == flag.get_string_entity():
|
if registered_flags.get_string_entity() == flag.get_string_entity():
|
||||||
is_valid = registered_flags.validate_input_flag_value(flag.get_value())
|
is_valid = registered_flags.validate_input_flag_value(flag.get_value())
|
||||||
if is_valid:
|
if is_valid:
|
||||||
return True
|
return 'Valid'
|
||||||
|
else:
|
||||||
|
return 'Invalid'
|
||||||
|
else:
|
||||||
|
return 'Undefined'
|
||||||
else:
|
else:
|
||||||
for registered_flag in registered_flags:
|
for registered_flag in registered_flags:
|
||||||
if registered_flag.get_string_entity() == flag.get_string_entity():
|
if registered_flag.get_string_entity() == flag.get_string_entity():
|
||||||
is_valid = registered_flag.validate_input_flag_value(flag.get_value())
|
is_valid = registered_flag.validate_input_flag_value(flag.get_value())
|
||||||
if is_valid:
|
if is_valid:
|
||||||
return True
|
return 'Valid'
|
||||||
return False
|
else:
|
||||||
|
return 'Invalid'
|
||||||
|
return 'Undefined'
|
||||||
|
return 'Undefined'
|
||||||
|
|
||||||
def get_description(self) -> str:
|
def get_description(self) -> str:
|
||||||
"""
|
"""
|
||||||
@@ -131,11 +139,11 @@ class InputCommand(BaseCommand, Generic[InputCommandType]):
|
|||||||
|
|
||||||
for k, _ in enumerate(list_of_tokens):
|
for k, _ in enumerate(list_of_tokens):
|
||||||
if _.startswith('-'):
|
if _.startswith('-'):
|
||||||
if current_flag_name or len(_) < 2 or len(_[:_.rfind('-')]) > 3:
|
if len(_) < 2 or len(_[:_.rfind('-')]) > 3:
|
||||||
raise UnprocessedInputFlagException()
|
raise UnprocessedInputFlagException()
|
||||||
current_flag_name = _
|
current_flag_name = _
|
||||||
else:
|
else:
|
||||||
if not current_flag_name:
|
if not current_flag_name or current_flag_value:
|
||||||
raise UnprocessedInputFlagException()
|
raise UnprocessedInputFlagException()
|
||||||
current_flag_value = _
|
current_flag_value = _
|
||||||
|
|
||||||
@@ -144,9 +152,9 @@ class InputCommand(BaseCommand, Generic[InputCommandType]):
|
|||||||
if not list_of_tokens[k+1].startswith('-'):
|
if not list_of_tokens[k+1].startswith('-'):
|
||||||
continue
|
continue
|
||||||
|
|
||||||
input_flag = InputFlag(name=current_flag_name[current_flag_name.rfind('-')+1:],
|
input_flag = InputFlag(name=current_flag_name[current_flag_name.rfind('-') + 1:],
|
||||||
prefix=cast(Literal['-', '--', '---'],
|
prefix=cast(Literal['-', '--', '---'],
|
||||||
current_flag_name[:current_flag_name.rfind('-')+1]),
|
current_flag_name[:current_flag_name.rfind('-')+1]),
|
||||||
value=current_flag_value)
|
value=current_flag_value)
|
||||||
|
|
||||||
all_flags = [flag.get_string_entity() for flag in input_flags.get_flags()]
|
all_flags = [flag.get_string_entity() for flag in input_flags.get_flags()]
|
||||||
|
|||||||
@@ -0,0 +1,5 @@
|
|||||||
|
__all__ = ["Response", "Status"]
|
||||||
|
|
||||||
|
|
||||||
|
from argenta.response.entity import Response
|
||||||
|
from argenta.response.status import Status
|
||||||
@@ -0,0 +1,27 @@
|
|||||||
|
from argenta.response.status import Status
|
||||||
|
from argenta.command.flags import (ValidInputFlags,
|
||||||
|
UndefinedInputFlags,
|
||||||
|
InvalidValueInputFlags)
|
||||||
|
|
||||||
|
|
||||||
|
class Response:
|
||||||
|
__slots__ = ('status',
|
||||||
|
'valid_flags',
|
||||||
|
'undefined_flags',
|
||||||
|
'invalid_value_flags')
|
||||||
|
|
||||||
|
def __init__(self, status: Status = None,
|
||||||
|
valid_flags: ValidInputFlags = ValidInputFlags(),
|
||||||
|
undefined_flags: UndefinedInputFlags = UndefinedInputFlags(),
|
||||||
|
invalid_value_flags: InvalidValueInputFlags = InvalidValueInputFlags()):
|
||||||
|
"""
|
||||||
|
Public. The entity of the user input sent to the handler
|
||||||
|
:param status: the status of the response
|
||||||
|
:param valid_flags: valid input flags
|
||||||
|
:param undefined_flags: undefined input flags
|
||||||
|
:param invalid_value_flags: input flags with invalid values
|
||||||
|
"""
|
||||||
|
self.status = status
|
||||||
|
self.valid_flags = valid_flags
|
||||||
|
self.undefined_flags = undefined_flags
|
||||||
|
self.invalid_value_flags = invalid_value_flags
|
||||||
@@ -0,0 +1,9 @@
|
|||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
|
||||||
|
class Status(Enum):
|
||||||
|
ALL_FLAGS_VALID = 'ALL_FLAGS_VALID'
|
||||||
|
UNDEFINED_FLAGS = 'UNDEFINED_FLAGS'
|
||||||
|
INVALID_VALUE_FLAGS = 'INVALID_VALUE_FLAGS'
|
||||||
|
UNDEFINED_AND_INVALID_FLAGS = 'UNDEFINED_AND_INVALID_FLAGS'
|
||||||
|
|
||||||
@@ -1,12 +1,11 @@
|
|||||||
from typing import Callable, Iterator
|
from typing import Callable, Iterator
|
||||||
|
|
||||||
from argenta.command import Command
|
from argenta.command import Command
|
||||||
from argenta.command.flag import InputFlags
|
from argenta.response import Response
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class CommandHandler:
|
class CommandHandler:
|
||||||
def __init__(self, handler: Callable[[], None] | Callable[[InputFlags], None], handled_command: Command):
|
def __init__(self, handler: Callable[[Response], None], handled_command: Command):
|
||||||
"""
|
"""
|
||||||
Private. Entity of the model linking the handler and the command being processed
|
Private. Entity of the model linking the handler and the command being processed
|
||||||
:param handler: the handler being called
|
:param handler: the handler being called
|
||||||
@@ -15,21 +14,18 @@ class CommandHandler:
|
|||||||
self._handler = handler
|
self._handler = handler
|
||||||
self._handled_command = handled_command
|
self._handled_command = handled_command
|
||||||
|
|
||||||
def handling(self, input_flags: InputFlags = None) -> None:
|
def handling(self, response: Response) -> None:
|
||||||
"""
|
"""
|
||||||
Private. Direct processing of an input command
|
Private. Direct processing of an input command
|
||||||
:param input_flags: the flags of the input command
|
:param response: the entity of response: various groups of flags and status of response
|
||||||
:return: None
|
:return: None
|
||||||
"""
|
"""
|
||||||
if input_flags is not None:
|
self._handler(response)
|
||||||
self._handler(input_flags)
|
|
||||||
else:
|
|
||||||
self._handler()
|
|
||||||
|
|
||||||
def get_handler(self) -> Callable[[], None] | Callable[[InputFlags], None]:
|
def get_handler(self) -> Callable[[Response], None]:
|
||||||
"""
|
"""
|
||||||
Private. Returns the handler being called
|
Private. Returns the handler being called
|
||||||
:return: the handler being called as Callable[[], None] or Callable[[InputFlags], None]
|
:return: the handler being called as Callable[[Response], None]
|
||||||
"""
|
"""
|
||||||
return self._handler
|
return self._handler
|
||||||
|
|
||||||
|
|||||||
@@ -1,9 +1,15 @@
|
|||||||
from typing import Callable
|
from typing import Callable, Literal, Type
|
||||||
from inspect import getfullargspec
|
from inspect import getfullargspec, get_annotations, getsourcefile, getsourcelines
|
||||||
|
from rich.console import Console
|
||||||
|
|
||||||
from argenta.command import Command
|
from argenta.command import Command
|
||||||
from argenta.command.models import InputCommand
|
from argenta.command.models import InputCommand
|
||||||
|
from argenta.response import Response, Status
|
||||||
from argenta.router.command_handler.entity import CommandHandlers, CommandHandler
|
from argenta.router.command_handler.entity import CommandHandlers, CommandHandler
|
||||||
from argenta.command.flag.models import Flag, Flags, InputFlags
|
from argenta.command.flags.models import (Flags, InputFlags,
|
||||||
|
UndefinedInputFlags,
|
||||||
|
ValidInputFlags,
|
||||||
|
InvalidValueInputFlags)
|
||||||
from argenta.router.exceptions import (RepeatedFlagNameException,
|
from argenta.router.exceptions import (RepeatedFlagNameException,
|
||||||
TooManyTransferredArgsException,
|
TooManyTransferredArgsException,
|
||||||
RequiredArgumentNotPassedException,
|
RequiredArgumentNotPassedException,
|
||||||
@@ -11,8 +17,7 @@ from argenta.router.exceptions import (RepeatedFlagNameException,
|
|||||||
|
|
||||||
|
|
||||||
class Router:
|
class Router:
|
||||||
def __init__(self,
|
def __init__(self, title: str = None):
|
||||||
title: str = None):
|
|
||||||
"""
|
"""
|
||||||
Public. Directly configures and manages handlers
|
Public. Directly configures and manages handlers
|
||||||
:param title: the title of the router, displayed when displaying the available commands
|
:param title: the title of the router, displayed when displaying the available commands
|
||||||
@@ -22,19 +27,20 @@ class Router:
|
|||||||
|
|
||||||
self._command_handlers: CommandHandlers = CommandHandlers()
|
self._command_handlers: CommandHandlers = CommandHandlers()
|
||||||
self._ignore_command_register: bool = False
|
self._ignore_command_register: bool = False
|
||||||
self._not_valid_flag_handler: Callable[[Flag], None] = lambda flag: print(f"Undefined or incorrect input flag: {flag.get_string_entity()}{(' '+flag.get_value()) if flag.get_value() else ''}")
|
|
||||||
|
|
||||||
|
|
||||||
def command(self, command: Command) -> Callable:
|
def command(self, command: Command | str) -> Callable:
|
||||||
"""
|
"""
|
||||||
Public. Registers handler
|
Public. Registers handler
|
||||||
:param command: Registered command
|
:param command: Registered command
|
||||||
:return: decorated handler as Callable[[Any], Any]
|
:return: decorated handler as Callable
|
||||||
"""
|
"""
|
||||||
self._validate_command(command)
|
self._validate_command(command)
|
||||||
|
if isinstance(command, str):
|
||||||
|
command = Command(command)
|
||||||
|
|
||||||
def command_decorator(func):
|
def command_decorator(func):
|
||||||
Router._validate_func_args(command, func)
|
Router._validate_func_args(func)
|
||||||
self._command_handlers.add_handler(CommandHandler(func, command))
|
self._command_handlers.add_handler(CommandHandler(func, command))
|
||||||
|
|
||||||
def wrapper(*args, **kwargs):
|
def wrapper(*args, **kwargs):
|
||||||
@@ -44,15 +50,6 @@ class Router:
|
|||||||
return command_decorator
|
return command_decorator
|
||||||
|
|
||||||
|
|
||||||
def set_invalid_input_flag_handler(self, func: Callable[[Flag], None]) -> None:
|
|
||||||
"""
|
|
||||||
Public. Registers handler for invalid input flag
|
|
||||||
:param func: registered handler
|
|
||||||
:return: None
|
|
||||||
"""
|
|
||||||
self._not_valid_flag_handler = func
|
|
||||||
|
|
||||||
|
|
||||||
def finds_appropriate_handler(self, input_command: InputCommand) -> None:
|
def finds_appropriate_handler(self, input_command: InputCommand) -> None:
|
||||||
"""
|
"""
|
||||||
Private. Finds the appropriate handler for given input command and passes control to it
|
Private. Finds the appropriate handler for given input command and passes control to it
|
||||||
@@ -78,73 +75,110 @@ class Router:
|
|||||||
:return: None
|
:return: None
|
||||||
"""
|
"""
|
||||||
handle_command = command_handler.get_handled_command()
|
handle_command = command_handler.get_handled_command()
|
||||||
|
response: Response = Response()
|
||||||
if handle_command.get_registered_flags().get_flags():
|
if handle_command.get_registered_flags().get_flags():
|
||||||
if input_command_flags.get_flags():
|
if input_command_flags.get_flags():
|
||||||
if self._validate_input_flags(handle_command, input_command_flags):
|
response: Response = self._structuring_input_flags(handle_command, input_command_flags)
|
||||||
command_handler.handling(input_command_flags)
|
command_handler.handling(response)
|
||||||
return
|
|
||||||
else:
|
else:
|
||||||
command_handler.handling(input_command_flags)
|
response.status = Status.ALL_FLAGS_VALID
|
||||||
return
|
command_handler.handling(response)
|
||||||
else:
|
else:
|
||||||
if input_command_flags.get_flags():
|
if input_command_flags.get_flags():
|
||||||
self._not_valid_flag_handler(input_command_flags[0])
|
response.status = Status.UNDEFINED_FLAGS
|
||||||
return
|
response.undefined_flags = UndefinedInputFlags()
|
||||||
|
response.undefined_flags.add_flags(input_command_flags.get_flags())
|
||||||
|
command_handler.handling(response)
|
||||||
else:
|
else:
|
||||||
command_handler.handling()
|
response.status = Status.ALL_FLAGS_VALID
|
||||||
return
|
command_handler.handling(response)
|
||||||
|
|
||||||
|
|
||||||
def _validate_input_flags(self, handled_command: Command, input_flags: InputFlags) -> bool:
|
@staticmethod
|
||||||
|
def _structuring_input_flags(handled_command: Command, input_flags: InputFlags) -> Response:
|
||||||
"""
|
"""
|
||||||
Private. Validates flags of input command
|
Private. Validates flags of input command
|
||||||
:param handled_command: entity of the handled command
|
:param handled_command: entity of the handled command
|
||||||
:param input_flags:
|
:param input_flags:
|
||||||
:return: is flags of input command valid as bool
|
:return: entity of response as Response
|
||||||
"""
|
"""
|
||||||
|
valid_input_flags: ValidInputFlags = ValidInputFlags()
|
||||||
|
invalid_value_input_flags: InvalidValueInputFlags = InvalidValueInputFlags()
|
||||||
|
undefined_input_flags: UndefinedInputFlags = UndefinedInputFlags()
|
||||||
for flag in input_flags:
|
for flag in input_flags:
|
||||||
is_valid: bool = handled_command.validate_input_flag(flag)
|
flag_status: Literal['Undefined', 'Valid', 'Invalid'] = handled_command.validate_input_flag(flag)
|
||||||
if not is_valid:
|
match flag_status:
|
||||||
self._not_valid_flag_handler(flag)
|
case 'Valid':
|
||||||
return False
|
valid_input_flags.add_flag(flag)
|
||||||
return True
|
case 'Undefined':
|
||||||
|
undefined_input_flags.add_flag(flag)
|
||||||
|
case 'Invalid':
|
||||||
|
invalid_value_input_flags.add_flag(flag)
|
||||||
|
|
||||||
|
if not invalid_value_input_flags.get_flags() and not undefined_input_flags.get_flags():
|
||||||
|
status = Status.ALL_FLAGS_VALID
|
||||||
|
elif invalid_value_input_flags.get_flags() and not undefined_input_flags.get_flags():
|
||||||
|
status = Status.INVALID_VALUE_FLAGS
|
||||||
|
elif not invalid_value_input_flags.get_flags() and undefined_input_flags.get_flags():
|
||||||
|
status = Status.UNDEFINED_FLAGS
|
||||||
|
else:
|
||||||
|
status = Status.UNDEFINED_AND_INVALID_FLAGS
|
||||||
|
|
||||||
|
return Response(invalid_value_flags=invalid_value_input_flags,
|
||||||
|
valid_flags=valid_input_flags,
|
||||||
|
status=status,
|
||||||
|
undefined_flags=undefined_input_flags)
|
||||||
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _validate_command(command: Command) -> None:
|
def _validate_command(command: Command | str) -> None:
|
||||||
"""
|
"""
|
||||||
Private. Validates the command registered in handler
|
Private. Validates the command registered in handler
|
||||||
:param command: validated command
|
:param command: validated command
|
||||||
:return: None if command is valid else raise exception
|
:return: None if command is valid else raise exception
|
||||||
"""
|
"""
|
||||||
command_name: str = command.get_trigger()
|
match type(command).__name__:
|
||||||
if command_name.find(' ') != -1:
|
case 'Command':
|
||||||
raise TriggerContainSpacesException()
|
command_name: str = command.get_trigger()
|
||||||
|
if command_name.find(' ') != -1:
|
||||||
flags: Flags = command.get_registered_flags()
|
raise TriggerContainSpacesException()
|
||||||
if flags:
|
flags: Flags = command.get_registered_flags()
|
||||||
flags_name: list = [x.get_string_entity().lower() for x in flags]
|
if flags:
|
||||||
if len(set(flags_name)) < len(flags_name):
|
flags_name: list = [x.get_string_entity().lower() for x in flags]
|
||||||
raise RepeatedFlagNameException()
|
if len(set(flags_name)) < len(flags_name):
|
||||||
|
raise RepeatedFlagNameException()
|
||||||
|
case 'str':
|
||||||
|
if command.find(' ') != -1:
|
||||||
|
raise TriggerContainSpacesException()
|
||||||
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _validate_func_args(command: Command, func: Callable) -> None:
|
def _validate_func_args(func: Callable) -> None:
|
||||||
"""
|
"""
|
||||||
Private. Validates the arguments of the handler
|
Private. Validates the arguments of the handler
|
||||||
:param command: registered command in handler
|
|
||||||
:param func: entity of the handler func
|
:param func: entity of the handler func
|
||||||
:return: None if func is valid else raise exception
|
:return: None if func is valid else raise exception
|
||||||
"""
|
"""
|
||||||
registered_args = command.get_registered_flags()
|
|
||||||
transferred_args = getfullargspec(func).args
|
transferred_args = getfullargspec(func).args
|
||||||
if registered_args.get_flags() and transferred_args:
|
if len(transferred_args) > 1:
|
||||||
if len(transferred_args) != 1:
|
|
||||||
raise TooManyTransferredArgsException()
|
|
||||||
elif registered_args.get_flags() and not transferred_args:
|
|
||||||
raise RequiredArgumentNotPassedException()
|
|
||||||
elif not registered_args.get_flags() and transferred_args:
|
|
||||||
raise TooManyTransferredArgsException()
|
raise TooManyTransferredArgsException()
|
||||||
|
elif len(transferred_args) == 0:
|
||||||
|
raise RequiredArgumentNotPassedException()
|
||||||
|
|
||||||
|
transferred_arg: str = transferred_args[0]
|
||||||
|
func_annotations: dict[str, Type] = get_annotations(func)
|
||||||
|
|
||||||
|
if arg_annotation := func_annotations.get(transferred_arg):
|
||||||
|
if arg_annotation is Response:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
file_path: str = getsourcefile(func)
|
||||||
|
source_line: int = getsourcelines(func)[1]+1
|
||||||
|
fprint = Console().print
|
||||||
|
fprint(f'\nFile "{file_path}", line {source_line}\n[b red]WARNING:[/b red] [i]The typehint '
|
||||||
|
f'of argument([green]{transferred_arg}[/green]) passed to the handler is [/i][bold blue]{Response}[/bold blue],'
|
||||||
|
f' [i]but[/i] [bold blue]{arg_annotation}[/bold blue] [i]is specified[/i]\n', highlight=False)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def set_command_register_ignore(self, _: bool) -> None:
|
def set_command_register_ignore(self, _: bool) -> None:
|
||||||
|
|||||||
@@ -1,32 +1,35 @@
|
|||||||
import _io
|
import _io
|
||||||
from unittest.mock import patch, MagicMock
|
from unittest.mock import patch, MagicMock
|
||||||
import unittest
|
from unittest import TestCase
|
||||||
import io
|
import io
|
||||||
import re
|
import re
|
||||||
|
|
||||||
from argenta.app import App
|
from argenta.app import App
|
||||||
from argenta.command import Command
|
from argenta.command import Command
|
||||||
from argenta.router import Router
|
from argenta.router import Router
|
||||||
from argenta.command.flag.models import Flags, InputFlags
|
from argenta.command.flags.models import Flags
|
||||||
from argenta.command.flag.defaults import PredefinedFlags
|
from argenta.command.flag.defaults import PredefinedFlags
|
||||||
|
from argenta.orchestrator import Orchestrator
|
||||||
|
from argenta.response import Response
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class TestSystemHandlerNormalWork(unittest.TestCase):
|
class TestSystemHandlerNormalWork(TestCase):
|
||||||
@patch("builtins.input", side_effect=["help", "q"])
|
@patch("builtins.input", side_effect=["help", "q"])
|
||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_incorrect_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_incorrect_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response: Response):
|
||||||
print('test command')
|
print('test command')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.set_unknown_command_handler(lambda command: print(f'Unknown command: {command.get_trigger()}'))
|
app.set_unknown_command_handler(lambda command: print(f'Unknown command: {command.get_trigger()}'))
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -37,9 +40,10 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_incorrect_command2(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_incorrect_command2(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response: Response):
|
||||||
print('test command')
|
print('test command')
|
||||||
|
|
||||||
app = App(ignore_command_register=False,
|
app = App(ignore_command_register=False,
|
||||||
@@ -47,7 +51,7 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.set_unknown_command_handler(lambda command: print(f'Unknown command: {command.get_trigger()}'))
|
app.set_unknown_command_handler(lambda command: print(f'Unknown command: {command.get_trigger()}'))
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -58,74 +62,80 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command_with_unregistered_flag(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command_with_unregistered_flag(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response: Response):
|
||||||
print(f'test command')
|
print(f'test command with undefined flag: {response.undefined_flags.get_flag('help').get_string_entity()}')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
self.assertIn('\nUndefined or incorrect input flag: --help\n', output)
|
self.assertIn('\ntest command with undefined flag: --help\n', output)
|
||||||
|
|
||||||
|
|
||||||
@patch("builtins.input", side_effect=["test --port 22", "q"])
|
@patch("builtins.input", side_effect=["test --port 22", "q"])
|
||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command_with_unregistered_flag2(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command_with_unregistered_flag2(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response: Response):
|
||||||
print('test command')
|
flag = response.undefined_flags.get_flag("port")
|
||||||
|
print(f'test command with undefined flag with value: {flag.get_string_entity()} {flag.get_value()}')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
self.assertIn('\nUndefined or incorrect input flag: --port 22\n', output)
|
self.assertIn('\ntest command with undefined flag with value: --port 22\n', output)
|
||||||
|
|
||||||
|
|
||||||
@patch("builtins.input", side_effect=["test --host 192.168.32.1 --port 132", "q"])
|
@patch("builtins.input", side_effect=["test --host 192.168.32.1 --port 132", "q"])
|
||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command_with_one_correct_flag_an_one_incorrect_flag(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command_with_one_correct_flag_an_one_incorrect_flag(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
flags = Flags(PredefinedFlags.HOST)
|
flags = Flags(PredefinedFlags.HOST)
|
||||||
|
|
||||||
@router.command(Command('test', flags=flags))
|
@router.command(Command('test', flags=flags))
|
||||||
def test(args: InputFlags):
|
def test(response: Response):
|
||||||
print(f'connecting to host {args.get_flag('host').get_value()}')
|
flag = response.undefined_flags.get_flag("port")
|
||||||
|
print(f'connecting to host with flag: {flag.get_string_entity()} {flag.get_value()}')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
self.assertIn('\nUndefined or incorrect input flag: --port 132\n', output)
|
self.assertIn('\nconnecting to host with flag: --port 132\n', output)
|
||||||
|
|
||||||
|
|
||||||
@patch("builtins.input", side_effect=["test", "some", "q"])
|
@patch("builtins.input", side_effect=["test", "some", "q"])
|
||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_one_correct_command_and_one_incorrect_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_one_correct_command_and_one_incorrect_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response: Response):
|
||||||
print(f'test command')
|
print(f'test command')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.set_unknown_command_handler(lambda command: print(f'Unknown command: {command.get_trigger()}'))
|
app.set_unknown_command_handler(lambda command: print(f'Unknown command: {command.get_trigger()}'))
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -136,20 +146,21 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_two_correct_commands_and_one_incorrect_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_two_correct_commands_and_one_incorrect_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response: Response):
|
||||||
print(f'test command')
|
print(f'test command')
|
||||||
|
|
||||||
@router.command(Command('more'))
|
@router.command(Command('more'))
|
||||||
def test():
|
def test(response: Response):
|
||||||
print(f'more command')
|
print(f'more command')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.set_unknown_command_handler(lambda command: print(f'Unknown command: {command.get_trigger()}'))
|
app.set_unknown_command_handler(lambda command: print(f'Unknown command: {command.get_trigger()}'))
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -160,16 +171,17 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command_with_incorrect_flag(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command_with_incorrect_flag(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response: Response):
|
||||||
print(f'test command')
|
print(f'test command')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.set_invalid_input_flags_handler(lambda command: print(f'Incorrect flag syntax: "{command}"'))
|
app.set_invalid_input_flags_handler(lambda command: print(f'Incorrect flag syntax: "{command}"'))
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -180,16 +192,17 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_empty_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_empty_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response: Response):
|
||||||
print(f'test command')
|
print(f'test command')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.set_empty_command_handler(lambda: print('Empty input command'))
|
app.set_empty_command_handler(lambda: print('Empty input command'))
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -200,16 +213,17 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command_with_repeated_flags(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command_with_repeated_flags(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test', flags=PredefinedFlags.PORT))
|
@router.command(Command('test', flags=PredefinedFlags.PORT))
|
||||||
def test(args: InputFlags):
|
def test(response: Response):
|
||||||
print('test command')
|
print('test command')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.set_repeated_input_flags_handler(lambda command: print(f'Repeated input flags: "{command}"'))
|
app.set_repeated_input_flags_handler(lambda command: print(f'Repeated input flags: "{command}"'))
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
|
|||||||
@@ -1,31 +1,35 @@
|
|||||||
import _io
|
import _io
|
||||||
from unittest.mock import patch, MagicMock
|
from unittest.mock import patch, MagicMock
|
||||||
import unittest
|
from unittest import TestCase
|
||||||
import io
|
import io
|
||||||
import re
|
import re
|
||||||
|
|
||||||
from argenta.app import App
|
from argenta.app import App
|
||||||
from argenta.command.models import Command
|
from argenta.command import Command
|
||||||
|
from argenta.response import Response
|
||||||
from argenta.router import Router
|
from argenta.router import Router
|
||||||
from argenta.command.flag.models import Flag, Flags, InputFlags
|
from argenta.orchestrator import Orchestrator
|
||||||
|
from argenta.command.flag import Flag
|
||||||
|
from argenta.command.flags import Flags
|
||||||
from argenta.command.flag.defaults import PredefinedFlags
|
from argenta.command.flag.defaults import PredefinedFlags
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class TestSystemHandlerNormalWork(unittest.TestCase):
|
class TestSystemHandlerNormalWork(TestCase):
|
||||||
@patch("builtins.input", side_effect=["test", "q"])
|
@patch("builtins.input", side_effect=["test", "q"])
|
||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response):
|
||||||
print('test command')
|
print('test command')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -36,16 +40,17 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command2(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command2(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response):
|
||||||
print('test command')
|
print('test command')
|
||||||
|
|
||||||
app = App(ignore_command_register=True,
|
app = App(ignore_command_register=True,
|
||||||
override_system_messages=True,
|
override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -56,16 +61,17 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command_with_custom_flag(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command_with_custom_flag(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
flag = Flag('help', '--', False)
|
flag = Flag('help', '--', False)
|
||||||
|
|
||||||
@router.command(Command('test', flags=flag))
|
@router.command(Command('test', flags=flag))
|
||||||
def test(args: InputFlags):
|
def test(response: Response):
|
||||||
print(f'\nhelp for {args.get_flag('help').get_name()} flag\n')
|
print(f'\nhelp for {response.valid_flags.get_flag('help').get_name()} flag\n')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -75,16 +81,18 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command_with_custom_flag2(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command_with_custom_flag2(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
flag = Flag('port', '--', re.compile(r'^\d{1,5}$'))
|
flag = Flag('port', '--', re.compile(r'^\d{1,5}$'))
|
||||||
|
|
||||||
@router.command(Command('test', flags=flag))
|
@router.command(Command('test', flags=flag))
|
||||||
def test(args: InputFlags):
|
def test(response: Response):
|
||||||
print(f'flag value for {args.get_flag('port').get_name()} flag : {args.get_flag('port').get_value()}')
|
input_flag = response.valid_flags.get_flag('port')
|
||||||
|
print(f'flag value for {input_flag.get_name()} flag : {input_flag.get_value()}')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -95,16 +103,17 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command_with_default_flag(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command_with_default_flag(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
flag = PredefinedFlags.SHORT_HELP
|
flag = PredefinedFlags.SHORT_HELP
|
||||||
|
|
||||||
@router.command(Command('test', flags=flag))
|
@router.command(Command('test', flags=flag))
|
||||||
def test(args: InputFlags):
|
def test(response: Response):
|
||||||
print(f'help for {args.get_flag('H').get_name()} flag')
|
print(f'help for {response.valid_flags.get_flag('H').get_name()} flag')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -115,17 +124,18 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command_with_default_flag2(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command_with_default_flag2(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
flag = PredefinedFlags.INFO
|
flag = PredefinedFlags.INFO
|
||||||
|
|
||||||
@router.command(Command('test', flags=flag))
|
@router.command(Command('test', flags=flag))
|
||||||
def test(args: InputFlags):
|
def test(response: Response):
|
||||||
if args.get_flag('info'):
|
if response.valid_flags.get_flag('info'):
|
||||||
print('info about test command')
|
print('info about test command')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -136,16 +146,17 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command_with_default_flag3(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command_with_default_flag3(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
flag = PredefinedFlags.HOST
|
flag = PredefinedFlags.HOST
|
||||||
|
|
||||||
@router.command(Command('test', flags=flag))
|
@router.command(Command('test', flags=flag))
|
||||||
def test(args: InputFlags):
|
def test(response: Response):
|
||||||
print(f'connecting to host {args[0].get_value()}')
|
print(f'connecting to host {response.valid_flags.get_flag('host').get_value()}')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -156,16 +167,18 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_correct_command_with_two_flags(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_correct_command_with_two_flags(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
flags = Flags(PredefinedFlags.HOST, PredefinedFlags.PORT)
|
flags = Flags(PredefinedFlags.HOST, PredefinedFlags.PORT)
|
||||||
|
|
||||||
@router.command(Command('test', flags=flags))
|
@router.command(Command('test', flags=flags))
|
||||||
def test(args: InputFlags):
|
def test(response: Response):
|
||||||
print(f'connecting to host {args[0].get_value()} and port {args[1].get_value()}')
|
valid_flags = response.valid_flags
|
||||||
|
print(f'connecting to host {valid_flags.get_flag('host').get_value()} and port {valid_flags.get_flag('port').get_value()}')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -176,19 +189,20 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_two_correct_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_two_correct_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response):
|
||||||
print(f'test command')
|
print(f'test command')
|
||||||
|
|
||||||
@router.command(Command('some'))
|
@router.command(Command('some'))
|
||||||
def test2():
|
def test2(response):
|
||||||
print(f'some command')
|
print(f'some command')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
@@ -199,23 +213,24 @@ class TestSystemHandlerNormalWork(unittest.TestCase):
|
|||||||
@patch("sys.stdout", new_callable=io.StringIO)
|
@patch("sys.stdout", new_callable=io.StringIO)
|
||||||
def test_input_three_correct_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
def test_input_three_correct_command(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
|
||||||
router = Router()
|
router = Router()
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
|
||||||
@router.command(Command('test'))
|
@router.command(Command('test'))
|
||||||
def test():
|
def test(response):
|
||||||
print(f'test command')
|
print(f'test command')
|
||||||
|
|
||||||
@router.command(Command('some'))
|
@router.command(Command('some'))
|
||||||
def test():
|
def test(response):
|
||||||
print(f'some command')
|
print(f'some command')
|
||||||
|
|
||||||
@router.command(Command('more'))
|
@router.command(Command('more'))
|
||||||
def test():
|
def test(response):
|
||||||
print(f'more command')
|
print(f'more command')
|
||||||
|
|
||||||
app = App(override_system_messages=True,
|
app = App(override_system_messages=True,
|
||||||
print_func=print)
|
print_func=print)
|
||||||
app.include_router(router)
|
app.include_router(router)
|
||||||
app.run_polling()
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
output = mock_stdout.getvalue()
|
output = mock_stdout.getvalue()
|
||||||
|
|
||||||
|
|||||||
@@ -1,10 +1,12 @@
|
|||||||
from argenta.command.flag import Flag, InputFlag, Flags
|
from argenta.command.flag import Flag, InputFlag
|
||||||
|
from argenta.command.flags import Flags
|
||||||
from argenta.command.models import InputCommand, Command
|
from argenta.command.models import InputCommand, Command
|
||||||
from argenta.command.exceptions import (UnprocessedInputFlagException,
|
from argenta.command.exceptions import (UnprocessedInputFlagException,
|
||||||
RepeatedInputFlagsException,
|
RepeatedInputFlagsException,
|
||||||
EmptyInputCommandException)
|
EmptyInputCommandException)
|
||||||
|
|
||||||
import unittest
|
import unittest
|
||||||
|
import re
|
||||||
|
|
||||||
|
|
||||||
class TestInputCommand(unittest.TestCase):
|
class TestInputCommand(unittest.TestCase):
|
||||||
@@ -23,25 +25,37 @@ class TestInputCommand(unittest.TestCase):
|
|||||||
with self.assertRaises(EmptyInputCommandException):
|
with self.assertRaises(EmptyInputCommandException):
|
||||||
InputCommand.parse('')
|
InputCommand.parse('')
|
||||||
|
|
||||||
def test_validate_correct_input_flag1(self):
|
def test_validate_valid_input_flag1(self):
|
||||||
command = Command('some', flags=Flag('test'))
|
command = Command('some', flags=Flag('test'))
|
||||||
self.assertEqual(command.validate_input_flag(InputFlag('test')), True)
|
self.assertEqual(command.validate_input_flag(InputFlag('test')), 'Valid')
|
||||||
|
|
||||||
def test_validate_correct_input_flag2(self):
|
def test_validate_valid_input_flag2(self):
|
||||||
command = Command('some', flags=Flags(Flag('test'), Flag('more')))
|
command = Command('some', flags=Flags(Flag('test'), Flag('more')))
|
||||||
self.assertEqual(command.validate_input_flag(InputFlag('more')), True)
|
self.assertEqual(command.validate_input_flag(InputFlag('more')), 'Valid')
|
||||||
|
|
||||||
def test_validate_incorrect_input_flag1(self):
|
def test_validate_undefined_input_flag1(self):
|
||||||
command = Command('some', flags=Flags(Flag('test')))
|
command = Command('some', flags=Flag('test'))
|
||||||
self.assertEqual(command.validate_input_flag(InputFlag('more')), False)
|
self.assertEqual(command.validate_input_flag(InputFlag('more')), 'Undefined')
|
||||||
|
|
||||||
def test_validate_incorrect_input_flag2(self):
|
def test_validate_undefined_input_flag2(self):
|
||||||
command = Command('some', flags=Flags(Flag('test'), Flag('more')))
|
command = Command('some', flags=Flags(Flag('test'), Flag('more')))
|
||||||
self.assertEqual(command.validate_input_flag(InputFlag('case')), False)
|
self.assertEqual(command.validate_input_flag(InputFlag('case')), 'Undefined')
|
||||||
|
|
||||||
def test_validate_incorrect_input_flag3(self):
|
def test_validate_undefined_input_flag3(self):
|
||||||
command = Command('some')
|
command = Command('some')
|
||||||
self.assertEqual(command.validate_input_flag(InputFlag('case')), False)
|
self.assertEqual(command.validate_input_flag(InputFlag('case')), 'Undefined')
|
||||||
|
|
||||||
|
def test_invalid_input_flag1(self):
|
||||||
|
command = Command('some', flags=Flag('test', possible_values=False))
|
||||||
|
self.assertEqual(command.validate_input_flag(InputFlag('test', value='example')), 'Invalid')
|
||||||
|
|
||||||
|
def test_invalid_input_flag2(self):
|
||||||
|
command = Command('some', flags=Flag('test', possible_values=['some', 'case']))
|
||||||
|
self.assertEqual(command.validate_input_flag(InputFlag('test', value='slay')), 'Invalid')
|
||||||
|
|
||||||
|
def test_invalid_input_flag3(self):
|
||||||
|
command = Command('some', flags=Flag('test', possible_values=re.compile(r'^ex\d{, 2}op$')))
|
||||||
|
self.assertEqual(command.validate_input_flag(InputFlag('test', value='example')), 'Invalid')
|
||||||
|
|
||||||
def test_isinstance_parse_correct_raw_command(self):
|
def test_isinstance_parse_correct_raw_command(self):
|
||||||
cmd = InputCommand.parse('ssh --host 192.168.0.3')
|
cmd = InputCommand.parse('ssh --host 192.168.0.3')
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
from argenta.command.flag.models import Flag, InputFlag, InputFlags, Flags
|
from argenta.command.flag import Flag, InputFlag
|
||||||
|
from argenta.command.flags import InputFlags, Flags
|
||||||
|
|
||||||
import unittest
|
import unittest
|
||||||
import re
|
import re
|
||||||
|
|||||||
@@ -1,4 +1,6 @@
|
|||||||
from argenta.command.flag import InputFlags, InputFlag, Flag, Flags
|
from argenta.command.flag import InputFlag, Flag
|
||||||
|
from argenta.command.flags import Flags, InputFlags, UndefinedInputFlags, InvalidValueInputFlags, ValidInputFlags
|
||||||
|
from argenta.response import Response
|
||||||
from argenta.router import Router
|
from argenta.router import Router
|
||||||
from argenta.command import Command
|
from argenta.command import Command
|
||||||
from argenta.router.exceptions import (TriggerContainSpacesException,
|
from argenta.router.exceptions import (TriggerContainSpacesException,
|
||||||
@@ -24,113 +26,95 @@ class TestRouter(unittest.TestCase):
|
|||||||
with self.assertRaises(RepeatedFlagNameException):
|
with self.assertRaises(RepeatedFlagNameException):
|
||||||
router._validate_command(Command(trigger='command', flags=Flags(Flag('test'), Flag('test'))))
|
router._validate_command(Command(trigger='command', flags=Flags(Flag('test'), Flag('test'))))
|
||||||
|
|
||||||
def test_validate_incorrect_input_flag1(self):
|
def test_structuring_input_flags1(self):
|
||||||
router = Router()
|
router = Router()
|
||||||
router.set_invalid_input_flag_handler(lambda flag: None)
|
cmd = Command('cmd')
|
||||||
self.assertEqual(router._validate_input_flags(Command('cmd'), InputFlags(InputFlag('ssh'))), False)
|
input_flags = InputFlags(InputFlag('ssh'))
|
||||||
|
self.assertEqual(router._structuring_input_flags(cmd, input_flags).undefined_flags, UndefinedInputFlags(InputFlag('ssh')))
|
||||||
|
|
||||||
def test_validate_incorrect_input_flag2(self):
|
def test_structuring_input_flags2(self):
|
||||||
router = Router()
|
router = Router()
|
||||||
router.set_invalid_input_flag_handler(lambda flag: None)
|
cmd = Command('cmd')
|
||||||
self.assertEqual(router._validate_input_flags(Command('cmd'), InputFlags(InputFlag('ssh', value='some'))), False)
|
input_flags = InputFlags(InputFlag('ssh', value='some'))
|
||||||
|
self.assertEqual(router._structuring_input_flags(cmd, input_flags).undefined_flags, UndefinedInputFlags(InputFlag('ssh', value='some')))
|
||||||
|
|
||||||
def test_validate_incorrect_input_flag3(self):
|
def test_structuring_input_flags3(self):
|
||||||
router = Router()
|
router = Router()
|
||||||
router.set_invalid_input_flag_handler(lambda flag: None)
|
cmd = Command('cmd', flags=Flag('port'))
|
||||||
command = Command('cmd', flags=Flag('port'))
|
|
||||||
input_flags = InputFlags(InputFlag('ssh', value='some2'))
|
input_flags = InputFlags(InputFlag('ssh', value='some2'))
|
||||||
self.assertEqual(router._validate_input_flags(command, input_flags), False)
|
self.assertEqual(router._structuring_input_flags(cmd, input_flags).undefined_flags, UndefinedInputFlags(InputFlag('ssh', value='some2')))
|
||||||
|
|
||||||
def test_validate_incorrect_input_flag4(self):
|
def test_structuring_input_flags4(self):
|
||||||
router = Router()
|
router = Router()
|
||||||
router.set_invalid_input_flag_handler(lambda flag: None)
|
|
||||||
command = Command('cmd', flags=Flag('ssh', possible_values=False))
|
command = Command('cmd', flags=Flag('ssh', possible_values=False))
|
||||||
input_flags = InputFlags(InputFlag('ssh', value='some3'))
|
input_flags = InputFlags(InputFlag('ssh', value='some3'))
|
||||||
self.assertEqual(router._validate_input_flags(command, input_flags), False)
|
self.assertEqual(router._structuring_input_flags(command, input_flags).invalid_value_flags, InvalidValueInputFlags(InputFlag('ssh', value='some3')))
|
||||||
|
|
||||||
def test_validate_incorrect_input_flag5(self):
|
def test_structuring_input_flags5(self):
|
||||||
router = Router()
|
router = Router()
|
||||||
router.set_invalid_input_flag_handler(lambda flag: None)
|
|
||||||
command = Command('cmd', flags=Flag('ssh', possible_values=re.compile(r'some[1-5]$')))
|
command = Command('cmd', flags=Flag('ssh', possible_values=re.compile(r'some[1-5]$')))
|
||||||
input_flags = InputFlags(InputFlag('ssh', value='some40'))
|
input_flags = InputFlags(InputFlag('ssh', value='some40'))
|
||||||
self.assertEqual(router._validate_input_flags(command, input_flags), False)
|
self.assertEqual(router._structuring_input_flags(command, input_flags).invalid_value_flags, InvalidValueInputFlags(InputFlag('ssh', value='some40')))
|
||||||
|
|
||||||
def test_validate_incorrect_input_flag6(self):
|
def test_structuring_input_flags6(self):
|
||||||
router = Router()
|
router = Router()
|
||||||
router.set_invalid_input_flag_handler(lambda flag: None)
|
|
||||||
command = Command('cmd', flags=Flag('ssh', possible_values=['example']))
|
command = Command('cmd', flags=Flag('ssh', possible_values=['example']))
|
||||||
input_flags = InputFlags(InputFlag('ssh', value='example2'))
|
input_flags = InputFlags(InputFlag('ssh', value='example2'))
|
||||||
self.assertEqual(router._validate_input_flags(command, input_flags), False)
|
self.assertEqual(router._structuring_input_flags(command, input_flags).invalid_value_flags, InvalidValueInputFlags(InputFlag('ssh', value='example2')))
|
||||||
|
|
||||||
def test_validate_incorrect_input_flag7(self):
|
def test_structuring_input_flags7(self):
|
||||||
router = Router()
|
|
||||||
router.set_invalid_input_flag_handler(lambda flag: None)
|
|
||||||
command = Command('cmd', flags=Flag('ssh', possible_values=['example']))
|
|
||||||
input_flags = InputFlags(InputFlag('ssh'))
|
|
||||||
self.assertEqual(router._validate_input_flags(command, input_flags), False)
|
|
||||||
|
|
||||||
def test_validate_correct_input_flag1(self):
|
|
||||||
command = Command('cmd', flags=Flag('port'))
|
command = Command('cmd', flags=Flag('port'))
|
||||||
input_flags = InputFlags(InputFlag('port', value='some2'))
|
input_flags = InputFlags(InputFlag('port', value='some2'))
|
||||||
self.assertEqual(Router()._validate_input_flags(command, input_flags), True)
|
self.assertEqual(Router()._structuring_input_flags(command, input_flags).valid_flags, ValidInputFlags(InputFlag('port', value='some2')))
|
||||||
|
|
||||||
def test_validate_correct_input_flag2(self):
|
def test_structuring_input_flags8(self):
|
||||||
command = Command('cmd', flags=Flag('port', possible_values=['some2', 'some3']))
|
command = Command('cmd', flags=Flag('port', possible_values=['some2', 'some3']))
|
||||||
input_flags = InputFlags(InputFlag('port', value='some2'))
|
input_flags = InputFlags(InputFlag('port', value='some2'))
|
||||||
self.assertEqual(Router()._validate_input_flags(command, input_flags), True)
|
self.assertEqual(Router()._structuring_input_flags(command, input_flags).valid_flags, ValidInputFlags(InputFlag('port', value='some2')))
|
||||||
|
|
||||||
def test_validate_correct_input_flag3(self):
|
def test_structuring_input_flags9(self):
|
||||||
command = Command('cmd', flags=Flag('ssh', possible_values=re.compile(r'more[1-5]$')))
|
command = Command('cmd', flags=Flag('ssh', possible_values=re.compile(r'more[1-5]$')))
|
||||||
input_flags = InputFlags(InputFlag('ssh', value='more5'))
|
input_flags = InputFlags(InputFlag('ssh', value='more5'))
|
||||||
self.assertEqual(Router()._validate_input_flags(command, input_flags), True)
|
self.assertEqual(Router()._structuring_input_flags(command, input_flags).valid_flags, ValidInputFlags(InputFlag('ssh', value='more5')))
|
||||||
|
|
||||||
def test_validate_correct_input_flag4(self):
|
def test_structuring_input_flags10(self):
|
||||||
command = Command('cmd', flags=Flag('ssh', possible_values=False))
|
command = Command('cmd', flags=Flag('ssh', possible_values=False))
|
||||||
input_flags = InputFlags(InputFlag('ssh'))
|
input_flags = InputFlags(InputFlag('ssh'))
|
||||||
self.assertEqual(Router()._validate_input_flags(command, input_flags), True)
|
self.assertEqual(Router()._structuring_input_flags(command, input_flags).valid_flags, ValidInputFlags(InputFlag('ssh')))
|
||||||
|
|
||||||
def test_validate_incorrect_func_args1(self):
|
def test_validate_incorrect_func_args1(self):
|
||||||
command = Command('cmd', flags=Flag('port'))
|
|
||||||
def handler():
|
def handler():
|
||||||
pass
|
pass
|
||||||
with self.assertRaises(RequiredArgumentNotPassedException):
|
with self.assertRaises(RequiredArgumentNotPassedException):
|
||||||
Router()._validate_func_args(command, handler)
|
Router()._validate_func_args(handler)
|
||||||
|
|
||||||
def test_validate_incorrect_func_args2(self):
|
def test_validate_incorrect_func_args2(self):
|
||||||
command = Command('cmd', flags=Flag('port'))
|
|
||||||
def handler(args, kwargs):
|
def handler(args, kwargs):
|
||||||
pass
|
pass
|
||||||
with self.assertRaises(TooManyTransferredArgsException):
|
with self.assertRaises(TooManyTransferredArgsException):
|
||||||
Router()._validate_func_args(command, handler)
|
Router()._validate_func_args(handler)
|
||||||
|
|
||||||
def test_validate_incorrect_func_args3(self):
|
|
||||||
command = Command('cmd')
|
|
||||||
def handler(args):
|
|
||||||
pass
|
|
||||||
with self.assertRaises(TooManyTransferredArgsException):
|
|
||||||
Router()._validate_func_args(command, handler)
|
|
||||||
|
|
||||||
def test_get_router_aliases(self):
|
def test_get_router_aliases(self):
|
||||||
router = Router()
|
router = Router()
|
||||||
@router.command(Command('some', aliases=['test', 'case']))
|
@router.command(Command('some', aliases=['test', 'case']))
|
||||||
def handler():
|
def handler(response):
|
||||||
pass
|
pass
|
||||||
self.assertListEqual(router.get_aliases(), ['test', 'case'])
|
self.assertListEqual(router.get_aliases(), ['test', 'case'])
|
||||||
|
|
||||||
def test_get_router_aliases2(self):
|
def test_get_router_aliases2(self):
|
||||||
router = Router()
|
router = Router()
|
||||||
@router.command(Command('some', aliases=['test', 'case']))
|
@router.command(Command('some', aliases=['test', 'case']))
|
||||||
def handler():
|
def handler(response):
|
||||||
pass
|
pass
|
||||||
@router.command(Command('ext', aliases=['more', 'foo']))
|
@router.command(Command('ext', aliases=['more', 'foo']))
|
||||||
def handler2():
|
def handler2(response):
|
||||||
pass
|
pass
|
||||||
self.assertListEqual(router.get_aliases(), ['test', 'case', 'more', 'foo'])
|
self.assertListEqual(router.get_aliases(), ['test', 'case', 'more', 'foo'])
|
||||||
|
|
||||||
def test_get_router_aliases3(self):
|
def test_get_router_aliases3(self):
|
||||||
router = Router()
|
router = Router()
|
||||||
@router.command(Command('some'))
|
@router.command(Command('some'))
|
||||||
def handler():
|
def handler(response):
|
||||||
pass
|
pass
|
||||||
self.assertListEqual(router.get_aliases(), [])
|
self.assertListEqual(router.get_aliases(), [])
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user