TrollRecorder/devkit/i18n_statistics.py at main · Lessica/TrollRecorder · GitHub

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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
#!/usr/bin/env python3

import matplotlib.pyplot as plt
import pathlib
import plistlib


lproj_base = pathlib.Path('res')

available_languages = [
    'en',
    'es',
    'fr',
    'ko',
    'ru',
    'tr',
    'ug-CN',
    'uk',
    'vi',
    'zh-Hans',
    'zh-Hant-HK',
    'zh-Hant-TW',
]

def get_lproj_path(lang) -> pathlib.Path:
    return lproj_base / f'{lang}.lproj'

def get_lproj_strings_paths(lang) -> list[pathlib.Path]:
    return [
        get_lproj_path(lang) / 'Localizable.strings',
        get_lproj_path(lang) / 'InfoPlist.strings',
    ]

def get_lproj_strings_count(lang) -> tuple[int, int]:
    all_lines = []
    for strings_path in get_lproj_strings_paths(lang):
        if not strings_path.exists():
            continue
        with open(strings_path, 'r') as f:
            all_lines.extend(f.readlines())
    todo_count = 0
    for line in all_lines:
        if line.find('/* TODO */') != -1 or line.find('/* Translated with ') != -1:
            todo_count += 1
    return int((len(all_lines) + 1) / 3), todo_count

def get_lproj_stringsdict_paths(lang) -> list[pathlib.Path]:
    return [
        get_lproj_path(lang) / 'Localizable.stringsdict',
    ]

def get_lproj_stringsdict_count(lang) -> int:
    all_count = 0
    for stringsdict_path in get_lproj_stringsdict_paths(lang):
        if not stringsdict_path.exists():
            continue
        stringsdict = plistlib.load(open(stringsdict_path, 'rb'))
        all_count += len(stringsdict)
    return all_count

if __name__ == '__main__':
    x_labels = available_languages
    y_values = []
    for lang in available_languages:
        total, todo = get_lproj_strings_count(lang)
        total += get_lproj_stringsdict_count(lang)
        percent = (total - todo) / total
        y_values.append((lang, percent, total - todo, total))
        print(f'{lang}: {total - todo}/{total} translated, {percent * 100:.2f}%')
    y_values.sort(key=lambda x: 1.0 - x[1])
    y_values.reverse()
    plt.figure().set_figwidth(15)
    plt.barh([x[0] for x in y_values], [x[2] for x in y_values])
    plt.xlabel('Strings Translated')
    plt.ylabel('Language')
    plt.title('Translation Progress')
    plt.savefig(lproj_base / 'stats.png')