| import yaml | |
| import os | |
| from .utils import get_files | |
| __all__ = ["load_fonts", "DSFont"] | |
| class DSFont: | |
| def __init__(self, path, language): | |
| self.path = path | |
| self.language = language | |
| def load_fonts(config_path="configs/dataset.yml") -> list[DSFont]: | |
| with open(config_path, "r", encoding="utf-8") as f: | |
| config = yaml.safe_load(f) | |
| ds_config = config["dataset"] | |
| ds_path = ds_config["path"] | |
| font_list = [] | |
| for spec in ds_config["specs"]: | |
| for spec_path in spec["path"]: | |
| spec_path = os.path.join(ds_path, spec_path) | |
| spec_files = get_files(spec_path) | |
| if spec.keys().__contains__("rule"): | |
| rule = eval(spec["rule"]) | |
| else: | |
| rule = None | |
| for file in spec_files: | |
| if rule is not None and not rule(file): | |
| print("skip: " + file) | |
| continue | |
| font_list.append(DSFont(file, spec["language"])) | |
| return font_list | |