mirror of
				https://github.com/cdryzun/tg_bot_collections.git
				synced 2025-11-04 16:56:43 +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