|
1 | 1 | import sys |
2 | 2 | import time |
3 | 3 |
|
| 4 | +from loguru import logger |
| 5 | + |
4 | 6 | from system import clear |
5 | 7 | from system.lib.config import config |
6 | 8 | from system.lib.console import Console |
7 | 9 | from system.lib.features.directories import clear_directories |
8 | 10 | from system.lib.features.initialization import initialize |
9 | | -from system.lib.features.update.check import check_update, check_for_outdated, get_tags |
10 | | -from system.lib.menu import menu, Menu |
| 11 | +from system.lib.features.update.check import check_for_outdated, check_update, get_tags |
| 12 | +from system.lib.menu import Menu, menu |
11 | 13 | from system.localization import locale |
12 | 14 |
|
13 | | -from loguru import logger |
14 | | - |
15 | 15 | logger.remove() |
16 | 16 | logger.add( |
17 | 17 | "./logs/info/{time:YYYY-MM-DD}.log", |
@@ -64,99 +64,121 @@ def refill_menu(): |
64 | 64 | import sc_compression |
65 | 65 |
|
66 | 66 | del sc_compression |
67 | | - |
| 67 | + except ImportError: |
| 68 | + logger.warning(locale.install_to_unlock % "sc-compression") |
| 69 | + else: |
68 | 70 | from system.lib.features.csv.compress import compress_csv |
69 | 71 | from system.lib.features.csv.decompress import decompress_csv |
70 | 72 |
|
71 | 73 | try: |
72 | 74 | import PIL |
73 | 75 |
|
74 | 76 | del PIL |
75 | | - |
76 | | - from system.lib.features.sc.decode import sc_decode |
77 | | - from system.lib.features.sc.encode import sc_encode |
78 | | - from system.lib.features.sc.decode_and_cut import decode_and_cut |
79 | | - from system.lib.features.sc.assembly_encode import sc1_encode |
| 77 | + except ImportError: |
| 78 | + logger.warning(locale.install_to_unlock % "PILLOW") |
| 79 | + else: |
| 80 | + from system.lib.features.sc.decode import ( |
| 81 | + decode_and_render_objects, |
| 82 | + decode_textures_only, |
| 83 | + ) |
| 84 | + from system.lib.features.sc.encode import ( |
| 85 | + collect_objects_and_encode, |
| 86 | + encode_textures_only, |
| 87 | + ) |
80 | 88 |
|
81 | 89 | sc_category = Menu.Category(0, locale.sc_label) |
82 | 90 | sc_category.add( |
83 | | - Menu.Item(locale.decode_sc, locale.decode_sc_description, sc_decode) |
| 91 | + Menu.Item( |
| 92 | + name=locale.decode_sc, |
| 93 | + description=locale.decode_sc_description, |
| 94 | + handler=decode_textures_only, |
| 95 | + ) |
84 | 96 | ) |
85 | 97 | sc_category.add( |
86 | | - Menu.Item(locale.encode_sc, locale.encode_sc_description, sc_encode) |
| 98 | + Menu.Item( |
| 99 | + name=locale.encode_sc, |
| 100 | + description=locale.encode_sc_description, |
| 101 | + handler=encode_textures_only, |
| 102 | + ) |
87 | 103 | ) |
88 | 104 | sc_category.add( |
89 | 105 | Menu.Item( |
90 | | - locale.decode_by_parts, |
91 | | - locale.decode_by_parts_description, |
92 | | - decode_and_cut, |
| 106 | + name=locale.decode_by_parts, |
| 107 | + description=locale.decode_by_parts_description, |
| 108 | + handler=decode_and_render_objects, |
93 | 109 | ) |
94 | 110 | ) |
95 | 111 | sc_category.add( |
96 | 112 | Menu.Item( |
97 | | - locale.encode_by_parts, |
98 | | - locale.encode_by_parts_description, |
99 | | - sc1_encode, |
| 113 | + name=locale.encode_by_parts, |
| 114 | + description=locale.encode_by_parts_description, |
| 115 | + handler=collect_objects_and_encode, |
100 | 116 | ) |
101 | 117 | ) |
102 | 118 | sc_category.add( |
103 | 119 | Menu.Item( |
104 | | - locale.overwrite_by_parts, |
105 | | - locale.overwrite_by_parts_description, |
106 | | - lambda: sc1_encode(True), |
| 120 | + name=locale.overwrite_by_parts, |
| 121 | + description=locale.overwrite_by_parts_description, |
| 122 | + handler=lambda: collect_objects_and_encode(True), |
107 | 123 | ) |
108 | 124 | ) |
109 | 125 | menu.add_category(sc_category) |
110 | | - except ImportError: |
111 | | - logger.warning(locale.install_to_unlock % "PILLOW") |
112 | 126 |
|
113 | 127 | csv_category = Menu.Category(1, locale.csv_label) |
114 | 128 | csv_category.add( |
115 | 129 | Menu.Item( |
116 | | - locale.decompress_csv, locale.decompress_csv_description, decompress_csv |
| 130 | + name=locale.decompress_csv, |
| 131 | + description=locale.decompress_csv_description, |
| 132 | + handler=decompress_csv, |
117 | 133 | ) |
118 | 134 | ) |
119 | 135 | csv_category.add( |
120 | 136 | Menu.Item( |
121 | | - locale.compress_csv, locale.compress_csv_description, compress_csv |
| 137 | + name=locale.compress_csv, |
| 138 | + description=locale.compress_csv_description, |
| 139 | + handler=compress_csv, |
122 | 140 | ) |
123 | 141 | ) |
124 | 142 | menu.add_category(csv_category) |
125 | | - except ImportError: |
126 | | - logger.warning(locale.install_to_unlock % "sc-compression") |
127 | 143 |
|
128 | 144 | other = Menu.Category(10, locale.other_features_label) |
129 | 145 | other.add( |
130 | | - Menu.Item(locale.check_update, locale.version % config.version, check_update) |
| 146 | + Menu.Item( |
| 147 | + name=locale.check_update, |
| 148 | + description=locale.version % config.version, |
| 149 | + handler=check_update, |
| 150 | + ) |
131 | 151 | ) |
132 | | - other.add(Menu.Item(locale.check_for_outdated, None, check_for_outdated)) |
| 152 | + other.add(Menu.Item(name=locale.check_for_outdated, handler=check_for_outdated)) |
133 | 153 | other.add( |
134 | 154 | Menu.Item( |
135 | | - locale.reinit, |
136 | | - locale.reinit_description, |
137 | | - lambda: (initialize(), refill_menu()), |
| 155 | + name=locale.reinit, |
| 156 | + description=locale.reinit_description, |
| 157 | + handler=lambda: (initialize(), refill_menu()), |
138 | 158 | ) |
139 | 159 | ) |
140 | 160 | other.add( |
141 | 161 | Menu.Item( |
142 | | - locale.change_language, |
143 | | - locale.change_lang_description % config.language, |
144 | | - lambda: (config.change_language(locale.change()), refill_menu()), |
| 162 | + name=locale.change_language, |
| 163 | + description=locale.change_lang_description % config.language, |
| 164 | + handler=lambda: (config.change_language(locale.change()), refill_menu()), |
145 | 165 | ) |
146 | 166 | ) |
147 | 167 | other.add( |
148 | 168 | Menu.Item( |
149 | | - locale.clear_directories, |
150 | | - locale.clean_dirs_description, |
151 | | - lambda: clear_directories() if Console.question(locale.clear_qu) else -1, |
| 169 | + name=locale.clear_directories, |
| 170 | + description=locale.clean_dirs_description, |
| 171 | + handler=lambda: clear_directories() |
| 172 | + if Console.question(locale.clear_qu) |
| 173 | + else -1, |
152 | 174 | ) |
153 | 175 | ) |
154 | 176 | other.add( |
155 | 177 | Menu.Item( |
156 | | - locale.toggle_update_auto_checking, |
157 | | - locale.enabled if config.auto_update else locale.disabled, |
158 | | - lambda: (config.toggle_auto_update(), refill_menu()), |
| 178 | + name=locale.toggle_update_auto_checking, |
| 179 | + description=locale.enabled if config.auto_update else locale.disabled, |
| 180 | + handler=lambda: (config.toggle_auto_update(), refill_menu()), |
159 | 181 | ) |
160 | 182 | ) |
161 | | - other.add(Menu.Item(locale.exit, None, lambda: (clear(), exit()))) |
| 183 | + other.add(Menu.Item(name=locale.exit, handler=lambda: (clear(), exit()))) |
162 | 184 | menu.add_category(other) |
0 commit comments