TreeWidget.__init__() - Code Metrics - FreeOpcUa/opcua-widgets - Measure and Improve Code Quality continuously with Scrutinizer

TreeWidget.init() A
last analyzed 2018-06-23 07:21 UTC

↳ Parent: TreeWidget

Complexity

Conditions

Size

Total Lines

Duplication

Lines	0
Ratio	0 %

Importance

Changes	3
Bugs	0	Features	0

Metric	Value
c	3
b	0
f	0
dl	0
loc	20
rs	9.4285
cc	2

from PyQt5.QtCore import pyqtSignal, QMimeData, QObject, Qt, QSettings
from PyQt5.QtGui import QStandardItemModel, QStandardItem, QIcon
from PyQt5.QtWidgets import QApplication, QAbstractItemView, QAction

from opcua import ua
from opcua import Node
from opcua.ua import UaError


class TreeWidget(QObject):

    error = pyqtSignal(Exception)

    def __init__(self, view):
        QObject.__init__(self, view)
        self.view = view
        self.model = TreeViewModel()
        self.model.clear()  # FIXME: do we need this?
        self.model.error.connect(self.error)
        self.view.setModel(self.model)

        #self.view.setUniformRowHeights(True)
        self.model.setHorizontalHeaderLabels(['DisplayName', "BrowseName", 'NodeId'])
        self.view.header().setSectionResizeMode(0)
        self.view.header().setStretchLastSection(True)
        self.view.setSelectionBehavior(QAbstractItemView.SelectRows)
        self.settings = QSettings()
        state = self.settings.value("tree_widget_state", None)
        if state is not None:
            self.view.header().restoreState(state)

        self.actionReload = QAction("Reload", self)
        self.actionReload.triggered.connect(self.reload_current)

    def save_state(self):
        self.settings.setValue("tree_widget_state", self.view.header().saveState())

    def clear(self):
        self.model.clear()

    def set_root_node(self, node):
        self.model.clear()
        self.model.set_root_node(node)
        self.view.expandToDepth(0)

    def copy_path(self):
        path = self.get_current_path()
        path_str = ",".join(path)
        QApplication.clipboard().setText(path_str)

    def expand_current_node(self, expand=True):
        idx = self.view.currentIndex()
        self.view.setExpanded(idx, expand)

    def expand_to_node(self, node):
        """
        Expand tree until given node and select it
        """
        if isinstance(node, str):
            idxlist = self.model.match(self.model.index(0, 0), Qt.DisplayRole, node, 1, Qt.MatchExactly|Qt.MatchRecursive)
            node = self.model.data(idxlist[0], Qt.UserRole)
        path = node.get_path()
        for node in path:
            # FIXME: this would be the correct way if it would work
            #idxlist = self.model.match(self.model.index(0, 0), Qt.UserRole, QVariantnode, 2, Qt.MatchExactly|Qt.MatchRecursive)
            try:
                text = node.get_display_name().Text
            except UaError as ex:
                return
            idxlist = self.model.match(self.model.index(0, 0), Qt.DisplayRole, text, 1, Qt.MatchExactly|Qt.MatchRecursive)
            if idxlist:
                idx = idxlist[0]
                self.view.setExpanded(idx, True)
                self.view.setCurrentIndex(idx)
                self.view.activated.emit(idx)

    def copy_nodeid(self):
        node = self.get_current_node()
        text = node.nodeid.to_string()
        QApplication.clipboard().setText(text)

    def get_current_path(self):
        idx = self.view.currentIndex()
        idx = idx.sibling(idx.row(), 0)
        it = self.model.itemFromIndex(idx)
        path = []
        while it and it.data(Qt.UserRole):
            node = it.data(Qt.UserRole)
            name = node.get_browse_name().to_string()
            path.insert(0, name)
            it = it.parent()
        return path

    def update_browse_name_current_item(self, bname):
        idx = self.view.currentIndex()
        idx = idx.sibling(idx.row(), 1)
        it = self.model.itemFromIndex(idx)
        it.setText(bname.to_string())

    def update_display_name_current_item(self, dname):
        idx = self.view.currentIndex()
        idx = idx.sibling(idx.row(), 0)
        it = self.model.itemFromIndex(idx)
        it.setText(dname.Text)

    def reload_current(self):
        idx = self.view.currentIndex()
        idx = idx.sibling(idx.row(), 0)
        it = self.model.itemFromIndex(idx)
        if not it:
            return None
        self.reload(it)

    def reload(self, item=None):
        if item is None:
            item = self.model.item(0, 0)
        for _ in range(item.rowCount()):
            child_it = item.child(0, 0)
            node = child_it.data(Qt.UserRole)
            if node:
                self.model.reset_cache(node)
            item.takeRow(0)
        node = item.data(Qt.UserRole)
        if node:
            self.model.reset_cache(node)
            idx = self.model.indexFromItem(item)
            #if self.view.isExpanded(idx):
            #self.view.setExpanded(idx, True)

    def remove_current_item(self):
        idx = self.view.currentIndex()
        self.model.removeRow(idx.row(), idx.parent())

    def get_current_node(self, idx=None):
        if idx is None:
            idx = self.view.currentIndex()
        idx = idx.sibling(idx.row(), 0)
        it = self.model.itemFromIndex(idx)
        if not it:
            return None
        node = it.data(Qt.UserRole)
        if not node:
            ex = RuntimeError("Item does not contain node data, report!")
            self.error.emit(ex)
            raise ex
        return node


