File size: 1,057 Bytes
61d733d 10fb84b 61d733d 89ec64c 10fb84b 61d733d 10fb84b 61d733d 10fb84b 61d733d 10fb84b 61d733d 10fb84b 61d733d 10fb84b 61d733d cf70bb7 61d733d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
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") -> 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"]))
font_list.sort(key=lambda x: x.path)
return font_list
|