class CommandCenter: def __init__(self, default_input_type, default_function=None, all_commands=None): self.commands = {} self.add_command("/default", default_input_type, default_function) if all_commands: for command, input_type, function in all_commands: self.add_command(command, input_type, function) def add_command(self, command, input_type, function=None): assert input_type in [None, str, int, float, bool, list], "Invalid input type" self.commands[command] = {"input_type": input_type, "function": function} def parse_command(self, input_string): # parsing the input string if not input_string.startswith("/"): command = "/default" argument = input_string.split(" ") else: inputs = input_string.split(" ") command = inputs[0] argument = inputs[1:] # type casting the arguments if self.commands[command]["input_type"] == str: argument = " ".join(argument) elif self.commands[command]["input_type"] == int: argument = int(" ".join(argument)) elif self.commands[command]["input_type"] == float: argument = float(" ".join(argument)) elif self.commands[command]["input_type"] == bool: argument = bool(" ".join(argument)) elif self.commands[command]["input_type"] == list: argument = argument else: argument = None return command, argument def execute_command(self, input_string): command, argument = self.parse_command(input_string) if command in self.commands: return self.commands[command]["function"](argument) else: return "Invalid command"