limited logs

This commit is contained in:
Guillermo 2024-09-13 05:25:33 +02:00
parent bac7373789
commit 2377e5e325
2 changed files with 5 additions and 3 deletions

3
bot.py
View file

@ -97,8 +97,5 @@ class Butler(object):
def run(self) -> None: def run(self) -> None:
"""Start the bot.""" """Start the bot."""
if self.whitelist:
self.logger.info("Current whitelist: {}".format(self.whitelist))
# Run the bot until the user presses Ctrl-C # Run the bot until the user presses Ctrl-C
self.application.run_polling(allowed_updates=Update.ALL_TYPES) self.application.run_polling(allowed_updates=Update.ALL_TYPES)

View file

@ -17,11 +17,14 @@ def main():
parser = argparse.ArgumentParser(description='Vylion\'s butler bot.') parser = argparse.ArgumentParser(description='Vylion\'s butler bot.')
parser.add_argument('token', metavar='TOKEN', parser.add_argument('token', metavar='TOKEN',
help='The Bot Token to work with the Telegram Bot API') 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]', parser.add_argument('-c', '--chats', nargs='*', default=[], metavar='[CHATS]',
help='A whitelist of allowed chats. Empty enables all.') help='A whitelist of allowed chats. Empty enables all.')
args = parser.parse_args() args = parser.parse_args()
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
loglevel = logging.getLevelName(args.loglevel)
bot = Butler(logger, args.token, args.chats) bot = Butler(logger, args.token, args.chats)
# Enable logging # Enable logging
@ -35,6 +38,8 @@ def main():
coloredlogs.install(level=logging.INFO, fmt=log_format) coloredlogs.install(level=logging.INFO, fmt=log_format)
logger.info("Bot started up!") logger.info("Bot started up!")
logger.info("Current whitelist: {}".format(args.chats))
logger.setLevel(loglevel)
bot.run() bot.run()