diff --git a/main.py b/main.py index d44336d..1083ea6 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,7 @@ import sys import asyncio import logging +from logging.handlers import RotatingFileHandler import os import psutil import time @@ -17,6 +18,13 @@ logging.basicConfig( ) logging.getLogger('trueconf.client.chatbot').propagate = False logger = logging.getLogger(__name__) +sd_workflow_logger = logging.getLogger('sd_workflow') +sd_workflow_logger.setLevel(logging.INFO) +if not sd_workflow_logger.handlers: + handler = RotatingFileHandler(os.path.join(os.path.dirname(__file__), 'logs', 'sd_workflow.log'), maxBytes=10*1024*1024, backupCount=5) + formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s') + handler.setFormatter(formatter) + sd_workflow_logger.addHandler(handler) # Добавляем путь к проекту в sys.path sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) @@ -123,7 +131,7 @@ async def main_menu(msg: Message): urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) # Импортируем SD_URL и SD_TOKEN из config import importlib.util - spec = importlib.util.spec_from_file_location("custom_config", "/opt/trueconf_bot/config/config.py") + spec = importlib.util.spec_from_file_location("custom_config", os.path.join(os.path.dirname(os.path.abspath(__file__)), "config", "config.py")) custom_config = importlib.util.module_from_spec(spec) spec.loader.exec_module(custom_config) SD_URL = custom_config.SD_URL