48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
import datetime
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import atexit
|
|
import threading
|
|
import logging
|
|
from logging.handlers import RotatingFileHandler
|
|
|
|
handler = RotatingFileHandler('comfy-deploy.log', maxBytes=500000, backupCount=5)
|
|
|
|
original_stdout = sys.stdout
|
|
original_stderr = sys.stderr
|
|
|
|
class StreamToLogger():
|
|
def __init__(self, log_level):
|
|
self.log_level = log_level
|
|
|
|
def write(self, buf):
|
|
if (self.log_level == logging.INFO):
|
|
original_stdout.write(buf)
|
|
original_stdout.flush()
|
|
elif (self.log_level == logging.ERROR):
|
|
original_stderr.write(buf)
|
|
original_stderr.flush()
|
|
|
|
for line in buf.rstrip().splitlines():
|
|
handler.handle(
|
|
logging.LogRecord(
|
|
name="comfy-deploy",
|
|
level=self.log_level,
|
|
pathname="prestartup_script.py",
|
|
lineno=1,
|
|
msg=line.rstrip(),
|
|
args=None,
|
|
exc_info=None
|
|
)
|
|
)
|
|
|
|
def flush(self):
|
|
if (self.log_level == logging.INFO):
|
|
original_stdout.flush()
|
|
elif (self.log_level == logging.ERROR):
|
|
original_stderr.flush()
|
|
|
|
# Redirect stdout and stderr to the logger
|
|
sys.stdout = StreamToLogger(logging.INFO)
|
|
sys.stderr = StreamToLogger(logging.ERROR) |