HYTOS

# coding: utf-8

"""

    This is area configuration module

"""

import os

import sys

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from functools import reduce, partial

from AppDocData import AppDocData, Config

import BMSetting_UI

class AttributeListModel(QStandardItemModel):

    def __init__(self, columListAll, parent=None, *args):

        QStandardItemModel.__init__(self, parent, *args)

        self.columListAll = columListAll

        for symbol in sorted(self.columListAll):

            items = [QStandardItem(symbol)]

            items[0].setEditable(False)

            self.appendRow(items)

        headers = [QStandardItem("Name"), QStandardItem("State")]

        for idx, header in enumerate(headers):

            header.setTextAlignment(Qt.AlignCenter)

            self.setHorizontalHeaderItem(idx, header)

class SymbolListModel(QStandardItemModel):

    def __init__(self, parent=None, *args):

        QStandardItemModel.__init__(self, parent, *args)

        self.SymbolList = [ "CAP", "DRAIN CUP", "EOL", "FLANGE", "FLANGE BLIND", "MONOFLANGE", "ORIFICE", "PLUG", "SCREW CAP",

            "3WAY", "4WAY", "ANGLE", "BALL", "BUTTERFLY", "CHECK", "DIAPHRAMVALVE", "GATE", "GLOBE", "NEEDLE", "VALVEPLUG",

            "CONTROL(ANGLE)", "CONTROL(BALL)", "CONTROL(BUTTERFLY)", "CONTROL(CHECK)", "CONTROL(GATE)", "CONTROL(GLOBE)", "CONTROL(NEEDLE)", "CONTROL(PLUG)",

            "MOV(ANGLE)", "MOV(BALL)", "MOV(BUTTERFLY)", "MOV(CHECK)", "MOV(GATE)", "MOV(GLOBE)", "MOV(NEEDLE)", "MOV(PLUG)", "CORIOLIS FLOW METER", "MAGNETIC FLOW METER", "MASS FLOW METER", "ULTRASONIC FLOW METER", "VORTEX FLOW METER", "OPC",

            "AIR TRAP", "TRAP", "GOOSE NECK VENT", "VENT", "SCREEN", "IP", "PG", "PP", "PSV", "RE", "RS", "SEAL", "EXPANSION JOINT", "FA", "SAMPLE CONNECTION", "SB", "SPACER", "STRAINER", "T STRAINER", "Y STRAINER", "BUCKET STRAINER", "CONE STRAINER", "TG", "HOSE CONNECTION", "OD" ]

        for symbol in self.SymbolList:

            items = [QStandardItem(symbol)]

            items[0].setEditable(False)

            self.appendRow(items)

        app_doc_data = AppDocData.instance()

        configs = app_doc_data.getConfigs(section='BM Symbol')

        for config in configs:

            if config.key not in self.SymbolList:

                items = [QStandardItem(config.key)]

                items[0].setEditable(False)

                self.appendRow(items)

        headers = [QStandardItem("Name")]

        for idx, header in enumerate(headers):

            header.setTextAlignment(Qt.AlignCenter)

            self.setHorizontalHeaderItem(idx, header)

class SymbolMappingModel(QStandardItemModel):

    """This is SymbolMapping Model class"""

    def __init__(self):

        """constructor"""

        QStandardItemModel.__init__(self)

        app_doc_data = AppDocData.instance()

        project = app_doc_data.getCurrentProject()

        if project is not None:

            self.clear()

            self.load_symbol_info()

            headers = [QStandardItem("Name"), QStandardItem("State")]

            for idx, header in enumerate(headers):

                header.setTextAlignment(Qt.AlignCenter)

                self.setHorizontalHeaderItem(idx, header)

    def load_symbol_info(self):

        """load symbol information and display it on tree view"""

        try:

            app_doc_data = AppDocData.instance()

            symbolTypeList = app_doc_data.getSymbolTypeList()

            for row, symbolType in enumerate(symbolTypeList):

                items = [QStandardItem(symbolType[2]), QStandardItem('')]

                items[0].setData(symbolType, Qt.UserRole)

                items[0].setEditable(False)

                items[0].setSelectable(False)

                items[1].setEditable(False)

                items[1].setSelectable(False)

                symbolList = app_doc_data.getSymbolListByType('UID', symbolType[0])

                for symbol in symbolList:

                    childs = [QStandardItem(symbol.getName()), QStandardItem('')]

                    childs[0].setData(symbol.getUid(), Qt.UserRole)

                    childs[0].setEditable(False)

                    _, svg = app_doc_data.read_symbol_shape(symbol.sName)

                    if svg:

                        symbol.pixmap = QPixmap()

                        symbol.pixmap.loadFromData(svg if isinstance(svg, bytes) else svg.encode())

                        icon = QIcon(symbol.pixmap)

                        childs[0].setIcon(icon)

                        childs[0].svgFilePath = None  # save svg file path

                    else:

                        svgPath = symbol.getSvgFileFullPath()

                        symbol.pixmap = QPixmap(svgPath)

                        icon = QIcon(symbol.pixmap)

                        childs[0].setIcon(icon)

                        childs[0].svgFilePath = svgPath  # save svg file path

                    items[0].appendRow(childs)

                items[0].sortChildren(0, Qt.AscendingOrder)

                self.appendRow(items)

        except Exception as ex:

            from App import App

            from AppDocData import MessageType

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

                      f'{sys.exc_info()[-1].tb_lineno}'

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

