fix upstream tree

Author: 3nids

src/tools/qgepmaptools.py

@@ -464,8 +464,7 @@ def getTree(self, node_id: str):
         self.rubberBand.reset()
 
         nodes, edges = self.network_analyzer.getTree(node_id, upstream)
-        polylines = self.network_analyzer.getEdgeGeometry(
-            [edge[2]['feature'] for edge in edges])
+        polylines = self.network_analyzer.getEdgeGeometry([edge[2]['feature'] for edge in edges])
 
         # Fix for QGIS < 2.0
         filtered_polylines = [pl for pl in polylines if pl]

src/tools/qgepnetwork.py

@@ -28,28 +28,25 @@
 """
 from __future__ import print_function
 
+import copyreg
+
 # pylint: disable=no-name-in-module
-from builtins import str
-from builtins import zip
-from builtins import next
-from builtins import object
+from builtins import str, zip, object
 from collections import defaultdict
 import time
 import re
-from qgis.PyQt.QtCore import pyqtSignal
+from qgis.PyQt.QtCore import pyqtSignal, QObject
 from qgis.PyQt.QtSql import QSqlDatabase, QSqlQuery
 
 from qgis.core import (
     Qgis,
     QgsMessageLog,
     QgsGeometry,
     QgsDataSourceUri,
+    QgsPointXY,
     NULL
 )
-from qgis.gui import (
-    QgsMessageBar
-)
-from qgis.PyQt.QtCore import QObject
+from qgis.gui import QgsMessageBar
 import networkx as nx
 
 
@@ -282,6 +279,11 @@ def getTree(self, node, upstream=False):
         if self.dirty:
             self.createGraph()
 
+        # fix point pickle
+        def pickle_point(p):
+            return QgsPointXY, (p.x(),p.y(),)
+        copyreg.pickle(QgsPointXY, pickle_point)
+
         if upstream:
             my_graph = self.graph.reverse()
         else: