From 7f997f04fb2acc0a7f535f4522ccec22905de859 Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Sat, 27 Jan 2024 19:06:04 +0800 Subject: :recycle: refactor(generator): rename Generator to TextGenerator and add TODO --- src/infini/generator.py | 2 +- src/infini/loader.py | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/infini/generator.py b/src/infini/generator.py index 3f386010..cc752d70 100644 --- a/src/infini/generator.py +++ b/src/infini/generator.py @@ -4,7 +4,7 @@ from infini.exceptions import UnknownEvent from jinja2 import Template -class Generator: +class TextGenerator: # TODO 兼容多类型事件 events: Dict[str, str] global_variables: Dict[str, str | Callable] diff --git a/src/infini/loader.py b/src/infini/loader.py index 1dbe7779..ba7eb546 100644 --- a/src/infini/loader.py +++ b/src/infini/loader.py @@ -1,6 +1,6 @@ from importlib.util import spec_from_file_location from infini.core import Core -from infini.generator import Generator +from infini.generator import TextGenerator from infini.handler import Handler from infini.interceptor import Interceptor from infini.register import Register @@ -147,7 +147,7 @@ class Loader: def inject_core(self, core: Core): pre_interceptor = Interceptor() handler = Handler() - generator = Generator() + generator = TextGenerator() interceptor = Interceptor() self.inject_pre_interceptor(pre_interceptor) @@ -192,12 +192,12 @@ class Loader: self.inject_handler(handler) return handler - def inject_generator(self, generator: Generator): + def inject_generator(self, generator: TextGenerator): generator.events = self.events generator.global_variables = self.global_variables - def into_generator(self) -> Generator: - generator = Generator() + def into_generator(self) -> TextGenerator: + generator = TextGenerator() self.inject_generator(generator) return generator -- cgit v1.2.3-70-g09d2