|
import os
|
|
import logging
|
|
import time
|
|
import sys
|
|
|
|
from rich.theme import Theme
|
|
from rich.logging import RichHandler
|
|
from rich.console import Console
|
|
from rich.pretty import install as pretty_install
|
|
from rich.traceback import install as traceback_install
|
|
|
|
log = None
|
|
|
|
|
|
def setup_logging(clean=False, debug=False):
|
|
global log
|
|
|
|
if log is not None:
|
|
return log
|
|
|
|
try:
|
|
if clean and os.path.isfile("setup.log"):
|
|
os.remove("setup.log")
|
|
time.sleep(0.1)
|
|
except:
|
|
pass
|
|
|
|
if sys.version_info >= (3, 9):
|
|
logging.basicConfig(
|
|
level=logging.DEBUG,
|
|
format="%(asctime)s | %(levelname)s | %(pathname)s | %(message)s",
|
|
filename="setup.log",
|
|
filemode="a",
|
|
encoding="utf-8",
|
|
force=True,
|
|
)
|
|
else:
|
|
logging.basicConfig(
|
|
level=logging.DEBUG,
|
|
format="%(asctime)s | %(levelname)s | %(pathname)s | %(message)s",
|
|
filename="setup.log",
|
|
filemode="a",
|
|
force=True,
|
|
)
|
|
|
|
console = Console(
|
|
log_time=True,
|
|
log_time_format="%H:%M:%S-%f",
|
|
theme=Theme(
|
|
{
|
|
"traceback.border": "black",
|
|
"traceback.border.syntax_error": "black",
|
|
"inspect.value.border": "black",
|
|
}
|
|
),
|
|
)
|
|
pretty_install(console=console)
|
|
traceback_install(
|
|
console=console,
|
|
extra_lines=1,
|
|
width=console.width,
|
|
word_wrap=False,
|
|
indent_guides=False,
|
|
suppress=[],
|
|
)
|
|
rh = RichHandler(
|
|
show_time=True,
|
|
omit_repeated_times=False,
|
|
show_level=True,
|
|
show_path=False,
|
|
markup=False,
|
|
rich_tracebacks=True,
|
|
log_time_format="%H:%M:%S-%f",
|
|
level=logging.DEBUG if debug else logging.INFO,
|
|
console=console,
|
|
)
|
|
rh.set_name(logging.DEBUG if debug else logging.INFO)
|
|
log = logging.getLogger("sd")
|
|
log.addHandler(rh)
|
|
|
|
return log
|
|
|
