diff --git a/bot.py b/bot.py index b55a71b..b145b8d 100644 --- a/bot.py +++ b/bot.py @@ -97,8 +97,5 @@ class Butler(object): def run(self) -> None: """Start the bot.""" - if self.whitelist: - self.logger.info("Current whitelist: {}".format(self.whitelist)) - # Run the bot until the user presses Ctrl-C self.application.run_polling(allowed_updates=Update.ALL_TYPES) diff --git a/main.py b/main.py index 8d4e4e7..6177c0c 100644 --- a/main.py +++ b/main.py @@ -17,11 +17,14 @@ def main(): parser = argparse.ArgumentParser(description='Vylion\'s butler bot.') parser.add_argument('token', metavar='TOKEN', help='The Bot Token to work with the Telegram Bot API') + parser.add_argument('-l', '--loglevel', default='INFO', metavar='LOG_LEVEL', + help='The level for logging.') parser.add_argument('-c', '--chats', nargs='*', default=[], metavar='[CHATS]', help='A whitelist of allowed chats. Empty enables all.') args = parser.parse_args() logger = logging.getLogger(__name__) + loglevel = logging.getLevelName(args.loglevel) bot = Butler(logger, args.token, args.chats) # Enable logging @@ -35,6 +38,8 @@ def main(): coloredlogs.install(level=logging.INFO, fmt=log_format) logger.info("Bot started up!") + logger.info("Current whitelist: {}".format(args.chats)) + logger.setLevel(loglevel) bot.run()