additionally log to file

Author: papuSpartan

scripts/spartan/World.py

@@ -11,13 +11,10 @@
 import time
 from typing import List, Dict, Union
 from threading import Thread
-from inspect import getsourcefile
-from os.path import abspath
-from pathlib import Path
 from modules.processing import process_images, StableDiffusionProcessingTxt2Img
 import modules.shared as shared
 from .Worker import Worker, State
-from .shared import logger, warmup_samples
+from .shared import logger, warmup_samples, extension_path
 from .pmodels import Config_Model, Benchmark_Payload
 from . import shared as sh
 
@@ -70,7 +67,6 @@ class World:
     """
 
     # I'd rather keep the sdwui root directory clean.
-    extension_path = Path(abspath(getsourcefile(lambda: 0))).parent.parent.parent
     config_path = shared.cmd_opts.distributed_config
     old_config_path = worker_info_path = extension_path.joinpath('workers.json')
 

scripts/spartan/shared.py

@@ -1,21 +1,31 @@
 import logging
+from inspect import getsourcefile
 from typing import Union
 from rich.logging import RichHandler
 from modules.shared import cmd_opts
 from pydantic import BaseModel, Field
+from os.path import abspath
+
+from pathlib import Path
+extension_path = Path(abspath(getsourcefile(lambda: 0))).parent.parent.parent
 
 # https://rich.readthedocs.io/en/stable/logging.html
 log_level = 'DEBUG' if cmd_opts.distributed_debug else 'INFO'
 logger = logging.getLogger("distributed")
-handler = RichHandler(
+rich_handler = RichHandler(
     rich_tracebacks=True,
     markup=True,
     show_time=False,
     keywords=["distributed", "Distributed", "worker", "Worker", "world", "World"]
 )
-logger.addHandler(handler)
-logger.setLevel(log_level)
 logger.propagate = False  # prevent log duplication by webui since it now uses the logging module
+logger.setLevel(log_level)
+log_path = extension_path.joinpath('distributed.log')
+file_handler = logging.FileHandler(log_path)
+file_handler.formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
+logger.addHandler(rich_handler)
+logger.addHandler(file_handler)
+# end logging
 
 warmup_samples = 2  # number of samples to do before recording a valid benchmark sample