limited logs
This commit is contained in:
parent
bac7373789
commit
2377e5e325
2 changed files with 5 additions and 3 deletions
3
bot.py
3
bot.py
|
@ -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)
|
||||||
|
|
5
main.py
5
main.py
|
@ -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()
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue