class Form(QFrame): def __init__(self, master=None, *campos, **kwargs): super().__init__(master) self.formlayout = QFormLayout() self.formulario = self.form() self.mainLayout = QVBoxLayout() self.mainLayout.addLayout(self.formlayout) self.setLayout(self.mainLayout) self.setWindowTitle(self.tr("Agenda")) self.rowid = 0 def form(self): self.nameEdit = QLineEdit() self.emailEdit = QLineEdit() self.phoneEdit = QLineEdit() self.phoneEdit.setInputMask("(00)0000-0000") self.dtanascEdit = QDateEdit() self.formlayout.addRow("&Nome",self.nameEdit) self.formlayout.addRow("&Email", self.emailEdit) self.formlayout.addRow("&Celular", self.phoneEdit) self.formlayout.addRow("&Data Nasc.", self.dtanascEdit) def lerdetalhes(self, index): records = dataset.selectbyid(model, index) for row in records: self.rowid = row.id self.nameEdit.setText(row.nomeperson) self.emailEdit.setText(row.emailperson) self.phoneEdit.setText(row.telefoneperson) self.dtanascEdit.setDate(row.dtanascimentoperson) def limparformulario(self): self.rowid = 0 self.nameEdit.setText('') self.nameEdit.setFocus() self.emailEdit.setText('') self.phoneEdit.setText('') self.dtanascEdit.setDate(QDate.currentDate()) def cancel(self): self.newBtn.setEnabled(True) def pegavalores(self): nnome = self.nameEdit.text() eemail = self.emailEdit.text() pphone = self.phoneEdit.text() ddtanascimento = self.dtanascEdit.text() if self.rowid == 0: values = {'nomeperson': nnome, 'emailperson': eemail, 'telefoneperson': pphone, 'dtanascimentoperson': ddtanascimento} else: values = {'id': self.rowid,'nomeperson': nnome, 'emailperson': eemail, 'telefoneperson': pphone, 'dtanascimentoperson': ddtanascimento} return values
class VentanaAdministradorDePersonas(QFrame): def __init__(self, parent=None): super(VentanaAdministradorDePersonas, self).__init__(parent) raiz.setWindowTitle("Administracion de personas") self.operacionesConPersonas = OperacionesConPersonas() self.instanciarVentana() def instanciarVentana(self): #Contenedor De todas las cosas de arriba a abajo layoutGeneral = QVBoxLayout() #Cosas de Derecha layoutDerecha = QVBoxLayout() #Ordena cosas de arriba a abajo layoutDerecha.setMargin(20) self.btnAgregarPersona = QPushButton("Agregar persona") self.btnAgregarPersona.setToolTip("Toma los datos ") self.btnAgregarPersona.clicked.connect(self.agregarPersona) layoutDerecha.addWidget(self.btnAgregarPersona) self.btnBuscarPersona = QPushButton("Buscar persona") self.btnBuscarPersona.setToolTip("Busca por cedula a una persona") self.btnBuscarPersona.clicked.connect(self.buscarPersona) layoutDerecha.addWidget(self.btnBuscarPersona) self.btnEliminarPersona = QPushButton("Eliminar persona") self.btnEliminarPersona.setToolTip( "Elimina a la persona que esta en busqueda") self.btnEliminarPersona.clicked.connect(self.eliminarPersona) layoutDerecha.addWidget(self.btnEliminarPersona) self.btnActualizarPersona = QPushButton("Actualizar persona") self.btnActualizarPersona.setToolTip( "Cambia los datos de la persona seleccionada") self.btnActualizarPersona.clicked.connect(self.actualizarPersona) layoutDerecha.addWidget(self.btnActualizarPersona) self.btnLimpiarTexto = QPushButton("Limpiar") self.btnLimpiarTexto.setToolTip( "Limpia los campos de texto para ingresar datos") self.btnLimpiarTexto.clicked.connect(self.limpiarCampos) layoutDerecha.addWidget(self.btnLimpiarTexto) self.btnVolverAlMenu = QPushButton("Volver") self.btnVolverAlMenu.setToolTip("Vuelve al menu principal") self.btnVolverAlMenu.clicked.connect(self.volverAlMenu) layoutDerecha.addWidget(self.btnVolverAlMenu) #Cosas de Izquierda layoutIzquierda = QVBoxLayout() #ordena cosas de arriba a abajo self.lblDNI = QLabel("DNI: ") layoutIzquierda.addWidget(self.lblDNI) self.DNI = QLineEdit() layoutIzquierda.addWidget(self.DNI) self.lblPrimerNombre = QLabel("Primer nombre: ") layoutIzquierda.addWidget(self.lblPrimerNombre) self.PrimerNombre = QLineEdit() layoutIzquierda.addWidget(self.PrimerNombre) self.lblSegundoNombre = QLabel("Segundo nombre: ") layoutIzquierda.addWidget(self.lblSegundoNombre) self.SegundoNombre = QLineEdit() layoutIzquierda.addWidget(self.SegundoNombre) self.lblPrimerApellido = QLabel("Primer apellido: ") layoutIzquierda.addWidget(self.lblPrimerApellido) self.PrimerApellido = QLineEdit() layoutIzquierda.addWidget(self.PrimerApellido) self.lblSegundoApellido = QLabel("Segundo apellido: ") layoutIzquierda.addWidget(self.lblSegundoApellido) self.SegundoApellido = QLineEdit() layoutIzquierda.addWidget(self.SegundoApellido) self.lblFechaDeNacimiento = QLabel("Fecha de nacimiento") layoutIzquierda.addWidget(self.lblFechaDeNacimiento) self.fechaDeNacimiento = QDateEdit() self.fechaDeNacimiento.setDisplayFormat("yyyy-MM-dd") self.fechaDeNacimiento.setCalendarPopup(True) self.fechaDeNacimiento.setDate(date.today()) layoutIzquierda.addWidget(self.fechaDeNacimiento) #Cosas de layout de arriba layoutSuperior = QHBoxLayout() #Ordena cosas de derecha a izquierda layoutSuperior.addLayout(layoutIzquierda) layoutSuperior.addLayout(layoutDerecha) #Cosas layout de abajo layoutInferior = QVBoxLayout() self.mostrarError = QPlainTextEdit(readOnly=True) layoutInferior.addWidget(self.mostrarError) #cosas del layout general layoutGeneral.addLayout(layoutSuperior) layoutGeneral.addLayout(layoutInferior) self.setLayout(layoutGeneral) #Redimensionar raiz.adjustSize() self.adjustSize() def agregarPersona(self): self.limpiarErrores() try: laPersona = self.llenarPersona() salida = self.operacionesConPersonas.insertarPersona(laPersona) self.mostrarError.insertPlainText(str(salida)) self.limpiarCampos( ) #habilitamos la edicion y evitamos dejar datos basura except Exception as e: self.mostrarError.insertPlainText(''.join(e.args)) pass def buscarPersona(self): self.limpiarErrores() try: laPersona = self.operacionesConPersonas.buscarPersonaPorDNI( int(''.join(self.DNI.text()))) if type(laPersona) == type(""): self.mostrarError.insertPlainText(''.join(laPersona)) return None self.llenarCampos(laPersona) except Exception as e: self.mostrarError.insertPlainText(''.join(e.args)) def eliminarPersona(self): self.limpiarErrores() try: laPersona = self.llenarPersona() mensajeDeConfirmacion = "DNI: " + self.DNI.text() opcionElegida = QMessageBox.question( self, "Desea eliminar a la persona?", mensajeDeConfirmacion, QMessageBox.Yes, QMessageBox.No) if opcionElegida == QMessageBox.Yes: respuesta = self.operacionesConPersonas.eliminarPersona( laPersona) if type(respuesta) == type(""): self.mostrarError.insertPlainText(''.join(respuesta)) self.limpiarCampos() return None except Exception as e: self.mostrarError.insertPlainText(''.join(e.args)) def actualizarPersona(self): self.limpiarErrores() try: laPersona = self.llenarPersona() mensajeDeConfirmacion = "DNI: " + self.DNI.text() opcionElegida = QMessageBox.question( self, "Desea eliminar a la persona?", mensajeDeConfirmacion, QMessageBox.Yes, QMessageBox.No) if opcionElegida == QMessageBox.Yes: respuesta = self.operacionesConPersonas.actualizarPersona( laPersona) if type(respuesta) == type(""): self.mostrarError.insertPlainText(''.join(respuesta)) self.limpiarCampos() return None except Exception as e: print(e.args) def llenarCampos(self, laPersona): self.limpiarCampos() self.DNI.setText(str(laPersona.getDNI())) self.DNI.setReadOnly(True) self.PrimerNombre.setText(laPersona.getPrimerNombre()) self.SegundoNombre.setText(laPersona.getSegundoNombre()) self.PrimerApellido.setText(laPersona.getPrimerApellido()) self.SegundoApellido.setText(laPersona.getSegundoApellido()) self.fechaDeNacimiento.setDate(laPersona.getFechaDeNacimiento()) def llenarPersona(self): return Persona(int(''.join(self.DNI.text())), ''.join(self.PrimerNombre.text()), ''.join(self.SegundoNombre.text()), ''.join(self.PrimerApellido.text()), ''.join(self.SegundoApellido.text()), ''.join(self.fechaDeNacimiento.text())) def limpiarCampos(self): self.DNI.clear() self.DNI.setReadOnly(False) self.PrimerNombre.clear() self.SegundoNombre.clear() self.PrimerApellido.clear() self.SegundoApellido.clear() self.fechaDeNacimiento.setDate(date.today()) #limpiar campos de texto y permitir la edicion de todos ellos def limpiarErrores(self): self.mostrarError.clear() def volverAlMenu(self): raiz.setCentralWidget(None) raiz.setCentralWidget(VentanaPrincipal())
class Window(QWidget): go_list = [] unit = None def __init__(self): super(Window, self).__init__() self.createCalendarBox() self.setup_database_dialog() self.database_utilities() self.ready_plots() layout = QGridLayout() layout.addWidget(self.db_dialog_groupbox, 0, 0) layout.addWidget(self.previewGroupBox, 0, 1) layout.addWidget(self.db_util_groupbox, 1, 0) layout.addWidget(self.ready_plt_groupbox, 1, 1) layout.setSizeConstraint(QLayout.SetFixedSize) self.setLayout(layout) self.previewLayout.setRowMinimumHeight(0, self.calendar.sizeHint().height()) self.previewLayout.setColumnMinimumWidth(0, self.calendar.sizeHint().width()) self.setWindowTitle("Data Visualization Application") def createCalendarBox(self): self.previewGroupBox = QGroupBox("Custom Plot") # db_entry_layout self.db_entry_layout = QHBoxLayout() self.db_entry_field = QLineEdit("tasos.db") self.db_entry_label = QLabel("Set Sensor") self.db_entry_set_button = QPushButton("Set") # Create available db QComboBox self.avail_db_combo = QComboBox() self.avail_db_combo.addItems(self.available_db_combo()) self.avail_db_combo_reload = QPushButton("Reload") # Create QHBoxLayout for avail_db_combo self.avail_db_combo_layout = QHBoxLayout() # Adds widgets to db_entry_ayout QHBoxLayout self.db_entry_layout.addWidget(self.db_entry_label) self.db_entry_layout.addWidget(self.avail_db_combo) self.db_entry_layout.addWidget(self.avail_db_combo_reload) self.db_entry_layout.addWidget(self.db_entry_set_button) self.db_entry_set_button.clicked.connect(self.set_db_entry_button_script) # Calendar Widget self.calendar = QCalendarWidget() self.calendar.setGridVisible(True) # Create main layout self.previewLayout = QGridLayout() # Creating sub layouts and adding their widgets self.cal_options = QGridLayout() self.chart_label = QLabel("&Chart") self.calendar_chart_field = QComboBox() self.calendar_chart_field.addItem("Lines") self.calendar_chart_field.addItem("Lines and Markers") self.calendar_chart_field.addItem("Scatter") self.calendar_chart_field.addItem("Bars") self.chart_label.setBuddy(self.calendar_chart_field) # Mode Layout self.h_mode_layout = QHBoxLayout() self.h_mode_layout.addWidget(self.chart_label) self.h_mode_layout.addWidget(self.calendar_chart_field) self.from_to_dates_layout = QVBoxLayout() # From Layout self.from_date_layout = QHBoxLayout() self.from_date_label = QLabel("From") self.from_date = QDateEdit() self.update_from_date = QPushButton("Set Date") self.from_date.setDisplayFormat('MMM d yyyy') self.from_date.setDate(self.calendar.selectedDate()) self.update_from_date.clicked.connect(self.from_selectedDateChanged) # Adds widgets to from_date_layout QHBoxLayout self.from_date_layout.addWidget(self.from_date) self.from_date_layout.addWidget(self.update_from_date) # To layout self.to_date_layout = QHBoxLayout() self.to_date_label = QLabel("To") self.to_date = QDateEdit() self.update_to_date = QPushButton("Set Date") self.to_date.setDisplayFormat('MMM d yyyy') self.to_date.setDate(self.calendar.selectedDate()) self.update_to_date.clicked.connect(self.to_selectedDateChanged) # Multiple graphic objects layout self.multi_go_label = QLabel("Plot multiple graph objects") self.multi_go_layout = QHBoxLayout() self.multi_go_add_button = QPushButton("Add Graph Object") self.multi_go_clear_button = QPushButton("Clear Graph Objects") self.multi_go_plot_button = QPushButton("Plot Graph Objects") # Adds widges to multi_go_layout self.multi_go_layout.addWidget(self.multi_go_add_button) self.multi_go_layout.addWidget(self.multi_go_clear_button) # Adds widgets to to_date_layout QHBoxLayout self.to_date_layout.addWidget(self.to_date) self.to_date_layout.addWidget(self.update_to_date) # Add widgets and QHBoxLayout to our QVBoxLayout self.from_to_dates_layout.addWidget(self.from_date_label) self.from_to_dates_layout.addLayout(self.from_date_layout) self.from_to_dates_layout.addWidget(self.to_date_label) self.from_to_dates_layout.addLayout(self.to_date_layout) self.custom_plot_button = QPushButton("Custom Plot") # self.available_db_combo() # self.cal_options.addLayout(self.avail_db_combo_layout, 0, 0) self.cal_options.addLayout(self.db_entry_layout, 0, 0) self.cal_options.addLayout(self.h_mode_layout, 1, 0) self.cal_options.addLayout(self.from_to_dates_layout, 2, 0) self.cal_options.addWidget(self.custom_plot_button, 3, 0) self.multi_qvbox_layout = QVBoxLayout() self.multi_qvbox_layout.addWidget(self.multi_go_label) self.multi_qvbox_layout.addLayout(self.multi_go_layout) self.multi_qvbox_layout.addWidget(self.multi_go_plot_button) self.avail_db_combo_reload.clicked.connect(self.reload_db_combo) self.custom_plot_button.clicked.connect(self.custom_plot_script) # Connect multi_go buttons self.multi_go_add_button.clicked.connect(self.add_go_script) self.multi_go_clear_button.clicked.connect(self.clear_go_script) self.multi_go_plot_button.clicked.connect(self.multi_go_plot_script) # Create QVBoxLayout that contains a QHBoxLayout with min,max,avg,count,sum self.stats_label = QLabel("Get Stats from The Selected Sensor and Dates") self.max = QPushButton("Get Max") self.min = QPushButton("Get Min") self.avg = QPushButton("Get Average") self.count = QPushButton("Get Count Of Measurements") self.sum = QPushButton("Get Total Sum") self.avg_to_max = QPushButton("Get Count Between Avg-Max") self.stats_first_qhbox_layout = QHBoxLayout() self.stats_first_qhbox_layout.addWidget(self.min) self.stats_first_qhbox_layout.addWidget(self.max) self.stats_first_qhbox_layout.addWidget(self.avg) self.stats_second_qhbox_layout = QHBoxLayout() self.stats_second_qhbox_layout.addWidget(self.count) self.stats_second_qhbox_layout.addWidget(self.sum) self.stats_second_qhbox_layout.addWidget(self.avg_to_max) self.stats_qvbox_layout = QVBoxLayout() self.stats_qvbox_layout.addWidget(self.stats_label) self.stats_qvbox_layout.addLayout(self.stats_first_qhbox_layout) self.stats_qvbox_layout.addLayout(self.stats_second_qhbox_layout) # Connect stats buttons when clicked self.min.clicked.connect(self.get_min_script) self.max.clicked.connect(self.get_max_script) self.avg.clicked.connect(self.get_avg_script) self.count.clicked.connect(self.get_count_script) self.sum.clicked.connect(self.get_sum_script) self.avg_to_max.clicked.connect(self.get_avg_to_max_count_script) # Add widgets and sub layout to main layout self.previewLayout.addWidget(self.calendar, 0, 0, Qt.AlignTop) # , Qt.AlignCenter self.previewLayout.addLayout(self.cal_options, 0, 1) self.previewLayout.addLayout(self.stats_qvbox_layout, 1, 0) self.previewLayout.addLayout(self.multi_qvbox_layout, 1, 1) self.previewGroupBox.setLayout(self.previewLayout) def get_avg_to_max_count_script(self): avgToMax_from_date = str(self.from_date.date().toPython()) avgToMax_to_date = str(self.to_date.date().toPython()) avgToMax_error = "Something went wrong while retrieving the percentage" avgToMax_db = self.avail_db_combo.currentText() + ".db" avgToMax_extr = Extractor(avgToMax_db) try: total_count = avgToMax_extr.get_count(avgToMax_from_date, avgToMax_to_date) avgToMax_avg = avgToMax_extr.get_avg(avgToMax_from_date, avgToMax_to_date) avgToMax_max = avgToMax_extr.get_max(avgToMax_from_date, avgToMax_to_date) atm_avg = avgToMax_avg[0][0] atm_max = avgToMax_max[0][0] avgToMax_count = avgToMax_extr.get_avg_to_max_count( str(atm_avg), str(atm_max), avgToMax_from_date, avgToMax_to_date) avgToMax_percentage = (avgToMax_count[0][0]/total_count[0][0]) print(avgToMax_count[0][0], total_count[0][0]) avgToMax_success = True except Exception as ex: avgToMax_success = False avgToMax_error = ex avgToMax_msg = QMessageBox() if avgToMax_success: avgToMax_msg.setIcon(QMessageBox.Information) avgToMax_msg.setText( "Percentage of values between average and max for the selected sensor and dates") avgToMax_msg.setInformativeText( "Avg to Max percentage: {0} %".format(avgToMax_percentage*100)) avgToMax_msg.setWindowTitle("Avg to Max percentage") copy_avgToMax_to_clip = avgToMax_msg.addButton( self.tr("Copy to clipboard"), QMessageBox.AcceptRole) avgToMax_msg.setStandardButtons(QMessageBox.Close) avgToMax_msg.setDefaultButton(copy_avgToMax_to_clip) else: avgToMax_msg.setIcon(QMessageBox.Critical) avgToMax_msg.setText("Getting Avg to Max Percentage Failed!") avgToMax_msg.setInformativeText( "Getting Avg to Max percentage from {0} failed!".format(self.avail_db_combo.currentText())) avgToMax_msg.setWindowTitle("ERROR!") avgToMax_msg.setDetailedText("ERROR:\n {0}".format(avgToMax_error)) avgToMax_msg.setStandardButtons(QMessageBox.Abort) avgToMax_retval = avgToMax_msg.exec_() copy_avgToMax_to_clip.clicked.connect(self.copy_to_clipboard(avgToMax_percentage)) avgToMax_msg.show() def get_sum_script(self): sum_from_date = str(self.from_date.date().toPython()) sum_to_date = str(self.to_date.date().toPython()) sum_error = "Something went wrong while retrieving sum" sum_db = self.avail_db_combo.currentText() + ".db" sum_extr = Extractor(sum_db) try: sum_stat = sum_extr.get_sum(sum_from_date, sum_to_date) sum_success = True except Exception as ex: sum_success = False sum_error = ex sum_msg = QMessageBox() if sum_success: sum_msg.setIcon(QMessageBox.Information) sum_msg.setText("Sum value for the selected sensor and dates") sum_msg.setInformativeText( "Sum value: {0}".format(sum_stat[0][0])) sum_msg.setWindowTitle("sum Value") copy_sum_to_clip = sum_msg.addButton( self.tr("Copy to clipboard"), QMessageBox.AcceptRole) sum_msg.setStandardButtons(QMessageBox.Close) sum_msg.setDefaultButton(copy_sum_to_clip) else: sum_msg.setIcon(QMessageBox.Critical) sum_msg.setText("Getting Sum Value Failed!") sum_msg.setInformativeText( "Getting sum value from {0} failed!".format(self.avail_db_combo.currentText())) sum_msg.setWindowTitle("ERROR!") sum_msg.setDetailedText("ERROR:\n {0}".format(sum_error)) sum_msg.setStandardButtons(QMessageBox.Abort) sum_retval = sum_msg.exec_() copy_sum_to_clip.clicked.connect(self.copy_to_clipboard(sum_stat[0][0])) sum_msg.show() def get_count_script(self): count_from_date = str(self.from_date.date().toPython()) count_to_date = str(self.to_date.date().toPython()) count_error = "Something went wrong while retrieving count" count_db = self.avail_db_combo.currentText() + ".db" count_extr = Extractor(count_db) try: count_stat = count_extr.get_count(count_from_date, count_to_date) count_success = True except Exception as ex: count_success = False count_error = ex count_msg = QMessageBox() if count_success: count_msg.setIcon(QMessageBox.Information) count_msg.setText("Count value for the selected sensor and dates") count_msg.setInformativeText( "count value: {0}".format(count_stat[0][0])) count_msg.setWindowTitle("Count Value") copy_count_to_clip = count_msg.addButton( self.tr("Copy to clipboard"), QMessageBox.AcceptRole) count_msg.setStandardButtons(QMessageBox.Close) count_msg.setDefaultButton(copy_count_to_clip) else: count_msg.setIcon(QMessageBox.Critical) count_msg.setText("Getting Count Value Failed!") count_msg.setInformativeText( "Getting count value from {0} failed!".format(self.avail_db_combo.currentText())) count_msg.setWindowTitle("ERROR!") count_msg.setDetailedText("ERROR:\n {0}".format(count_error)) count_msg.setStandardButtons(QMessageBox.Abort) count_retval = count_msg.exec_() copy_count_to_clip.clicked.connect(self.copy_to_clipboard(count_stat[0][0])) count_msg.show() def get_avg_script(self): avg_from_date = str(self.from_date.date().toPython()) avg_to_date = str(self.to_date.date().toPython()) avg_error = "Something went wrong while retrieving avg" avg_db = self.avail_db_combo.currentText() + ".db" avg_extr = Extractor(avg_db) try: avg_stat = avg_extr.get_avg(avg_from_date, avg_to_date) avg_success = True except Exception as ex: avg_success = False avg_error = ex avg_msg = QMessageBox() if avg_success: avg_msg.setIcon(QMessageBox.Information) avg_msg.setText("Avg value for the selected sensor and dates") avg_msg.setInformativeText( "Avg value: {0}".format(avg_stat[0][0])) avg_msg.setWindowTitle("Avg Value") copy_avg_to_clip = avg_msg.addButton( self.tr("Copy to clipboard"), QMessageBox.AcceptRole) avg_msg.setStandardButtons(QMessageBox.Close) avg_msg.setDefaultButton(copy_avg_to_clip) else: avg_msg.setIcon(QMessageBox.Critical) avg_msg.setText("Getting Avg Value Failed!") avg_msg.setInformativeText( "Getting avg value from {0} failed!".format(self.avail_db_combo.currentText())) avg_msg.setWindowTitle("ERROR!") avg_msg.setDetailedText("ERROR:\n {0}".format(avg_error)) avg_msg.setStandardButtons(QMessageBox.Abort) avg_retval = avg_msg.exec_() copy_avg_to_clip.clicked.connect(self.copy_to_clipboard(avg_stat[0][0])) avg_msg.show() def get_max_script(self): max_from_date = str(self.from_date.date().toPython()) max_to_date = str(self.to_date.date().toPython()) max_error = "Something went wrong while retrieving max" max_db = self.avail_db_combo.currentText() + ".db" max_extr = Extractor(max_db) try: max_stat = max_extr.get_max(max_from_date, max_to_date) max_success = True except Exception as ex: max_success = False max_error = ex max_msg = QMessageBox() if max_success: max_msg.setIcon(QMessageBox.Information) max_msg.setText("Max value for the selected sensor and dates") max_msg.setInformativeText( "Max value: {0}".format(max_stat[0][0])) max_msg.setWindowTitle("Max Value") copy_max_to_clip = max_msg.addButton( self.tr("Copy to clipboard"), QMessageBox.AcceptRole) max_msg.setStandardButtons(QMessageBox.Close) max_msg.setDefaultButton(copy_max_to_clip) else: max_msg.setIcon(QMessageBox.Critical) max_msg.setText("Getting Max Value Failed!") max_msg.setInformativeText( "Getting max value from {0} failed!".format(self.avail_db_combo.currentText())) max_msg.setWindowTitle("ERROR!") max_msg.setDetailedText("ERROR:\n {0}".format(max_error)) max_msg.setStandardButtons(QMessageBox.Abort) max_retval = max_msg.exec_() copy_max_to_clip.clicked.connect(self.copy_to_clipboard(max_stat[0][0])) max_msg.show() def get_min_script(self): min_from_date = str(self.from_date.date().toPython()) min_to_date = str(self.to_date.date().toPython()) min_error = "Something went wrong while retrieving min" min_db = self.avail_db_combo.currentText() + ".db" min_extr = Extractor(min_db) try: min_stat = min_extr.get_min(min_from_date, min_to_date) min_success = True except Exception as ex: min_success = False min_error = ex min_msg = QMessageBox() if min_success: min_msg.setIcon(QMessageBox.Information) min_msg.setText("Min value for the selected sensor and dates") min_msg.setInformativeText( "Min value: {0}".format(min_stat[0][0])) min_msg.setWindowTitle("Min Value") copy_min_to_clip = min_msg.addButton( self.tr("Copy to clipboard"), QMessageBox.AcceptRole) min_msg.setStandardButtons(QMessageBox.Close) min_msg.setDefaultButton(copy_min_to_clip) else: min_msg.setIcon(QMessageBox.Critical) min_msg.setText("Getting Min Value Failed!") min_msg.setInformativeText( "Getting min value from {0} failed!".format(self.avail_db_combo.currentText())) min_msg.setWindowTitle("ERROR!") min_msg.setDetailedText("ERROR:\n {0}".format(min_error)) min_msg.setStandardButtons(QMessageBox.Abort) min_retval = min_msg.exec_() copy_min_to_clip.clicked.connect(self.copy_to_clipboard(min_stat[0][0])) min_msg.show() def copy_to_clipboard(self, var): pyperclip.copy(var) def add_go_script(self): current_index_chart = self.calendar_chart_field.currentIndex() if current_index_chart == 0: chart_mode = "lines" elif current_index_chart == 1: chart_mode = "lines+markers" elif current_index_chart == 2: chart_mode = "markers" elif current_index_chart == 3: chart_mode = "Bars" else: cm_error = "Something went wrong" print(cm_error) multi_go_db = self.avail_db_combo.currentText() + ".db" multi_go_extr = Extractor(multi_go_db) multi_go_plt = Plotter() multi_go_lists = [] # converts Qdate object to python date object go_from_date = str(self.from_date.date().toPython()) go_to_date = str(self.to_date.date().toPython()) multi_go_lists = multi_go_extr.custom_select(go_from_date, go_to_date) if chart_mode == "lines" or chart_mode == "lines+markers" or chart_mode == "markers": go_data = multi_go_plt.go_scatter( chart_mode, multi_go_lists[0], multi_go_lists[1], multi_go_db) Window.go_list.append(go_data) if Window.unit is None: Window.unit = multi_go_lists[2][0] elif chart_mode == "Bars": go_data = multi_go_plt.go_bar( chart_mode, multi_go_lists[0], multi_go_lists[1], multi_go_db) Window.go_list.append(go_data) if Window.unit is None: Window.unit = multi_go_lists[2][0] else: print("Something went wrong") def clear_go_script(self): Window.go_list.clear() Window.unit = None def multi_go_plot_script(self): multi_go_plt = Plotter() current_index_chart = self.calendar_chart_field.currentIndex() if current_index_chart == 0: chart_mode = "lines" elif current_index_chart == 1: chart_mode = "lines+markers" elif current_index_chart == 2: chart_mode = "markers" elif current_index_chart == 3: chart_mode = "Bars" else: cm_error = "Something went wrong" print(cm_error) if chart_mode == "lines" or chart_mode == "lines+markers" or chart_mode == "markers": multi_go_plt.go_plot("Multiple Graphs", "TIME", Window.unit, Window.go_list) elif chart_mode == "Bars": multi_go_plt.go_bar_plot("Multiple Graphs", "TIME", Window.unit, Window.go_list) else: print("Something went wrong") Window.go_list.clear() Window.unit = None def custom_plot_script(self): current_index_chart = self.calendar_chart_field.currentIndex() if current_index_chart == 0: chart_mode = "lines" elif current_index_chart == 1: chart_mode = "lines+markers" elif current_index_chart == 2: chart_mode = "markers" elif current_index_chart == 3: chart_mode = "Bars" else: cm_error = "Something went wrong" print(cm_error) custom_db = self.avail_db_combo.currentText() + ".db" custom_extr = Extractor(custom_db) custom_plt = Plotter() custom_plot_lists = [] custom_plot_title = custom_db[:-3] # converts Qdate object to python date object custom_from_date = str(self.from_date.date().toPython()) custom_to_date = str(self.to_date.date().toPython()) custom_plot_lists = custom_extr.custom_select(custom_from_date, custom_to_date) if chart_mode == "lines" or chart_mode == "lines+markers" or chart_mode == "markers": custom_plt.scatter_plot( chart_mode, custom_plot_lists[0], custom_plot_lists[1], custom_plot_title, "TIME", custom_plot_lists[2][0]) elif chart_mode == "Bars": custom_plt.bar_plot(custom_plot_lists[0], custom_plot_lists[1], custom_plot_title, "TIME", custom_plot_lists[2][0]) else: print("Something went wrong") def setup_database_dialog(self): # Create Groupbox self.db_dialog_groupbox = QGroupBox("Sensor Declaration and Data Import") # Create widgets self.db_label = QLabel("Provide a name for the sensor") self.path_label = QLabel("Provide path to search for .xml files") self.db_field = QLineEdit("") self.path_field = QLineEdit(str(os.path.dirname(os.path.realpath(__file__)))) self.button = QPushButton("Get started") # Create layout and add widgets self.db_dialog_layout = QVBoxLayout() self.db_dialog_layout.addWidget(self.db_label) self.db_dialog_layout.addWidget(self.db_field) self.db_dialog_layout.addWidget(self.path_label) self.db_dialog_layout.addWidget(self.path_field) self.db_dialog_layout.addWidget(self.button) # Set dialog layout self.db_dialog_groupbox.setLayout(self.db_dialog_layout) self.db_field.textChanged.connect(self.db_field_changed) self.button.clicked.connect(self.run_script) def run_script(self): error = "Error parsing files or path" db = self.db_field.text() + ".db" db_interaction = DatabaseInteraction(db) # returns object of class DatabaseInteraction path = self.path_field.text() xml_importer = XMLImporter(db_interaction.name, path) try: success = xml_importer.import_xml() except Exception as ex: success = False error = ex msg = QMessageBox() if success: msg.setIcon(QMessageBox.Information) msg.setText("Sensor declaration successful") msg.setInformativeText("The sensor: {0} has been created.".format(self.db_field.text())) msg.setWindowTitle("Sensor declared!") msg.setDetailedText("The details are as follows:\nSensor name: {0} \nPath of .xml " "files: {1}".format(self.db_field.text(), path)) msg.setStandardButtons(QMessageBox.Ok) else: msg.setIcon(QMessageBox.Critical) msg.setText("Sensor declaration failed!") msg.setInformativeText( "Declaration of the sensor: {0} failed!".format(self.db_field.text())) msg.setWindowTitle("Sensor declaration failed!") msg.setDetailedText("ERROR:\n {0}".format(error)) msg.setStandardButtons(QMessageBox.Abort) retval = msg.exec_() msg.show() def database_utilities(self): # Create Groupbox self.db_util_groupbox = QGroupBox("Utilities") # Create Widgets self.db_update_label = QLabel("Select the Sensor you wish to update") self.path_label = QLabel("Xml File Path") self.update_path_field = QLineEdit(str(os.path.dirname(os.path.realpath(__file__)))) self.update_button = QPushButton("Update") self.update_title = QLabel("Update Sensor") # self.db_clear_field = QLineEdit("example.db") self.db_clear_label = QLabel("Select Sensor") self.clear_button = QPushButton("Remove") self.clear_title = QLabel("Remove a sensor") # Create available db QComboBox for update self.update_avail_db_combo = QComboBox() self.update_avail_db_combo.addItems(self.available_db_combo()) self.update_avail_db_combo_reload = QPushButton("Reload") # Create available db QComboBox for clear table self.clear_avail_db_combo = QComboBox() self.clear_avail_db_combo.addItems(self.available_db_combo()) self.clear_avail_db_combo_reload = QPushButton("Reload") # Add Widgets self.db_util_db_update = QHBoxLayout() self.db_util_db_update.addWidget(self.db_update_label) self.db_util_db_update.addWidget(self.update_avail_db_combo) self.db_util_db_update.addWidget(self.update_avail_db_combo_reload) self.db_util_path = QHBoxLayout() self.db_util_path.addWidget(self.path_label) self.db_util_path.addWidget(self.update_path_field) self.db_util_clear = QHBoxLayout() self.db_util_clear.addWidget(self.db_clear_label) self.db_util_clear.addWidget(self.clear_avail_db_combo) self.db_util_clear.addWidget(self.clear_avail_db_combo_reload) self.db_util_vbox_layout = QVBoxLayout() self.db_util_vbox_layout.addWidget(self.update_title) self.db_util_vbox_layout.addLayout(self.db_util_db_update) self.db_util_vbox_layout.addLayout(self.db_util_path) self.db_util_vbox_layout.addWidget(self.update_button) self.db_util_vbox_layout.addWidget(self.clear_title) self.db_util_vbox_layout.addLayout(self.db_util_clear) self.db_util_vbox_layout.addWidget(self.clear_button) # Set dialog layout self.db_util_groupbox.setLayout(self.db_util_vbox_layout) self.update_button.clicked.connect(self.update_script) self.clear_button.clicked.connect(self.clear_table_script) self.update_avail_db_combo_reload.clicked.connect(self.reload_db_combo) self.clear_avail_db_combo_reload.clicked.connect(self.reload_db_combo) def update_script(self): db = self.update_avail_db_combo.currentText() + ".db" database_interaction = DatabaseInteraction(db) path = self.update_path_field.text() xml_importer = XMLImporter(database_interaction.name, path) update_error = "Something went wrong during update" try: success = xml_importer.import_xml() except Exception as ex: success = False update_error = ex update_msg = QMessageBox() if success: update_msg.setIcon(QMessageBox.Information) update_msg.setText("Sensor update successful") update_msg.setInformativeText("The sensor: {0} has been updated.".format(db[:-3])) update_msg.setWindowTitle("Sensor updated!") update_msg.setDetailedText("The details are as follows:\Sensor name: {0} \nPath of .xml " "files: {1}".format(db[:-3], path)) update_msg.setStandardButtons(QMessageBox.Ok) else: update_msg.setIcon(QMessageBox.Critical) update_msg.setText("Sensor update failed!") update_msg.setInformativeText("Update of the sensor: {0} failed!".format(db[:-3])) update_msg.setWindowTitle("Sensor update failed!") update_msg.setDetailedText("ERROR:\n {0}".format(update_error)) update_msg.setStandardButtons(QMessageBox.Abort) update_retval = update_msg.exec_() update_msg.show() def ready_plots(self): # Create Groupbox self.ready_plt_groupbox = QGroupBox("Ready-to-use plots") # Create Widgets self.choose_db_label = QLabel( "Provide the database that contains the data you wish to plot") self.choose_db_field = QLineEdit("") self.chart_label = QLabel("Chart") self.chart_field = QComboBox() self.chart_field.addItem("Lines") self.chart_field.addItem("Lines and Markers") self.chart_field.addItem("Scatter") self.chart_field.addItem("Bars") self.plot_mode_label = QLabel("Choose what time period to plot") self.plot_mode_field = QComboBox() self.plot_mode_field.addItem("Daily") self.plot_mode_field.addItem("Weekly") self.plot_mode_field.addItem("Monthly") self.plot_mode_field.addItem("Yearly") self.plot_button = QPushButton("Plot") # Create available db QComboBox for ready-plots self.ready_plot_avail_db_combo = QComboBox() self.ready_plot_avail_db_combo.addItems(self.available_db_combo()) self.ready_plot_avail_db_combo_reload = QPushButton("Reload") # Add QComboBox and QPushButton to a QHBoxLayout self.ready_plot_qhbox_layout = QHBoxLayout() self.ready_plot_qhbox_layout.addWidget(self.ready_plot_avail_db_combo) self.ready_plot_qhbox_layout.addWidget(self.ready_plot_avail_db_combo_reload) # Add Widgets self.plot_vbox_layout = QVBoxLayout() self.plot_vbox_layout.addWidget(self.choose_db_label) self.plot_vbox_layout.addLayout(self.ready_plot_qhbox_layout) self.plot_vbox_layout.addWidget(self.chart_label) self.plot_vbox_layout.addWidget(self.chart_field) self.plot_vbox_layout.addWidget(self.plot_mode_label) self.plot_vbox_layout.addWidget(self.plot_mode_field) self.plot_vbox_layout.addWidget(self.plot_button) # Set layout to groupbox self.ready_plt_groupbox.setLayout(self.plot_vbox_layout) self.plot_button.clicked.connect(self.ready_plot_script) self.ready_plot_avail_db_combo_reload.clicked.connect(self.reload_db_combo) def db_field_changed(self): self.db_field.setText(self.db_field.text()) def db_clear_field_changed(self): self.db_clear_field.setText(self.db_clear_field.text()) def db_update_field_changed(self): self.db_update_field.setText(self.db_update_field.text()) def from_selectedDateChanged(self): self.from_date.setDate(self.calendar.selectedDate()) def to_selectedDateChanged(self): self.to_date.setDate(self.calendar.selectedDate()) def clear_table_script(self): clear_error = "Something went wrong during clearing sensor" db = self.clear_avail_db_combo.currentText() + ".db" try: # db_int.clear_table() os.remove(db) clear_success = True except Exception as ex: clear_success = False clear_error = ex clear_msg = QMessageBox() if clear_success: clear_msg.setIcon(QMessageBox.Information) clear_msg.setText("Sensor has been removed") clear_msg.setInformativeText( "The {0} sensor have been removed.".format(db[:-3])) clear_msg.setWindowTitle("Sensor has been removed!") clear_msg.setStandardButtons(QMessageBox.Ok) else: clear_msg.setIcon(QMessageBox.Critical) clear_msg.setText("Removing sensor has failed!") clear_msg.setInformativeText("Removing the sensor: {0} has failed!".format(db[:-3])) clear_msg.setWindowTitle("Removing this sensor has failed!") clear_msg.setDetailedText("ERROR:\n {0}".format(clear_error)) clear_msg.setStandardButtons(QMessageBox.Abort) clear_retval = clear_msg.exec_() clear_msg.show() def set_db_entry_button_script(self): db_name_fixed = self.avail_db_combo.currentText() + ".db" self.db = DatabaseInteraction(db_name_fixed) extr = Extractor(self.db.name) # Create datetime objects of the first and last strings of date self.datetime_first = datetime.strptime(extr.select_first_row()[ 0][0], "%Y-%m-%dT%H:%M:%S.%fZ") self.datetime_last = datetime.strptime(extr.select_last_row()[ 0][0], "%Y-%m-%dT%H:%M:%S.%fZ") self.first_row = QDate.fromString(str(self.datetime_first.date()), "yyyy-MM-dd") self.last_row = QDate.fromString(str(self.datetime_last.date()), "yyyy-MM-dd") self.calendar.setMinimumDate(self.first_row) self.calendar.setMaximumDate(self.last_row) self.first_row = QDate.fromString(str(self.datetime_first.date()), "yyyy-MM-dd") self.last_row = QDate.fromString(str(self.datetime_last.date()), "yyyy-MM-dd") self.calendar.setMinimumDate(self.first_row) self.calendar.setMaximumDate(self.last_row) def reload_db_combo(self): self.avail_db_combo.clear() self.avail_db_combo.addItems(self.available_db_combo()) self.clear_avail_db_combo.clear() self.clear_avail_db_combo.addItems(self.available_db_combo()) self.update_avail_db_combo.clear() self.update_avail_db_combo.addItems(self.available_db_combo()) self.ready_plot_avail_db_combo.clear() self.ready_plot_avail_db_combo.addItems(self.available_db_combo()) def available_db_combo(self): db_files = [] db_list_model = QStringListModel() for subdir, dirs, files in os.walk(os.path.dirname(os.path.realpath(__file__))): for file in files: filepath = subdir + os.path.sep + file if filepath.endswith(".db"): if filepath not in db_files: db_files.append(file[:-3]) db_list_model.setStringList(db_files) return db_list_model.stringList() def ready_plot_script(self): db = self.ready_plot_avail_db_combo.currentText() + ".db" extr = Extractor(db) plt = Plotter() ready_plot_title = db[:-3] if self.chart_field.currentIndex() == 0: chart_mode = "lines" elif self.chart_field.currentIndex() == 1: chart_mode = "lines+markers" elif self.chart_field.currentIndex() == 2: chart_mode = "markers" elif self.chart_field.currentIndex() == 3: chart_mode = "Bars" else: print("Something went wrong") if self.plot_mode_field.currentIndex() == 0: plot_mode = "daily_select" elif self.plot_mode_field.currentIndex() == 1: plot_mode = "weekly_select" elif self.plot_mode_field.currentIndex() == 2: plot_mode = "monthly_select" elif self.plot_mode_field.currentIndex() == 3: plot_mode = "yearly_select" else: print("Something went wrong") plot_lists = [] # Extract from the database the data for x-y axis and the unit plot_lists = extr.extract(plot_mode) if chart_mode == "lines" or chart_mode == "lines+markers" or chart_mode == "markers": plt.scatter_plot(chart_mode, plot_lists[0], plot_lists[1], ready_plot_title, "TIME", plot_lists[2][0]) elif chart_mode == "Bars": plt.bar_plot(plot_lists[0], plot_lists[1], ready_plot_title, "TIME", plot_lists[2][0]) else: print("Something went wrong")
class SpotPriceUI(QTabWidget): def __init__(self, *args, **kwargs): super(SpotPriceUI, self).__init__(*args, **kwargs) self.extra_data_widget = QWidget(self) # 提取数据tab layout = QVBoxLayout() # 现货价格源数据 source_layout = QHBoxLayout() self.current_date = QDateEdit(self) self.current_date.setDisplayFormat("yyyy-MM-dd") self.current_date.setDate(QDate.currentDate()) self.current_date.setCalendarPopup(True) source_layout.addWidget(self.current_date) source_layout.addWidget(QLabel("源数据:", self)) self.source_edit = QLineEdit(self) source_layout.addWidget(self.source_edit) layout.addLayout(source_layout) # 分析 analysis_layout = QHBoxLayout() self.analysis_button = QPushButton("提取数据", self) analysis_layout.addWidget(self.analysis_button) self.tip_label = QLabel("输入源数据后,点击提取数据进行数据提取.", self) analysis_layout.addWidget(self.tip_label) analysis_layout.addStretch() layout.addLayout(analysis_layout) # 预览数据的表格 self.preview_table = QTableWidget(self) self.preview_table.setColumnCount(5) self.preview_table.setHorizontalHeaderLabels(["日期", "品种", "交易代码", "现货价", "增减"]) layout.addWidget(self.preview_table) # 提交按钮 commit_layout = QHBoxLayout() commit_layout.addStretch() self.commit_button = QPushButton("确认提交", self) commit_layout.addWidget(self.commit_button) layout.addLayout(commit_layout) self.extra_data_widget.setLayout(layout) self.addTab(self.extra_data_widget, "数据提取") # 提取数据tab # 修改数据的tab self.modify_data_widget = QWidget(self) modify_layout = QVBoxLayout() params_layout = QHBoxLayout() params_layout.addWidget(QLabel("选择日期:", self)) self.modify_date_edit = QDateEdit(self) self.modify_date_edit.setDate(QDate.currentDate()) self.modify_date_edit.setCalendarPopup(True) self.modify_date_edit.setDisplayFormat("yyyy-MM-dd") params_layout.addWidget(self.modify_date_edit) self.modify_query_button = QPushButton("查询", self) params_layout.addWidget(self.modify_query_button) self.modify_tip_label = QLabel("选择日期查询出数据,双击要修改的数据编辑正确的数据,点击行尾修改.", self) params_layout.addWidget(self.modify_tip_label) params_layout.addStretch() modify_layout.addLayout(params_layout) # 数据表格 self.modify_table = QTableWidget(self) self.modify_table.setColumnCount(6) self.modify_table.setHorizontalHeaderLabels(["ID", "日期", "品种", "现货价", "增减", "修改"]) modify_layout.addWidget(self.modify_table) self.modify_data_widget.setLayout(modify_layout) self.addTab(self.modify_data_widget, "修改数据") self.setStyleSheet( "#modifyButton{border:none;color:rgb(180,30,50)}#modifyButton:hover{color:rgb(20,50,160)}" )
class PersonEditDialog(QDialog): GROUP_NAME = '' ORGANIZATION_NAME = '' def __init__(self, person, is_new=False): super().__init__(GlobalAccess().get_main_window()) self.is_ok = {} assert (isinstance(person, Person)) self.current_object = person self.is_new = is_new self.time_format = 'hh:mm:ss' time_accuracy = race().get_setting('time_accuracy', 0) if time_accuracy: self.time_format = 'hh:mm:ss.zzz' def exec_(self): self.init_ui() self.set_values_from_model() return super().exec_() def init_ui(self): self.setWindowTitle(_('Entry properties')) self.setWindowIcon(QIcon(config.ICON)) self.setSizeGripEnabled(False) self.setModal(True) self.layout = QFormLayout(self) self.label_surname = QLabel(_('Last name')) self.item_surname = QLineEdit() self.layout.addRow(self.label_surname, self.item_surname) self.label_name = QLabel(_('First name')) self.item_name = AdvComboBox() self.item_name.addItems(get_names()) self.layout.addRow(self.label_name, self.item_name) self.label_group = QLabel(_('Group')) self.item_group = AdvComboBox() self.item_group.addItems(get_race_groups()) self.layout.addRow(self.label_group, self.item_group) self.label_team = QLabel(_('Team')) self.item_team = AdvComboBox() self.item_team.addItems(get_race_teams()) self.layout.addRow(self.label_team, self.item_team) use_birthday = Config().configuration.get('use_birthday', False) if use_birthday: self.label_birthday = QLabel(_('Birthday')) self.item_birthday = QDateEdit() self.item_birthday.setDate(date.today()) self.item_birthday.setMaximumDate(date.today()) self.layout.addRow(self.label_birthday, self.item_birthday) else: self.label_year = QLabel(_('Year of birth')) self.item_year = QSpinBox() self.item_year.setMinimum(0) self.item_year.setMaximum(date.today().year) self.item_year.editingFinished.connect(self.year_change) self.layout.addRow(self.label_year, self.item_year) self.label_qual = QLabel(_('Qualification')) self.item_qual = AdvComboBox() for i in list(Qualification): self.item_qual.addItem(i.get_title()) self.layout.addRow(self.label_qual, self.item_qual) self.is_ok['bib'] = True self.label_bib = QLabel(_('Bib')) self.item_bib = QSpinBox() self.item_bib.setMinimum(0) self.item_bib.setMaximum(Limit.BIB) self.item_bib.valueChanged.connect(self.check_bib) self.layout.addRow(self.label_bib, self.item_bib) self.label_bib_info = QLabel('') self.layout.addRow(QLabel(''), self.label_bib_info) self.label_start = QLabel(_('Start time')) self.item_start = QTimeEdit() self.item_start.setDisplayFormat(self.time_format) self.layout.addRow(self.label_start, self.item_start) self.label_start_group = QLabel(_('Start group')) self.item_start_group = QSpinBox() self.item_start_group.setMinimum(0) self.item_start_group.setMaximum(99) self.layout.addRow(self.label_start_group, self.item_start_group) self.is_ok['card'] = True self.label_card = QLabel(_('Punch card #')) self.item_card = QSpinBox() self.item_card.setMinimum(0) self.item_card.setMaximum(9999999) self.item_card.valueChanged.connect(self.check_card) self.layout.addRow(self.label_card, self.item_card) self.label_card_info = QLabel('') self.layout.addRow(QLabel(''), self.label_card_info) self.item_rented = QCheckBox(_('rented card')) self.item_paid = QCheckBox(_('is paid')) self.item_out_of_competition = QCheckBox(_('out of competition')) self.item_personal = QCheckBox(_('personal participation')) self.layout.addRow(self.item_rented, self.item_out_of_competition) self.layout.addRow(self.item_paid, self.item_personal) self.label_comment = QLabel(_('Comment')) self.item_comment = QTextEdit() self.item_comment.setTabChangesFocus(True) self.layout.addRow(self.label_comment, self.item_comment) def cancel_changes(): self.close() def apply_changes(): try: self.apply_changes_impl() except Exception as e: logging.error(str(e)) self.close() button_box = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel) self.button_ok = button_box.button(QDialogButtonBox.Ok) self.button_ok.setText(_('OK')) self.button_ok.clicked.connect(apply_changes) self.button_cancel = button_box.button(QDialogButtonBox.Cancel) self.button_cancel.setText(_('Cancel')) self.button_cancel.clicked.connect(cancel_changes) self.layout.addRow(button_box) self.show() def year_change(self): """ Convert 2 digits of year to 4 2 -> 2002 11 - > 2011 33 -> 1933 56 -> 1956 98 - > 1998 0 -> 0 exception! """ widget = self.sender() assert isinstance(widget, QSpinBox) year = widget.value() if 0 < year < 100: cur_year = date.today().year new_year = cur_year - cur_year % 100 + year if new_year > cur_year: new_year -= 100 widget.setValue(new_year) def items_ok(self): ret = True for item_name in self.is_ok.keys(): if self.is_ok[item_name] is not True: ret = False break return ret def check_bib(self): bib = self.item_bib.value() self.label_bib_info.setText('') if bib: person = find(race().persons, bib=bib) if person: if person.bib == self.current_object.bib: self.button_ok.setEnabled(True) return self.button_ok.setDisabled(True) self.is_ok['bib'] = False info = '{}\n{}'.format(_('Number already exists'), person.full_name) if person.group: info = '{}\n{}: {}'.format(info, _('Group'), person.group.name) self.label_bib_info.setText(info) else: self.label_bib_info.setText(_('Number is unique')) self.is_ok['bib'] = True if self.items_ok(): self.button_ok.setEnabled(True) else: self.button_ok.setEnabled(True) def check_card(self): number = self.item_card.value() self.label_card_info.setText('') if number: person = None for _p in race().persons: if _p.card_number and _p.card_number == number: person = _p break if person: if person.card_number == self.current_object.card_number: self.button_ok.setEnabled(True) return self.button_ok.setDisabled(True) self.is_ok['card'] = False info = '{}\n{}'.format(_('Card number already exists'), person.full_name) if person.group: info = '{}\n{}: {}'.format(info, _('Group'), person.group.name) if person.bib: info = '{}\n{}: {}'.format(info, _('Bib'), person.bib) self.label_card_info.setText(info) else: self.label_card_info.setText(_('Card number is unique')) self.is_ok['card'] = True if self.items_ok(): self.button_ok.setEnabled(True) else: self.button_ok.setEnabled(True) def set_values_from_model(self): self.item_surname.setText(self.current_object.surname) self.item_surname.selectAll() self.item_name.setCurrentText(self.current_object.name) if self.current_object.group is not None: self.item_group.setCurrentText(self.current_object.group.name) else: self.item_group.setCurrentText(self.GROUP_NAME) if self.current_object.organization is not None: self.item_team.setCurrentText( self.current_object.organization.name) else: self.item_team.setCurrentText(self.ORGANIZATION_NAME) if self.current_object.qual: self.item_qual.setCurrentText(self.current_object.qual.get_title()) if self.current_object.bib: self.item_bib.setValue(int(self.current_object.bib)) if self.current_object.start_time is not None: time = time_to_qtime(self.current_object.start_time) self.item_start.setTime(time) if self.current_object.start_group is not None: self.item_start_group.setValue(int( self.current_object.start_group)) if self.current_object.card_number: self.item_card.setValue(self.current_object.card_number) self.item_out_of_competition.setChecked( self.current_object.is_out_of_competition) self.item_paid.setChecked(self.current_object.is_paid) self.item_paid.setChecked(self.current_object.is_paid) self.item_personal.setChecked(self.current_object.is_personal) self.item_rented.setChecked(self.current_object.is_rented_card) self.item_comment.setText(self.current_object.comment) use_birthday = Config().configuration.get('use_birthday', False) if use_birthday: if self.current_object.birth_date: self.item_birthday.setDate(self.current_object.birth_date) else: if self.current_object.get_year(): self.item_year.setValue(self.current_object.get_year()) def apply_changes_impl(self): person = self.current_object assert (isinstance(person, Person)) if self.is_new: race().persons.insert(0, person) if person.name != self.item_name.currentText(): person.name = self.item_name.currentText() if person.surname != self.item_surname.text(): person.surname = self.item_surname.text() if (person.group is not None and person.group.name != self.item_group.currentText()) or\ (person.group is None and len(self.item_group.currentText()) > 0): person.group = find(race().groups, name=self.item_group.currentText()) if (person.organization is not None and person.organization.name != self.item_team.currentText()) or \ (person.organization is None and len(self.item_team.currentText()) > 0): organization = find(race().organizations, name=self.item_team.currentText()) if organization is None: organization = Organization() organization.name = self.item_team.currentText() race().organizations.append(organization) Teamwork().send(organization.to_dict()) person.organization = organization if person.qual.get_title() != self.item_qual.currentText(): person.qual = Qualification.get_qual_by_name( self.item_qual.currentText()) if person.bib != self.item_bib.value(): person.bib = self.item_bib.value() new_time = time_to_otime(self.item_start.time()) if person.start_time != new_time: person.start_time = new_time if person.start_group != self.item_start_group.value( ) and self.item_start_group.value(): person.start_group = self.item_start_group.value() if (not person.card_number or int(person.card_number) != self.item_card.value()) \ and self.item_card.value: race().person_card_number(person, self.item_card.value()) if person.is_out_of_competition != self.item_out_of_competition.isChecked( ): person.is_out_of_competition = self.item_out_of_competition.isChecked( ) if person.is_paid != self.item_paid.isChecked(): person.is_paid = self.item_paid.isChecked() if person.is_rented_card != self.item_rented.isChecked(): person.is_rented_card = self.item_rented.isChecked() if person.is_personal != self.item_personal.isChecked(): person.is_personal = self.item_personal.isChecked() if person.comment != self.item_comment.toPlainText(): person.comment = self.item_comment.toPlainText() use_birthday = Config().configuration.get('use_birthday', False) if use_birthday: new_birthday = qdate_to_date(self.item_birthday.date()) if person.birth_date != new_birthday and new_birthday: if person.birth_date or new_birthday != date.today(): person.birth_date = new_birthday else: if person.get_year() != self.item_year.value(): person.set_year(self.item_year.value()) ResultCalculation(race()).process_results() Teamwork().send(person.to_dict())
def createButtonView(self): buttons = [] datePickers = [] buttonPanel = QWidget() buttonPanel.setContentsMargins(0, 0, 0, 0) buttonCurrent = QPushButton('Current tasks') buttonCurrent.setCheckable(True) buttonFinished = QPushButton('Finished tasks') buttonFinished.setCheckable(True) buttonFilter = QPushButton("Filter by end date") buttonApply = QPushButton("Apply") buttonFilter.setCheckable(True) buttonCurrent.setMaximumWidth(100) buttonFinished.setMaximumWidth(100) buttonFilter.setMaximumWidth(100) buttonApply.setMaximumWidth(100) buttonCurrent.setMinimumWidth(100) buttonFinished.setMinimumWidth(100) buttonFilter.setMinimumWidth(100) buttonApply.setMinimumWidth(100) buttons.append(buttonFilter) # labelStart = QLabel("Date start") datePickers.append(labelStart) dateStart = QDateEdit() dateStart.setDisplayFormat('dd/MM/yyyy') dateStart.setCalendarPopup(True) dateStart.setDate(QDate.currentDate()) datePickers.append(dateStart) labelEnd = QLabel("Date end") datePickers.append(labelEnd) dateEnd = QDateEdit() dateEnd.setDisplayFormat('dd/MM/yyyy') dateEnd.setCalendarPopup(True) dateEnd.setDate(QDate.currentDate()) datePickers.append(dateEnd) datePickers.append(buttonApply) buttonCurrent.clicked.connect( self.switchCurrentTask(buttonFinished, buttonCurrent, buttons)) buttonFinished.clicked.connect( self.switchFinishedTask(buttonCurrent, buttonFinished, buttons)) buttonFilter.clicked.connect( self.openDatePickerFilter(datePickers, buttonFilter)) buttonApply.clicked.connect( self.ApplyDateFilterArchive(dateStart, dateEnd)) buttonCurrent.setChecked(True) vBox = QVBoxLayout() vBox.addWidget(buttonCurrent) vBox.addWidget(buttonFinished) vBox.addSpacing(30) for bnt in buttons: bnt.setVisible(False) vBox.addWidget(bnt) for datePicker in datePickers: datePicker.setVisible(False) vBox.addWidget(datePicker) vBox.addStretch(1) buttonPanel.setLayout(vBox) return buttonPanel