mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-18 02:30:08 +08:00
init🎉: 首次提交
This commit is contained in:
HibiKier
@@ -0,0 +1,307 @@
|
||||
from datetime import datetime, timedelta
|
||||
from typing import Any, Dict, overload
|
||||
|
||||
from nonebot import require
|
||||
|
||||
require("nonebot_plugin_session")
|
||||
from loguru import logger as logger_
|
||||
from nonebot.log import default_filter, default_format
|
||||
from nonebot_plugin_session import Session
|
||||
|
||||
from zhenxun.configs.path_config import LOG_PATH
|
||||
|
||||
logger_.add(
|
||||
LOG_PATH / f"{datetime.now().date()}.log",
|
||||
level="INFO",
|
||||
rotation="00:00",
|
||||
format=default_format,
|
||||
filter=default_filter,
|
||||
retention=timedelta(days=30),
|
||||
)
|
||||
|
||||
logger_.add(
|
||||
LOG_PATH / f"error_{datetime.now().date()}.log",
|
||||
level="ERROR",
|
||||
rotation="00:00",
|
||||
format=default_format,
|
||||
filter=default_filter,
|
||||
retention=timedelta(days=30),
|
||||
)
|
||||
|
||||
|
||||
class logger:
|
||||
TEMPLATE_A = "Adapter[{}] {}"
|
||||
TEMPLATE_B = "Adapter[{}] [<u><c>{}</c></u>]: {}"
|
||||
TEMPLATE_C = "Adapter[{}] 用户[<u><e>{}</e></u>] 触发 [<u><c>{}</c></u>]: {}"
|
||||
TEMPLATE_D = "Adapter[{}] 群聊[<u><e>{}</e></u>] 用户[<u><e>{}</e></u>] 触发 [<u><c>{}</c></u>]: {}"
|
||||
TEMPLATE_E = "Adapter[{}] 群聊[<u><e>{}</e></u>] 用户[<u><e>{}</e></u>] 触发 [<u><c>{}</c></u>] [Target](<u><e>{}</e></u>): {}"
|
||||
|
||||
TEMPLATE_ADAPTER = "Adapter[<m>{}</m>] "
|
||||
TEMPLATE_USER = "用户[<u><e>{}</e></u>] "
|
||||
TEMPLATE_GROUP = "群聊[<u><e>{}</e></u>] "
|
||||
TEMPLATE_COMMAND = "CMD[<u><c>{}</c></u>] "
|
||||
TEMPLATE_TARGET = "[Target]([<u><e>{}</e></u>]) "
|
||||
|
||||
SUCCESS_TEMPLATE = "[<u><c>{}</c></u>]: {} | 参数[{}] 返回: [<y>{}</y>]"
|
||||
|
||||
WARNING_TEMPLATE = "[<u><y>{}</y></u>]: {}"
|
||||
|
||||
ERROR_TEMPLATE = "[<u><r>{}</r></u>]: {}"
|
||||
|
||||
@overload
|
||||
@classmethod
|
||||
def info(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: int | str | None = None,
|
||||
group_id: int | str | None = None,
|
||||
adapter: str | None = None,
|
||||
target: Any = None,
|
||||
):
|
||||
...
|
||||
|
||||
@overload
|
||||
@classmethod
|
||||
def info(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: Session | None = None,
|
||||
target: Any = None,
|
||||
):
|
||||
...
|
||||
|
||||
@classmethod
|
||||
def info(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: int | str | Session | None = None,
|
||||
group_id: int | str | None = None,
|
||||
adapter: str | None = None,
|
||||
target: Any = None,
|
||||
):
|
||||
user_id: str | None = session # type: ignore
|
||||
group_id = None
|
||||
if type(session) == Session:
|
||||
user_id = session.id1
|
||||
adapter = session.bot_type
|
||||
if session.id3 or session.id2:
|
||||
group_id = f"{session.id3}:{session.id2}"
|
||||
template = cls.__parser_template(
|
||||
info, command, user_id, group_id, adapter, target
|
||||
)
|
||||
logger_.opt(colors=True).info(template)
|
||||
|
||||
@classmethod
|
||||
def success(
|
||||
cls,
|
||||
info: str,
|
||||
command: str,
|
||||
param: Dict[str, Any] | None = None,
|
||||
result: str = "",
|
||||
):
|
||||
param_str = ""
|
||||
if param:
|
||||
param_str = ",".join([f"<m>{k}</m>:<g>{v}</g>" for k, v in param.items()])
|
||||
logger_.opt(colors=True).success(
|
||||
cls.SUCCESS_TEMPLATE.format(command, info, param_str, result)
|
||||
)
|
||||
|
||||
@overload
|
||||
@classmethod
|
||||
def warning(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: int | str | None = None,
|
||||
group_id: int | str | None = None,
|
||||
adapter: str | None = None,
|
||||
target: Any = None,
|
||||
e: Exception | None = None,
|
||||
):
|
||||
...
|
||||
|
||||
@overload
|
||||
@classmethod
|
||||
def warning(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: Session | None = None,
|
||||
adapter: str | None = None,
|
||||
target: Any = None,
|
||||
e: Exception | None = None,
|
||||
):
|
||||
...
|
||||
|
||||
@classmethod
|
||||
def warning(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: int | str | Session | None = None,
|
||||
group_id: int | str | None = None,
|
||||
adapter: str | None = None,
|
||||
target: Any = None,
|
||||
e: Exception | None = None,
|
||||
):
|
||||
user_id: str | None = session # type: ignore
|
||||
group_id = None
|
||||
if type(session) == Session:
|
||||
user_id = session.id1
|
||||
adapter = session.bot_type
|
||||
if session.id3 or session.id2:
|
||||
group_id = f"{session.id3}:{session.id2}"
|
||||
template = cls.__parser_template(
|
||||
info, command, user_id, group_id, adapter, target
|
||||
)
|
||||
if e:
|
||||
template += f" || 错误<r>{type(e)}: {e}</r>"
|
||||
logger_.opt(colors=True).warning(template)
|
||||
|
||||
@overload
|
||||
@classmethod
|
||||
def error(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: int | str | None = None,
|
||||
group_id: int | str | None = None,
|
||||
adapter: str | None = None,
|
||||
target: Any = None,
|
||||
e: Exception | None = None,
|
||||
):
|
||||
...
|
||||
|
||||
@overload
|
||||
@classmethod
|
||||
def error(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: Session | None = None,
|
||||
target: Any = None,
|
||||
e: Exception | None = None,
|
||||
):
|
||||
...
|
||||
|
||||
@classmethod
|
||||
def error(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: int | str | Session | None = None,
|
||||
group_id: int | str | None = None,
|
||||
adapter: str | None = None,
|
||||
target: Any = None,
|
||||
e: Exception | None = None,
|
||||
):
|
||||
user_id: str | None = session # type: ignore
|
||||
group_id = None
|
||||
if type(session) == Session:
|
||||
user_id = session.id1
|
||||
adapter = session.bot_type
|
||||
if session.id3 or session.id2:
|
||||
group_id = f"{session.id3}:{session.id2}"
|
||||
template = cls.__parser_template(
|
||||
info, command, user_id, group_id, adapter, target
|
||||
)
|
||||
if e:
|
||||
template += f" || 错误 <r>{type(e)}: {e}</r>"
|
||||
logger_.opt(colors=True).error(template)
|
||||
|
||||
@overload
|
||||
@classmethod
|
||||
def debug(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: int | str | None = None,
|
||||
group_id: int | str | None = None,
|
||||
adapter: str | None = None,
|
||||
target: Any = None,
|
||||
e: Exception | None = None,
|
||||
):
|
||||
...
|
||||
|
||||
@overload
|
||||
@classmethod
|
||||
def debug(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: Session | None = None,
|
||||
target: Any = None,
|
||||
e: Exception | None = None,
|
||||
):
|
||||
...
|
||||
|
||||
@classmethod
|
||||
def debug(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
*,
|
||||
session: int | str | Session | None = None,
|
||||
group_id: int | str | None = None,
|
||||
adapter: str | None = None,
|
||||
target: Any = None,
|
||||
e: Exception | None = None,
|
||||
):
|
||||
user_id: str | None = session # type: ignore
|
||||
group_id = None
|
||||
if type(session) == Session:
|
||||
user_id = session.id1
|
||||
adapter = session.bot_type
|
||||
if session.id3 or session.id2:
|
||||
group_id = f"{session.id3}:{session.id2}"
|
||||
template = cls.__parser_template(
|
||||
info, command, user_id, group_id, adapter, target
|
||||
)
|
||||
if e:
|
||||
template += f" || 错误 <r>{type(e)}: {e}</r>"
|
||||
logger_.opt(colors=True).debug(template)
|
||||
|
||||
@classmethod
|
||||
def __parser_template(
|
||||
cls,
|
||||
info: str,
|
||||
command: str | None = None,
|
||||
user_id: int | str | None = None,
|
||||
group_id: int | str | None = None,
|
||||
adapter: str | None = None,
|
||||
target: Any = None,
|
||||
) -> str:
|
||||
arg_list = []
|
||||
template = ""
|
||||
if adapter is not None:
|
||||
template += cls.TEMPLATE_ADAPTER
|
||||
arg_list.append(adapter)
|
||||
if group_id is not None:
|
||||
template += cls.TEMPLATE_GROUP
|
||||
arg_list.append(group_id)
|
||||
if user_id is not None:
|
||||
template += cls.TEMPLATE_USER
|
||||
arg_list.append(user_id)
|
||||
if command is not None:
|
||||
template += cls.TEMPLATE_COMMAND
|
||||
arg_list.append(command)
|
||||
if target is not None:
|
||||
template += cls.TEMPLATE_TARGET
|
||||
arg_list.append(target)
|
||||
arg_list.append(info)
|
||||
template += "{}"
|
||||
return template.format(*arg_list)
|
||||
Reference in New Issue
Block a user