88 lines
1.7 KiB
Python
88 lines
1.7 KiB
Python
"""
|
|
Python's default logger may be causing lock contention issues.
|
|
|
|
We've seen tremendous slowdown in so-vits-svc, and the logging crate
|
|
seems to be one of the big contributors.
|
|
|
|
Other projects are seeing this as well:
|
|
https://github.com/kubernetes-client/python/issues/1867
|
|
"""
|
|
|
|
import os
|
|
|
|
# From the original crate
|
|
CRITICAL = 50
|
|
FATAL = CRITICAL
|
|
ERROR = 40
|
|
WARNING = 30
|
|
WARN = WARNING
|
|
INFO = 20
|
|
DEBUG = 10
|
|
NOTSET = 0
|
|
|
|
# From the original crate
|
|
_levelToName = {
|
|
CRITICAL: 'CRITICAL',
|
|
ERROR: 'ERROR',
|
|
WARNING: 'WARNING',
|
|
INFO: 'INFO',
|
|
DEBUG: 'DEBUG',
|
|
NOTSET: 'NOTSET',
|
|
}
|
|
_nameToLevel = {
|
|
'CRITICAL': CRITICAL,
|
|
'FATAL': FATAL,
|
|
'ERROR': ERROR,
|
|
'WARN': WARNING,
|
|
'WARNING': WARNING,
|
|
'INFO': INFO,
|
|
'DEBUG': DEBUG,
|
|
'NOTSET': NOTSET,
|
|
}
|
|
|
|
|
|
def _get_log_level():
|
|
level = os.getenv("PYTHON_LOG_LEVEL")
|
|
if not level or type(level) is not str:
|
|
return WARN
|
|
level = level.strip().upper()
|
|
if level in _nameToLevel:
|
|
return _nameToLevel[level]
|
|
return WARN
|
|
|
|
|
|
class Logger:
|
|
def __init__(self):
|
|
self.level = _get_log_level()
|
|
self.handlers = [] # We don't care about this complexity
|
|
self.flush = True
|
|
pass
|
|
|
|
# noinspection PyPep8Naming
|
|
def setLevel(self, level):
|
|
self.level = level
|
|
|
|
# noinspection PyPep8Naming
|
|
def addHandler(self, *args):
|
|
pass
|
|
|
|
def warning(self, *args):
|
|
print(args, flush=self.flush)
|
|
|
|
def warn(self, *args):
|
|
print(args, flush=self.flush)
|
|
|
|
def info(self, *args):
|
|
print(args, flush=self.flush)
|
|
|
|
def debug(self, *args):
|
|
print(args, flush=self.flush)
|
|
|
|
|
|
LOGGER = Logger()
|
|
|
|
|
|
# noinspection PyPep8Naming
|
|
def getLogger(*args):
|
|
return LOGGER
|