diff --git a/log.py b/log.py index ae671e3..6beb356 100644 --- a/log.py +++ b/log.py @@ -2,32 +2,33 @@ import logging def getlogger(): - # create a custom logger + # create a custom logger if no log handler logger = logging.getLogger(__name__) + if not logger.hasHandlers(): + logger.setLevel(logging.INFO) + # create handlers + warn_handler = logging.StreamHandler() + info_handler = logging.StreamHandler() + error_handler = logging.FileHandler('bot.log', mode='a') + warn_handler.setLevel(logging.WARNING) + error_handler.setLevel(logging.ERROR) + info_handler.setLevel(logging.INFO) - # create handlers - warn_handler = logging.StreamHandler() - info_handler = logging.StreamHandler() - error_handler = logging.FileHandler('bot.log', mode='a') - warn_handler.setLevel(logging.WARNING) - error_handler.setLevel(logging.ERROR) - info_handler.setLevel(logging.INFO) + # create formatters + warn_format = logging.Formatter( + '%(name)s - %(funcName)s - %(levelname)s - %(message)s') + error_format = logging.Formatter( + '%(asctime)s - %(name)s - %(funcName)s - %(levelname)s - %(message)s') + info_format = logging.Formatter('%(message)s') - # create formatters - warn_format = logging.Formatter( - '%(name)s - %(funcName)s - %(levelname)s - %(message)s') - error_format = logging.Formatter( - '%(asctime)s - %(name)s - %(funcName)s - %(levelname)s - %(message)s') - info_format = logging.Formatter('%(message)s') + # set formatter + warn_handler.setFormatter(warn_format) + error_handler.setFormatter(error_format) + info_handler.setFormatter(info_format) - # set formatter - warn_handler.setFormatter(warn_format) - error_handler.setFormatter(error_format) - info_handler.setFormatter(info_format) - - # add handlers to logger - logger.addHandler(warn_handler) - logger.addHandler(error_handler) - logger.addHandler(info_handler) + # add handlers to logger + logger.addHandler(warn_handler) + logger.addHandler(error_handler) + logger.addHandler(info_handler) return logger diff --git a/main.py b/main.py index 4853a82..b71e5d8 100644 --- a/main.py +++ b/main.py @@ -1,8 +1,6 @@ import asyncio import json import os -import signal -from functools import partial from bot import Bot from log import getlogger @@ -53,6 +51,4 @@ async def main(): if __name__ == "__main__": logger.info("matrix chatgpt bot start.....") - print("matrix chatgpt bot start.....") asyncio.run(main()) -