mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-04-21 05:29:57 +08:00
37 lines
1.1 KiB
Python
Executable File
37 lines
1.1 KiB
Python
Executable File
import json
|
|
import logging.config
|
|
import os
|
|
|
|
|
|
def log_dir():
|
|
fnm = os.path.join(os.path.dirname(__file__), '../log/')
|
|
if not os.path.exists(fnm):
|
|
fnm = os.path.join(os.path.dirname(__file__), '../../log/')
|
|
assert os.path.exists(fnm), f"Can't locate log dir: {fnm}"
|
|
return fnm
|
|
|
|
|
|
def setup_logging(default_path="conf/logging.json",
|
|
default_level=logging.INFO,
|
|
env_key="LOG_CFG"):
|
|
path = default_path
|
|
value = os.getenv(env_key, None)
|
|
if value:
|
|
path = value
|
|
if os.path.exists(path):
|
|
with open(path, "r") as f:
|
|
config = json.load(f)
|
|
fnm = log_dir()
|
|
|
|
config["handlers"]["info_file_handler"]["filename"] = fnm + "info.log"
|
|
config["handlers"]["error_file_handler"]["filename"] = fnm + "error.log"
|
|
logging.config.dictConfig(config)
|
|
else:
|
|
logging.basicConfig(level=default_level)
|
|
|
|
|
|
__fnm = os.path.join(os.path.dirname(__file__), 'conf/logging.json')
|
|
if not os.path.exists(__fnm):
|
|
__fnm = os.path.join(os.path.dirname(__file__), '../../conf/logging.json')
|
|
setup_logging(__fnm)
|