Merge pull request #17 from myDevicesIoT/feature/restore-actuator-state

Feature/restore actuator state

Author: jburhenn

myDevices/devices/digital/gpio.py

@@ -37,6 +37,8 @@
 class NativeGPIO(Singleton, GPIOPort):
     IN = 0
     OUT = 1
+    OUT_LOW = 2
+    OUT_HIGH = 3
 
     ASUS_GPIO = 44
 
@@ -295,11 +297,9 @@ def __setFunction__(self, channel, value):
         self.checkDigitalChannelExported(channel)
         self.checkPostingFunctionAllowed()
         try:
-            if value == self.IN:
-                value = 'in'
-            else:
-                value = 'out'
-            try:               
+            value_dict = {self.IN: 'in', self.OUT: 'out', self.OUT_LOW: 'low', self.OUT_HIGH: 'high'}
+            value = value_dict[value]
+            try:
                 self.functionFile[channel].write(value)
                 self.functionFile[channel].seek(0)
             except:

myDevices/devices/digital/helper.py

@@ -63,9 +63,17 @@ def __str__(self):
         return "MotionSensor"
 
 class DigitalActuator(DigitalSensor):
-    def __init__(self, gpio, channel, invert=False):
+    def __init__(self, gpio, channel, invert=False, last_state=None):
         DigitalSensor.__init__(self, gpio, channel, invert)
-        self.gpio.setFunction(self.channel, GPIO.OUT)
+        function = GPIO.OUT
+        if gpio == 'GPIO' and last_state is not None:
+            if self.invert:
+                last_state = int(not last_state)
+            if last_state == 1:
+                function = GPIO.OUT_HIGH
+            elif last_state == 0:
+                function = GPIO.OUT_LOW
+        self.gpio.setFunction(self.channel, function)
 
     def __str__(self):
         return "DigitalActuator"
@@ -82,30 +90,29 @@ def write(self, value):
         return self.read()
 
 class LightSwitch(DigitalActuator):        
-    def __init__(self, gpio, channel, invert=False):
-        DigitalActuator.__init__(self, gpio, channel, invert)
+    def __init__(self, gpio, channel, invert=False, last_state=None):
+        DigitalActuator.__init__(self, gpio, channel, invert, last_state)
 
     def __str__(self):
         return "LightSwitch"
 
 class MotorSwitch(DigitalActuator):        
-    def __init__(self, gpio, channel, invert=False):
-        DigitalActuator.__init__(self, gpio, channel, invert)
+    def __init__(self, gpio, channel, invert=False, last_state=None):
+        DigitalActuator.__init__(self, gpio, channel, invert, last_state)
 
     def __str__(self):
         return "MotorSwitch"
 
 class RelaySwitch(DigitalActuator):        
-    def __init__(self, gpio, channel, invert=False):
-        DigitalActuator.__init__(self, gpio, channel, invert)
+    def __init__(self, gpio, channel, invert=False, last_state=None):
+        DigitalActuator.__init__(self, gpio, channel, invert, last_state)
 
     def __str__(self):
         return "RelaySwitch"
 
 class ValveSwitch(DigitalActuator):        
-    def __init__(self, gpio, channel, invert=False):
-        DigitalActuator.__init__(self, gpio, channel, invert)
+    def __init__(self, gpio, channel, invert=False, last_state=None):
+        DigitalActuator.__init__(self, gpio, channel, invert, last_state)
 
     def __str__(self):
         return "ValveSwitch"
-

myDevices/devices/manager.py

@@ -58,7 +58,7 @@ def findDeviceClass(name):
                     return getattr(module, name)
     return None
 
-def saveDevice(name, install_date):
+def saveDevice(name, install_date=None):
     with mutex:
         logger.debug('saveDevice: ' + str(name))
         if name not in DEVICES:
@@ -67,7 +67,8 @@ def saveDevice(name, install_date):
         if DEVICES[name]['origin'] == 'manual':
             return
         DYNAMIC_DEVICES[name] = DEVICES[name]
-        DEVICES[name]['install_date'] = install_date
+        if install_date:
+            DEVICES[name]['install_date'] = install_date
         json_devices = getJSON(DYNAMIC_DEVICES)
         with open(DEVICES_JSON_FILE, 'w') as outfile:
             outfile.write(json_devices)
@@ -138,6 +139,19 @@ def updateDevice(name, json):
 
         return (c, d, t)
 
+def updateDeviceState(name, value):
+    with mutex:
+        try:
+            if not name in DEVICES:
+                return
+            device = DEVICES[name]
+            if 'last_state' not in device['args'] or device['args']['last_state'] != value:
+                logger.info('Saving state {} for device {}'.format(value, name))
+                device['args'].update({'last_state': value})
+                saveDevice(name)
+        except:
+            pass
+
 def addDevice(name, device, description, args, origin):
     with mutex:
         if name in DEVICES:

myDevices/sensors/sensors.py

@@ -181,7 +181,10 @@ def SensorsInfo(self):
                                 channel = '{}:{}'.format(device['name'], device_type.lower())
                             else:
                                 channel = device['name']
-                            cayennemqtt.DataChannel.add(sensors_info, cayennemqtt.DEV_SENSOR, channel, value=self.CallDeviceFunction(func), name=display_name, **sensor_type['data_args'])
+                            value = self.CallDeviceFunction(func)
+                            cayennemqtt.DataChannel.add(sensors_info, cayennemqtt.DEV_SENSOR, channel, value=value, name=display_name, **sensor_type['data_args'])
+                            if 'DigitalActuator' == device_type and value in (0, 1):
+                                manager.updateDeviceState(device['name'], value)
                         except:
                             exception('Failed to get sensor data: {} {}'.format(device_type, device['name']))
                     # else:
@@ -368,13 +371,16 @@ def SensorCommand(self, command, sensorId, channel, value):
                     info('Sensor not found')
                     return result
                 if command in commands:
-                    info('Sensor found: {}'.format(instance.DEVICES[sensorId]))
+                    device = instance.DEVICES[sensorId]
+                    info('Sensor found: {}'.format(device))
                     func = getattr(sensor, commands[command]['function'])
                     value = commands[command]['value_type'](value)
                     if channel:
                         result = self.CallDeviceFunction(func, int(channel), value)
                     else:
                         result = self.CallDeviceFunction(func, value)
+                    if 'DigitalActuator' in device['type']:
+                        manager.updateDeviceState(sensorId, value)
                     return result
                 warn('Command not implemented: {}'.format(command))
                 return result