Update actors/pill.py

actors/pill.py

@@ -1,5 +1,5 @@
 import pygame
-from .enums import PillType
+from enums import PillType
 
 class Pill(pygame.sprite.Sprite):
     def __init__(self, x, y, pill_type: PillType):
@@ -7,16 +7,38 @@ class Pill(pygame.sprite.Sprite):
         self.x = x
         self.y = y
         self.pill_type = pill_type
+        self.eaten = False
         self.radius = 2 if pill_type == PillType.NORMAL else 6
         self.color = (255, 255, 255)
         self.image = pygame.Surface((self.radius * 2, self.radius * 2), pygame.SRCALPHA)
         pygame.draw.circle(self.image, self.color, (self.radius, self.radius), self.radius)
         self.rect = self.image.get_rect(center=(x, y))
 
+    def check_eaten(self, pacman_rect, game_state):
+        if not self.eaten and self.rect.colliderect(pacman_rect):
+            self.eaten = True
+            self.kill()
+            return self.on_eaten(game_state)
+        return 0
+
+    def on_eaten(self, game_state):
+        """Override this in subclasses to apply effects when eaten"""
+        return 0
+
 class NormalPill(Pill):
     def __init__(self, x, y):
         super().__init__(x, y, PillType.NORMAL)
 
+    def on_eaten(self, game_state):
+        game_state["score"] += 100
+        return 100
+
 class PowerPill(Pill):
     def __init__(self, x, y):
-        super().__init__(x, y, PillType.POWER)
+        super().__init__(x, y, PillType.POWER)
+
+    def on_eaten(self, game_state):
+        game_state["score"] += 500
+        game_state["ghost_mode"] = "frightened"
+        game_state["frightened_timer"] = pygame.time.get_ticks()  # Optional
+        return 500