Update: Add Arena Component Number Setting GUI

pur1fying · Aug 24, 2024 · 922acbf · 922acbf
1 parent a15e2dd
commit 922acbf
Showing 1 changed file with 21 additions and 2 deletions.
diff --git a/gui/components/expand/arenaPriority.py b/gui/components/expand/arenaPriority.py
@@ -1,7 +1,7 @@
 from PyQt5.QtCore import Qt
 from PyQt5.QtGui import QIntValidator
 from PyQt5.QtWidgets import QWidget, QHBoxLayout, QLabel, QPushButton, QVBoxLayout
-from qfluentwidgets import LineEdit
+from qfluentwidgets import LineEdit, ComboBox
 
 from gui.util import notification
 
@@ -10,17 +10,21 @@ class Layout(QWidget):
     def __init__(self, parent=None, config=None):
         super().__init__(parent=parent)
         self.config = config
-        self.setFixedHeight(120)
+        self.setFixedHeight(150)
         self.info_widget = self.parent()
         self.hBoxLayout = QVBoxLayout(self)
         self.lay_1 = QHBoxLayout(self)
         self.lay_2 = QHBoxLayout(self)
+        self.lay_3 = QHBoxLayout(self)
         self.label_1 = QLabel(self.tr('输入你需要对手比你低几级，高几级则填负数：'), self)
         self.label_2 = QLabel(self.tr('输入你最多需要刷新几次：'), self)
+        self.label_3 = QLabel(self.tr('自定义对手的编号：'), self)
         self.input_1 = LineEdit(self)
         self.input_2 = LineEdit(self)
+        self.input_3 = ComboBox(self)
         self.accept_1 = QPushButton(self.tr('确定'), self)
         self.accept_2 = QPushButton(self.tr('确定'), self)
+        self.accept_3 = QPushButton(self.tr('确定'), self)
 
         self.level_diff = self.config.get('ArenaLevelDiff')
         validator_1 = QIntValidator(-50, 50)
@@ -33,8 +37,13 @@ def __init__(self, parent=None, config=None):
         self.input_2.setText(str(self.refresh_times))
         self.input_2.setValidator(validator_2)
 
+        self.arena_component_number = self.config.get('ArenaComponentNumber')
+        self.input_3.addItems([str(i) for i in range(1, 4)])
+        self.input_3.setCurrentIndex(int(self.arena_component_number) - 1)
+
         self.accept_1.clicked.connect(self.__accept_1)
         self.accept_2.clicked.connect(self.__accept_2)
+        self.accept_3.clicked.connect(self.__accept_3)
 
         self.lay_1.addWidget(self.label_1, 20, Qt.AlignLeft)
         self.lay_1.addWidget(self.input_1, 0, Qt.AlignRight)
@@ -44,12 +53,17 @@ def __init__(self, parent=None, config=None):
         self.lay_2.addWidget(self.input_2, 0, Qt.AlignRight)
         self.lay_2.addWidget(self.accept_2, 0, Qt.AlignCenter)
 
+        self.lay_3.addWidget(self.label_3, 20, Qt.AlignLeft)
+        self.lay_3.addWidget(self.input_3, 0, Qt.AlignRight)
+        self.lay_3.addWidget(self.accept_3, 0, Qt.AlignCenter)
+
         self.hBoxLayout.setContentsMargins(24, 0, 24, 0)
         self.hBoxLayout.addSpacing(16)
         self.hBoxLayout.setAlignment(Qt.AlignCenter)
 
         self.hBoxLayout.addLayout(self.lay_1)
         self.hBoxLayout.addLayout(self.lay_2)
+        self.hBoxLayout.addLayout(self.lay_3)
 
     def __accept_1(self):
         self.level_diff = int(self.input_1.text())
@@ -60,3 +74,8 @@ def __accept_2(self, _=None):
         self.refresh_times = int(self.input_2.text())
         self.config.set('maxArenaRefreshTimes', self.refresh_times)
         notification.success(self.tr('设置成功'), f'{self.tr("你最大刷新次数设置为：")}{self.refresh_times}', self.config)
+
+    def __accept_3(self, _=None):
+        self.arena_component_number = int(self.input_3.currentText())
+        self.config.set('arenaComponentNumber', self.arena_component_number)
+        notification.success(self.tr('设置成功'), f'{self.tr("你的对手编号设置为：")}{self.arena_component_number}', self.config)