cdryzun/tg_bot_collections
1
0
Fork 0
mirror of https://github.com/cdryzun/tg_bot_collections.git synced 2025-11-04 16:56:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add --disable-command option to specify a command to disable

Browse Source
This commit is contained in:
Pagliacii
2023-12-15 21:39:21 +08:00
parent 384cc4656f
commit 93fb98f7b5
3 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
handlers/__init__.py
View File

@ -55,12 +55,14 @@ def wrap_handler(handler: T, bot: TeleBot) -> T:
return update_wrapper(wrapper, handler)
def load_handlers(bot: TeleBot) -> None:
def load_handlers(bot: TeleBot, disable_commands: list[str]) -> None:
# import all submodules
this_path = Path(__file__).parent
for child in this_path.iterdir():
if child.name.startswith("_"):
continue
if child.stem in disable_commands:
continue
module = importlib.import_module(f".{child.stem}", __package__)
if hasattr(module, "register"):
print(f"Loading {child.stem} handlers.")