hakkoso/actuators.py

import atexit
from time import perf_counter
try:
    import RPi.GPIO as GPIO
    GPIO.setmode(GPIO.BCM)
    rpi = True
except:
    rpi = False
    class GPIO():
        LOW = 0
        HIGH = 1
        

class Actuator():
    def __init__(self, *args, **kwargs):
        self.wattage = kwargs.pop('wattage')
        self.total_seconds_on = 0
        self.logging = False
        self.started_at = 0

    def log_on(self):
        if self.logging:
            return
        self.started_at = perf_counter()
        self.logging = True

    def elapsed(self):
        return perf_counter() - self.started_at

    def log_off(self):
        if not self.logging:
            return
        self.total_seconds_on += self.elapsed()
        self.logging = False

    def s_to_hr(self, s):
        return s / 60 / 60

    def consumption(self):
        total_time = self.total_seconds_on
        if self.logging:
            # If we are on, count this time too
            total_time += self.elapsed()
        return self.s_to_hr(total_time) * self.wattage

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()
        if enable:
            self.log_on()
        else:
            self.log_off()

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

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):
        if enable:
            self.log_on()
        else:
            self.log_off()
        # print('FAKE ENABLE') if enable else self.disable()

    def disable(self):
        self.log_off()
        # print('FAKE DISABLE')

def PIN(*args, **kwargs):
    return (GPIOPin if rpi else MockPIN)(*args, **kwargs)

actuators = {
    'heater': PIN(pin=22, initial=GPIO.HIGH, onstate=GPIO.LOW, offstate=GPIO.HIGH,
                  wattage=100),
    'fan': PIN(pin=27, initial=GPIO.HIGH, onstate=GPIO.LOW, offstate=GPIO.HIGH,
               wattage=12*0.4),
    'humidifier': PIN(pin=17, initial=GPIO.HIGH, onstate=GPIO.LOW, offstate=GPIO.HIGH,
                      wattage=25),
    'heater2': PIN(pin=17, initial=GPIO.HIGH, onstate=GPIO.LOW, offstate=GPIO.HIGH,
                      wattage=400),
    'freezer': PIN(pin=14, initial=GPIO.LOW, onstate=GPIO.HIGH, offstate=GPIO.LOW,
                      wattage=165),
}