Pathfinding: fixed some bugs

Piratmac · Piratmac · commit 3b4a38a8cca6 · 2020-08-27T20:31:42.000+02:00
diff --git a/2019/pathfinding.py b/2019/pathfinding.py
@@ -12,14 +12,9 @@ class NegativeWeightCycle(Exception):
 
 
 class Graph:
-    vertices = []
-    edges = {}
-    distance_from_start = {}
-    came_from = {}
-
     def __init__(self, vertices=[], edges={}):
-        self.vertices = vertices
-        self.edges = edges
+        self.vertices = vertices.copy()
+        self.edges = edges.copy()
 
     def neighbors(self, vertex):
         """
@@ -57,6 +52,7 @@ def grid_to_vertices(self, grid, diagonals_allowed=False, wall="#"):
         :return: True if the grid was converted
         """
         self.vertices = []
+        self.edges = {}
         y = 0
 
         for line in grid.splitlines():
@@ -151,15 +147,15 @@ def add_traps(self, vertices):
 
         return changed
 
-    def add_walls(self, vertices):
+    def add_walls(self, walls):
         """
         Adds walls - useful for modification of map
 
         :param Any vertex: The vertices to consider
         :return: True if successful, False if no vertex found
         """
         changed = False
-        for vertex in vertices:
+        for vertex in walls:
             if vertex in self.edges:
                 del self.edges[vertex]
                 if isinstance(self.vertices, list):
@@ -169,7 +165,7 @@ def add_walls(self, vertices):
                 changed = True
 
         self.edges = {
-            source: [target for target in self.edges[source] if target not in vertices]
+            source: [target for target in self.edges[source] if target not in walls]
             for source in self.edges
         }
 
