diff --git a/handlers/__init__.py b/handlers/__init__.py index c24b9f4..3ea566c 100644 --- a/handlers/__init__.py +++ b/handlers/__init__.py @@ -80,3 +80,13 @@ def load_handlers(bot: TeleBot, disable_commands: list[str]) -> None: if all_commands: bot.set_my_commands(all_commands) print("Setting commands done.") + + +def available_commands() -> list[str]: + commands = [] + this_path = Path(__file__).parent + for child in this_path.iterdir(): + if child.name.startswith("_"): + continue + commands.append(child.stem) + return commands diff --git a/tg.py b/tg.py index a8f77e9..678a534 100644 --- a/tg.py +++ b/tg.py @@ -2,7 +2,7 @@ import argparse from telebot import TeleBot -from handlers import load_handlers +from handlers import available_commands, load_handlers def main(): @@ -19,6 +19,7 @@ def main(): dest="disable_commands", help="Specify a command to disable. Can be used multiple times.", default=[], + choices=available_commands(), ) options = parser.parse_args()