Фикс спотыкания о шлейф

paperio/local_runner/game_objects/player.py

@@ -8,6 +8,7 @@
 class Player:
     speed = SPEED
     direction = None
+    prev_direction = None
 
     def __init__(self, id, x, y, name, color, client):
         self.id = id
@@ -27,6 +28,8 @@ def __init__(self, id, x, y, name, color, client):
         self.is_disconnected = False
 
     def change_direction(self, command):
+        self.prev_direction = self.direction
+
         if command == UP and self.direction != DOWN:
             self.direction = UP
 
@@ -157,8 +160,38 @@ def get_direction_line(self):
         if self.direction == RIGHT:
             return self._get_line(WIDTH, 0)
 
+    def diff_position(self, direction, x, y, val):
+        if direction == UP:
+            return x, y - val
+
+        if direction == DOWN:
+            return x, y + val
+
+        if direction == LEFT:
+            return x + val, y
+
+        if direction == RIGHT:
+            return x - val, y
+
+    def get_position(self):
+        if self.direction is None:
+            return self.x, self.y
+
+        x, y = self.x, self.y
+        while not ((x - round(WIDTH / 2)) % WIDTH == 0 and (y - round(WIDTH / 2)) % WIDTH == 0):
+            x, y = self.diff_position(self.direction, x, y, self.speed)
+
+        return (x, y), (x, y) != (self.x, self.y)
+
+    def get_prev_position(self):
+        if self.prev_direction is None:
+            return self.x, self.y
+        return self.diff_position(self.prev_direction, self.x, self.y, WIDTH)
+
     def is_ate(self, players_to_captured):
         for p, captured in players_to_captured.items():
-            if self != p and (self.x, self.y) in captured:
+            position, is_move = self.get_position()
+            if self != p and position in captured and \
+                    (is_move or self.get_prev_position() in captured):
                 return True
         return False