From 12800db25fafc0cc9d0e4993e0121ed2750551e3 Mon Sep 17 00:00:00 2001 From: Ben Nuttall Date: Wed, 23 Sep 2015 16:32:00 +0100 Subject: [PATCH] Refactor PiLiter into generic LEDBoard class --- gpiozero/boards.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/gpiozero/boards.py b/gpiozero/boards.py index 9b92602..e981713 100644 --- a/gpiozero/boards.py +++ b/gpiozero/boards.py @@ -101,9 +101,8 @@ class TrafficHat(FishDish): self._all = self._leds + (self.buzzer,) -class PiLiter(object): - def __init__(self): - leds = (4, 17, 27, 18, 22, 23, 24, 25) +class LEDBoard(object): + def __init__(self, leds): self._leds = tuple(LED(led) for led in leds) @property @@ -133,3 +132,9 @@ class PiLiter(object): self.off() if i+1 < n: # don't sleep on final iteration sleep(off_time) + + +class PiLiter(LEDBoard): + def __init__(self): + leds = (4, 17, 27, 18, 22, 23, 24, 25) + super(PiLiter, self).__init__(leds)