| 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/font.yml"): | |
| 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"])) | |
| font_list.sort(key=lambda x: x.path) | |
| exclusion_list = ds_config["exclusion"] | |
| exclusion_list = [os.path.join(ds_path, path) for path in exclusion_list] | |
| def exclusion_rule(font: DSFont): | |
| for exclusion in exclusion_list: | |
| if os.path.samefile(font.path, exclusion): | |
| return True | |
| return False | |
| return font_list, exclusion_rule | |