Spaces:
Runtime error
Runtime error
| import json | |
| import os | |
| from qt.components.config import ConfigWidget | |
| from qt.components.equipments import EquipmentsWidget | |
| from utils.parser import Parser | |
| if not os.path.exists("config"): | |
| CONFIG = {} | |
| else: | |
| CONFIG = json.load(open("config", encoding="utf-8")) | |
| def config_script(parser: Parser, config_widget: ConfigWidget, equipments_widget: EquipmentsWidget): | |
| def load_config(): | |
| config_name = config_widget.config_select.combo_box.currentText() | |
| config = CONFIG.get(parser.school.school, {}).get(config_name, {}) | |
| if not config: | |
| return | |
| for label, equipment in equipments_widget.items(): | |
| if 'equipment' not in config[label]: | |
| continue | |
| else: | |
| index = equipment.equipment.combo_box.findText(config[label]['equipment']) | |
| equipment.equipment.combo_box.setCurrentIndex(index) | |
| equipment.strength_level.combo_box.setCurrentIndex(config[label]['strength_level']) | |
| if 'enchant' in config[label]: | |
| index = equipment.enchant.combo_box.findText(config[label]['enchant']) | |
| equipment.enchant.combo_box.setCurrentIndex(index) | |
| if 'special_enchant' in config[label]: | |
| if equipment.special_enchant.radio_button.isChecked() != config[label]['special_enchant']: | |
| equipment.special_enchant.radio_button.click() | |
| if 'embed_levels' in config[label]: | |
| for i, embed_level in enumerate(equipment.embed_levels): | |
| embed_level.combo_box.setCurrentIndex(config[label]['embed_levels'][i]) | |
| if 'stone_level' in config[label]: | |
| equipment.stone_level.combo_box.setCurrentIndex(config[label]['stone_level']) | |
| if 'stone_attrs' in config[label]: | |
| for i, stone_attr in enumerate(equipment.stone_attrs): | |
| index = equipment.stone_attrs[i].combo_box.findText(config[label]['stone_attrs'][i]) | |
| stone_attr.combo_box.setCurrentIndex(index) | |
| config_widget.config_name.text_browser.setText(config_name) | |
| config_widget.load_config.clicked.connect(load_config) | |
| def save_config(): | |
| config_name = config_widget.config_name.text_browser.text() | |
| if parser.school.school not in CONFIG: | |
| CONFIG[parser.school.school] = {} | |
| if config_name not in CONFIG[parser.school.school]: | |
| CONFIG[parser.school.school][config_name] = {} | |
| config = CONFIG[parser.school.school][config_name] | |
| for label, equipment in equipments_widget.items(): | |
| config[label] = {} | |
| if not (text := equipment.equipment.combo_box.currentText()): | |
| continue | |
| else: | |
| config[label]['equipment'] = text | |
| config[label]['strength_level'] = equipment.strength_level.combo_box.currentIndex() | |
| if equipment.enchant: | |
| config[label]['enchant'] = equipment.enchant.combo_box.currentText() | |
| if equipment.special_enchant: | |
| config[label]['special_enchant'] = equipment.special_enchant.radio_button.isChecked() | |
| if equipment.embed_levels: | |
| config[label]['embed_levels'] = [ | |
| embed_level.combo_box.currentIndex() for embed_level in equipment.embed_levels | |
| ] | |
| if equipment.stone_level: | |
| config[label]['stone_level'] = equipment.stone_level.combo_box.currentIndex() | |
| if equipment.stone_attrs: | |
| config[label]['stone_attrs'] = [ | |
| stone_attr.combo_box.currentText() for stone_attr in equipment.stone_attrs | |
| ] | |
| json.dump(CONFIG, open("config", "w", encoding="utf-8"), ensure_ascii=False) | |
| config_choices = list(CONFIG.get(parser.school.school, {})) | |
| if current_select := config_widget.config_select.combo_box.currentText(): | |
| default_index = config_choices.index(current_select) | |
| else: | |
| default_index = -1 | |
| config_widget.config_select.set_items(config_choices, default_index=default_index) | |
| config_widget.save_config.clicked.connect(save_config) | |