class Setting(QWidget, Ui_widget_setting):
signal_core_number = pyqtSignal(int)
signal_suspend_status = pyqtSignal(bool)
signal_waiting_min = pyqtSignal(int)
signal_schedule_status = pyqtSignal(bool)
signal_cancel_plan = pyqtSignal(str)
signal_change_language = pyqtSignal(str)
def __init__(self, suspend, cores, schedule_status, waiting_min, language):
super().__init__()
self.setupUi(self)
# ----------init variable------------
self.suspend = suspend
self.cores = cores
self.schedule_status = schedule_status
self.waiting_min = waiting_min
self.language = language
self.language_list = ['English', 'Chinese']
# ----------init widget--------------
self.label_suspend = QTreeWidgetItem()
self.checkbox_suspend = QCheckBox()
self.label_cores = QTreeWidgetItem()
self.edit_cores = QSpinBox()
self.label_plan = QTreeWidgetItem()
self.edit_plan_datetime = QDateTimeEdit()
self.label_about = QTreeWidgetItem()
self.label_language = QTreeWidgetItem()
self.combobox_lauguage = QComboBox()
# -----------init function----------------
self.ui_set()
self.btn()
self.init_data_show()
self.self_translate(language)
self.show()
def btn(self):
self.tree_setting.itemClicked.connect(self.effect_expand)
self.tree_setting.itemChanged.connect(self.enable_schedule)
self.combobox_lauguage.activated.connect(self.choose_language)
self.checkbox_suspend.stateChanged.connect(self.change_suspend_status)
def ui_set(self):
# style
self.setStyleSheet(
"QSpinBox{border:1.5px solid #778899;border-radius:4px; padding:2px 2px}"
"QDateTimeEdit{border:1.5px solid #778899;border-radius:4px; padding:2px 2px}"
)
self.tree_setting.setColumnWidth(0, 180)
self.tree_setting.expandItem(self.tree_setting.topLevelItem(0))
self.tree_setting.topLevelItem(1).setCheckState(1, 0)
self.edit_cores.setMaximumSize(50, 25)
self.edit_cores.setContentsMargins(0, 4, 0, 0)
self.edit_cores.setMinimum(1)
self.edit_cores.setMaximum(132)
self.label_plan.setDisabled(True)
self.edit_plan_datetime.setMaximumSize(135, 28)
self.edit_plan_datetime.setDisabled(True)
self.combobox_lauguage.setMaximumSize(135, 28)
self.combobox_lauguage.addItems(self.language_list)
self.combobox_lauguage.setCurrentText(self.language)
# function
self.add_tree_item(0, self.label_suspend, "暂停队列",
self.checkbox_suspend)
self.add_tree_item(0, self.label_cores, "使用核数", self.edit_cores)
self.add_tree_item(1, self.label_plan, "计划启动于",
self.edit_plan_datetime)
self.add_tree_item(2, self.label_language, "语言选择",
self.combobox_lauguage)
def add_tree_item(self, top_level_index, label, label_name, input_edit):
label.setText(0, label_name)
self.tree_setting.topLevelItem(top_level_index).addChild(label)
self.tree_setting.setItemWidget(label, 1, input_edit)
def effect_expand(self, item, column):
index = self.tree_setting.indexOfTopLevelItem(item)
if index >= 0:
if item.isExpanded():
item.setExpanded(False)
else:
item.setExpanded(True)
def enable_schedule(self, item, column):
index = self.tree_setting.indexOfTopLevelItem(item)
if index == 1:
check_state = item.checkState(column)
self.schedule_status = bool(check_state)
self.label_plan.setDisabled(2 - check_state)
self.edit_plan_datetime.setEnabled(check_state)
item.setExpanded(2 - check_state)
self.reset_date_edit()
self.suspend = self.checkbox_suspend.checkState()
self.checkbox_suspend.setCheckState(self.suspend + check_state)
self.checkbox_suspend.setDisabled(check_state)
def init_data_show(self):
self.edit_cores.setValue(self.cores)
self.checkbox_suspend.setCheckState(self.suspend * 2)
if self.schedule_status:
self.tree_setting.topLevelItem(1).setCheckState(1, 2)
self.label_plan.setDisabled(False)
self.edit_plan_datetime.setEnabled(True)
self.tree_setting.topLevelItem(1).setExpanded(True)
waiting_seconds = self.waiting_min * 60
self.edit_plan_datetime.setDisplayFormat("yyyy/MM/dd HH:mm")
self.edit_plan_datetime.setDateTime(
QDateTime.currentDateTime().addSecs(waiting_seconds))
else:
self.reset_date_edit()
def reset_date_edit(self):
self.edit_plan_datetime.setDisplayFormat("yyyy/MM/dd HH:mm")
self.edit_plan_datetime.setDateTime(QDateTime.currentDateTime())
self.edit_plan_datetime.setMinimumDateTime(QDateTime.currentDateTime())
self.edit_plan_datetime.setMaximumDateTime(
QDateTime.currentDateTime().addDays(5))
self.edit_plan_datetime.setCalendarPopup(True)
def change_suspend_status(self, status):
self.suspend = bool(status)
def plan_start(self):
curr_time = QDateTime.currentSecsSinceEpoch()
schedule_time = self.edit_plan_datetime.dateTime().toSecsSinceEpoch()
self.waiting_min = int(round((schedule_time - curr_time) / 60, 0))
print('waiting min', self.waiting_min)
self.signal_waiting_min.emit(self.waiting_min)
def choose_language(self):
language = self.combobox_lauguage.currentText()
self.signal_change_language.emit(language)
self.self_translate(language)
def self_translate(self, language):
if language == 'English':
self.label_suspend.setText(0, 'Suspend next')
self.label_cores.setText(0, 'Threads number')
self.label_plan.setText(0, 'Scheduled in')
self.label_language.setText(0, 'Language')
else:
self.label_suspend.setText(0, "暂停队列")
self.label_cores.setText(0, "使用核数")
self.label_plan.setText(0, "计划启动于")
self.label_language.setText(0, "语言选择")
self.retranslateUi(self)
def closeEvent(self, event):
self.cores = self.edit_cores.value()
suspend_status = self.checkbox_suspend.checkState()
self.signal_core_number.emit(self.cores)
self.signal_schedule_status.emit(self.schedule_status)
if self.schedule_status:
self.plan_start()
else:
self.signal_cancel_plan.emit(' ')
self.signal_suspend_status.emit(bool(suspend_status))
self.close()
