From 7e72f3fd2700b2447cb8278f5cf530785ca95197 Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Sat, 27 Jan 2024 16:55:34 +0800 Subject: :sparkles: feat(logging): add logging module --- src/infini/loader.py | 6 ++++-- src/infini/logging.py | 23 +++++++++++++++++++++++ 2 files changed, 27 insertions(+), 2 deletions(-) create mode 100644 src/infini/logging.py 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") -- cgit v1.2.3-70-g09d2