add left

Stegallo · Dec 8, 2024 · 588c92f · 588c92f
1 parent f567a13
commit 588c92f
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/common/grid.py b/common/grid.py
@@ -25,16 +25,27 @@ def __hash__(self) -> int:
         return hash((self.x, self.y))
 
     def right(self):
-        if (self.y, self.x) == (-1, 0):
+        if self.icon == "^":
             return Direction(0, 1, ">")
-        if (self.y, self.x) == (0, 1):
+        if self.icon == ">":
             return Direction(1, 0, "v")
-        if (self.y, self.x) == (1, 0):
+        if self.icon == "v":
             return Direction(0, -1, "<")
-        if (self.y, self.x) == (0, -1):
+        if self.icon == "<":
             return Direction(-1, 0, "^")
         raise Exception
 
+    def left(self):
+        if self.icon == "^":
+            return Direction(0, -1, "<")
+        if self.icon == ">":
+            return Direction(-1, 0, "^")
+        if self.icon == "v":
+            return Direction(0, 1, ">")
+        if self.icon == "<":
+            return Direction(1, 0, "v")
+        raise Exception
+
     @staticmethod
     def from_symbol(symbol: str):
         return Direction(