class TreeViewModel(QStandardItemModel):

    error = pyqtSignal(Exception)

    def __init__(self):
        super(TreeViewModel, self).__init__()
        self._fetched = []

    def clear(self):
        # remove all rows but not header!!
        self.removeRows(0, self.rowCount())
        self._fetched = []

    def set_root_node(self, node):
        desc = self._get_node_desc(node)
        self.add_item(desc, node=node)

    def _get_node_desc(self, node):
        attrs = node.get_attributes([ua.AttributeIds.DisplayName, ua.AttributeIds.BrowseName, ua.AttributeIds.NodeId, ua.AttributeIds.NodeClass])
        desc = ua.ReferenceDescription()
        desc.DisplayName = attrs[0].Value.Value
        desc.BrowseName = attrs[1].Value.Value
        desc.NodeId = attrs[2].Value.Value
        desc.NodeClass = attrs[3].Value.Value
        desc.TypeDefinition = ua.TwoByteNodeId(ua.ObjectIds.FolderType)
        return desc

    def add_item(self, desc, parent=None, node=None):
        dname = bname = nodeid = "No Value"
        if desc.DisplayName:
            dname = desc.DisplayName.to_string()
        if desc.BrowseName:
            bname = desc.BrowseName.to_string()
        nodeid = desc.NodeId.to_string()
        item = [QStandardItem(dname), QStandardItem(bname), QStandardItem(nodeid)]
        if desc.NodeClass == ua.NodeClass.Object:
            if desc.TypeDefinition == ua.TwoByteNodeId(ua.ObjectIds.FolderType):
                item[0].setIcon(QIcon(":/folder.svg"))
            else:
                item[0].setIcon(QIcon(":/object.svg"))
        elif desc.NodeClass == ua.NodeClass.Variable:
            if desc.TypeDefinition == ua.TwoByteNodeId(ua.ObjectIds.PropertyType):
                item[0].setIcon(QIcon(":/property.svg"))
            else:
                item[0].setIcon(QIcon(":/variable.svg"))
        elif desc.NodeClass == ua.NodeClass.Method:
            item[0].setIcon(QIcon(":/method.svg"))
        elif desc.NodeClass == ua.NodeClass.ObjectType:
            item[0].setIcon(QIcon(":/object_type.svg"))
        elif desc.NodeClass == ua.NodeClass.VariableType:
            item[0].setIcon(QIcon(":/variable_type.svg"))
        elif desc.NodeClass == ua.NodeClass.DataType:
            item[0].setIcon(QIcon(":/data_type.svg"))
        elif desc.NodeClass == ua.NodeClass.ReferenceType:
            item[0].setIcon(QIcon(":/reference_type.svg"))
        if node:
            item[0].setData(node, Qt.UserRole)
        else:
            parent_node = parent.data(Qt.UserRole)
            item[0].setData(Node(parent_node.server, desc.NodeId), Qt.UserRole)
        if parent:
            return parent.appendRow(item)
        else:
            return self.appendRow(item)

    def reset_cache(self, node):
        if node in self._fetched:
            self._fetched.remove(node)

    def canFetchMore(self, idx):
        item = self.itemFromIndex(idx)
        if not item:
            return False
        node = item.data(Qt.UserRole)
        if node not in self._fetched:
            self._fetched.append(node)
            return True
        return False

    def hasChildren(self, idx):
        item = self.itemFromIndex(idx)
        if not item:
            return True
        node = item.data(Qt.UserRole)
        if node in self._fetched:
            return QStandardItemModel.hasChildren(self, idx)
        return True

    def fetchMore(self, idx):
        parent = self.itemFromIndex(idx)
        if parent:
            self._fetchMore(parent)

    def _fetchMore(self, parent):
        try:
            node = parent.data(Qt.UserRole)
            descs = node.get_children_descriptions()
            descs.sort(key=lambda x: x.BrowseName)
            added = []
            for desc in descs:
                if not desc.NodeId in added:
                    self.add_item(desc, parent)
                    added.append(desc.NodeId)
        except Exception as ex:
            self.error.emit(ex)
            raise

    def mimeData(self, idxs):
        mdata = QMimeData()
        nodes = []
        for idx in idxs:
            item = self.itemFromIndex(idx)
            if item:
                node = item.data(Qt.UserRole)
                if node:
                    nodes.append(node.nodeid.to_string())
        mdata.setText(", ".join(nodes))
        return mdata




