Spaces:
Runtime error
Runtime error
File size: 4,160 Bytes
970efde |
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 78 79 80 81 82 83 84 85 86 87 |
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, {})
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)
|