Files
Argenta/argenta/router/command_handler/entity.py
T

21 lines
619 B
Python

from typing import Callable
from argenta.command import Command
from argenta.command.flag.models import InputFlags
class CommandHandler:
def __init__(self, handler: Callable[[], None] | Callable[[InputFlags], None], handled_command: Command):
self.handler = handler
self.handled_command = handled_command
def handling(self, input_flags: InputFlags):
if input_flags.get_flags():
self.handler(input_flags)
else:
self.handler()
def get_handler(self):
return self.handler
def get_handled_command(self):
return self.handled_command