Source code for tools.UI_databank
#!/usr/bin/python3
# -*- coding: utf-8 -*-
"""Pychemqt, Chemical Engineering Process simulator
Copyright (C) 2009-2025, Juan José Gómez Romera <jjgomera@gmail.com>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
.. include:: UI_databank.rst
Module with chemical compounds database management
* :class:`UI_databank_widget`: Database widget, to use in whatever need: \
database dialog, proyect component list definition
* :class:`UI_databank`: Database dialog
API reference
-------------
"""
import os
from tools.qt import QtGui, QtWidgets
from lib.config import IMAGE_PATH
from lib import sql
from UI import viewComponents
[docs]
class UI_databank_widget(QtWidgets.QWidget):
"""Database widget, to use in whatever need: database dialog, proyect
component list definnition"""
[docs]
def __init__(self, parent=None):
super().__init__(parent)
self.__TEXTSTATUS__ = self.tr("Launched component database aplication")
gridLayout = QtWidgets.QGridLayout(self)
layoutTitle = QtWidgets.QHBoxLayout()
layoutTitle.setSpacing(5)
self.buttonNew = QtWidgets.QToolButton(self)
self.buttonNew.setToolTip(self.tr("Create new element"))
self.buttonNew.setIcon(QtGui.QIcon(QtGui.QPixmap(
os.path.join(IMAGE_PATH, "button", "fileNew.png"))))
self.buttonNew.clicked.connect(self.newComponent)
layoutTitle.addWidget(self.buttonNew)
self.buttonCopy = QtWidgets.QToolButton(self)
self.buttonCopy.setEnabled(False)
self.buttonCopy.setToolTip(self.tr("Clone this element"))
self.buttonCopy.setIcon(QtGui.QIcon(QtGui.QPixmap(
os.path.join(IMAGE_PATH, "button", "editCopy.png"))))
self.buttonCopy.clicked.connect(self.copyComponent)
layoutTitle.addWidget(self.buttonCopy)
self.buttonDelete = QtWidgets.QToolButton(self)
self.buttonDelete.setToolTip(self.tr("Delete element"))
self.buttonDelete.setIcon(QtGui.QIcon(QtGui.QPixmap(
os.path.join(IMAGE_PATH, "button", "editDelete.png"))))
self.buttonDelete.clicked.connect(self.deleteComponent)
self.buttonDelete.setEnabled(False)
layoutTitle.addWidget(self.buttonDelete)
gridLayout.addItem(layoutTitle, 1, 1)
gridLayout.addWidget(QtWidgets.QLabel(self.tr("Find")+": "), 1, 2)
self.Busqueda = QtWidgets.QLineEdit()
self.Busqueda.textChanged.connect(self.buscar)
gridLayout.addWidget(self.Busqueda, 1, 3)
self.siguiente = QtWidgets.QPushButton(self.tr("Next"))
self.siguiente.clicked.connect(self.Next)
gridLayout.addWidget(self.siguiente, 1, 4)
self.BaseDatos = QtWidgets.QTableWidget()
self.BaseDatos.setMinimumWidth(375)
self.BaseDatos.verticalHeader().hide()
self.BaseDatos.setEditTriggers(
QtWidgets.QAbstractItemView.EditTrigger.NoEditTriggers)
self.BaseDatos.setShowGrid(False)
self.BaseDatos.setRowCount(0)
self.BaseDatos.setColumnCount(3)
self.BaseDatos.setHorizontalHeaderItem(
0, QtWidgets.QTableWidgetItem("Id"))
self.BaseDatos.setHorizontalHeaderItem(
1, QtWidgets.QTableWidgetItem(self.tr("Name")))
self.BaseDatos.setHorizontalHeaderItem(
2, QtWidgets.QTableWidgetItem(self.tr("Formula")))
self.BaseDatos.setSelectionBehavior(
QtWidgets.QAbstractItemView.SelectionBehavior.SelectRows)
self.BaseDatos.horizontalHeader().setStretchLastSection(True)
self.BaseDatos.currentCellChanged.connect(self.checkButton)
self.BaseDatos.doubleClicked.connect(self.mostrarPropiedades)
gridLayout.addWidget(self.BaseDatos, 2, 1, 1, 4)
self.correctos = []
self.indice = 0
self.rellenar()
[docs]
def rellenar(self):
"""Fill in list with component from database"""
self.BaseDatos.setRowCount(0)
sql.databank.execute("select * from compuestos")
for i in sql.databank:
self.BaseDatos.setRowCount(self.BaseDatos.rowCount()+1)
self.BaseDatos.setItem(
i[0]-1, 0, QtWidgets.QTableWidgetItem(str(i[0])))
self.BaseDatos.setItem(i[0]-1, 1, QtWidgets.QTableWidgetItem(i[2]))
self.BaseDatos.setItem(i[0]-1, 2, QtWidgets.QTableWidgetItem(i[1]))
self.BaseDatos.setRowHeight(self.BaseDatos.rowCount()-1, 20)
sql.databank_Custom.execute("select * from compuestos")
for i in sql.databank_Custom:
filas = self.BaseDatos.rowCount()
self.BaseDatos.setRowCount(filas+1)
self.BaseDatos.setItem(
filas, 0, QtWidgets.QTableWidgetItem(str(i[0])))
self.BaseDatos.setItem(filas, 1, QtWidgets.QTableWidgetItem(i[2]))
self.BaseDatos.setItem(filas, 2, QtWidgets.QTableWidgetItem(i[1]))
self.BaseDatos.setRowHeight(self.BaseDatos.rowCount()-1, 20)
self.BaseDatos.resizeColumnsToContents()
[docs]
def buscar(self):
"""Search str at database"""
self.indice = 0
texto = self.Busqueda.text()
query = "select * from compuestos where "
query += f"name LIKE '{texto}' or formula LIKE '{texto}'"
sql.databank.execute(query)
self.correctos = []
for i in sql.databank:
self.correctos.append(i[0])
if self.correctos:
self.BaseDatos.setCurrentCell(self.correctos[self.indice]-1, 0)
[docs]
def Next(self):
"""Show next coincidence with search string"""
if self.indice < len(self.correctos)-1:
self.indice += 1
else:
self.indice = 0
self.BaseDatos.setCurrentCell(self.correctos[self.indice]-1, 0)
[docs]
def checkButton(self, indice):
"""Edit action are only available in custom database elements"""
self.buttonCopy.setEnabled(indice>=0)
if indice >= sql.N_comp:
self.buttonDelete.setEnabled(True)
else:
self.buttonDelete.setEnabled(False)
[docs]
def mostrarPropiedades(self):
"""Show properties of selected component"""
indice = self.currentIndice
if indice > 0:
dlg = viewComponents.View_Component(indice)
dlg.exec()
@property
def currentIndice(self):
"""Return index of selected element"""
value = self.BaseDatos.item(self.BaseDatos.currentRow(), 0).text()
return int(value)
[docs]
def newComponent(self):
"""Define new component"""
dlg = viewComponents.View_Component(0)
if dlg.exec():
self.rellenar()
[docs]
def copyComponent(self):
"""Copy component from database to can edit its values"""
sql.copyElement(self.currentIndice)
self.rellenar()
[docs]
def deleteComponent(self):
"""Delete selected component if its user defined"""
sql.deleteElement(self.currentIndice)
self.rellenar()
[docs]
class UI_databank(QtWidgets.QDialog):
"""Database dialog"""
[docs]
def __init__(self, parent=None):
super().__init__(parent)
self.setWindowTitle(self.tr("Components database"))
layout = QtWidgets.QVBoxLayout(self)
self.databank = UI_databank_widget()
layout.addWidget(self.databank)
self.buttonBox = QtWidgets.QDialogButtonBox(
QtWidgets.QDialogButtonBox.StandardButton.Cancel
| QtWidgets.QDialogButtonBox.StandardButton.Ok)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
layout.addWidget(self.buttonBox)
if __name__ == "__main__":
import sys
app = QtWidgets.QApplication(sys.argv)
Dialog = UI_databank()
Dialog.show()
sys.exit(app.exec())
