diff options
| -rw-r--r-- | src/infini/generator.py | 2 | ||||
| -rw-r--r-- | 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 |