feat(strategy/iron2023): implement ball get item dunder method

project-neon · Mar 1, 2023 · 058002f · 058002f
1 parent c59a17f
commit 058002f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/entities/Ball.py b/entities/Ball.py
@@ -56,3 +56,12 @@ def _update_speeds(self):
 
         self.vx = speed(self._frames['x'], self.game.vision._fps)
         self.vy = speed(self._frames['y'], self.game.vision._fps)
+
+    def __getitem__(self, item):
+        if item == 0:
+            return self.x
+
+        if item == 1:
+            return self.y
+
+        raise IndexError("Ball only has 2 coordinates")