Example #1
0
 def _setModel(self, model):
     self._model = model
     self._listModel = ListModel(self._model, self, parent = self._window)
     self._tableModel = TableModel(self._model, self, parent = self._window)
     self._window.ui.listView.setModel(self._listModel)
     self._window.ui.tableView.setModel(self._tableModel)
     self._window.ui.pathEdit.clear()
     self._window.ui.ratioEdit.clear()
Example #2
0
class ModelManager:
    def __init__(self, window):
        self._window = window
        self._setModel(Model())
    
    def _setModel(self, model):
        self._model = model
        self._listModel = ListModel(self._model, self, parent = self._window)
        self._tableModel = TableModel(self._model, self, parent = self._window)
        self._window.ui.listView.setModel(self._listModel)
        self._window.ui.tableView.setModel(self._tableModel)
        self._window.ui.pathEdit.clear()
        self._window.ui.ratioEdit.clear()

    def addCurrency(self, cur):
        if self._model.hasCurrency(cur):
            return QtWidgets.QMessageBox.warning(self._window, "Некорректный ввод",
                "Данная валюта уже есть в списке.")
        num = self._model.numCurrencies()
        self._listModel.beginInsertRows(QtCore.QModelIndex(), num, num)
        self._tableModel.beginInsertRows(QtCore.QModelIndex(), num, num)
        self._tableModel.beginInsertColumns(QtCore.QModelIndex(), num, num)
        self._model.addCurrency(cur)
        self._tableModel.endInsertRows()
        self._tableModel.endInsertColumns()
        self._listModel.endInsertRows()
        self._window.ui.tableView.resizeColumnsToContents()

    def deleteCurrencyNum(self, num):
        self._listModel.beginRemoveRows(QtCore.QModelIndex(), num, num)
        self._tableModel.beginRemoveRows(QtCore.QModelIndex(), num, num)
        self._tableModel.beginRemoveColumns(QtCore.QModelIndex(), num, num)
        self._model.deleteCurrencyNum(num)
        self._tableModel.endRemoveRows()
        self._tableModel.endRemoveColumns()
        self._listModel.endRemoveRows()

    def renameCurrency(self, old, new):
        if old == new: return
        if self._model.hasCurrency(new):
            QtWidgets.QMessageBox.warning(self._window, "Некорректный ввод",
                "Данная валюта уже есть в списке.")
            return False
        self._model.renameCurrency(old, new)
        num = self._model.currencyIndex(new)
        self._listModel.dataChanged.emit(self._listModel.createIndex(num, 0),
            self._listModel.createIndex(num, 0))
        self._tableModel.headerDataChanged.emit(QtCore.Qt.Vertical, num, num)
        self._tableModel.headerDataChanged.emit(QtCore.Qt.Horizontal, num, num)
        self._window.ui.tableView.resizeColumnsToContents()
        return True

    def setRatio(self, cur1, cur2, ratio):
        num1 = self._model.currencyIndex(cur1)
        num2 = self._model.currencyIndex(cur2)
        self._model.setRatio(cur1, cur2, ratio)
        self._tableModel.dataChanged.emit(self._tableModel.createIndex(num1, num2),
            self._listModel.createIndex(num1, num2))
        self._tableModel.dataChanged.emit(self._tableModel.createIndex(num2, num1),
            self._listModel.createIndex(num2, num1))
        self._window.ui.tableView.resizeColumnsToContents()
        return True

    def speculation(self):
        if self._model.numCurrencies() > 0:
            return Speculation(self._model)
        return None

    def loadModel(self, filename):
        try:
            model = pickle.load(open(filename, "rb"))
            if not isinstance(model, Model):
                raise ValueError
        except:
            QtWidgets.QMessageBox.critical(self._window, "Ошибка чтения",
                "Не удалось открыть файл - неверный формат.")
        else:
            self._setModel(model)

    def saveModel(self, filename):
        try:
            pickle.dump(self._model, open(filename, "wb"))
        except:
            QtWidgets.QMessageBox.critical(self._window, "Ошибка записи",
                "Не удалось записать таблицу в файл.")
            raise