HYTOS

# coding: utf-8

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

import os

import sys

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

import sqlite3

from AppDocData import AppDocData

from AppDocData import Config

import TextItemEdit_UI

from EngineeringTextItem import QEngineeringTextItem

from SymbolSvgItem import SymbolSvgItem

from EngineeringLineItem import QEngineeringLineItem

class QTextItemEditDialog(QDialog):

    """ This is text item edit dialog class """

    def __init__(self, parent):

        QDialog.__init__(self, parent)

        self.mainWindow = parent

        self.ui = TextItemEdit_UI.Ui_TextItemEditDialog()

        self.ui.setupUi(self)

        self.ui.pushButtonFindText.clicked.connect(self.find_text)

        self.ui.pushButtonReplaceText.clicked.connect(self.replaceTextClicked)

        self.ui.checkBoxUID.stateChanged.connect(self.checkBoxChanged)

        self.ui.lineEditFindText.setFocus()

    def is_valid_uuid(self, text, version=4):

        """ Check if uuid_to_test is a valid UUID. """

        import uuid

        try:

            uuid_obj = uuid.UUID(text)  # , version=version)

        except:

            return False

        return str(uuid_obj) == text

    def checkBoxChanged(self, checkState):

        if checkState is int(Qt.Checked):

            self.ui.pushButtonReplaceText.setEnabled(False)

            self.ui.lineEditReplaceText.setEnabled(False)

        else:

            self.ui.pushButtonReplaceText.setEnabled(True)

            self.ui.lineEditReplaceText.setEnabled(True)

    def find_text(self):

        """find user input text"""

        self.ui.tableViewResult.setModel(None)

        text = self.ui.lineEditFindText.text()

        if not text: return

        text_items = []

        if int(self.ui.checkBoxUID.checkState()) is int(Qt.Checked):

            text_items = [item for item in self.mainWindow.graphicsView.scene.items()

                              if hasattr(item, 'uid') and str(item.uid).count(text)]

        else:

            text_items = [item for item in self.mainWindow.graphicsView.scene.items()

                              if issubclass(type(item), QEngineeringTextItem) and item.text().count(text)]

        if not text_items:

            QMessageBox.about(self, self.tr("Notice"), self.tr('Nothing Found.'))

            return

        model = QStandardItemModel(len(text_items), 1)

        model.setHorizontalHeaderLabels(['UID'])

        for row in range(len(text_items)):

            item = QStandardItem(str(text_items[row]))

            item.setData(text_items[row])

            model.setItem(row, 0, item)

        self.ui.tableViewResult.setModel(model)

        self.ui.tableViewResult.selectionModel().selectionChanged.connect(self.selection_changed)

        self.selectTextItem(text_items[0])

    def replaceTextClicked(self):

        """replace text"""

        self.find_text()

        text_find = self.ui.lineEditFindText.text()

        replace = self.ui.lineEditReplaceText.text()

        model = self.ui.tableViewResult.model()

        if replace and model.rowCount():

            messages = []

            for row in range(model.rowCount()):

                item = self.ui.tableViewResult.model().item(row, 0)

                data = item.data()

                if data and data.text() != replace:

                    text = data.text()

                    messages.append(f"replaced: {text} --> {text.replace(text_find, replace)}")

                    data.setPlainText(text.replace(text_find, replace))

            if messages:

                QMessageBox.information(self, self.tr('Information'), '\r\n'.join(messages))

    '''

        @brief      key press event

        @author     kyouho 

        @date       2018.08.27

    '''

    def selectTextItem(self, item):

        rect = item.sceneBoundingRect()

        self.mainWindow.graphicsView.centerOn(rect.center())

        self.mainWindow.graphicsView.zoomImage(True, QMouseEvent(QEvent.MouseButtonPress,

                                                                 self.mainWindow.graphicsView.mapFromScene(

                                                                     QPointF(rect.left(), rect.top())), Qt.LeftButton,

                                                                 Qt.LeftButton, Qt.NoModifier), 3)

        item.setSelected(True)

    def replaceTextItem(self, item):

        pass

    def keyPressEvent(self, event):

        """key press event"""

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

            if self.ui.lineEditFindText.hasFocus():

                self.find_text()

            elif self.ui.lineEditReplaceText.hasFocus():

                self.replaceTextClicked()

        elif event.key() == Qt.Key_Escape:

            self.reject()

    def selection_changed(self, selection):

        """row selection is changed"""

        try:

            if selection.indexes():

                row = selection.indexes()[0].row()

                item = self.ui.tableViewResult.model().item(row, 0)

                data = item.data()

                self.selectTextItem(data)

        except Exception as ex:

            from App import App

            from AppDocData import MessageType

            message = 'error occurred({}) in {}:{}'.format(repr(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 reject(self):

        """close a dialog"""

        QDialog.reject(self)