class AddAssetDialog(QDialog, Ui_AddAssetDialog):
def __init__(self, symbol, isin='', name=''):
QDialog.__init__(self)
self.setupUi(self)
self.asset_id = None
self.SymbolEdit.setText(symbol)
self.isinEdit.setText(isin)
self.NameEdit.setText(name)
self.type_model = QSqlTableModel(db=db_connection())
self.type_model.setTable('asset_types')
self.type_model.select()
self.TypeCombo.setModel(self.type_model)
self.TypeCombo.setModelColumn(1)
self.data_src_model = QSqlTableModel(db=db_connection())
self.data_src_model.setTable('data_sources')
self.data_src_model.select()
self.DataSrcCombo.setModel(self.data_src_model)
self.DataSrcCombo.setModelColumn(1)
# center dialog with respect to main application window
parent = None
for widget in QApplication.topLevelWidgets():
if widget.objectName() == Setup.MAIN_WND_NAME:
parent = widget
if parent:
x = parent.x() + parent.width() / 2 - self.width() / 2
y = parent.y() + parent.height() / 2 - self.height() / 2
self.setGeometry(x, y, self.width(), self.height())
def accept(self):
self.asset_id = JalDB().add_asset(
self.SymbolEdit.text(), self.NameEdit.text(),
self.type_model.record(self.TypeCombo.currentIndex()).value("id"),
self.isinEdit.text(),
self.data_src_model.record(
self.DataSrcCombo.currentIndex()).value("id"))
super().accept()
class CurrencyComboBox(QComboBox):
changed = Signal(int)
def __init__(self, parent):
QComboBox.__init__(self, parent)
self.p_selected_id = 0
self.model = None
self.activated.connect(self.OnUserSelection)
self.query = QSqlQuery(db=db_connection())
self.query.prepare(f"SELECT id, symbol FROM currencies")
self.query.exec()
self.model = QSqlTableModel(db=db_connection())
self.model.setQuery(self.query)
self.model.select()
self.setModel(self.model)
self.setModelColumn(self.model.fieldIndex("symbol"))
def isCustom(self):
return True
def getId(self):
return self.p_selected_id
def setId(self, new_id):
if self.p_selected_id == new_id:
return
self.p_selected_id = new_id
name = readSQL("SELECT symbol FROM currencies WHERE id=:id",
[(":id", self.p_selected_id)])
if self.currentIndex() == self.findText(name):
return
self.setCurrentIndex(self.findText(name))
selected_id = Property(int, getId, setId, notify=changed, user=True)
def setIndex(self, index):
if index is not None:
self.selected_id = index
self.changed.emit(self.selected_id)
@Slot()
def OnUserSelection(self, _selected_index):
self.selected_id = self.model.record(self.currentIndex()).value("id")
self.changed.emit(self.selected_id)
class AbstractOperationDetails(QWidget):
dbUpdated = Signal()
def __init__(self, parent=None):
QWidget.__init__(self, parent)
self.model = None
self.table_name = ''
self.mapper = None
self.modified = False
self.name = "N/A"
self.operation_type = None
self.layout = QGridLayout(self)
self.layout.setContentsMargins(2, 2, 2, 2)
self.bold_font = QFont()
self.bold_font.setBold(True)
self.main_label = QLabel(self)
self.main_label.setFont(self.bold_font)
self.layout.addWidget(self.main_label, 0, 0, 1, 1, Qt.AlignLeft)
self.commit_button = QPushButton(load_icon("accept.png"), '', self)
self.commit_button.setToolTip(self.tr("Commit changes"))
self.commit_button.setEnabled(False)
self.revert_button = QPushButton(load_icon("cancel.png"), '', self)
self.revert_button.setToolTip(self.tr("Cancel changes"))
self.revert_button.setEnabled(False)
self.verticalSpacer = QSpacerItem(20, 40, QSizePolicy.Minimum,
QSizePolicy.Expanding)
self.horizontalSpacer = QSpacerItem(40, 20, QSizePolicy.Expanding,
QSizePolicy.Minimum)
def _init_db(self, table_name):
self.table_name = table_name
self.model = QSqlTableModel(parent=self, db=db_connection())
self.model.setTable(table_name)
self.model.setEditStrategy(QSqlTableModel.OnManualSubmit)
self.mapper = QDataWidgetMapper(self.model)
self.mapper.setModel(self.model)
self.mapper.setSubmitPolicy(QDataWidgetMapper.AutoSubmit)
self.model.dataChanged.connect(self.onDataChange)
self.commit_button.clicked.connect(self.saveChanges)
self.revert_button.clicked.connect(self.revertChanges)
def isCustom(self):
return True
def setId(self, id):
self.model.setFilter(f"id={id}")
self.mapper.setCurrentModelIndex(self.model.index(0, 0))
@Slot()
def onDataChange(self, _index_start, _index_stop, _role):
self.modified = True
self.commit_button.setEnabled(True)
self.revert_button.setEnabled(True)
@Slot()
def saveChanges(self):
if not self.model.submitAll():
logging.fatal(
self.tr("Operation submit failed: ") +
self.model.lastError().text())
return False
self.modified = False
self.commit_button.setEnabled(False)
self.revert_button.setEnabled(False)
self.dbUpdated.emit()
@Slot()
def revertChanges(self):
self.model.revertAll()
self.modified = False
self.commit_button.setEnabled(False)
self.revert_button.setEnabled(False)
def createNew(self, account_id=0):
if self.modified:
self.revertChanges()
logging.warning(
self.tr(
"Unsaved changes were reverted to create new operation"))
self.model.setFilter(f"{self.table_name}.id = 0")
new_record = self.prepareNew(account_id)
assert self.model.insertRows(0, 1)
self.model.setRecord(0, new_record)
self.mapper.toLast()
def prepareNew(self, account_id):
new_record = self.model.record()
new_record.setNull("id")
new_record.setValue("op_type", self.operation_type)
return new_record
def copyNew(self):
row = self.mapper.currentIndex()
new_record = self.copyToNew(row)
self.model.setFilter(f"{self.table_name}.id = 0")
assert self.model.insertRows(0, 1)
self.model.setRecord(0, new_record)
self.mapper.toLast()
def copyToNew(self, row):
new_record = self.model.record(row)
return new_record