+class MultiColourLamp(BaseLight):
+
+ FITTING_IMG = "lamp.png"
+ DEFAULT_COLOURS = sorted(BaseLight.COLOURS.keys())
+
+ def __init__(self, **kw):
+ self.colour_cycle = kw.pop("colours", None)
+ self.colour_pos = 0
+ kw["colour"] = self.colour_cycle[0]
+ super(MultiColourLamp, self).__init__(**kw)
+
+ def toggle(self):
+ self.colour_pos += 1
+ if self.colour_pos >= len(self.colour_cycle):
+ self.colour = self.colour_cycle[0]
+ self.colour_pos = -1
+ self.on = False
+ else:
+ self.colour = self.colour_cycle[self.colour_pos]
+ self.on = True
+ self.invalidate_fitting_image()
+
+