1			from PyQt5.QtCore import pyqtSignal, QMimeData, QObject, Qt, QSettings
2			from PyQt5.QtGui import QStandardItemModel, QStandardItem, QIcon
3			from PyQt5.QtWidgets import QApplication, QAbstractItemView, QAction
4
5			from opcua import ua
6			from opcua import Node
7			from opcua.ua import UaError
8
9
10			class TreeWidget(QObject):
11
12			error = pyqtSignal(Exception)
13
14			def __init__(self, view):
15			QObject.__init__(self, view)
16			self.view = view
17			self.model = TreeViewModel()
18			self.model.clear() # FIXME: do we need this?
19			self.model.error.connect(self.error)
20			self.view.setModel(self.model)
21
22			#self.view.setUniformRowHeights(True)
23			self.model.setHorizontalHeaderLabels(['DisplayName', "BrowseName", 'NodeId'])
24			self.view.header().setSectionResizeMode(0)
25			self.view.header().setStretchLastSection(True)
26			self.view.setSelectionBehavior(QAbstractItemView.SelectRows)
27			self.settings = QSettings()
28			state = self.settings.value("tree_widget_state", None)
29			if state is not None:
30			self.view.header().restoreState(state)
31
32			self.actionReload = QAction("Reload", self)
33			self.actionReload.triggered.connect(self.reload_current)
34
35			def save_state(self):
36			self.settings.setValue("tree_widget_state", self.view.header().saveState())
37
38			def clear(self):
39			self.model.clear()
40
41			def set_root_node(self, node):
42			self.model.clear()
43			self.model.set_root_node(node)
44			self.view.expandToDepth(0)
45
46			def copy_path(self):
47			path = self.get_current_path()
48			path_str = ",".join(path)
49			QApplication.clipboard().setText(path_str)
50
51			def expand_current_node(self, expand=True):
52			idx = self.view.currentIndex()
53			self.view.setExpanded(idx, expand)
54
55			def expand_to_node(self, node):
56			"""
57			Expand tree until given node and select it
58			"""
59			if isinstance(node, str):
60			idxlist = self.model.match(self.model.index(0, 0), Qt.DisplayRole, node, 1, Qt.MatchExactly\|Qt.MatchRecursive)
61			node = self.model.data(idxlist[0], Qt.UserRole)
62			path = node.get_path()
63			for node in path:
64			# FIXME: this would be the correct way if it would work
65			#idxlist = self.model.match(self.model.index(0, 0), Qt.UserRole, QVariantnode, 2, Qt.MatchExactly\|Qt.MatchRecursive)
66			try:
67			text = node.get_display_name().Text
68			except UaError as ex:
69			return
70			idxlist = self.model.match(self.model.index(0, 0), Qt.DisplayRole, text, 1, Qt.MatchExactly\|Qt.MatchRecursive)
71			if idxlist:
72			idx = idxlist[0]
73			self.view.setExpanded(idx, True)
74			self.view.setCurrentIndex(idx)
75			self.view.activated.emit(idx)
76
77			def copy_nodeid(self):
78			node = self.get_current_node()
79			text = node.nodeid.to_string()
80			QApplication.clipboard().setText(text)
81
82			def get_current_path(self):
83			idx = self.view.currentIndex()
84			idx = idx.sibling(idx.row(), 0)
85			it = self.model.itemFromIndex(idx)
86			path = []
87			while it and it.data(Qt.UserRole):
88			node = it.data(Qt.UserRole)
89			name = node.get_browse_name().to_string()
90			path.insert(0, name)
91			it = it.parent()
92			return path
93
94			def update_browse_name_current_item(self, bname):
95			idx = self.view.currentIndex()
96			idx = idx.sibling(idx.row(), 1)
97			it = self.model.itemFromIndex(idx)
98			it.setText(bname.to_string())
99
100			def update_display_name_current_item(self, dname):
101			idx = self.view.currentIndex()
102			idx = idx.sibling(idx.row(), 0)
103			it = self.model.itemFromIndex(idx)
104			it.setText(dname.Text)
105
106			def reload_current(self):
107			idx = self.view.currentIndex()
108			idx = idx.sibling(idx.row(), 0)
109			it = self.model.itemFromIndex(idx)
110			if not it:
111			return None
112			self.reload(it)
113
114			def reload(self, item=None):
115			if item is None:
116			item = self.model.item(0, 0)
117			for _ in range(item.rowCount()):
118			child_it = item.child(0, 0)
119			node = child_it.data(Qt.UserRole)
120			if node:
121			self.model.reset_cache(node)
122			item.takeRow(0)
123			node = item.data(Qt.UserRole)
124			if node:
125			self.model.reset_cache(node)
126			idx = self.model.indexFromItem(item)
127			#if self.view.isExpanded(idx):
128			#self.view.setExpanded(idx, True)
129
130			def remove_current_item(self):
131			idx = self.view.currentIndex()
132			self.model.removeRow(idx.row(), idx.parent())
133
134			def get_current_node(self, idx=None):
135			if idx is None:
136			idx = self.view.currentIndex()
137			idx = idx.sibling(idx.row(), 0)
138			it = self.model.itemFromIndex(idx)
139			if not it:
140			return None
141			node = it.data(Qt.UserRole)
142			if not node:
143			ex = RuntimeError("Item does not contain node data, report!")
144			self.error.emit(ex)
145			raise ex
146			return node
147
148
149			class TreeViewModel(QStandardItemModel):
150
151			error = pyqtSignal(Exception)
152
153			def __init__(self):
154			super(TreeViewModel, self).__init__()
155			self._fetched = []
156
157			def clear(self):
158			# remove all rows but not header!!
159			self.removeRows(0, self.rowCount())
160			self._fetched = []
161
162			def set_root_node(self, node):
163			desc = self._get_node_desc(node)
164			self.add_item(desc, node=node)
165
166			def _get_node_desc(self, node):
167			attrs = node.get_attributes([ua.AttributeIds.DisplayName, ua.AttributeIds.BrowseName, ua.AttributeIds.NodeId, ua.AttributeIds.NodeClass])
168			desc = ua.ReferenceDescription()
169			desc.DisplayName = attrs[0].Value.Value
170			desc.BrowseName = attrs[1].Value.Value
171			desc.NodeId = attrs[2].Value.Value
172			desc.NodeClass = attrs[3].Value.Value
173			desc.TypeDefinition = ua.TwoByteNodeId(ua.ObjectIds.FolderType)
174			return desc
175
176			def add_item(self, desc, parent=None, node=None):
177			dname = bname = nodeid = "No Value"
178			if desc.DisplayName:
179			dname = desc.DisplayName.to_string()
180			if desc.BrowseName:
181			bname = desc.BrowseName.to_string()
182			nodeid = desc.NodeId.to_string()
183			item = [QStandardItem(dname), QStandardItem(bname), QStandardItem(nodeid)]
184			if desc.NodeClass == ua.NodeClass.Object:
185			if desc.TypeDefinition == ua.TwoByteNodeId(ua.ObjectIds.FolderType):
186			item[0].setIcon(QIcon(":/folder.svg"))
187			else:
188			item[0].setIcon(QIcon(":/object.svg"))
189			elif desc.NodeClass == ua.NodeClass.Variable:
190			if desc.TypeDefinition == ua.TwoByteNodeId(ua.ObjectIds.PropertyType):
191			item[0].setIcon(QIcon(":/property.svg"))
192			else:
193			item[0].setIcon(QIcon(":/variable.svg"))
194			elif desc.NodeClass == ua.NodeClass.Method:
195			item[0].setIcon(QIcon(":/method.svg"))
196			elif desc.NodeClass == ua.NodeClass.ObjectType:
197			item[0].setIcon(QIcon(":/object_type.svg"))
198			elif desc.NodeClass == ua.NodeClass.VariableType:
199			item[0].setIcon(QIcon(":/variable_type.svg"))
200			elif desc.NodeClass == ua.NodeClass.DataType:
201			item[0].setIcon(QIcon(":/data_type.svg"))
202			elif desc.NodeClass == ua.NodeClass.ReferenceType:
203			item[0].setIcon(QIcon(":/reference_type.svg"))
204			if node:
205			item[0].setData(node, Qt.UserRole)
206			else:
207			parent_node = parent.data(Qt.UserRole)
208			item[0].setData(Node(parent_node.server, desc.NodeId), Qt.UserRole)
209			if parent:
210			return parent.appendRow(item)
211			else:
212			return self.appendRow(item)
213
214			def reset_cache(self, node):
215			if node in self._fetched:
216			self._fetched.remove(node)
217
218			def canFetchMore(self, idx):
219			item = self.itemFromIndex(idx)
220			if not item:
221			return False
222			node = item.data(Qt.UserRole)
223			if node not in self._fetched:
224			self._fetched.append(node)
225			return True
226			return False
227
228			def hasChildren(self, idx):
229			item = self.itemFromIndex(idx)
230			if not item:
231			return True
232			node = item.data(Qt.UserRole)
233			if node in self._fetched:
234			return QStandardItemModel.hasChildren(self, idx)
235			return True
236
237			def fetchMore(self, idx):
238			parent = self.itemFromIndex(idx)
239			if parent:
240			self._fetchMore(parent)
241
242			def _fetchMore(self, parent):
243			try:
244			node = parent.data(Qt.UserRole)
245			descs = node.get_children_descriptions()
246			descs.sort(key=lambda x: x.BrowseName)
247			added = []
248			for desc in descs:
249			if not desc.NodeId in added:
250			self.add_item(desc, parent)
251			added.append(desc.NodeId)
252			except Exception as ex:
253			self.error.emit(ex)
254			raise
255
256			def mimeData(self, idxs):
257			mdata = QMimeData()
258			nodes = []
259			for idx in idxs:
260			item = self.itemFromIndex(idx)
261			if item:
262			node = item.data(Qt.UserRole)
263			if node:
264			nodes.append(node.nodeid.to_string())
265			mdata.setText(", ".join(nodes))
266			return mdata
267
268
269

FreeOpcUa / opcua-widgets

TreeWidget.__init__() A last analyzed 2018-06-23 07:21 UTC

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like

TreeWidget.init() A
last analyzed 2018-06-23 07:21 UTC