HYTOS

# coding: utf-8

""" This is text item edit dialog module """

import os

import sys, math

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from AppDocData import AppDocData

import ReplaceSymbol_UI

from EngineeringTextItem import QEngineeringTextItem

from SymbolSvgItem import SymbolSvgItem

from EngineeringLineItem import QEngineeringLineItem

from EngineeringVendorItem import QEngineeringVendorItem

from QtImageViewer import QtImageViewer

class QReplaceSymbolDialog(QDialog):

    """ This is symbol item replace and insert dialog class """

    #CONDITION_REPLACE = {'radioButtonHasCon':['item.has_connection'], 'radioButtonHasNoCon':['not item.has_connection'], 'radioButtonConLine':['False'], 'radioButtonAll':['True']}

    #CONDITION_INSERT = {'radioButtonHasCon':['conn.connetionItem'], 'radioButtonHasNoCon':['not conn.connectedItem'], 'radioButtonConLine':['conn.connectedItem and type(conn.connectedItem) is QEngineeringLineItem', 'conn.connectedItem.is_piping(True)'], 'radioButtonAll':['True']}

    def __init__(self, parent):

        QDialog.__init__(self, parent)

        self.mainWindow = parent

        self.view = self.mainWindow.graphicsView

        self.ui = ReplaceSymbol_UI.Ui_ReplaceSymbolDialog()

        self.ui.setupUi(self)

        self.ui.buttonBox.button(QDialogButtonBox.Save).setIcon(QIcon(':/newPrefix/OK.svg'))

        self.ui.buttonBox.button(QDialogButtonBox.Close).setIcon(QIcon(':/newPrefix/Remove.svg'))

        self.ui.pushButtonRun.clicked.connect(self.run)

        self.ui.buttonBox.rejected.connect(self.reject)

        self.ui.buttonBox.button(QDialogButtonBox.Save).clicked.connect(self.save_preset)

        self.ui.pushButtonPreset.clicked.connect(self.add_preset)

        # hide custom condition

        self.ui.radioButtonCustom.setHidden(True)

        self.ui.plainTextEdit.setHidden(True)

        self.ui.tableWidgetPreset.horizontalHeader().setStretchLastSection(True)

        self.ui.tableWidgetPreset.setColumnCount(5)

        self.ui.tableWidgetPreset.setHorizontalHeaderLabels(['', 'Find', 'Target', 'Action', 'Condition'])

        self.init_symbol_list()

        self.init_preset()

    def init_preset(self):

        """ load pesets """

        app_doc_data = AppDocData.instance()

        presets = app_doc_data.getSymbolPreset()

        self.ui.tableWidgetPreset.setRowCount(len(presets))

        rowCount = 0

        for preset in presets:

            item = QTableWidgetItem()

            item.setFlags(Qt.ItemIsUserCheckable | Qt.ItemIsSelectable | Qt.ItemIsEnabled)

            item.setCheckState(Qt.Unchecked)

            self.ui.tableWidgetPreset.setItem(rowCount, 0, item)

            item = QTableWidgetItem(preset[0])

            item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)

            self.ui.tableWidgetPreset.setItem(rowCount, 1, item)

            item = QTableWidgetItem(preset[1])

            item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)

            self.ui.tableWidgetPreset.setItem(rowCount, 2, item)

            item = QTableWidgetItem(preset[2])

            item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)

            self.ui.tableWidgetPreset.setItem(rowCount, 3, item)

            item = QTableWidgetItem(preset[3])

            item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)

            self.ui.tableWidgetPreset.setItem(rowCount, 4, item)

            rowCount += 1

    def add_preset(self):

        """ save current action into preset """

        find_symbol = self.ui.comboBoxFind.currentText()

        replace_symbol = self.ui.comboBoxReplace.currentText()

        condition = self.ui.buttonGroup_2.checkedButton().objectName()

        if condition == 'radioButtonCustom' or condition == 'radioButtonPreset':

            return False

            #conditions = [text for text in self.ui.plainTextEdit.toPlainText().split('\n')]

        else:

            pass

            #conditions = QReplaceSymbolDialog.CONDITION_REPLACE[condition] if self.ui.radioButtonReplace.isChecked() else QReplaceSymbolDialog.CONDITION_INSERT[condition]

        rowCount = self.ui.tableWidgetPreset.rowCount()

        self.ui.tableWidgetPreset.setRowCount(rowCount + 1)

        item = QTableWidgetItem()

        item.setFlags(Qt.ItemIsUserCheckable | Qt.ItemIsSelectable | Qt.ItemIsEnabled)

        item.setCheckState(Qt.Unchecked)

        self.ui.tableWidgetPreset.setItem(rowCount, 0, item)

        item = QTableWidgetItem(find_symbol)

        item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)

        self.ui.tableWidgetPreset.setItem(rowCount, 1, item)

        item = QTableWidgetItem(replace_symbol)

        item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)

        self.ui.tableWidgetPreset.setItem(rowCount, 2, item)

        item = QTableWidgetItem('Replace' if self.ui.radioButtonReplace.isChecked() else 'Insert')

        item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)

        self.ui.tableWidgetPreset.setItem(rowCount, 3, item)

        item = QTableWidgetItem(condition.replace('radioButton', ''))

        item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)

        self.ui.tableWidgetPreset.setItem(rowCount, 4, item)

    def init_symbol_list(self):

        app_doc_data = AppDocData.instance()

        symbol_list = app_doc_data.getTargetSymbolList(all=True)

        # for nozzle and equipment package

        self.ui.comboBoxFind.addItem('Equipment Package')

        for symbol in symbol_list:

            self.ui.comboBoxFind.addItem(symbol.getName())

            self.ui.comboBoxReplace.addItem(symbol.getName())

    def keyPressEvent(self, event):

        try:

            if event.key() == Qt.Key_Delete:

                try:

                    row = self.ui.tableWidgetPreset.selectedIndexes()[0].row()

                    self.ui.tableWidgetPreset.removeRow(row)

                except Exception as ex:

                    pass

            QDialog.keyPressEvent(self, event)

        except Exception as ex:

            from App import App

            from AppDocData import MessageType

            message = 'error occurred({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename,

                                                           sys.exc_info()[-1].tb_lineno)

            App.mainWnd().addMessage.emit(MessageType.Error, message)

            return None

    def run(self):

        from ReplaceInsertCommand import ReplaceInsertCommand

        from App import App

        scene = self.view.scene()

        find_symbol = self.ui.comboBoxFind.currentText()

        replace_symbol = self.ui.comboBoxReplace.currentText()

        replace_action = 'Replace' if self.ui.radioButtonReplace.isChecked() else 'Insert'

        condition = self.ui.buttonGroup_2.checkedButton().objectName().replace('radioButton', '')

        if condition == 'Custom':

            custom = self.ui.plainTextEdit.toPlainText()

        else:

            custom = None

        if condition == 'Preset':

            table = self.ui.tableWidgetPreset

            for index in range(table.rowCount()):

                if int(table.item(index, 0).checkState()) is int(Qt.Checked):

                    find_symbol = table.item(index, 1).text()

                    replace_symbol = table.item(index, 2).text()

                    replace_action = table.item(index, 3).text()

                    condition = table.item(index, 4).text()

                    break

        cmd = ReplaceInsertCommand()

        cmd.display_message.connect(App.mainWnd().onAddMessage)

        cmd.execute(scene, find_symbol, replace_symbol, replace_action, condition, custom=custom)

    def reject(self):

        """close a dialog"""

        QDialog.reject(self)

    def save_preset(self):

        """ save preset """

        presets = []

        table = self.ui.tableWidgetPreset

        for index in range(table.rowCount()):

            preset = [table.item(index, 1).text(), table.item(index, 2).text(), table.item(index, 3).text(), table.item(index, 4).text()]

            presets.append(preset)

        app_doc_data = AppDocData.instance()

        app_doc_data.saveSymbolPreset(presets)

        QDialog.accept(self)