Fixing game to working state

fruit_catcher/fruit.py

@@ -0,0 +1,65 @@
+from random import randint
+
+SHAPE_DEFINITIONS = [
+    [(0,0)],
+    [(0, 0), (1, 0), (0, 1), (1, 1)],
+]
+
+class FruitPiece:
+    character = "@"
+    color = "green_on_indigo"
+    display = True
+    def __init__(self, position):
+        self.position = position
+
+class Fruit:
+    width = 2
+    height = 1
+    display = False
+    pieces = []
+    name = "fruit"
+    character = "@"
+    color = "green_on_indigo"
+
+    def __init__(self, position, game, shape_offsets):
+        self.position = position
+        self.pieces = {}
+        for offset in shape_offsets:
+            self.create_shape(game, offset)
+
+    def play_turn(self, game):
+        if game.turn_number % 3 == 0:
+            x, y = self.position
+            if y >= 29:
+                for piece in self.pieces.values():
+                    game.remove_agent(piece)
+                game.remove_agent(self)
+                game.end()
+            else:
+                self.position = (x, y + 1)
+                self.update_piece_positions()
+                self.check_catcher_collision(game)
+
+    def create_shape(self, game, offset):
+        x, y = self.position
+        ox, oy = offset
+        piece = FruitPiece((x + ox, y + oy))
+        self.pieces[offset] = piece
+        game.add_agent(piece)
+
+    def update_piece_positions(self):
+        x, y = self.position
+        for offset, piece in self.pieces.items():
+            ox, oy = offset
+            piece.position = (x + ox, y + oy)
+
+    def check_catcher_collision(self, game):
+        catcher = game.get_agent_by_name("catcher")
+        catcher_positions = {p.position for p in catcher.pieces}
+        for piece in self.pieces.values():
+            if piece.position in catcher_positions:
+                game.state['Score'] += 1
+                for p in self.pieces.values():
+                    game.remove_agent(p)
+                game.remove_agent(self)
+                return