init: 全局日志管理
chenzhirong
5 months ago
parent
bd1d5d6636
commit
93190ac0e5
@ -1,3 +1,27 @@
|
||||
from collections import namedtuple
|
||||
|
||||
from beartype.claw import beartype_this_package
|
||||
|
||||
beartype_this_package()
|
||||
HttpCode = namedtuple('HttpResp', ['code', 'msg'])
|
||||
|
||||
class HttpResp:
|
||||
"""HTTP响应结果
|
||||
"""
|
||||
SUCCESS = HttpCode(200, '成功')
|
||||
FAILED = HttpCode(300, '失败')
|
||||
PARAMS_VALID_ERROR = HttpCode(310, '参数校验错误')
|
||||
PARAMS_TYPE_ERROR = HttpCode(311, '参数类型错误')
|
||||
REQUEST_METHOD_ERROR = HttpCode(312, '请求方法错误')
|
||||
ASSERT_ARGUMENT_ERROR = HttpCode(313, '断言参数错误')
|
||||
|
||||
LOGIN_ACCOUNT_ERROR = HttpCode(330, '登录账号或密码错误')
|
||||
LOGIN_DISABLE_ERROR = HttpCode(331, '登录账号已被禁用了')
|
||||
TOKEN_EMPTY = HttpCode(332, 'token参数为空')
|
||||
TOKEN_INVALID = HttpCode(333, 'token参数无效')
|
||||
|
||||
NO_PERMISSION = HttpCode(403, '无相关权限')
|
||||
REQUEST_404_ERROR = HttpCode(404, '请求接口不存在')
|
||||
DATA_ALREADY_EXISTS = HttpCode(409, '数据已存在')
|
||||
SYSTEM_ERROR = HttpCode(500, '系统错误')
|
||||
SYSTEM_TIMEOUT_ERROR = HttpCode(504, '请求超时')
|
||||
@ -0,0 +1,78 @@
|
||||
import logging
|
||||
import os
|
||||
import os.path
|
||||
from logging.handlers import RotatingFileHandler
|
||||
|
||||
initialized_root_logger = False
|
||||
|
||||
|
||||
def get_project_base_directory():
|
||||
PROJECT_BASE = os.path.abspath(
|
||||
os.path.join(
|
||||
os.path.dirname(os.path.realpath(__file__)),
|
||||
os.pardir,
|
||||
os.pardir,
|
||||
)
|
||||
)
|
||||
return PROJECT_BASE
|
||||
|
||||
|
||||
def init_root_logger(logfile_basename: str, log_format: str = "%(asctime)-15s %(levelname)-8s %(process)d %(message)s"):
|
||||
global initialized_root_logger
|
||||
if initialized_root_logger:
|
||||
return
|
||||
initialized_root_logger = True
|
||||
|
||||
logger = logging.getLogger()
|
||||
logger.handlers.clear()
|
||||
log_path = os.path.abspath(os.path.join(get_project_base_directory(), "logs", f"{logfile_basename}.log"))
|
||||
|
||||
os.makedirs(os.path.dirname(log_path), exist_ok=True)
|
||||
formatter = logging.Formatter(log_format)
|
||||
|
||||
handler1 = RotatingFileHandler(log_path, maxBytes=10 * 1024 * 1024, backupCount=5)
|
||||
handler1.setFormatter(formatter)
|
||||
logger.addHandler(handler1)
|
||||
|
||||
handler2 = logging.StreamHandler()
|
||||
handler2.setFormatter(formatter)
|
||||
logger.addHandler(handler2)
|
||||
|
||||
logging.captureWarnings(True)
|
||||
|
||||
LOG_LEVELS = os.environ.get("LOG_LEVELS", "")
|
||||
pkg_levels = {}
|
||||
for pkg_name_level in LOG_LEVELS.split(","):
|
||||
terms = pkg_name_level.split("=")
|
||||
if len(terms) != 2:
|
||||
continue
|
||||
pkg_name, pkg_level = terms[0], terms[1]
|
||||
pkg_name = pkg_name.strip()
|
||||
pkg_level = logging.getLevelName(pkg_level.strip().upper())
|
||||
if not isinstance(pkg_level, int):
|
||||
pkg_level = logging.INFO
|
||||
pkg_levels[pkg_name] = logging.getLevelName(pkg_level)
|
||||
|
||||
# for pkg_name in ['peewee', 'pdfminer']:
|
||||
# if pkg_name not in pkg_levels:
|
||||
# pkg_levels[pkg_name] = logging.getLevelName(logging.WARNING)
|
||||
if 'root' not in pkg_levels:
|
||||
pkg_levels['root'] = logging.getLevelName(logging.INFO)
|
||||
|
||||
for pkg_name, pkg_level in pkg_levels.items():
|
||||
pkg_logger = logging.getLogger(pkg_name)
|
||||
pkg_logger.setLevel(pkg_level)
|
||||
|
||||
msg = f"{logfile_basename} log path: {log_path}, log levels: {pkg_levels}"
|
||||
logger.info(msg)
|
||||
|
||||
|
||||
def log_exception(e, *args):
|
||||
logging.exception(e)
|
||||
for a in args:
|
||||
if hasattr(a, "text"):
|
||||
logging.error(a.text)
|
||||
raise Exception(a.text)
|
||||
else:
|
||||
logging.error(str(a))
|
||||
raise e
|
||||
Loading…
Reference in New Issue