HYTOS

try:

    from PyQt5.QtCore import *

    from PyQt5.QtGui import *

    from PyQt5.QtWidgets import *

except ImportError:

    try:

        from PyQt4.QtCore import *

        from PyQt4.QtGui import *

    except ImportError:

        raise ImportError("ImageViewerQt: Requires PyQt5 or PyQt4.")

from AppDocData import AppDocData

import os

import sys

import SymbolBase

import symbol

import QSymbolEditorDialog

import QSymbolDisplayDialog

class QDirTreeWidget(QTreeWidget):

    #Add signal

    singleClicked = pyqtSignal(SymbolBase.SymbolBase)

    TREE_DATA_ROLE = Qt.ToolTipRole

    def __init__(self):

        QTreeWidget.__init__(self)

        self.setDragEnabled(True)   # enable drag

        self.initDirTreeWidget()

        self.isDoubleClicked = False

        self.itemDoubleClicked.connect(self.itemDoubleClickEvent)

        self.itemClicked.connect(self.itemClickEvent)

        self.setContextMenuPolicy(Qt.CustomContextMenu)

        self.customContextMenuRequested.connect(self.openContextMenu)

    def openContextMenu(self, position):

        indexes = self.selectedIndexes()

        itemPosition = self.mapTo(self, position)

        item = self.itemAt(itemPosition)

        sym = self.getSymbolByItemName(item, 0)

        text = item.text(0)

        if len(indexes) > 0:

            level = 0

            index = indexes[0]

            while index.parent().isValid():

                index = index.parent()

                level += 1

        if text.lower().endswith(".png"):

            menu = QMenu()

            editSymbolAction = QAction(self.tr("Edit Symbol"))

            editSymbolAction.triggered.connect(lambda: self.editSymbolActionClickEvent(item, 0))

            menu.addAction(editSymbolAction)

            displaySymbolAction = QAction(self.tr("Display Symbol"))

            displaySymbolAction.triggered.connect(lambda: self.displaySymbolActionClickEvent(sym.getType(), text))

            menu.addAction(displaySymbolAction)

            deleteSymbolAction = QAction(self.tr("Delete Symbol"))

            deleteSymbolAction.triggered.connect(lambda: self.deleteSymbolActionClickEvent(sym.getType(), text))

            menu.addAction(deleteSymbolAction)

            menu.exec_(self.viewport().mapToGlobal(position))

    def editSymbolActionClickEvent(self, item, columNo):

        self.showSymbolEditorDialog(item, columNo)

    def displaySymbolActionClickEvent(self, itemType, itemName):

        project = AppDocData.instance().getCurrentProject()

        image = QImage(project.getImageFilePath()+"/"+itemType+"/"+itemName, "PNG") #itemName includes ".png"

        dialog = QSymbolDisplayDialog.QSymbolDisplayDialog(image)

        dialog.showDialog()

    def deleteSymbolActionClickEvent(self, itemType, itemName):

        msg = QMessageBox()

        msg.setIcon(QMessageBox.Critical)

        msg.setText("선택한 심볼을 삭제하시겠습니까?\n삭제된 심볼은 복구할 수 없습니다.")

        msg.setWindowTitle("심볼 삭제")

        msg.setStandardButtons(QMessageBox.Ok|QMessageBox.Cancel)

        #msg.buttonClicked.connect(lambda: self.handleDeleteSymbolAction(itemName))

        result = msg.exec_()

        self.handleDeleteSymbolAction(result, itemType, itemName)

    def handleDeleteSymbolAction(self, result, itemType, itemName):

        print("handle")

        if result == QMessageBox.Ok:

            project = AppDocData.instance().getCurrentProject()

            imagePath = project.getImageFilePath() + "/" + itemType + "/" + itemName # itemName includes ".png"

            if os.path.exists(imagePath):

                os.remove(imagePath)

            svgPath = AppDocData.instance().getSvgFilePath() + "/" + itemType + "/"  + itemName.replace(".png", ".svg")

            if os.path.exists(svgPath):

                os.remove(svgPath)

            AppDocData.instance().deleteSymbol(itemName.replace(".png", ""))

            self.initDirTreeWidget()

        else:

            pass

    def initDirTreeWidget(self):

        project = AppDocData.instance().getCurrentProject()

        if project is not None:

            self.clear()

            print("Project : " + project.getName())

            projectPath = project.getPath().replace("\\", "/")

            self.makeChildDir()

            self.loadSymbolInfo()

            #self.loadDirectoryInfo(project.getImageFilePath(), self)

            self.expandAll()

    '''

        @brief      Load Symbol Info and add TreeItem with DB

        @author     Jeongwoo

        @date       18.04.20

    '''

    def loadSymbolInfo(self):

        '''ss'''

        SVG_PATH_ROOT = AppDocData.instance().getCurrentProject().getSvgFilePath()

        symbolTypeList = AppDocData.instance().getSymbolTypeList()

        for symbolType in symbolTypeList:

            parent = QTreeWidgetItem(self, [symbolType])

            symbolList = AppDocData.instance().getSymbolListByQuery('type', symbolType)

            #symbolNameList = AppDocData.instance().getSymbolNameListByType(symbolType)

            for symbol in symbolList:

                symbolItem = QTreeWidgetItem(parent, [symbol.getName()])

                symbolItem.setData(0, self.TREE_DATA_ROLE, symbol) ## ADD DATA

                svgPath = SVG_PATH_ROOT + "/" + symbolType + "/" + symbol.getName() + ".svg"

                icon = QIcon(svgPath)

                symbolItem.setIcon(0, icon)

                symbolItem.svgFilePath = svgPath # save svg file path

    '''

        @brief      Make Directory

        @author     Jeongwoo

        @date       18.04.??

        @history    18.04.12    Jeongwoo    Add output, temp Directory

    '''

    def makeChildDir(self):

        project = AppDocData.instance().getCurrentProject()

        dbDir = project.getDbFilePath()

        if not os.path.exists(dbDir):

            os.makedirs(dbDir)

        imgDir = project.getImageFilePath()

        if not os.path.exists(imgDir):

            os.makedirs(imgDir)

        svgDir = project.getSvgFilePath()

        if not os.path.exists(svgDir):

            os.makedirs(svgDir)

        outputDir = project.getOutputPath()

        if not os.path.exists(outputDir):

            os.makedirs(outputDir)

        tempDir = project.getTempPath()

        if not os.path.exists(tempDir):

            os.makedirs(tempDir)

    #def loadDirectoryInfo(self, startPath, tree):

    #    for element in os.listdir(startPath):

    #        pathInfo = os.path.join(startPath, element)

    #        parentItem = QTreeWidgetItem(tree, [os.path.basename(element)])

    #        if pathInfo.endswith(".png"):

    #            icon = QIcon(pathInfo)

    #            parentItem.setIcon(0, icon)

    #            parentItem.svgFilePath = pathInfo  # save svg file path

    #        if os.path.isdir(pathInfo):

    #            self.loadDirectoryInfo(pathInfo, parentItem)

    def showSymbolEditorDialog(self, item, columnNo):

        try:

            sym = self.getSymbolByItemName(item, columnNo)

            if sym is not None:

                path = sym.getPath()

                image = QImage(path, "PNG")

                print("after image")

                symbolEditorDialog = QSymbolEditorDialog.QSymbolEditorDialog(self, image, AppDocData.instance().getCurrentProject(), sym)

                _ret = symbolEditorDialog.showDialog()

                self.initDirTreeWidget()

            else:

                QMessageBox.about(self, "알림", "심볼 데이터를 불러오는 중 에러가 발생했습니다.")

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

    def itemDoubleClickEvent(self, item, columnNo):

        self.isDoubleClicked = True

        itemName = item.text(columnNo)

        print(itemName + " : Double Clicked")

        self.showSymbolEditorDialog(item, columnNo)

        self.isDoubleClicked = False

    def itemClickEvent(self, item, columnNo):

        if self.isDoubleClicked == False:

            print(item.text(columnNo) + " : Single Clicked")

            sym = self.getSymbolByItemName(item, columnNo)

            if sym is not None:

                self.singleClicked.emit(sym)

            else:

                QMessageBox.about(self, "알림", "심볼 데이터를 불러오는 중 에러가 발생했습니다.")

    '''

        @brief      Get Symbol data by symbol name

        @author     Jeongwoo

        @date       18.04.20

    '''

    def getSymbolByItemName(self, item, columnNo):

        tmpItem = item

        itemName = item.text(columnNo)

        #if itemName.lower().endswith(".png"):

        #path = itemName

        #while tmpItem.parent() is not None:

        #    path = tmpItem.parent().text(columnNo) + "/" + path

        #    tmpItem = tmpItem.parent()

        #fullPath = AppDocData.instance().getCurrentProject().getImageFilePath() + "/" + path # path includes ".png"

        #name = itemName.replace(".png", "")

        name = itemName

        sym = AppDocData.instance().getSymbolByQuery("name", name)

        return sym

        #else:

        #    return None

    '''

        @brief  start drag

        @author humkyung

        @date   2018.04.17

        @history    18.04.20    Jeongwoo    Change Path in QPixmap

    '''

    def startDrag(self, dropAction):

        try:

            items = self.selectedItems()

            if items:

                symData = items[0].data(0, self.TREE_DATA_ROLE)

                pixmap = QPixmap(AppDocData.instance().getCurrentProject().getSvgFilePath() + '/'+symData.getType()+'/' +symData.getName()+ '.svg')

                mime = QMimeData()

                mime.setText('BALL VALVE WITH PLUG')

                drag = QDrag(self)

                drag.setMimeData(mime) 

                originalPoint = symData.getOriginalPoint()

                drag.setHotSpot(QPoint(int(originalPoint.split(",")[0]), int(originalPoint.split(",")[1])))

                drag.setPixmap(pixmap)        

                drag.exec(Qt.CopyAction)

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