注册自定义生成器 Args: name: 生成器名称 generator_class: 生成器类
(cls, name: str, generator_class: type)
| 48 | |
| 49 | @classmethod |
| 50 | def register_generator(cls, name: str, generator_class: type): |
| 51 | """ |
| 52 | 注册自定义生成器 |
| 53 | |
| 54 | Args: |
| 55 | name: 生成器名称 |
| 56 | generator_class: 生成器类 |
| 57 | """ |
| 58 | if not issubclass(generator_class, ImageGeneratorBase): |
| 59 | raise TypeError( |
| 60 | f"注册失败:生成器类必须继承自 ImageGeneratorBase。\n" |
| 61 | f"提供的类: {generator_class.__name__}\n" |
| 62 | f"基类: ImageGeneratorBase" |
| 63 | ) |
| 64 | |
| 65 | cls.GENERATORS[name] = generator_class |
nothing calls this directly
no outgoing calls
no test coverage detected