aboutsummaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
author苏向夜 <fu050409@163.com>2024-01-27 16:55:34 +0800
committer苏向夜 <fu050409@163.com>2024-01-27 16:55:34 +0800
commit7e72f3fd2700b2447cb8278f5cf530785ca95197 (patch)
treed12d98bf5fd73ce5f0174fdead0a1f5253be0679
parent9a11c90db9a80769275fb4de8313ff6dcfe760ae (diff)
downloadinfini-7e72f3fd2700b2447cb8278f5cf530785ca95197.tar.gz
infini-7e72f3fd2700b2447cb8278f5cf530785ca95197.zip
:sparkles: feat(logging): add logging module
-rw-r--r--src/infini/loader.py6
-rw-r--r--src/infini/logging.py23
2 files changed, 27 insertions, 2 deletions
diff --git a/src/infini/loader.py b/src/infini/loader.py
index 7d6bbd61..1dbe7779 100644
--- a/src/infini/loader.py
+++ b/src/infini/loader.py
@@ -5,6 +5,7 @@ from infini.handler import Handler
from infini.interceptor import Interceptor
from infini.register import Register
from infini.typing import List, Dict, Sequence, ModuleType, RouterType, Callable
+from infini.logging import logger
from pathlib import Path
import inspect
@@ -123,8 +124,9 @@ class Loader:
registers = self._find_register_variables(module)
self.load_from_registers(registers)
if not registers:
- # TODO 警告内容
- ...
+ logger.warning(
+ f"Infini 装载器未能在规则包 [bold green]{module.__name__}[/bold green] 中找到注册器."
+ )
return module
diff --git a/src/infini/logging.py b/src/infini/logging.py
new file mode 100644
index 00000000..df4f818d
--- /dev/null
+++ b/src/infini/logging.py
@@ -0,0 +1,23 @@
+from rich.logging import RichHandler
+from rich.console import Console
+
+import logging
+
+console = Console()
+
+
+logging.basicConfig(
+ level="INFO",
+ format="%(message)s",
+ handlers=[
+ RichHandler(
+ console=console,
+ show_time=False,
+ show_path=False,
+ show_level=True,
+ markup=True,
+ )
+ ],
+)
+
+logger = logging.getLogger("rich")