class BMSettingDialog(QDialog):

    def __init__(self, parent, columListAll):

        QDialog.__init__(self, parent)

        self.ui = BMSetting_UI.Ui_BMSettingDialog()

        self.ui.setupUi(self)

        self.delimiter = '!-!'

        self.delimiterCombine = '!@!'

        self.isAccepted = False

        self.symbol_mapping = {}

        self.attributes = []

        self.columListAll = columListAll

        self.symbolListModel = SymbolListModel()

        self.ui.tableViewSymbolList.setModel(self.symbolListModel)

        # for test

        '''

        self.lineEdit = QLineEdit()

        self.comboBox = QComboBox()

        self.ui.gridLayout_3.addWidget(self.lineEdit, 0, 1, 1, 1)

        self.ui.gridLayout_3.addWidget(self.comboBox, 0, 2, 1, 1)

        self.proxy = QSortFilterProxyModel(self)

        self.proxy.setSourceModel(self.symbolListModel)

        self.ui.tableViewSymbolList.setModel(self.proxy)

        self.comboBox.addItems(["Column {0}".format(x) for x in range(self.symbolListModel.columnCount())])

        self.lineEdit.textChanged.connect(self.on_lineEdit_textChanged)

        self.comboBox.currentIndexChanged.connect(self.on_comboBox_currentIndexChanged)

        self.horizontalHeader = self.ui.tableViewSymbolList.horizontalHeader()

        self.horizontalHeader.sectionClicked.connect(self.on_view_horizontalHeader_sectionClicked)

        '''

        # uo to here

        self.symbolMappingModel = SymbolMappingModel()

        self.symbolMappingModel.invisibleRootItem()

        self.ui.treeViewSymbolMapping.setModel(self.symbolMappingModel)

        self.ui.treeViewSymbolMapping.expandAll()

        self.attributeListModel = AttributeListModel(self.columListAll)

        self.ui.tableViewAttributeList.setModel(self.attributeListModel)

        self.add_check_box()

        self.load_BM_setting()

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

        self.ui.treeViewSymbolMapping.header().setStretchLastSection(True)

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

        self.ui.tableViewSymbolList.setSizeAdjustPolicy(QAbstractScrollArea.AdjustToContents)

        self.ui.treeViewSymbolMapping.setSizeAdjustPolicy(QAbstractScrollArea.AdjustToContents)

        self.ui.tableViewAttributeList.setSizeAdjustPolicy(QAbstractScrollArea.AdjustToContents)

        self.ui.treeViewSymbolMapping.resizeColumnToContents(0)

        self.ui.treeViewSymbolMapping.setColumnWidth(1, 50)

        self.ui.tableViewSymbolList.resizeColumnsToContents()

        self.ui.tableViewAttributeList.resizeColumnsToContents()

        self.ui.tableViewAttributeList.setColumnWidth(1, 50)

        self.ui.tableViewSymbolList.clicked.connect(self.on_symbol_changed)

        self.ui.pushButtonAdd.clicked.connect(self.on_add_symbol)

        self.ui.pushButtonDelete.clicked.connect(self.on_delete_symbol)

        self.ui.buttonBox.accepted.connect(self.accept)

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

        self.ui.tableViewSymbolList.selectionModel().setCurrentIndex(self.symbolListModel.index(0, 0), QItemSelectionModel.Select)

        # for test

        #self.on_symbol_changed(self.proxy.index(0, 0))

        self.on_symbol_changed(self.symbolListModel.index(0, 0))

    # for test

    '''

    @pyqtSlot(int)

    def on_view_horizontalHeader_sectionClicked(self, logicalIndex):

        self.logicalIndex = logicalIndex

        self.menuValues = QMenu(self)

        self.signalMapper = QSignalMapper(self)  

        self.comboBox.blockSignals(True)

        self.comboBox.setCurrentIndex(self.logicalIndex)

        self.comboBox.blockSignals(True)

        valuesUnique = [self.symbolListModel.item(row, self.logicalIndex).text() for row in range(self.symbolListModel.rowCount())]

        actionAll = QAction("All", self)

        actionAll.triggered.connect(self.on_actionAll_triggered)

        self.menuValues.addAction(actionAll)

        self.menuValues.addSeparator()

        for actionNumber, actionName in enumerate(sorted(list(set(valuesUnique)))):              

            action = QAction(actionName, self)

            self.signalMapper.setMapping(action, actionNumber)  

            action.triggered.connect(self.signalMapper.map)  

            self.menuValues.addAction(action)

        self.signalMapper.mapped.connect(self.on_signalMapper_mapped)  

        headerPos = self.ui.tableViewSymbolList.mapToGlobal(self.horizontalHeader.pos())        

        posY = headerPos.y() + self.horizontalHeader.height()

        posX = headerPos.x() + self.horizontalHeader.sectionPosition(self.logicalIndex)

        self.menuValues.exec_(QPoint(posX, posY))

    @pyqtSlot()

    def on_actionAll_triggered(self):

        filterColumn = self.logicalIndex

        filterString = QRegExp("", Qt.CaseInsensitive, QRegExp.RegExp)

        self.proxy.setFilterRegExp(filterString)

        self.proxy.setFilterKeyColumn(filterColumn)

    @pyqtSlot(int)

    def on_signalMapper_mapped(self, i):

        stringAction = self.signalMapper.mapping(i).text()

        filterColumn = self.logicalIndex

        filterString = QRegExp(stringAction, Qt.CaseSensitive, QRegExp.FixedString)

        self.proxy.setFilterRegExp(filterString)

        self.proxy.setFilterKeyColumn(filterColumn)

    @pyqtSlot(str)

    def on_lineEdit_textChanged(self, text):

        search = QRegExp(text, Qt.CaseInsensitive, QRegExp.RegExp)

        self.proxy.setFilterRegExp(search)

    @pyqtSlot(int)

    def on_comboBox_currentIndexChanged(self, index):

        self.proxy.setFilterKeyColumn(index)

    '''

    # up to here

    def on_add_symbol(self):

        from BMSymbolAddDialog import QBMSymbolAddDialog

        dialog = QBMSymbolAddDialog(self, list(self.symbol_mapping.keys()))

        isAccepted, name = dialog.showDialog()

        if isAccepted:

            items = [QStandardItem(name)]

            items[0].setEditable(False)

            self.symbolListModel.appendRow(items)

            self.symbol_mapping[name] = []

    def on_delete_symbol(self):

        index = self.ui.tableViewSymbolList.currentIndex()

        selected_symbol = self.symbolListModel.itemFromIndex(index)

        if selected_symbol.text() not in self.symbolListModel.SymbolList:

            self.symbol_mapping.pop(selected_symbol.text())

            self.symbolListModel.removeRow(index.row())

    def load_BM_setting(self):

        app_doc_data = AppDocData.instance()

        for symbol in self.symbolListModel.SymbolList:

            self.symbol_mapping[symbol] = []

        configs = app_doc_data.getConfigs(section='BM Symbol')

        for config in configs:

            self.symbol_mapping[config.key] = config.value.split(',')

        configs = app_doc_data.getConfigs(section='Order', key='BM_LIST')

        if configs and len(configs) == 1:

            self.attributes = [attr.split(self.delimiterCombine)[0] for attr in configs[0].value.split(self.delimiter)]

        for row in range(self.attributeListModel.rowCount()):

            index = self.attributeListModel.index(row, 1)

            attr_index = self.attributeListModel.index(row, 0)

            attr = self.attributeListModel.itemFromIndex(attr_index).text()

            if attr in self.attributes:

                checkBox = self.ui.tableViewAttributeList.indexWidget(index)

                checkBox.setCheckState(Qt.Checked)

    def add_check_box(self):

        for row in range(self.symbolMappingModel.rowCount()):

            parent_index = self.symbolMappingModel.index(row, 0)

            child_count = self.symbolMappingModel.rowCount(parent_index)

            for child_row in range(child_count):

                checkBox = QCheckBox()

                index = self.symbolMappingModel.index(child_row, 1, parent_index)

                symbol_index = self.symbolMappingModel.index(child_row, 0, parent_index)

                symbol_id = self.symbolMappingModel.itemFromIndex(symbol_index).data(Qt.UserRole)

                checkBox.stateChanged.connect(partial(self.on_symbol_checkBox_changed, str(symbol_id)))

                self.ui.treeViewSymbolMapping.setIndexWidget(index, checkBox)

        for row in range(self.attributeListModel.rowCount()):

            index = self.attributeListModel.index(row, 1)

            checkBox = QCheckBox()

            attr_index = self.attributeListModel.index(row, 0)

            attr = self.attributeListModel.itemFromIndex(attr_index).text()

            checkBox.stateChanged.connect(partial(self.on_attribute_checkBox_changed, attr))

            self.ui.tableViewAttributeList.setIndexWidget(index, checkBox)

    def on_symbol_checkBox_changed(self, symbol_id, checkState):

        # for test

        #index = self.ui.tableViewSymbolList.currentIndex()

        #source_index = self.proxy.mapToSource(self.proxy.index(index.row(), index.column()))

        #selected_symbol = self.symbolListModel.itemFromIndex(source_index)

        # up to here

        index = self.ui.tableViewSymbolList.currentIndex()

        selected_symbol = self.symbolListModel.itemFromIndex(index)

        if checkState is int(Qt.Checked) and not symbol_id in self.symbol_mapping[selected_symbol.text()]:

            self.symbol_mapping[selected_symbol.text()].append(symbol_id)

        elif checkState is int(Qt.Unchecked) and symbol_id in self.symbol_mapping[selected_symbol.text()]:

            self.symbol_mapping[selected_symbol.text()].remove(symbol_id)

    def on_attribute_checkBox_changed(self, attr, checkState):

        if checkState is int(Qt.Checked) and not attr in self.attributes:

            self.attributes.append(attr)

        elif checkState is int(Qt.Unchecked) and attr in self.attributes:

            self.attributes.remove(attr)

    def on_symbol_changed(self, index):

        # for test

        #source_index = self.proxy.mapToSource(index)

        #selected_symbol = self.symbolListModel.itemFromIndex(source_index)

        # up to here

        selected_symbol = self.symbolListModel.itemFromIndex(index)

        for row in range(self.symbolMappingModel.rowCount()):

            category_index = self.symbolMappingModel.index(row, 0)

            child_count = self.symbolMappingModel.rowCount(category_index)

            for child_row in range(child_count):

                child_index = self.symbolMappingModel.index(child_row, 0, category_index)

                id2_symbol_item = self.symbolMappingModel.itemFromIndex(child_index)

                id2_symbol_uid = str(id2_symbol_item.data(Qt.UserRole))

                checkBox_index = self.symbolMappingModel.index(child_row, 1, category_index)

                checkBox = self.ui.treeViewSymbolMapping.indexWidget(checkBox_index)

                checkBox.setEnabled(True)

                if id2_symbol_uid in self.symbol_mapping[selected_symbol.text()]:

                    checkBox.setCheckState(Qt.Checked)

                else:

                    checkBox.setCheckState(Qt.Unchecked)

                    matches = [key for key, value in self.symbol_mapping.items() if id2_symbol_uid in value] # key != id2_symbol_item.text() and 

                    if matches:

                        checkBox.setEnabled(False)

    def reject(self):

        QDialog.reject(self)

    def accept(self):

        try:

            app_doc_data = AppDocData.instance()

            configs = []

            for key, value in self.symbol_mapping.items():

                if value:

                    configs.append(Config('BM Symbol', key, ','.join(value)))

            #self.attributes = ['UID', 'ITEM_NO', 'Quantity'] + self.attributes

            configs.append(Config('Order', 'BM_LIST', self.delimiter.join([attr + self.delimiterCombine + attr for attr in self.attributes])))

            app_doc_data.deleteConfigs(section='BM Symbol')

            app_doc_data.deleteConfigs(section='Order', key='BM_LIST')

            app_doc_data.saveConfigs(configs)

            self.isAccepted = True

            QDialog.accept(self)

        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)

    def showDialog(self):

        self.setWindowFlags(self.windowFlags() & ~Qt.WindowCloseButtonHint & ~Qt.WindowContextHelpButtonHint)

        self.exec_()

        return self.isAccepted