Update actors/pill.py

actors/pill.py

@@ -1,5 +1,22 @@
+import pygame
 from .enums import PillType
 
-class Pill:
+class Pill(pygame.sprite.Sprite):
-    def __init__(self, typ: PillType):
+    def __init__(self, x, y, pill_type: PillType):
-        self.pilltype = typ
+        super().__init__()
+        self.x = x
+        self.y = y
+        self.pill_type = pill_type
+        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))
+
+class NormalPill(Pill):
+    def __init__(self, x, y):
+        super().__init__(x, y, PillType.NORMAL)
+
+class PowerPill(Pill):
+    def __init__(self, x, y):
+        super().__init__(x, y, PillType.POWER)