mirror of
https://github.com/cdryzun/tg_bot_collections.git
synced 2025-08-05 13:16:42 +08:00
feat: add --disable-command option to specify a command to disable
This commit is contained in:
14
tg.py
14
tg.py
@ -9,12 +9,24 @@ def main():
|
||||
# Init args
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("tg_token", help="tg token")
|
||||
|
||||
# 'disable-command' option
|
||||
# The action 'append' will allow multiple entries to be saved into a list
|
||||
# The variable name is changed to 'disable_commands'
|
||||
parser.add_argument(
|
||||
"--disable-command",
|
||||
action="append",
|
||||
dest="disable_commands",
|
||||
help="Specify a command to disable. Can be used multiple times.",
|
||||
default=[],
|
||||
)
|
||||
|
||||
options = parser.parse_args()
|
||||
print("Arg parse done.")
|
||||
|
||||
# Init bot
|
||||
bot = TeleBot(options.tg_token)
|
||||
load_handlers(bot)
|
||||
load_handlers(bot, options.disable_commands)
|
||||
print("Bot init done.")
|
||||
|
||||
# Start bot
|
||||
|
Reference in New Issue
Block a user