Add roach_is_serumless helper.

[koperkapel.git] / koperkapel / serums.py

diff --git a/koperkapel/serums.py b/koperkapel/serums.py
index c2a52bba6ff0e10525597d2b1a1760108ef95036..40ebfbac59f11950847f662e80906e6107e6c4a7 100644 (file)
--- a/koperkapel/serums.py
+++ b/koperkapel/serums.py
@@ -27,6 +27,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