X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=koperkapel%2Fserums.py;h=38794fd4926834bba5129b650f836e944c534490;hb=9d3fee60d4fb75d183d6db5af8468c2d9dbf9e85;hp=c2a52bba6ff0e10525597d2b1a1760108ef95036;hpb=9b4ed6a78343aca0bdcfb347730142458c911e3f;p=koperkapel.git

diff --git a/koperkapel/serums.py b/koperkapel/serums.py
index c2a52bb..38794fd 100644
--- a/koperkapel/serums.py
+++ b/koperkapel/serums.py
@@ -1,7 +1,9 @@
 """ Tools for creating serum actors. """
 
+import random
 from pgzero.loaders import images
 from pygame.constants import BLEND_RGBA_MULT
+from pygame.transform import rotate
 from .actors.surf import SurfActor
 
 SERUMS = ["smart", "fast", "strong"]
@@ -27,6 +29,13 @@ def roach_serum_color(roach):
     return SERUM_OVERLAY_COLORS["none"]
 
 
+def roach_is_serumless(roach):
+    for serum_name in SERUMS:
+        if getattr(roach, serum_name):
+            return False
+    return True
+
+
 class SerumFactory:
     def __init__(self, suffix):
         self.suffix = suffix
@@ -34,6 +43,7 @@ class SerumFactory:
     def assemble(self, name):
         assert name in SERUMS
         puddle = images.load("serum%s/serum" % (self.suffix,))
+        puddle = rotate(puddle, 90 * random.randint(0, 3))
         serum_icon = images.load("serum%s/%s" % (
             self.suffix, SERUM_TILENAME_MAP[name],))
         frame = puddle.copy()