Spaces:
Running on Zero
Running on Zero
| from pathlib import Path | |
| from typing import Optional, Union | |
| import toml | |
| from pydantic import BaseModel | |
| from .compat import model_dump, type_validate_python | |
| from .dirs import get_config_file | |
| config_file_path = get_config_file("config.toml") | |
| class MemeConfig(BaseModel): | |
| load_builtin_memes: bool = True | |
| meme_dirs: list[Path] = [] | |
| meme_disabled_list: list[str] = [] | |
| class ResourceConfig(BaseModel): | |
| resource_url: Optional[str] = None | |
| resource_urls: list[str] = [ | |
| "https://raw.githubusercontent.com/ClarityJS/meme-generator/", | |
| "https://mirror.ghproxy.com/https://raw.githubusercontent.com/ClarityJS/meme-generator/", | |
| "https://cdn.jsdelivr.net/gh/ClarityJS/meme-generator@", | |
| "https://fastly.jsdelivr.net/gh/ClarityJS/meme-generator@", | |
| "https://raw.gitmirror.com/ClarityJS/meme-generator/", | |
| ] | |
| class GifConfig(BaseModel): | |
| gif_max_size: float = 10 | |
| gif_max_frames: int = 100 | |
| # 已由大佬提供翻译相关技术，无需配置了 | |
| # class TranslatorConfig(BaseModel): | |
| # baidu_trans_appid: str = "" | |
| # baidu_trans_apikey: str = "" | |
| class ServerConfig(BaseModel): | |
| host: str = "127.0.0.1" | |
| port: int = 2233 | |
| class LogConfig(BaseModel): | |
| log_level: Union[int, str] = "INFO" | |
| class Config(BaseModel): | |
| meme: MemeConfig = MemeConfig() | |
| resource: ResourceConfig = ResourceConfig() | |
| gif: GifConfig = GifConfig() | |
| # translate: TranslatorConfig = TranslatorConfig() | |
| server: ServerConfig = ServerConfig() | |
| log: LogConfig = LogConfig() | |
| def load(cls) -> "Config": | |
| return type_validate_python(cls, toml.load(config_file_path)) | |
| def dump(self): | |
| with open(config_file_path, "w", encoding="utf-8") as f: | |
| toml.dump(model_dump(self), f) | |
| if not config_file_path.exists(): | |
| meme_config = Config() | |
| config_file_path.write_text("", encoding="utf8") | |
| else: | |
| meme_config = Config.load() | |