koperkapel/serums.py

   1 """ Tools for creating serum actors. """
   2
   3
   4 SERUMS = ["smart", "fast", "strong"]
   5
   6 SERUM_OVERLAY_COLORS = {
   7     "smart": (0, 0, 255, 255),  # blue
   8     "fast": (0, 255, 0, 255),  # green
   9     "strong": (255, 0, 255, 255),  # purple
  10     "none": (170, 68, 0, 255),  # brown
  11 }
  12
  13
  14 def roach_serum_color(roach):
  15     for serum_name in SERUMS:
  16         if getattr(roach, serum_name):
  17             return SERUM_OVERLAY_COLORS[serum_name]
  18     return SERUM_OVERLAY_COLORS["none"]