HYTOS

# coding: utf-8

""" This is Drawing Tree Widget module """

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("DrawingTreeWidget: Requires PyQt5 or PyQt4.")

from AppDocData import *

class SearchProxyModel(QSortFilterProxyModel):

    def setFilterRegExp(self, pattern):

        if isinstance(pattern, str):

            pattern = QRegExp( pattern, Qt.CaseInsensitive, QRegExp.FixedString)

        super(SearchProxyModel, self).setFilterRegExp(pattern)

    def _accept_index(self, idx):

        if idx.isValid():

            text = idx.data(Qt.DisplayRole)

            if self.filterRegExp().indexIn(text) >= 0:

                return True

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

                if self._accept_index(idx.model().index(row, 0, idx)):

                    return True

        return False

    def filterAcceptsRow(self, source_row, source_parent):

        idx = self.sourceModel().index(source_row, 0, source_parent)

        return self._accept_index(idx)

class QDrawingTreeWidget(QTreeWidget):

    def __init__(self):

        QTreeWidget.__init__(self)

    def load_drawing_list(self, reverse=False):

        """load p&id drawing list"""

        from Drawing import Drawing

        app_doc_data = AppDocData.instance()

        drawings = app_doc_data.getDrawings()

        new_drawings = []

        self.clear()

        self.root = QTreeWidgetItem(self, [self.tr('P&ID Drawings'), ''])

        self.root.setFlags(self.root.flags() | Qt.ItemIsTristate | Qt.ItemIsUserCheckable)

        self.root.setCheckState(0, Qt.Unchecked)

        files = app_doc_data.getDrawingFileList()

        count = 0

        for file in files if not reverse else reversed(files):

            x = [drawing for drawing in drawings if drawing.name == file]

            if not x or not x[0].UID:

                drawing = Drawing(None, file, None)

                new_drawings.append(drawing)

            else:

                drawing = x[0]

            item = QTreeWidgetItem(self.root, [file, drawing.datetime])

            item.setIcon(0, QIcon(':newPrefix/image.png'))

            item.setFlags(item.flags() | Qt.ItemIsUserCheckable)

            item.setCheckState(0, Qt.Unchecked)

            item.setData(Qt.UserRole, 0, drawing)

            count += 1

        self.root.setText(0, self.tr('P&ID Drawings') + f"({self.root.childCount()})")

        self.expandItem(self.root)

        self.resizeColumnToContents(0)

        app_doc_data.saveDrawings(new_drawings)

    def set_filter_regexp(self, regexp: QRegExp) -> None:

        """set filter string which is regular expression"""

        if isinstance(regexp, QRegExp):

            for i in range(0, self.root.childCount()):

                item = self.root.child(i)

                drawing_name = item.text(0)

                item.setHidden(regexp.indexIn(drawing_name) < 0)

        msg.setText('Process check fail \nRunning Process List : ' + '\n' + '\n'.join(names))

from DrawingTreeWidget import QDrawingTreeWidget

        # Add Symbol TreeWidget

        self.verticalLayoutSymbolTree.addWidget(self.symbolTreeWidget)

        self.treeWidgetDrawingList = QDrawingTreeWidget()

        self.drawingFilter.textChanged.connect(self.on_drawing_filter_changed)

        self.drawingLayout.addWidget(self.treeWidgetDrawingList)

            self.treeWidgetDrawingList.load_drawing_list(reverse)

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

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

    def on_drawing_filter_changed(self):

        """filter drawing tree view"""

        regexp = QRegExp(self.drawingFilter.text(), Qt.CaseInsensitive, QRegExp.FixedString)

        self.treeWidgetDrawingList.set_filter_regexp(regexp)

        self.treeWidgetDrawingList.expandAll()

# Form implementation generated from reading ui file '.\UI\MainWindow.ui'

# Created by: PyQt5 UI code generator 5.15.4

# WARNING: Any manual changes made to this file will be lost when pyuic5 is

# run again.  Do not edit this file unless you know what you are doing.

        self.drawingFilter = QtWidgets.QLineEdit(self.tabDrawingList)

        self.drawingFilter.setObjectName("drawingFilter")

        self.horizontalLayout_4.addWidget(self.drawingFilter)

        self.drawingLayout = QtWidgets.QVBoxLayout()

        self.drawingLayout.setObjectName("drawingLayout")

        self.gridLayout_7.addLayout(self.drawingLayout, 1, 0, 1, 1)

        spacerItem1 = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)

        self.horizontalLayout_5.addItem(spacerItem1)

        spacerItem2 = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)

        self.horizontalLayout_3.addItem(spacerItem2)

        self.drawingFilter.setPlaceholderText(_translate("MainWindow", "Search..."))

              <widget class="QLineEdit" name="drawingFilter">

               <property name="placeholderText">

                <string>Search...</string>

              </widget>

         <item row="1" column="0">

          <layout class="QVBoxLayout" name="drawingLayout"/>

         </item>