hakkoso/actuators.py

import atexit
try:
    import RPi.GPIO as GPIO
    GPIO.setmode(GPIO.BCM)
    rpi = True
except:
    rpi = False
        

class Actuator():
    def __init__(self):
        pass


class GPIOPin(Actuator):
    def __init__(self, *args, **kwargs):
        self.pin = (kwargs.pop('pin'))
        self.initial = (kwargs.pop('initial'))
        self.onstate = (kwargs.pop('onstate'))
        self.offstate = (kwargs.pop('offstate'))
        GPIO.setup(self.pin, GPIO.OUT, initial=self.initial)
        atexit.register(self.deinit)
        super().__init__(*args, **kwargs)
        
    def deinit(self):
        # Be sure everything is off at exit
        print('Cleaning before exit')
        GPIO.output(self.pin, self.initial)
        GPIO.cleanup()

    def enable(self, enable=True):
        GPIO.output(self.pin, self.onstate) if enable else self.disable()

    def disable(self):
        GPIO.output(self.pin, self.offstate)

class MockPIN(Actuator):
    def __init__(self, *args, **kwargs):
        self.pin = (kwargs.pop('pin'))
        self.initial = (kwargs.pop('initial'))
        self.onstate = (kwargs.pop('onstate'))
        self.offstate = (kwargs.pop('offstate'))
        super().__init__(*args, **kwargs)
        
    def enable(self, enable=True):
        pass
        # print('FAKE ENABLE') if enable else self.disable()


    def disable(self):
        pass
        # print('FAKE DISABLE')

actuators = {
    'heater': (GPIOPin if rpi else MockPIN)(pin=22, initial=GPIO.HIGH, onstate=GPIO.LOW, offstate=GPIO.HIGH),
}