HYTOS

# coding: utf-8

"""

    This is fluid code dialog module

"""

import sys

import sqlite3

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from AppDocData import AppDocData

import FluidCode_UI

class QFluidCodeDialog(QDialog):

    '''

        @brief  constructor

        @author humkyung

        @date   2018.07.03

        @history    2018.07.03  kyouho      Add getFluidCodeData logic

    '''

    def __init__(self, parent):

        QDialog.__init__(self, parent)

        self.ui = FluidCode_UI.Ui_DialogFluidCode()

        self.ui.setupUi(self)

        self.removeFluidCode = []

        self.currentCode = {}

        docData = AppDocData.instance()

        fluidCodes = docData.getFluidCodeData()

        table = self.ui.tableWidgetFluidCode

        table.setColumnCount(3)

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

        table.setRowCount(len(fluidCodes))

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

        row = 0

        for fluidCode in fluidCodes:

            table.setItem(row, 0, QTableWidgetItem(fluidCode.uid))

            table.setItem(row, 1, QTableWidgetItem(fluidCode.code))

            table.setItem(row, 2, QTableWidgetItem(fluidCode.description))

            row += 1

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

        if row != 0:

            table.resizeColumnToContents(2)

        table.hideColumn(0)

        table.cellChanged.connect(self.cellValueChanged)

        self.checkRowAndAddRow()

        self.setCurrentCode()

    '''

        @brief  save fluid codes to sqlite

        @author humkyungS

        @date   2018.07.03

        @history    2018.07.03 kyouho 

    '''

    def accept(self):

        table = self.ui.tableWidgetFluidCode

        fluidCodes = []

        try:

            rowCount = table.rowCount()

            for row in range(rowCount):

                data = self.getFluidCodeDataInRow(row)

                if data.code:

                    fluidCodes.append(data)   

            docData = AppDocData.instance()

            docData.setFluidCodeData(fluidCodes, self.removeFluidCode)

        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))

        QDialog.accept(self)

    '''

        @brief      key press event

        @author     kyouho

        @date       2018.07.03

    '''

    def keyPressEvent(self, e):

        try:

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

                table = self.ui.tableWidgetFluidCode

                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()

                for row in rowsIndex:

                    data = self.getFluidCodeDataInRow(row)

                    self.removeFluidCode.append(data)

                    model.removeRow(row)

                self.checkRowAndAddRow()

        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      get FluidCodeData in QtTable

        @author     kyouho

        @date       2018.07.03

    '''

    def getFluidCodeDataInRow(self, rowIndex):

        try:

            from FluidCodeData import FluidCodeData

            table = self.ui.tableWidgetFluidCode

            uid = table.item(rowIndex, 0)

            codeItem = table.item(rowIndex, 1)

            descriptionItem = table.item(rowIndex, 2)

            data = FluidCodeData(uid.text() ,codeItem.text(), descriptionItem.text())

        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))

        return data

    '''

        @brief      cellValueChange event

        @author     kyouho

        @date       2018.07.03

    '''

    def cellValueChanged(self, row, column):

        try:

            table = self.ui.tableWidgetFluidCode

            item = table.item(row, 1)

            code = item.text()

            if column == 1:

                result = self.isExistCode(code)

                if result:

                    self.checkRowAndAddRow()

                    self.setCurrentCode()

                else:

                    item.setText(self.currentCode[row])

            else:

                table.resizeColumnToContents(2)

        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.03

    '''

    def isExistCode(self, editCode):

        try:

            if not editCode:

                return False

            table = self.ui.tableWidgetFluidCode

            rowCount = table.rowCount()

            codes = []

            for row in range(rowCount):

                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.03

    '''

    def setCurrentCode(self):

        try:

            self.currentCode.clear()

            table = self.ui.tableWidgetFluidCode

            rowCount = table.rowCount()

            for row in range(rowCount):

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

                self.currentCode[row] = code

        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.04

    '''

    def checkRowAndAddRow(self):

        try:

            table = self.ui.tableWidgetFluidCode

            rowCount = table.rowCount()

            result = True

            for row in range(rowCount):

                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.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))