def __init__(self, parent, patient):
QDialog.__init__(self)
self.patient = patient
self.parent = parent
self.save_button = QPushButton("Zapisz")
self.save_button.clicked.connect(self.saveForm)
self.spin_boxes = []
self.form_layout = QFormLayout()
treatment_layout = QHBoxLayout()
self.date_field = QDateEdit()
spin_box = QDoubleSpinBox()
spin_box.setRange(0.0, 1.0)
spin_box.setSingleStep(0.1)
self.treatment_field = spin_box
treatment_layout.addWidget(self.date_field)
treatment_layout.addWidget(self.treatment_field)
self.form_layout.addRow(treatment_layout)
buttons_layout = QHBoxLayout()
buttons_layout.addWidget(self.save_button)
self.layout = QVBoxLayout(self)
self.layout.addLayout(self.form_layout)
self.layout.addLayout(buttons_layout)
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)
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
def addRow(self) -> (QDateEdit, QTimeEdit):
# New row should be the successor of the last one
row: int = max(self.__layoutRows) + 1 if self.__layoutRows else 0
a = QDateEdit(self)
b = QTimeEdit(self)
a.setFixedHeight(30)
b.setFixedHeight(30)
a.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Fixed)
b.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Fixed)
self.body.gLayout.addWidget(a, row, 0)
self.body.gLayout.addWidget(b, row, 1)
# Create a button to remove row
removeBut = QPushButton(QIcon(QPixmap(':/resources/icons/close.png')),
'', self)
removeBut.setFixedSize(30, 30)
removeBut.setToolTip('Remove row')
# Lambda here is ok since it's called from main thread, so even if it's not a slot it's safe
removeBut.clicked.connect(lambda: self.removeRow(row))
self.body.gLayout.addWidget(removeBut, row, 2)
# Hide time if it's not wanted
if not self.__timeVisible:
b.hide()
if not self.__dateVisible:
a.hide()
self.__layoutRows.append(row)
return a, b
def __init__(self, parent=None): # + parent=None
super().__init__(parent) # + parent
self.setLayout(QFormLayout())
self.inputs = dict()
self.inputs['Customer Name'] = QLineEdit()
self.inputs['Customer Address'] = QPlainTextEdit()
self.inputs['Invoice Date'] = QDateEdit(date=QDate.currentDate(), calendarPopup=True)
self.inputs['Days until Due'] = QSpinBox()
for label, widget in self.inputs.items():
self.layout().addRow(label, widget)
self.line_items = QTableWidget(rowCount=10, columnCount=3)
self.line_items.setHorizontalHeaderLabels(['Job', 'Rate', 'Hours'])
self.line_items.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
self.layout().addRow(self.line_items)
for row in range(self.line_items.rowCount()):
for col in range(self.line_items.columnCount()):
if col > 0:
w = QSpinBox()
self.line_items.setCellWidget(row, col, w)
submit = QPushButton('Create Invoice', clicked=self.on_submit)
# + vvvvvv vvvvvvvvvvvvv
_print = QPushButton('Print Invoice', clicked=self.window().printpreviewDialog) # + _print, + self.window()
self.layout().addRow(submit, _print) # + _print
def __init__(self, conn):
'''
Parameters:
conn : connection
Connection to the database.
'''
super().__init__()
self.setWindowTitle('Aggiungi Cliente')
self.setWindowFlag(QtCore.Qt.WindowContextHelpButtonHint, False)
self.conn = conn
self.cursor = conn.cursor()
self.gen_layout = QVBoxLayout()
self.form_layout = QFormLayout()
self.form_layout.setRowWrapPolicy(QFormLayout.WrapLongRows)
# Widgets
self.nome_field = QLineEdit()
self.form_layout.addRow('Nome: ', self.nome_field)
self.cognome_field = QLineEdit()
self.form_layout.addRow('Cognome: ', self.cognome_field)
self.date_field = QDateEdit(QDate.currentDate())
self.date_field.setDisplayFormat("MM/dd/yyyy")
self.date_field.setCalendarPopup(True)
self.form_layout.addRow('Data di nascita (mm/gg/aaaa): ',
self.date_field)
self.phone_field = QLineEdit()
self.form_layout.addRow('Telefono (solo cifre): ', self.phone_field)
self.email_field = QLineEdit()
self.form_layout.addRow('E-Mail: ', self.email_field)
self.confirm_button = QPushButton('Conferma')
self.confirm_button.clicked.connect(self.post_cliente)
self.gen_layout.addLayout(self.form_layout)
self.gen_layout.addWidget(self.confirm_button)
self.setLayout(self.gen_layout)
class MeasurementsForm(QDialog):
def __init__(self, parent, patient):
QDialog.__init__(self)
self.patient = patient
self.parent = parent
self.setWindowTitle("Dodaj pomiar")
self.save_button = QPushButton("Zapisz")
self.save_button.clicked.connect(self.saveForm)
self.exit_button = QPushButton("Wyjdź")
self.exit_button.clicked.connect(self.accept)
self.spin_boxes = []
self.form_layout = QFormLayout()
measurement_layout = QHBoxLayout()
self.date_field = QDateEdit()
spin_box = QDoubleSpinBox()
spin_box.setRange(0.0, 10000.0)
spin_box.setSingleStep(0.1)
self.measurement_field = spin_box
measurement_layout.addWidget(self.date_field)
measurement_layout.addWidget(self.measurement_field)
self.form_layout.addRow(measurement_layout)
buttons_layout = QHBoxLayout()
buttons_layout.addWidget(self.exit_button)
buttons_layout.addWidget(self.save_button)
self.layout = QVBoxLayout(self)
self.layout.addLayout(self.form_layout)
self.layout.addLayout(buttons_layout)
def saveForm(self):
self.date = self.date_field.date().toPython()
self.value = round(self.measurement_field.value(), 1)
if any([self.date == m.date for m in self.patient.measurements]):
QMessageBox.warning(
self,
"Dodawanie pomiaru",
"Istnieje pomiar z tą datą",
)
return
measurement = create_measurement_for_patient(self.patient, self.date,
self.value)
self.parent.addMeasurement(measurement)
def addFutureTreatment(self):
row = QHBoxLayout()
row.addWidget(QLabel("Wybierz punkt podania lekarstwa"))
treatment_date = QDateEdit()
treatment_date.setMinimumDate(self.patient.measurements[-1].date)
row.addWidget(treatment_date)
remove_treatment_button = QToolButton()
remove_treatment_button.setIcon(QIcon("resources/icon/minus.png"))
remove_treatment_button.setStatusTip("Usuń")
row.addWidget(remove_treatment_button)
self.treatments.append(treatment_date)
self.form_layout.addRow(row)
def remove_treatment():
self.form_layout.removeRow(row)
self.treatments.remove(treatment_date)
remove_treatment_button.clicked.connect(remove_treatment)
def __init__(self, parent, patient):
QDialog.__init__(self)
self.patient = patient
self.parent = parent
self.setWindowTitle("Dodaj pomiar")
self.save_button = QPushButton("Zapisz")
self.save_button.clicked.connect(self.saveForm)
self.exit_button = QPushButton("Wyjdź")
self.exit_button.clicked.connect(self.accept)
self.spin_boxes = []
self.form_layout = QFormLayout()
measurement_layout = QHBoxLayout()
self.date_field = QDateEdit()
spin_box = QDoubleSpinBox()
spin_box.setRange(0.0, 10000.0)
spin_box.setSingleStep(0.1)
self.measurement_field = spin_box
measurement_layout.addWidget(self.date_field)
measurement_layout.addWidget(self.measurement_field)
self.form_layout.addRow(measurement_layout)
buttons_layout = QHBoxLayout()
buttons_layout.addWidget(self.exit_button)
buttons_layout.addWidget(self.save_button)
self.layout = QVBoxLayout(self)
self.layout.addLayout(self.form_layout)
self.layout.addLayout(buttons_layout)
def __init__(self, title, header, types, parameters, parent=None):
super().__init__(parent)
self.setWindowTitle(title)
self.inputs = []
self.types = types
layout = QFormLayout()
for desc, parameter, ty in zip(header, parameters, types):
if ty == 'foreignkey':
combobox = QComboBox()
for id, text in parameter:
combobox.addItem(text, userData=id)
layout.addRow(desc, combobox)
self.inputs.append(combobox)
elif ty == 'date':
date_edit = QDateEdit(parameter)
layout.addRow(desc, date_edit)
self.inputs.append(date_edit)
elif ty == 'text':
line_edit = QLineEdit(parameter)
layout.addRow(desc, line_edit)
self.inputs.append(line_edit)
elif ty == 'integer':
line_edit = QLineEdit(parameter)
line_edit.setValidator(QIntValidator())
layout.addRow(desc, line_edit)
self.inputs.append(line_edit)
else:
assert False
buttons = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel, Qt.Horizontal, self)
buttons.accepted.connect(self.accept)
buttons.rejected.connect(self.reject)
layout.addWidget(buttons)
self.setLayout(layout)
def new_date_input(parent, title_str, date=None):
"""
creates input with title
:param parent:
:param title_str:
:param hint_str:
:param only_numbers:
:return:
"""
widget = QWidget(parent)
widget.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Maximum)
# layout
layout = QVBoxLayout()
layout.setMargin(0)
layout.setSpacing(2)
# label
title = QLabel(widget)
title.setText(title_str)
title.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Maximum)
# text edit
widget.date_input = QDateEdit(widget)
if date is None:
widget.date_input.setDate(QDate.currentDate())
else:
widget.date_input.setDate(date)
# add to layout
layout.addWidget(title)
layout.addWidget(widget.date_input)
widget.setLayout(layout)
return widget
def __init__(self, *args, **kwargs):
super(ExchangeSpiderUI, self).__init__(*args, **kwargs)
layout = QVBoxLayout()
layout.setContentsMargins(QMargins(2, 0, 2, 1))
main_splitter = QSplitter(self)
main_splitter.setHandleWidth(1)
self.tree_widget = ExchangeLibTree(self)
main_splitter.addWidget(self.tree_widget)
action_splitter = QSplitter(Qt.Vertical, self)
action_splitter.setHandleWidth(1)
spider_widget = QWidget(self)
spider_widget.setAutoFillBackground(True)
palette = QPalette()
pix = QPixmap("images/spider_bg.png")
pix = pix.scaled(QSize(700, 700), Qt.KeepAspectRatio)
palette.setBrush(QPalette.Background, QBrush(pix))
spider_widget.setPalette(palette)
spider_layout = QVBoxLayout()
tips_layout = QHBoxLayout()
tips_layout.setSpacing(1)
tips_layout.addWidget(QLabel("当前交易所:", self))
self.spider_exchange_button = QPushButton("未选择", self)
tips_layout.addWidget(self.spider_exchange_button)
tips_layout.addWidget(QLabel(self))
tips_layout.addWidget(QLabel("当前操作:", self))
self.spider_action_button = QPushButton("未选择", self)
tips_layout.addWidget(self.spider_action_button)
tips_layout.addWidget(QLabel(self))
tips_layout.addWidget(QLabel("选择日期:", self))
self.spider_date_edit = QDateEdit(QDate.currentDate(), self)
self.spider_date_edit.setCalendarPopup(True)
self.spider_date_edit.setDisplayFormat("yyyy-MM-dd")
tips_layout.addWidget(self.spider_date_edit)
tips_layout.addWidget(QLabel(self))
self.spider_start_button = QPushButton("开始", self)
tips_layout.addWidget(self.spider_start_button)
tips_layout.addStretch()
spider_layout.addLayout(tips_layout)
self.spider_status = QLabel("等待开始抓取", self)
self.spider_status.setWordWrap(True)
self.spider_status.setAlignment(Qt.AlignCenter)
spider_layout.addWidget(self.spider_status)
spider_widget.setLayout(spider_layout)
action_splitter.addWidget(spider_widget)
# 解析部分
parser_widget = QWidget(self)
parser_widget.setAutoFillBackground(True)
palette = QPalette()
pix = QPixmap("images/parser_bg.png")
pix = pix.scaled(QSize(700, 700), Qt.KeepAspectRatio)
palette.setBrush(QPalette.Background, QBrush(pix))
parser_widget.setPalette(palette)
parser_layout = QVBoxLayout()
tips_layout = QHBoxLayout()
tips_layout.setSpacing(1)
tips_layout.addWidget(QLabel("当前交易所:", self))
self.parser_exchange_button = QPushButton("未选择", self)
tips_layout.addWidget(self.parser_exchange_button)
tips_layout.addWidget(QLabel(self))
tips_layout.addWidget(QLabel("当前操作:", self))
self.parser_action_button = QPushButton("未选择", self)
tips_layout.addWidget(self.parser_action_button)
tips_layout.addWidget(QLabel(self))
tips_layout.addWidget(QLabel("选择日期:", self))
self.parser_date_edit = QDateEdit(QDate.currentDate(), self)
self.parser_date_edit.setCalendarPopup(True)
self.parser_date_edit.setDisplayFormat("yyyy-MM-dd")
tips_layout.addWidget(self.parser_date_edit)
tips_layout.addWidget(QLabel(self))
self.parser_start_button = QPushButton("开始", self)
tips_layout.addWidget(self.parser_start_button)
tips_layout.addStretch()
parser_layout.addLayout(tips_layout)
self.parser_status = QLabel("等待开始解析", self)
self.parser_status.setAlignment(Qt.AlignCenter)
parser_layout.addWidget(self.parser_status)
parser_widget.setLayout(parser_layout)
action_splitter.addWidget(parser_widget)
main_splitter.addWidget(action_splitter)
main_splitter.setStretchFactor(0, 4)
main_splitter.setStretchFactor(1, 6)
layout.addWidget(main_splitter)
self.setLayout(layout)
main_splitter.setObjectName("mainSplitter")
action_splitter.setObjectName("actionSplitter")
self.spider_exchange_button.setObjectName("tipButton")
self.spider_action_button.setObjectName("tipButton")
self.spider_status.setObjectName("spiderStatus")
self.parser_exchange_button.setObjectName("tipButton")
self.parser_action_button.setObjectName("tipButton")
self.parser_status.setObjectName("parserStatus")
self.setStyleSheet(
"#mainSplitter::handle{background-color:rgba(50,50,50,100)}"
"#actionSplitter::handle{background-color:rgba(50,50,50,100)}"
"#tipButton{border:none;color:rgb(220,100,100)}"
"#spiderStatus,#parserStatus{font-size:16px;font-weight:bold;color:rgb(230,50,50)}"
)
class AcquistoForm(QDialog):
'''
Widget per l'inserimento di un acquisto immediato.
'''
def __init__(self, conn):
'''
Parameters:
conn : connection
Connection to the database.
'''
super().__init__()
self.setWindowTitle('Aggiungi Acquisto')
self.setWindowFlag(QtCore.Qt.WindowContextHelpButtonHint, False)
self.conn = conn
self.cursor = conn.cursor()
self.books = dict()
self.books_qt = dict()
self.gen_layout = QVBoxLayout()
self.form_layout = QFormLayout()
self.form_layout.setRowWrapPolicy(QFormLayout.WrapLongRows)
# Widgets
self.client_field = QLineEdit()
self.form_layout.addRow('Numero Cliente (facoltativo): ',
self.client_field)
self.dip_field = QLineEdit()
self.form_layout.addRow('Dipendente (CF): ', self.dip_field)
self.date_picker = QDateEdit(QDate.currentDate())
self.date_picker.setDisplayFormat("MM/dd/yyyy")
self.date_picker.setCalendarPopup(True)
self.form_layout.addRow('Data (mm/gg/aaaa): ', self.date_picker)
self.importo_field = QDoubleSpinBox()
self.importo_field.setMaximum(999999999.99)
self.form_layout.addRow('Importo: ', self.importo_field)
self.ins_layout = QHBoxLayout()
self.libro_field = QLineEdit()
self.quantita_field = QSpinBox()
self.quantita_field.setMinimum(1)
self.libro_button = QPushButton('Aggiungi')
self.libro_button.clicked.connect(self.aggiungi_libro)
self.ins_layout.addWidget(QLabel('Libro (ISBN): '))
self.ins_layout.addWidget(self.libro_field)
self.ins_layout.addWidget(QLabel('Quantità: '))
self.ins_layout.addWidget(self.quantita_field)
self.ins_layout.addWidget(self.libro_button)
self.labels = ['ISBN', 'Quantità', '']
self.table = QTableWidget(0, 3)
self.table.setHorizontalHeaderLabels(self.labels)
self.table.horizontalHeader().setSectionResizeMode(
0, QHeaderView.Stretch)
self.table.horizontalHeader().setSectionResizeMode(
1, QHeaderView.Stretch)
self.table.horizontalHeader().setSectionResizeMode(
2, QHeaderView.ResizeToContents)
self.confirm_button = QPushButton('Conferma')
self.confirm_button.clicked.connect(self.post_acquisto)
self.gen_layout.addLayout(self.form_layout)
self.gen_layout.addLayout(self.ins_layout)
self.gen_layout.addWidget(self.table)
self.gen_layout.addWidget(self.confirm_button)
self.setLayout(self.gen_layout)
def aggiungi_libro(self):
self.books[self.libro_field.text()] = self.quantita_field.value()
self.update_table()
def remove_libro(self, book):
self.books.pop(book, None)
self.update_table()
def update_table(self):
self.table.clearContents()
while self.table.rowCount() > 0:
self.table.removeRow(0)
for book in self.books.keys():
row = self.table.rowCount()
button = QPushButton('Rimuovi')
button.clicked.connect(lambda: self.remove_libro(book))
self.table.insertRow(row)
self.table.setItem(row, 0, QTableWidgetItem(book))
self.table.setItem(row, 1, QTableWidgetItem(str(self.books[book])))
self.table.setCellWidget(row, 2, button)
def verif_qty(self):
'''
Shows error messages based on the validity of inserted books, and
returns False, or returns True if all the books are ok to sell right now.
'''
if len(self.books.items()) == 0:
self._show_error('Non ci sono libri nell\'acquisto')
return 1
for book, qty in self.books.items():
self.cursor.execute(FIND_QUANTITA, (book, ))
result = self.cursor.fetchall()
if not result:
self._show_error('\'{}\' non è un libro valido.'.format(book))
return 2
stored = result[0][0]
if stored < qty:
return self.__show_are_you_sure(
'L\'acquisto richiede {} libri {}, ma ne sono presenti solo {}.\nNon sarà possibile sottrarre i libri acquistati.\nVuoi proseguire ugualmente?'
.format(qty, book, stored))
return 0
def verif_client_dip(self):
'''
Returns false and displays and error message if cliente and dipendente are
not valid tuples in the database, returns true if they are ok
'''
cliente = self.client_field.text()
if cliente:
if not cliente.isdigit():
self._show_error('Il codice del Cliente deve essere numerico.')
return 1
self.cursor.execute(FIND_CLIENTE, (cliente, ))
result = self.cursor.fetchall()
if not result or not result[0][0]:
self._show_error('Cliente {} non esiste.'.format(cliente))
return 2
dipendente = self.dip_field.text()
if not dipendente:
self._show_error('Il Dipendente non può essere vuoto.')
return 3
self.cursor.execute(FIND_DIPENDENTE, (dipendente, ))
result = self.cursor.fetchall()
if not result or not result[0][0]:
self._show_error('Dipendente {} non esiste.'.format(dipendente))
return 4
return 0
def post_acquisto(self):
if self.verif_client_dip():
return
should_show = self.verif_qty()
if should_show > 0:
return
cliente = self.client_field.text().strip() if self.client_field.text(
).strip() else None
importo = self.importo_field.value()
self.cursor.execute(
INSERT_ACQUISTO,
(self.date_picker.date().toString('MM/dd/yyyy'),
self.importo_field.value(), self.dip_field.text()))
new_id = self.cursor.fetchall()[0][0]
self.cursor.execute(INSERT_IMMEDIATO, (new_id, cliente))
for book in self.books.keys():
self.cursor.execute(INSERT_COMPRENDE,
(new_id, book, self.books[book]))
self.conn.commit()
if should_show == 0: self._show_confirm()
self.accept()
def __show_are_you_sure(self, msg=''):
dialog = _AreYouSureDialog(msg)
dialog.setWindowTitle('ATTENZIONE')
result = dialog.exec_()
return -1 if result == QDialog.Accepted else 3
def _show_confirm(self):
dialog = _ScalaAcquistiDialog(self.books, self.conn)
dialog.setWindowTitle('Rimozione libri')
dialog.exec_()
def _show_error(self, msg=''):
dialog = QMessageBox()
dialog.setWindowTitle('ERRORE')
dialog.setText(msg)
dialog.exec_()
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 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
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")
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)
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 setup(self):
for label, value in self.data:
if label is None and value is None:
# Separator: (None, None)
separator = QFrame()
separator.setFrameShape(QFrame.HLine)
separator.setFrameShadow(QFrame.Sunken)
self.formlayout.addRow(separator)
self.widgets.append(None)
continue
if label is None:
if isinstance(value, (list, tuple)):
field = PushLayout(value, self)
self.formlayout.addRow(field)
else:
img_fmt = tuple(['.'+str(bytes(ext).decode()) for ext in QImageReader.supportedImageFormats()])
if value.endswith(img_fmt):
# Image
pixmap = QPixmap(value)
lab = QLabel()
lab.setPixmap(pixmap)
self.formlayout.addRow(lab)
else:
# Comment
self.formlayout.addRow(QLabel(value))
self.widgets.append(None)
continue
if tuple_to_qfont(value) is not None:
field = FontLayout(value, self)
elif text_to_qcolor(value).isValid():
field = ColorLayout(QColor(value), self)
elif is_text_string(value):
if value in ['file', 'dir'] or value.startswith('file:'):
field = FileLayout(value, self)
elif value == 'slider' or value.startswith('slider:') \
or value.startswith('slider@'):
field = SliderLayout(value, self)
elif value == 'password':
field = QLineEdit(self)
field.setEchoMode(QLineEdit.Password)
elif value in ['calendar', 'calendarM'] \
or value.startswith(('calendar:', 'calendarM:')) \
or value.startswith(('calendar@', 'calendarM@')):
index = value.find('@')
if index != -1:
value, date = value[:index], value[index+1:]
else:
date = False
field = QCalendarWidget(self)
field.setVerticalHeaderFormat(field.NoVerticalHeader)
parsed = value.split(':')
if parsed[-1] == '':
field.setGridVisible(True)
parsed.pop(-1)
if parsed[0] == 'calendarM':
field.setFirstDayOfWeek(Qt.Monday)
if len(parsed) == 2:
field.setMaximumDate(datetime.date(*eval(parsed[1])))
elif len(parsed) == 3:
field.setMinimumDate(datetime.date(*eval(parsed[1])))
field.setMaximumDate(datetime.date(*eval(parsed[2])))
if date:
field.setSelectedDate(datetime.date(*eval(date)))
elif '\n' in value:
if value == '\n':
value = ''
for linesep in (os.linesep, '\n'):
if linesep in value:
value = value.replace(linesep, "\u2029") # paragraph separator
field = QTextEdit(value, self)
else:
field = QLineEdit(value, self)
elif isinstance(value, (list, tuple)) and is_text_string(value[0])\
and (value[0].startswith('0v')
or value[0].startswith('0h')
or value[0].startswith('1v')
or value[0].startswith('1h')):
if value[0][2:]:
binary = value[0][2:]
else:
binary = value[0][0] * len(value[1:])
if value[0][1] == 'v':
field = CheckVLayout(value[1:], binary, self)
elif value[0][1] == 'h':
field = CheckHLayout(value[1:], binary, self)
elif isinstance(value, (list, tuple)):
save_value = value
value = list(value) # always needed to protect self.data
selindex = value.pop(0)
if isinstance(selindex, int):
selindex = selindex - 1
if isinstance(value[0], (list, tuple)):
keys = [ key for key, _val in value ]
value = [ val for _key, val in value ]
else:
keys = value
if selindex in value:
selindex = value.index(selindex)
elif selindex in keys:
selindex = keys.index(selindex)
elif not isinstance(selindex, int):
print(f"Warning: '{selindex}' index is invalid (label: {label}, value: {value})" ,file=sys.stderr)
selindex = -1
if isinstance(save_value, list):
field = QComboBox(self)
field.addItems(value)
field.setCurrentIndex(selindex)
elif isinstance(save_value, tuple):
field = RadioLayout(value, selindex, self)
elif isinstance(value, bool):
field = QCheckBox(self)
field.setChecked(value)
elif isinstance(value, float):
field = QLineEdit(QLocale().toString(value), self)
field.setValidator(QDoubleValidator(field))
dialog = self.get_dialog()
dialog.register_float_field(field)
if SIGNAL is None:
field.textChanged.connect(dialog.float_valid)
else:
self.connect(field, SIGNAL('textChanged(QString)'), dialog.float_valid)
elif isinstance(value, int):
field = QSpinBox(self)
field.setRange(-1e9, 1e9)
field.setValue(value)
elif isinstance(value, datetime.datetime):
field = QDateTimeEdit(self)
field.setDateTime(value)
elif isinstance(value, datetime.date):
field = QDateEdit(self)
field.setDate(value)
elif isinstance(value, datetime.time):
field = QTimeEdit(self)
field.setTime(value)
else:
field = QLineEdit(repr(value), self)
# Eventually catching the 'countfield' feature and processing it
if label.startswith('n '):
label = label[2:]
if isinstance(field, QLineEdit) and is_text_string(value) or isinstance(field, QComboBox):
field = CountLayout(field)
else:
print(f"Warning: '{label}' doesn't support 'nfield' feature", file=sys.stderr)
# Eventually extracting tooltip from label and processing it
index = label.find('::')
if index != -1:
label, tooltip = label[:index], label[index+2:]
field.setToolTip(tooltip)
# Eventually catching the 'required' feature and processing it
if label.endswith(' *'):
label = label[:-1] + '<font color="red">*</font>'
if isinstance(field, (QLineEdit, QTextEdit, QComboBox, FileLayout, RadioLayout)):
dialog = self.get_dialog()
dialog.register_required_field(field)
else:
print(f"Warning: '{type(field)}' doesn't support 'required' feature", file=sys.stderr)
if isinstance(field, QLineEdit):
if SIGNAL is None:
field.textChanged.connect(dialog.required_valid)
else:
self.connect(field, SIGNAL('textChanged(QString)'),dialog.required_valid)
elif isinstance(field, QTextEdit):
if SIGNAL is None:
field.textChanged.connect(dialog.required_valid)
else:
self.connect(field, SIGNAL('textChanged()'),dialog.required_valid)
elif isinstance(field, QComboBox):
if SIGNAL is None:
field.currentIndexChanged.connect(dialog.required_valid)
else:
self.connect(field,SIGNAL('currentIndexChanged(QString)'),dialog.required_valid)
elif isinstance(field, FileLayout):
if SIGNAL is None:
field.lineedit.textChanged.connect(dialog.required_valid)
else:
self.connect(field.lineedit,SIGNAL('textChanged(QString)'),dialog.required_valid)
elif isinstance(field, RadioLayout):
if SIGNAL is None:
field.group.buttonClicked.connect(dialog.required_valid)
else:
self.connect(field.group, SIGNAL('buttonClicked(int)'),dialog.required_valid)
# Eventually setting the widget_color
if self.widget_color:
style = "background-color:" + self.widget_color + ";"
field.setStyleSheet(style)
if self.type == 'form':
self.formlayout.addRow(label, field)
elif self.type == 'questions':
self.formlayout.addRow(QLabel(label))
self.formlayout.addRow(field)
self.widgets.append(field)
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 setupUi(self):
self.setObjectName(_fromUtf8("MainWindow"))
self.resize(565, 358)
self.centralwidget = QWidget(self)
self.centralwidget.setObjectName(_fromUtf8("centralwidget"))
self.verticalLayout = QVBoxLayout(self.centralwidget)
self.verticalLayout.setObjectName(_fromUtf8("verticalLayout"))
self.horizontalLayout = QHBoxLayout()
self.horizontalLayout.setSizeConstraint(QLayout.SetDefaultConstraint)
self.horizontalLayout.setObjectName(_fromUtf8("horizontalLayout"))
spacerItem = QSpacerItem(40, 20, QSizePolicy.MinimumExpanding,
QSizePolicy.Minimum)
self.horizontalLayout.addItem(spacerItem)
self.searchTE = QLineEdit(self.centralwidget)
self.searchTE.setObjectName(_fromUtf8("searchTE"))
self.horizontalLayout.addWidget(self.searchTE)
self.searchBtn = QPushButton(self.centralwidget)
sizePolicy = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Fixed)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.searchBtn.sizePolicy().hasHeightForWidth())
self.searchBtn.setSizePolicy(sizePolicy)
self.searchBtn.setStyleSheet(_fromUtf8("border: none"))
self.searchBtn.setText(_fromUtf8(""))
icon = QIcon()
icon.addPixmap(
QPixmap(
_fromUtf8(
"../.designer/gitlab/ExcelToSql/icons/searchBtn.png")),
QIcon.Normal, QIcon.Off)
self.searchBtn.setIcon(icon)
self.searchBtn.setIconSize(QtCore.QSize(48, 24))
self.searchBtn.setObjectName(_fromUtf8("searchBtn"))
self.horizontalLayout.addWidget(self.searchBtn)
self.pushButton_2 = QPushButton(self.centralwidget)
sizePolicy = QSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.pushButton_2.sizePolicy().hasHeightForWidth())
self.pushButton_2.setSizePolicy(sizePolicy)
self.pushButton_2.setMaximumSize(QtCore.QSize(32, 16777215))
self.pushButton_2.setStyleSheet(_fromUtf8("color: blue"))
self.pushButton_2.setObjectName(_fromUtf8("pushButton_2"))
self.horizontalLayout.addWidget(self.pushButton_2)
# self.pushButton_2.clicked.connect(self.onRunBtnClick())
self.verticalLayout.addLayout(self.horizontalLayout)
self.gridLayout = QGridLayout()
self.gridLayout.setObjectName(_fromUtf8("gridLayout"))
self.dateEdit = QDateEdit(self.centralwidget)
self.dateEdit.setObjectName(_fromUtf8("dateEdit"))
self.gridLayout.addWidget(self.dateEdit, 0, 0, 1, 1)
self.dateEdit_2 = QDateEdit(self.centralwidget)
self.dateEdit_2.setObjectName(_fromUtf8("dateEdit_2"))
self.gridLayout.addWidget(self.dateEdit_2, 0, 1, 1, 1)
self.verticalLayout.addLayout(self.gridLayout)
self.tableWidget = QTableWidget(self.centralwidget)
sizePolicy = QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Minimum)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.tableWidget.sizePolicy().hasHeightForWidth())
self.tableWidget.setSizePolicy(sizePolicy)
self.tableWidget.setObjectName(_fromUtf8("tableWidget"))
self.tableWidget.setColumnCount(3)
self.tableWidget.setRowCount(0)
item = QTableWidgetItem()
self.tableWidget.setHorizontalHeaderItem(0, item)
item = QTableWidgetItem()
self.tableWidget.setHorizontalHeaderItem(1, item)
item = QTableWidgetItem()
self.tableWidget.setHorizontalHeaderItem(2, item)
self.verticalLayout.addWidget(self.tableWidget)
self.setCentralWidget(self.centralwidget)
self.statusbar = QStatusBar(self)
self.statusbar.setObjectName(_fromUtf8("statusbar"))
self.setStatusBar(self.statusbar)
item = QTableWidgetItem()
self.tableWidget.setHorizontalHeaderItem(0, item)
item = QTableWidgetItem()
self.tableWidget.setHorizontalHeaderItem(1, item)
item = QTableWidgetItem()
self.tableWidget.setHorizontalHeaderItem(2, item)
self.retranslateUi()
QtCore.QMetaObject.connectSlotsByName(self)
print("A")
self.pushButton_2.clicked.connect(self, QtCore.SLOT("onRunBtnClick()"))
print("B")
conn = sqlite3.connect("/home/eamon/Desktop/test.sqlite")
result = conn.execute("SELECT * FROM Sheet1")
for raw_number, raw_data in enumerate(result):
self.tableWidget.insertRow(raw_number)
# print(raw_number)
for column_number, data in enumerate(raw_data):
item = QTableWidgetItem(str(data))
self.tableWidget.setItem(raw_number, column_number, item)
# print("\t", column_number)
conn.close()
def createEditor(self, parent: QWidget, option: QStyleOptionViewItem,
index: QModelIndex):
return QDateEdit(parent)
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())
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)}"
)
def __init__(self, db_session, **kwargs):
super(MainPage, self).__init__()
self.db_session = db_session
# Create central widget that will contain everything
# (Bit odd design decision from QT, I would say, to separate widgets and
# layouts. Now we need to have central _widget_ to contain layout to contain
# widgets)
central_widget = QWidget()
central_widget.setStyleSheet("background: white;")
# Add central widget to main window
# Create main grid layout that will organize everything
main_grid = qt.QGridLayout(central_widget)
self.setLayout(main_grid)
main_grid.setSpacing(0)
main_grid.setMargin(0)
# Yet again interesting design decision. On QT you cannot set style for layout.
# Thus, we create widget that will contain layout and now we can set style for the widget.
widget_that_would_not_be_needed_if_qt_would_not_be_so_stupid = QWidget(
)
widget_that_would_not_be_needed_if_qt_would_not_be_so_stupid.setStyleSheet(
'''
QObject { background: #FF005AA1; }
QPushButton {
background: #FF005AA1;
color: white;
text-align: center;
border: none;
font-weight: 500;
font-size: 15px;
height: 48px;
width: 120px;
}
QPushButton:hover {
background-color: #FF014a82;
}
QPushButton:checked {
background: #FF01508c;
}
QPushButton:pressed {
color: #FF005AA1;
background: white;
}
QWidget{
background: #FF005AA1;
}
''')
# On left side we will have some (or at least one) button
left_side_buttons = QVBoxLayout(
widget_that_would_not_be_needed_if_qt_would_not_be_so_stupid)
left_side_buttons.setSpacing(0)
left_side_buttons.setMargin(0)
# And add it to main box the left most cell
main_grid.addWidget(
widget_that_would_not_be_needed_if_qt_would_not_be_so_stupid, 0, 0)
main_grid.setRowStretch(0, 1)
# Column 0 must not stretch
main_grid.setColumnStretch(0, 0)
# Column 1 must stretch
main_grid.setColumnStretch(1, 1)
# Separator line
line = qt.QFrame()
line.setFrameShape(qt.QFrame.HLine)
line.setFrameShadow(qt.QFrame.Sunken)
# Add scan button
self.scan_button = qt.QPushButton("Luo laskut")
self.scan_button.clicked.connect(self.luo_lasku_pressed_handler)
# Add widgets to container
left_side_buttons.addWidget(self.scan_button)
# Add separator line
line = qt.QFrame()
line.setFrameShape(qt.QFrame.HLine)
line.setFrameShadow(qt.QFrame.Sunken)
left_side_buttons.addWidget(line)
# Add stretch that will stretch to fill rest of the space
left_side_buttons.addStretch()
self.list_of_visits = qt.QTreeWidget()
self.list_of_visits.setColumnCount(3)
self.list_of_visits.setHeaderLabels(
["Käyntipäivä", "Nimi", "Terapian laji", "Lisätiedot"])
self.list_of_visits.setSortingEnabled(True)
# Allow multiselect
self.list_of_visits.setSelectionMode(
qt.QAbstractItemView.ExtendedSelection)
third_stupid_widget = QWidget()
right_side_layout = qt.QVBoxLayout(third_stupid_widget)
main_grid.addWidget(third_stupid_widget, 0, 1)
new_visit_date = QDateEdit(datetime.datetime.now())
new_visit_date.setCalendarPopup(True)
new_visit_date.setDisplayFormat('dd.MM.yyyy')
self.new_visit_customers_combobox = QComboBox()
new_visit_type = QLineEdit()
new_visit_type.setText("Terapia")
self.update_page()
self.new_visit_button = qt.QPushButton("Lisää käynti")
second_stupid_widget = QWidget()
new_visit_layout = QHBoxLayout(second_stupid_widget)
right_side_layout.addWidget(second_stupid_widget)
new_visit_layout.addWidget(new_visit_date)
new_visit_layout.addWidget(self.new_visit_customers_combobox)
new_visit_layout.addWidget(new_visit_type)
new_visit_layout.addWidget(self.new_visit_button)
right_side_layout.addWidget(self.list_of_visits)
# Add properties view
# main_grid.addWidget(qt.QPlainTextEdit(), 0, 2)
def new_visit_handler():
for _customer in self.customers:
if _customer.name == self.new_visit_customers_combobox.currentText(
):
_customer.new_visit(
cost=_customer.hour_price,
visit_length_min=45,
visit_date=new_visit_date.date().toPython(),
visit_type=new_visit_type.text(),
)
self.update_page()
db_session.commit()
self.new_visit_button.clicked.connect(new_visit_handler)
# create context menu
self.popMenu = qt.QMenu(self)
self.popMenu.addAction(
qt.QAction('Poista käynti',
self.list_of_visits,
triggered=self._delete_visit))
# set button context menu policy
self.list_of_visits.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.list_of_visits.customContextMenuRequested.connect(
self.on_context_menu)
class PrenotazioneForm(QDialog):
'''
Widget per l'inserimento di un acquisto immediato.
'''
def __init__(self, conn):
'''
Parameters:
conn : connection
Connection to the database.
'''
super().__init__()
self.setWindowTitle('Aggiungi Prenotazione')
self.setWindowFlag(QtCore.Qt.WindowContextHelpButtonHint, False)
self.conn = conn
self.cursor = conn.cursor()
self.books = dict()
self.books_qt = dict()
self.gen_layout = QVBoxLayout()
self.form_layout = QFormLayout()
self.form_layout.setRowWrapPolicy(QFormLayout.WrapLongRows)
# Widgets
self.client_field = QLineEdit()
self.form_layout.addRow('Numero Cliente: ', self.client_field)
self.dip_field = QLineEdit()
self.form_layout.addRow('Dipendente (CF): ', self.dip_field)
self.date_picker = QDateEdit(QDate.currentDate())
self.date_picker.setDisplayFormat("MM/dd/yyyy")
self.date_picker.setCalendarPopup(True)
self.form_layout.addRow('Data (mm/gg/aaaa): ', self.date_picker)
self.importo_field = QDoubleSpinBox()
self.importo_field.setMaximum(999999999.99)
self.form_layout.addRow('Importo: ', self.importo_field)
self.ins_layout = QHBoxLayout()
self.libro_field = QLineEdit()
self.quantita_field = QSpinBox()
self.quantita_field.setMinimum(1)
self.libro_button = QPushButton('Aggiungi')
self.libro_button.clicked.connect(self.aggiungi_libro)
self.ins_layout.addWidget(QLabel('Libro (ISBN): '))
self.ins_layout.addWidget(self.libro_field)
self.ins_layout.addWidget(QLabel('Quantità: '))
self.ins_layout.addWidget(self.quantita_field)
self.ins_layout.addWidget(self.libro_button)
self.labels = ['ISBN', 'Quantità', '']
self.table = QTableWidget(0, 3)
self.table.setHorizontalHeaderLabels(self.labels)
self.table.horizontalHeader().setSectionResizeMode(
0, QHeaderView.Stretch)
self.table.horizontalHeader().setSectionResizeMode(
1, QHeaderView.Stretch)
self.table.horizontalHeader().setSectionResizeMode(
2, QHeaderView.ResizeToContents)
self.info_label = QLabel(
'Le prenotazioni non riducono i libri disponibili.')
self.confirm_button = QPushButton('Conferma')
self.confirm_button.clicked.connect(self.post_prenotazione)
self.gen_layout.addLayout(self.form_layout)
self.gen_layout.addLayout(self.ins_layout)
self.gen_layout.addWidget(self.table)
self.gen_layout.addWidget(self.info_label)
self.gen_layout.addWidget(self.confirm_button)
self.setLayout(self.gen_layout)
def aggiungi_libro(self):
self.books[self.libro_field.text()] = self.quantita_field.value()
self.update_table()
def remove_libro(self, book):
self.books.pop(book, None)
self.update_table()
def update_table(self):
self.table.clearContents()
while self.table.rowCount() > 0:
self.table.removeRow(0)
for book in self.books.keys():
row = self.table.rowCount()
button = QPushButton('Rimuovi')
button.clicked.connect(lambda: self.remove_libro(book))
self.table.insertRow(row)
self.table.setItem(row, 0, QTableWidgetItem(book))
self.table.setItem(row, 1, QTableWidgetItem(str(self.books[book])))
self.table.setCellWidget(row, 2, button)
def verif_libri(self):
'''
Shows error messages based on the validity of inserted books, and
returns False, or returns True if all the books are ok to sell right now.
'''
if len(self.books.items()) == 0:
self._show_error('Non ci sono libri nell\'acquisto')
return False
for book in self.books.keys():
self.cursor.execute('SELECT * FROM libro WHERE isbn = %s',
(book, ))
result = self.cursor.fetchall()
if not result:
self._show_error('\'{}\' non è un libro valido.'.format(book))
return False
return True
def verif_client_dip(self):
'''
Returns false and displays and error message if cliente and dipendente are
not valid tuples in the database, returns true if they are ok
'''
cliente = self.client_field.text()
if not cliente:
self._show_error('Il Cliente è necessario per una Prenotazione.')
return False
if cliente:
if not cliente.isdigit():
self._show_error('Il codice del Cliente deve essere numerico.')
return False
self.cursor.execute(FIND_CLIENTE, (cliente, ))
result = self.cursor.fetchall()
if not result or not result[0][0]:
self._show_error('Cliente {} non esiste.'.format(cliente))
return False
dipendente = self.dip_field.text()
if not dipendente:
self._show_error('Il Dipendente non può essere vuoto.')
return False
self.cursor.execute(FIND_DIPENDENTE, (dipendente, ))
result = self.cursor.fetchall()
if not result or not result[0][0]:
self._show_error('Dipendente {} non esiste.'.format(dipendente))
return False
return True
def post_prenotazione(self):
if not self.verif_libri():
return
if not self.verif_client_dip():
return
cliente = self.client_field.text().strip() if self.client_field.text(
).strip() else None
importo = self.importo_field.value()
self.cursor.execute(
INSERT_ACQUISTO,
(self.date_picker.date().toString('MM/dd/yyyy'),
self.importo_field.value(), self.dip_field.text()))
new_id = self.cursor.fetchall()[0][0]
self.cursor.execute(INSERT_PRENOTAZIONE, (new_id, cliente))
for book in self.books.keys():
self.cursor.execute(INSERT_COMPRENDE,
(new_id, book, self.books[book]))
self.conn.commit()
self._show_confirm()
self.accept()
def _show_confirm(self):
dialog = QMessageBox()
dialog.setWindowTitle('Conferma Prenotazione')
msg = 'Registrato prenotazione per i seguenti libri:\n{}'
dialog.setText(
msg.format('\n'.join(
['{} x {}'.format(k, v) for k, v in self.books.items()])))
dialog.exec_()
def _show_error(self, msg=''):
dialog = QMessageBox()
dialog.setWindowTitle('ERRORE')
dialog.setText(msg)
dialog.exec_()
def __init__(self, conn):
'''
Parameters:
conn : connection
Connection to the database.
'''
super().__init__()
self.setWindowTitle('Aggiungi Acquisto')
self.setWindowFlag(QtCore.Qt.WindowContextHelpButtonHint, False)
self.conn = conn
self.cursor = conn.cursor()
self.books = dict()
self.books_qt = dict()
self.gen_layout = QVBoxLayout()
self.form_layout = QFormLayout()
self.form_layout.setRowWrapPolicy(QFormLayout.WrapLongRows)
# Widgets
self.client_field = QLineEdit()
self.form_layout.addRow('Numero Cliente (facoltativo): ',
self.client_field)
self.dip_field = QLineEdit()
self.form_layout.addRow('Dipendente (CF): ', self.dip_field)
self.date_picker = QDateEdit(QDate.currentDate())
self.date_picker.setDisplayFormat("MM/dd/yyyy")
self.date_picker.setCalendarPopup(True)
self.form_layout.addRow('Data (mm/gg/aaaa): ', self.date_picker)
self.importo_field = QDoubleSpinBox()
self.importo_field.setMaximum(999999999.99)
self.form_layout.addRow('Importo: ', self.importo_field)
self.ins_layout = QHBoxLayout()
self.libro_field = QLineEdit()
self.quantita_field = QSpinBox()
self.quantita_field.setMinimum(1)
self.libro_button = QPushButton('Aggiungi')
self.libro_button.clicked.connect(self.aggiungi_libro)
self.ins_layout.addWidget(QLabel('Libro (ISBN): '))
self.ins_layout.addWidget(self.libro_field)
self.ins_layout.addWidget(QLabel('Quantità: '))
self.ins_layout.addWidget(self.quantita_field)
self.ins_layout.addWidget(self.libro_button)
self.labels = ['ISBN', 'Quantità', '']
self.table = QTableWidget(0, 3)
self.table.setHorizontalHeaderLabels(self.labels)
self.table.horizontalHeader().setSectionResizeMode(
0, QHeaderView.Stretch)
self.table.horizontalHeader().setSectionResizeMode(
1, QHeaderView.Stretch)
self.table.horizontalHeader().setSectionResizeMode(
2, QHeaderView.ResizeToContents)
self.confirm_button = QPushButton('Conferma')
self.confirm_button.clicked.connect(self.post_acquisto)
self.gen_layout.addLayout(self.form_layout)
self.gen_layout.addLayout(self.ins_layout)
self.gen_layout.addWidget(self.table)
self.gen_layout.addWidget(self.confirm_button)
self.setLayout(self.gen_layout)
def __init__(self, parent=None):
AbstractOperationDetails.__init__(self, parent)
self.name = "Trade"
self.date_label = QLabel(self)
self.settlement_label = QLabel()
self.number_label = QLabel(self)
self.account_label = QLabel(self)
self.symbol_label = QLabel(self)
self.qty_label = QLabel(self)
self.price_label = QLabel(self)
self.fee_label = QLabel(self)
self.comment_label = QLabel(self)
self.main_label.setText(g_tr("TradeWidget", "Buy / Sell"))
self.date_label.setText(g_tr("TradeWidget", "Date/Time"))
self.settlement_label.setText(g_tr("TradeWidget", "Settlement"))
self.number_label.setText(g_tr("TradeWidget", "#"))
self.account_label.setText(g_tr("TradeWidget", "Account"))
self.symbol_label.setText(g_tr("TradeWidget", "Asset"))
self.qty_label.setText(g_tr("TradeWidget", "Qty"))
self.price_label.setText(g_tr("TradeWidget", "Price"))
self.fee_label.setText(g_tr("TradeWidget", "Fee"))
self.comment_label.setText(g_tr("TradeWidget", "Note"))
self.timestamp_editor = QDateTimeEdit(self)
self.timestamp_editor.setCalendarPopup(True)
self.timestamp_editor.setTimeSpec(Qt.UTC)
self.timestamp_editor.setFixedWidth(self.timestamp_editor.fontMetrics().width("00/00/0000 00:00:00") * 1.25)
self.timestamp_editor.setDisplayFormat("dd/MM/yyyy hh:mm:ss")
self.settlement_editor = QDateEdit(self)
self.settlement_editor.setCalendarPopup(True)
self.settlement_editor.setTimeSpec(Qt.UTC)
self.settlement_editor.setFixedWidth(self.settlement_editor.fontMetrics().width("00/00/0000") * 1.5)
self.settlement_editor.setDisplayFormat("dd/MM/yyyy")
self.account_widget = AccountSelector(self)
self.asset_widget = AssetSelector(self)
self.qty_edit = AmountEdit(self)
self.qty_edit.setAlignment(Qt.AlignRight)
self.price_edit = AmountEdit(self)
self.price_edit.setAlignment(Qt.AlignRight)
self.fee_edit = AmountEdit(self)
self.fee_edit.setAlignment(Qt.AlignRight)
self.number = QLineEdit(self)
self.comment = QLineEdit(self)
self.layout.addWidget(self.date_label, 1, 0, 1, 1, Qt.AlignLeft)
self.layout.addWidget(self.account_label, 2, 0, 1, 1, Qt.AlignLeft)
self.layout.addWidget(self.symbol_label, 3, 0, 1, 1, Qt.AlignLeft)
self.layout.addWidget(self.comment_label, 4, 0, 1, 1, Qt.AlignLeft)
self.layout.addWidget(self.timestamp_editor, 1, 1, 1, 1, Qt.AlignLeft)
self.layout.addWidget(self.account_widget, 2, 1, 1, 4)
self.layout.addWidget(self.asset_widget, 3, 1, 1, 4)
self.layout.addWidget(self.comment, 4, 1, 1, 4)
self.layout.addWidget(self.settlement_label, 1, 2, 1, 1, Qt.AlignRight)
self.layout.addWidget(self.settlement_editor, 1, 3, 1, 1, Qt.AlignLeft)
self.layout.addWidget(self.number_label, 1, 5, 1, 1, Qt.AlignRight)
self.layout.addWidget(self.qty_label, 2, 5, 1, 1, Qt.AlignRight)
self.layout.addWidget(self.price_label, 3, 5, 1, 1, Qt.AlignRight)
self.layout.addWidget(self.fee_label, 4, 5, 1, 1, Qt.AlignRight)
self.layout.addWidget(self.number, 1, 6, 1, 1)
self.layout.addWidget(self.qty_edit, 2, 6, 1, 1)
self.layout.addWidget(self.price_edit, 3, 6, 1, 1)
self.layout.addWidget(self.fee_edit, 4, 6, 1, 1)
self.layout.addWidget(self.commit_button, 0, 8, 1, 1)
self.layout.addWidget(self.revert_button, 0, 9, 1, 1)
self.layout.addItem(self.verticalSpacer, 6, 6, 1, 1)
self.layout.addItem(self.horizontalSpacer, 1, 6, 1, 1)
super()._init_db("trades")
self.mapper.setItemDelegate(TradeWidgetDelegate(self.mapper))
self.account_widget.changed.connect(self.mapper.submit)
self.asset_widget.changed.connect(self.mapper.submit)
self.mapper.addMapping(self.timestamp_editor, self.model.fieldIndex("timestamp"))
self.mapper.addMapping(self.settlement_editor, self.model.fieldIndex("settlement"))
self.mapper.addMapping(self.account_widget, self.model.fieldIndex("account_id"))
self.mapper.addMapping(self.asset_widget, self.model.fieldIndex("asset_id"))
self.mapper.addMapping(self.number, self.model.fieldIndex("number"))
self.mapper.addMapping(self.qty_edit, self.model.fieldIndex("qty"))
self.mapper.addMapping(self.price_edit, self.model.fieldIndex("price"))
self.mapper.addMapping(self.fee_edit, self.model.fieldIndex("fee"))
self.mapper.addMapping(self.comment, self.model.fieldIndex("note"))
self.model.select()
def getDateRangeWidget(self, minDate, maxDate):
dateRange = QDateEdit()
dateRange.setCalendarPopup(True)
dateRange.setDisplayFormat('MMM d yy')
dateRange.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter)
dateRange.setMaximumDate(maxDate)
dateRange.setMinimumDate(minDate)
return dateRange
class ClienteForm(QDialog):
'''
Dialog per l'inserimento di un ordine.
'''
def __init__(self, conn):
'''
Parameters:
conn : connection
Connection to the database.
'''
super().__init__()
self.setWindowTitle('Aggiungi Cliente')
self.setWindowFlag(QtCore.Qt.WindowContextHelpButtonHint, False)
self.conn = conn
self.cursor = conn.cursor()
self.gen_layout = QVBoxLayout()
self.form_layout = QFormLayout()
self.form_layout.setRowWrapPolicy(QFormLayout.WrapLongRows)
# Widgets
self.nome_field = QLineEdit()
self.form_layout.addRow('Nome: ', self.nome_field)
self.cognome_field = QLineEdit()
self.form_layout.addRow('Cognome: ', self.cognome_field)
self.date_field = QDateEdit(QDate.currentDate())
self.date_field.setDisplayFormat("MM/dd/yyyy")
self.date_field.setCalendarPopup(True)
self.form_layout.addRow('Data di nascita (mm/gg/aaaa): ',
self.date_field)
self.phone_field = QLineEdit()
self.form_layout.addRow('Telefono (solo cifre): ', self.phone_field)
self.email_field = QLineEdit()
self.form_layout.addRow('E-Mail: ', self.email_field)
self.confirm_button = QPushButton('Conferma')
self.confirm_button.clicked.connect(self.post_cliente)
self.gen_layout.addLayout(self.form_layout)
self.gen_layout.addWidget(self.confirm_button)
self.setLayout(self.gen_layout)
def verif_data(self):
if not self.nome_field.text().strip():
self._show_error('Il nome non può essere vuoto.')
return False
if not self.cognome_field.text().strip():
self._show_error('Il cognome non può essere vuoto.')
return False
phone = self.phone_field.text().strip()
email = self.email_field.text().strip()
if not phone and not email:
self._show_error(
'Almeno uno fra telefono ed e-mail deve essere non vuoto.')
return False
if phone and not re.match('[0-9]{3,15}', phone):
self._show_error('Numero di telefono non valido.')
return False
if email and not re.match('[A-Za-z0-9\.]{1,}@[A-Za-z0-9\.]{1,}',
email):
self._show_error('E-Mail non valida.')
return False
return True
def post_cliente(self):
if not self.verif_data():
return
nome = self.nome_field.text().strip()
cognome = self.cognome_field.text().strip()
data = self.date_field.date().toString('MM/dd/yyyy')
phone = self.phone_field.text().strip()
if not phone: phone = None
email = self.email_field.text().strip()
if not email: email = None
self.cursor.execute(INSERT_CLIENTE,
(nome, cognome, data, phone, email))
self.conn.commit()
self.accept()
def _show_error(self, msg=''):
dialog = QMessageBox()
dialog.setWindowTitle('ERRORE')
dialog.setText(msg)
dialog.exec_()