HYTOS

# coding: utf-8

"""

    This is code table dialog module

"""

import os

import sys

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from AppDocData import AppDocData, MessageType

import CodeTable_UI

class QCodeTableDialog(QDialog):

    """

    This code table dialog class

    """

    CODE_TABLES = ('Nominal Diameter', 'Fluid Code', 'Insulation Purpose', 'PnID Number', 'Piping Materials Class', 'Unit Number', 'ValveOperCodes', 'EqpTagNames')

    def __init__(self, parent):

        QDialog.__init__(self, parent)

        self.currentCode = {}

        self.ui = CodeTable_UI.Ui_CodeTableDialog()

        self.ui.setupUi(self)

        self.ui.tableWidgetNominalDiameter.setSortingEnabled(True)

        self.ui.tableWidgetFluidCode.setSortingEnabled(True)

        self.ui.tableWidgetInsulationPurpose.setSortingEnabled(True)

        self.ui.tableWidgetPnIDNumber.setSortingEnabled(True)

        self.ui.tableWidgetPipingMaterialsClass.setSortingEnabled(True)

        self.ui.tableWidgetUnitNumber.setSortingEnabled(True)

        self.ui.tableWidgetValveOperCodes.setSortingEnabled(True)

        #DB Table명 기준으로 작성

        for table in QCodeTableDialog.CODE_TABLES:

            self.settingTable(table)

    '''

        @brief      Setting Table

        @author     kyouho

        @date       2018.07.10

    '''

    def settingTable(self, tableName):

        try:

            tableName = self.replaceText(tableName)

            docData = AppDocData.instance()

            table = self.findTableWidget(tableName)

            if table: table.horizontalHeader().setStretchLastSection(True)

            if tableName == "NominalDiameter":

                tableDatas = docData.getNomialPipeSizeData()

            else:

                tableDatas = docData.getCodeTable(tableName)

            if tableName == "NominalDiameter":

                self.insertTableWidgetNominalPipeSizeRow(tableDatas)

                table.cellChanged.connect(self.cellValueChanged)

                self.checkRowAndAddRow(tableName, table)

            else:

                table.setColumnCount(4)

                table.setHorizontalHeaderLabels(['uid', 'Code', 'Desc.', 'Allowables'])

                table.hideColumn(0)

                self.insertTableWidgetCommonRow(table, tableDatas)

                table.horizontalHeaderItem(1).setSizeHint(QSize(30, 30))

                table.cellChanged.connect(self.cellValueChanged)

                self.checkRowAndAddRow(tableName, table)

                self.setCurrentCode(table, tableName)

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

    '''

        @brief      Insert row NominalPipeSzie Tablewidget

        @author     kyouho

        @date       2018.07.10

    '''

    def insertTableWidgetNominalPipeSizeRow(self, pipeSizes):

        try:

            self.ui.tableWidgetNominalDiameter.setColumnCount(7)

            self.ui.tableWidgetNominalDiameter.setHorizontalHeaderLabels(['Code', 'Metric', 'Inch', 'InchStr', 'Allowables', 'MetricStr', 'Allowables'])

            self.ui.tableWidgetNominalDiameter.setRowCount(len(pipeSizes))

            row = 0

            for pipeSize in pipeSizes:

                item = QTableWidgetItem(pipeSize.code)

                item.setData(Qt.UserRole, pipeSize)

                self.ui.tableWidgetNominalDiameter.setItem(row, 0, item)

                self.ui.tableWidgetNominalDiameter.setItem(row, 1, QTableWidgetItem('' if pipeSize.metric is None else str(pipeSize.metric)))

                self.ui.tableWidgetNominalDiameter.setItem(row, 2, QTableWidgetItem('' if pipeSize.inch is None else str(pipeSize.inch)))

                self.ui.tableWidgetNominalDiameter.setItem(row, 3, QTableWidgetItem('' if pipeSize.inchStr is None else pipeSize.inchStr))

                self.ui.tableWidgetNominalDiameter.setItem(row, 4, QTableWidgetItem('' if pipeSize.allowable_inch_str is None else pipeSize.allowable_inch_str))

                self.ui.tableWidgetNominalDiameter.setItem(row, 5, QTableWidgetItem('' if pipeSize.metricStr is None else pipeSize.metricStr))

                self.ui.tableWidgetNominalDiameter.setItem(row, 6, QTableWidgetItem('' if pipeSize.allowable_metric_str is None else pipeSize.allowable_metric_str))

                row += 1

            self.ui.tableWidgetNominalDiameter.horizontalHeaderItem(0).setSizeHint(QSize(30, 30))

        except Exception as ex:

            from App import App

            message = 'error occured({}) 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)

    '''

        @brief      Insert row Common Tablewidget

        @author     kyouho

        @date       2018.07.10

    '''

    def insertTableWidgetCommonRow(self, table, tableDatas):

        try:

            table.setRowCount(len(tableDatas))

            row = 0

            for tableData in tableDatas:

                table.setItem(row, 0, QTableWidgetItem(tableData[0]))   # UID

                table.setItem(row, 1, QTableWidgetItem(tableData[1]))   # Name

                table.setItem(row, 2, QTableWidgetItem(tableData[2]))   # Description

                table.setItem(row, 3, QTableWidgetItem(tableData[3]))   # Allowables

                row += 1

        except Exception as ex:

            from App import App

            message = 'error occured({}) 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)

    '''

        @brief      Find TableWidget with Name

        @author     kyouhokyouho

        @date       2018.07.10

    '''

    def findTableWidget(self, tableName):

        tableName = self.replaceText(tableName)

        return self.findChild(QTableWidget, 'tableWidget' + tableName)

    '''

        @brief      key press event

        @author     kyouho

        @date       2018.07.10

    '''

    def keyPressEvent(self, e):

        try:

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

                _tabWidget = self.ui.tabWidget

                currentTabIndex = _tabWidget.currentIndex()

                tabText = self.replaceText(_tabWidget.tabText(currentTabIndex))

                table = self.findTableWidget(tabText)

                if table:

                    selectedIndexes = table.selectedIndexes()

                    selectedRows = [item.row() for item in selectedIndexes]

                    model = table.model()

                    rowsIndex = []

                    for row in selectedRows:

                        rowsIndex.append(row)

                    #중복 제거

                    rowsIndex = list(set(rowsIndex))

                    rowsIndex.reverse()

                    if tabText != "NominalDiameter":

                        for row in rowsIndex:

                            table.hideRow(row)

                            """

                            uid = table.item(row, 0).text()

                            self.removeUID[uid] = tabText

                            model.removeRow(row)

                            """

                        self.checkRowAndAddRow(tabText, table)

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

    '''

        @brief      Add new row

        @author     kyouho

        @date       2018.07.10

    '''

    def checkRowAndAddRow(self, tableName, table):

        try:

            rowCount = table.rowCount()

            result = True

            if tableName != "NominalDiameter":

                for row in range(rowCount):

                    if table.isRowHidden(row): continue

                    code = table.item(row, 1).text()

                    if not code:

                        result = False

                if result:

                    table.cellChanged.disconnect(self.cellValueChanged)

                    table.setRowCount(rowCount + 1)

                    table.setItem(rowCount, 0, QTableWidgetItem(''))

                    table.setItem(rowCount, 1, QTableWidgetItem(''))

                    table.setItem(rowCount, 2, QTableWidgetItem(''))

                    table.setItem(rowCount, 3, QTableWidgetItem(''))

                    table.cellChanged.connect(self.cellValueChanged)

            else:

                columnCount = table.columnCount()

                for row in range(rowCount):

                    if not result:

                        break

                    for columnIndex in range(columnCount):

                        if not table.item(row, columnIndex).text():

                            result = False

                            break

                if result:

                    table.setRowCount(rowCount + 1)

                    table.cellChanged.disconnect(self.cellValueChanged)

                    for columnIndex in range(columnCount):

                        table.setItem(rowCount, columnIndex, QTableWidgetItem(''))    

                    table.cellChanged.connect(self.cellValueChanged)

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

    '''

        @brief      cellValueChange event

        @author     kyouho

        @date       2018.07.10

    '''

    def cellValueChanged(self, row, column):

        try:

            _tabWidget = self.ui.tabWidget

            currentTabIndex = _tabWidget.currentIndex()

            tabText = self.replaceText(_tabWidget.tabText(currentTabIndex))

            table = self.findTableWidget(tabText)

            if tabText != "NominalDiameter":

                item = table.item(row, 1)

                code = item.text()

                if column == 1:

                    result = self.isExistCode(table, code)

                    if result:

                        self.checkRowAndAddRow(tabText, table)

                        self.setCurrentCode(table, tabText)

                    else:

                        QMessageBox.warning(self, '알림', '같은 코드가 존재합니다.')

                        item.setText(self.currentCode[tabText][row])

                elif column == 2:

                    table.resizeColumnToContents(2)

                else:

                    table.resizeColumnToContents(3)

            else:

                self.checkRowAndAddRow(tabText, table)

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

    '''

        @brief      Check Duplicate Code 

        @author     kyouho

        @date       2018.07.10

    '''

    def isExistCode(self, table, editCode):

        try:

            if not editCode:

                return False

            rowCount = table.rowCount()

            codes = []

            for row in range(rowCount):

                if table.isRowHidden(row): continue

                code = table.item(row, 1).text()

                codes.append(code)

            count = codes.count(editCode)

            if count >=2:

                return False

            else:

                return True

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

    '''

        @brief      save current Code (self.currentCode)

        @author     kyouho

        @date       2018.07.10

    '''

    def setCurrentCode(self, table, tabText):

        try:

            self.currentCode[tabText] = {}

            rowCount = table.rowCount()

            res = {}

            for row in range(rowCount):

                code = table.item(row, 1).text()

                res[row] = code

            self.currentCode[tabText] = res

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

    '''

        @brief      replaceTextForCodeTable

        @author     kyouho

        @date       2018.07.12

    '''

    def replaceText(self, text):

        return text.replace(' ','').replace('&&', 'n')

    '''

        @brief      save codes

        @author     kyouho

        @date       2018.07.12

    '''

    def accept(self):

        from CodeTables import CodeTable

        for table in QCodeTableDialog.CODE_TABLES:

            if table != 'Nominal Diameter':

                self.saveCommonCodeData(table)

        self.saveNomialPipeSize()

        CodeTable.clearTables()

        QDialog.accept(self)

    '''

        @brief      save common code data

        @author     kyouho

        @date       2018.07.12

    '''

    def saveCommonCodeData(self, tableName):

        datas = []

        try:

            tableName = self.replaceText(tableName)

            table = self.findTableWidget(tableName)

            rowCount = table.rowCount()

            for row in range(rowCount):

                if table.isRowHidden(row):

                    uid, code, description, allowables = '-1',table.item(row, 1).text(),'',table.item(row, 0).text()

                else:

                    uid = table.item(row, 0).text()

                    code = table.item(row, 1).text()

                    description = table.item(row, 2).text() if table.item(row, 2) is not None else ''

                    allowables = table.item(row, 3).text() if table.item(row, 3) is not None else ''

                if code:

                    datas.append((uid, code, description, allowables))

            docData = AppDocData.instance()

            docData.saveCommonCodeData(tableName.replace(' ', ''), datas)

        except Exception as ex:

            from App import App

            message = 'error occured({}) 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)

    '''

        @brief      save common code data

        @author     kyouho

        @date       2018.07.16

    '''

    def saveNomialPipeSize(self):

        pipeSizes = []

        try:

            docData = AppDocData.instance()

            from NominalPipeSize import NominalPipeSize

            table = self.ui.tableWidgetNominalDiameter

            rowCount = table.rowCount()

            for row in range(rowCount):

                pipe_size = table.item(row, 0).data(Qt.UserRole)

                pipe_size.code = table.item(row, 0).text()

                pipe_size.metric = float(table.item(row, 1).text()) if table.item(row, 1).text() != '' else None

                pipe_size.inch = float(table.item(row, 2).text()) if table.item(row, 2).text() != '' else None

                pipe_size.inchStr = table.item(row, 3).text()

                pipe_size.allowable_inch_str = table.item(row, 4).text()

                pipe_size.metricStr = table.item(row, 5).text()

                pipe_size.allowable_metric_str = table.item(row, 6).text()

                pipeSizes.append(pipe_size)

            docData.insertNomialPipeSize(pipeSizes)

        except Exception as ex:

            from App import App

            message = 'error occured({}) 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)