File size: 1,366 Bytes
61d733d
 
2046da1
61d733d
 
 
 
10fb84b
61d733d
 
 
 
 
 
 
 
a9b34cd
10fb84b
61d733d
 
10fb84b
 
61d733d
 
 
10fb84b
 
61d733d
 
 
10fb84b
 
61d733d
 
 
 
 
10fb84b
61d733d
10fb84b
61d733d
cf70bb7
a9b34cd
 
 
 
 
 
 
 
 
 
 
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
44
45
46
47
48
49
50
51
52
53
54
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