Fix ZeroDivisionError if plug is powered off

README.md

@@ -1,11 +1,3 @@
-# SEM6000 Python Library
-
-SEM6000 is a energy meter and power switch with Bluetooth 4.0.
-
-This library provides a Python module for these devices
-
-## Run the example code
-
 ```
 $ git clone … sem6000
 $ cd sem6000
@@ -14,51 +6,3 @@ $ . ./python3_venv/bin/activate
 $ pip3 install -r requirements.txt
 $ python3 example.py
 ```
-
-## Collectd Plugin
-
-You can find a Plugin for [collectd](https://collectd.org) in the `collectd`
-subdirectory.
-
-Installation procedure (the target directory may be changed of course):
-
-```shell
-# mkdir -p /usr/local/lib/collectd/python
-# cp collectd/collectd_sem6000.py /usr/local/lib/collectd/python
-# cp sem6000.py /usr/local/lib/collectd/python
-```
-
-Add or adjust the configuration for your collectd’s Python plugin as follows:
-
-```
-<Plugin python>
-  ModulePath "/usr/local/share/collectd/python"
-  LogTraces true
-  Interactive false
-  Import "collectd_sem6000"
-
-  <Module collectd_sem6000>
-    Address "12:34:56:78:90:ab"
-    SocketName "FirstSocket"
-    ReadTimeout 30
-    SuspendTime 300
-  </Module>
-  <Module collectd_sem6000>
-    Address "ab:cd:ef:13:37:42"
-    SocketName "ASecondSocket"
-  </Module>
-  # ...
-</Plugin>
-```
-
-`ReadTimeout` and `SuspendTime` control what’s happening when a device is
-unavailable. If no value could be retrieved for `ReadTimeout` seconds, the
-plugin does not retry for `SuspendTime` seconds. After that, normal operation
-is resumed. This procedure ensures that an unreachable device does not block
-other devices (too often) in the current single-threaded architecture.
-
-If not specified, `ReadTimeout` is 30 seconds and `SuspendTime` is 5 minutes.
-
-Make sure that everything listed in `requirements.txt` is available to the user
-running collectd.
-

collectd/collectd_sem6000.py

@@ -1,144 +0,0 @@
-#!/usr/bin/env python3
-# coding: utf-8
-# vim: noet ts=2 sw=2 sts=2
-
-import os
-import time
-import collectd
-
-from sem6000 import SEMSocket
-from bluepy.btle import BTLEDisconnectError
-
-instances = []
-
-def init_func():
-	pass
-
-def config_func(cfg):
-	global instances
-
-	config = {}
-
-	for node in cfg.children:
-		key = node.key.lower()
-		value = node.values[0]
-
-		if key in ['address', 'socketname']:
-			config[key] = value
-
-		if key == 'readtimeout':
-			config['readtimeout'] = int(value)
-
-		if key == 'suspendtime':
-			config['suspendtime'] = int(value)
-
-	if 'address' not in config.keys():
-		collectd.error('sem6000: address must be set')
-		return
-
-	if 'socketname' not in config.keys():
-		config['socketname'] = config['address'].replace(':', '')
-
-	if 'readtimeout' not in config.keys():
-		config['readtimeout'] = 30
-
-	if 'suspendtime' not in config.keys():
-		config['suspendtime'] = 300
-
-	instances.append( {
-		'config': config,
-		'socket': None,
-		'suspended': False,
-		'lastsuccess': 0,
-		'resumetime': 0
-		} )
-
-def read_func():
-	global instances
-
-	for inst in instances:
-		config = inst['config']
-
-		if inst['suspended']:
-			if time.time() < inst['resumetime']:
-				continue
-			else:
-				collectd.info("sem6000: Device {} waking up.".format(config['address']))
-				inst['suspended'] = False
-				inst['lastsuccess'] = time.time()
-
-		try:
-			if inst['socket'] == None:
-				collectd.info("sem6000: Connecting to {}...".format(config['address']))
-
-				inst['socket'] = SEMSocket(config['address'])
-				collectd.info("sem6000: Connected.")
-
-			inst['socket'].getStatus()
-		except (SEMSocket.NotConnectedException, BTLEDisconnectError, BrokenPipeError) as e:
-			collectd.warning("sem6000: Exception caught: {}".format(e))
-			collectd.warning("sem6000: Restarting on next cycle...")
-
-			if inst['lastsuccess'] < time.time() - config['readtimeout']:
-				collectd.error("sem6000: no successful communication with {} for {:.1f} seconds. Suspending device for {:.1f} seconds.".format(
-					config['address'], config['readtimeout'], config['suspendtime']))
-
-				inst['suspended'] = True
-				inst['resumetime'] = time.time() + config['suspendtime']
-
-			if inst['socket'] != None:
-				inst['socket'].disconnect()
-				inst['socket'] = None
-
-		socket = inst['socket']
-
-		if socket != None and socket.voltage != 0:
-			collectd.debug("Uploading values for {}".format(socket.mac_address))
-
-			inst['lastsuccess'] = time.time()
-
-			val = collectd.Values(plugin = 'sem6000-{}'.format(config['socketname']))
-
-			val.type = 'voltage'
-			val.type_instance = 'grid'
-			val.values = [ socket.voltage ]
-			val.dispatch()
-
-			val.type = 'current'
-			val.type_instance = 'load'
-			val.values = [ socket.current ]
-			val.dispatch()
-
-			val.type = 'power'
-			val.type_instance = 'real_power'
-			val.values = [ socket.power ]
-			val.dispatch()
-
-			val.type = 'gauge'
-			val.type_instance = 'power_factor'
-			val.values = [ socket.power_factor ]
-			val.dispatch()
-
-			val.type = 'gauge'
-			val.type_instance = 'load_on'
-			val.values = [ socket.powered ]
-			val.dispatch()
-
-			val.type = 'frequency'
-			val.type_instance = 'grid'
-			val.values = [ socket.frequency ]
-			val.dispatch()
-
-def shutdown_func():
-	global instances
-
-	for inst in instances:
-		if inst['socket'] != None:
-			inst['socket'].disconnect()
-
-	instances = []
-
-collectd.register_config(config_func)
-collectd.register_init(init_func)
-collectd.register_read(read_func)
-collectd.register_shutdown(shutdown_func)

example.py

@@ -11,25 +11,22 @@ while True:
     time.sleep(1)
     try:
         if socket == None:
-            print("Connecting... ", end="")
+            print("Connecting...")
+
+            # auto_reconnect_timeout enabled auto reconnect if sending a command fails. Valid values:
+            # None (default): everything that fails throws NotConnectedException's
+            # -1: infinite retries
+            # integer: seconds before exception is thrown
+
             socket = SEMSocket('f0:c7:7f:0d:e7:17')
-            print("Success!")
-            print("You're now connected to: {} (Icon: {})".format(socket.name, socket.icons[0]))
-            if socket.login("1234") and socket.authenticated:
-                print("Login successful!")
-                socket.getSynConfig()
-                print()
-                print("=== Tariff settings ===")
-                print("Default charge:", socket.default_charge)
-                print("Night charge:", socket.night_charge)
-                print("Night tariff from {} to {}".format(socket.night_charge_start_time.tm_hour, socket.night_charge_end_time.tm_hour))
-                print()
-                print("=== Other settings ===")
-                print("Night mode:", "active" if socket.night_mode else "inactive")
-                print("Power protection:", socket.power_protect)
-                print()
+            print("Connected.")
+
+            #socket.login("1337")
+            #socket.changePassword("1234")
+            #socket.login("1234")
 
         socket.getStatus()
+        socket.setStatus(True)
         print("=== {} ({}) ===".format(socket.mac_address, "on" if socket.powered else "off"))
         print("\t{}V {}A → {}W@{}Hz (PF: {})".format(socket.voltage, socket.current, socket.power, socket.frequency, socket.power_factor))
     except (SEMSocket.NotConnectedException, bluepy.btle.BTLEDisconnectError, BrokenPipeError):

sem6000.py

@@ -1,10 +1,8 @@
 from bluepy import btle
 import time
-import datetime
 import uuid
 
 class SEMSocket():
-    icons = ["plug", "speaker", "flatscreen", "desk lamp", "oven", "kitchen machine", "canning pot", "stanging lamp", "kettle", "mixer", "hanging lamp", "toaster", "washing machine", "fan", "fridge", "iron", "printer", "monitor", "notebook", "workstation", "video recorder", "curling iron", "heater"]
     password = "0000"
     powered = False
     voltage = 0
@@ -15,12 +13,9 @@ class SEMSocket():
     frequency = 0
     mac_address = ""
     custom_service = None
-    authenticated = False
-    _icon_idx = None
-    _name = None
-    _version_char = None
+    _read_char = None
     _write_char = None
-    _name_char = None
+    _notify_char = None
     _btle_device = None
 
     def __init__(self, mac):
@@ -37,13 +32,6 @@ class SEMSocket():
         cmd = bytearray([0x04])
         payload = bytearray([0x00, 0x00, 0x00])
         msg = self.BTLEMessage(self, cmd, payload)
-        return msg.send()
-
-    def getSynConfig(self):
-        #15, 5, 16, 0, 0, 0, 17, -1, -1
-        cmd = bytearray([0x10])
-        payload = bytearray([0x00, 0x00, 0x00])
-        msg = self.BTLEMessage(self, cmd, payload)
         msg.send()
 
     def setStatus(self, status):
@@ -52,18 +40,7 @@ class SEMSocket():
         cmd = bytearray([0x03])
         payload = bytearray([0x00, status, 0x00, 0x00])
         msg = self.BTLEMessage(self, cmd, payload)
-        return msg.send()
-
-    def syncTime(self):
-        #15, 12, 1, 0, SECOND, MINUTE, HOUR_OF_DAY, DAY_OF_MONTH, MONTH (+1), int(YEAR/256), YEAR%256, 0, 0, CHKSUM, 255, 255
-        now = datetime.datetime.now()
-        cmd = bytearray([0x01])
-        payload = bytearray([0x00])
-        payload += bytearray([now.second, now.minute, now.hour])
-        payload += bytearray([now.day, now.month, int(now.year/256), now.year%256])
-        payload += bytearray([0x00, 0x00])
-        msg = self.BTLEMessage(self, cmd, payload)
-        return msg.send()
+        msg.send()
 
     def login(self, password):
         self.password = password
@@ -78,9 +55,7 @@ class SEMSocket():
         payload.append(0x00)
         payload.append(0x00)
         msg = self.BTLEMessage(self, cmd, payload)
-        success = msg.send()
-        self.authenticated = self.authenticated and success
-        return success
+        msg.send()
 
     def changePassword(self, newPassword):
         cmd = bytearray([0x17])
@@ -88,81 +63,20 @@ class SEMSocket():
         payload.append(0x00)
         payload.append(0x01)
         for i in range(4):
-            payload.append(int(newPassword[i]))
+            payload.append(int(self.newPassword[i]))
         for i in range(4):
             payload.append(int(self.password[i]))
         self.password = newPassword
         msg = self.BTLEMessage(self, cmd, payload)
-        success = msg.send()
-        self.authenticated = self.authenticated and success
-        return success
-
-    def setIcon(self, iconIdx):
-        cmd = bytearray([0x0f])
-        payload = bytearray([0x00, 0x03, iconIdx, 0x00, 0x00, 0x00, 0x00])
-        msg = self.BTLEMessage(self, cmd, payload)
-
-        success = msg.send()
-        return success
-
-    def enableLED(self, status):
-        cmd = bytearray([0x0f])
-        payload = bytearray([0x00, 0x05, status, 0x00, 0x00, 0x00, 0x00])
-        msg = self.BTLEMessage(self, cmd, payload)
-
-        success = msg.send()
-        return success
-
-    @property
-    def name(self):
-        self._name = self._name_char.read().decode("UTF-8")
-        return self._name
-
-    @name.setter
-    def name(self, newName):
-        newNameBytes = newName.encode("UTF-8")
-        cmd = bytearray([0x02])
-        payload = bytearray()
-        payload.append(0x02)
-        for i in range(20):
-            if i <= (len(newNameBytes) - 1):
-                payload.append(newNameBytes[i])
-            else:
-                payload.append(0x00)
-        msg = self.BTLEMessage(self, cmd, payload)
-        success = msg.send()
-        # For some reason the original app sets the first 7 bytes of the payload to zero and sends it again.
-        # However, a first test showed, that this really doesn't change anything. If it becomes neccessary here's a first draft:
-        #for i in range(7):
-        #    payload[i+1] = 0x00
-        #msg = self.BTLEMessage(self, cmd, payload)
-        if not success:
-            raise self.SendMessageFailed
-        if self.name != newName:
-            raise self.NotLoggedIn
-
-    @property
-    def serial(self):
-        # 15, 5, 17, 0, 0, 0, 18, -1, -1
-        cmd = bytearray([0x11])
-        payload = bytearray([0x00, 0x00, 0x00])
-        msg = self.BTLEMessage(self, cmd, payload)
-        success = msg.send()
-        if not success:
-            raise self.SendMessageFailed
-        return self._serial
-
-    @property
-    def firmware_version(self):
-        char_value = self._version_char.read()
-        major = char_value[11]
-        minor = char_value[12]
-        return "{}.{}".format(major, minor)
+        msg.send()
 
     @property
     def connected(self):
         try:
-            return "conn" in self._btle_device.status().get("state")
+            if "conn" in self._btle_device.status().get("state"):
+                return True
+            else:
+                return False
         except:
             return False
 
@@ -176,30 +90,44 @@ class SEMSocket():
         self.disconnect()
         self._btle_device.connect(self.mac_address)
         self._btle_handler = self.BTLEHandler(self)
-        self._btle_device.setDelegate(self._btle_handler)
 
         self._custom_service = self._btle_device.getServiceByUUID(0xfff0)
-        self._version_char   = self._custom_service.getCharacteristics("0000fff1-0000-1000-8000-00805f9b34fb")[0] #contains firmware version info
-        self._write_char     = self._custom_service.getCharacteristics("0000fff3-0000-1000-8000-00805f9b34fb")[0] #is used to write commands
-        self._name_char      = self._custom_service.getCharacteristics("0000fff6-0000-1000-8000-00805f9b34fb")[0]
+        self._read_char      = self._custom_service.getCharacteristics("0000fff1-0000-1000-8000-00805f9b34fb")[0]
+        self._write_char     = self._custom_service.getCharacteristics("0000fff3-0000-1000-8000-00805f9b34fb")[0]
+        self._notify_char    = self._custom_service.getCharacteristics("0000fff4-0000-1000-8000-00805f9b34fb")[0]
+        self._btle_device.setDelegate(self._btle_handler)
 
     def disconnect(self):
         if self.connected == True:
             self._btle_device.disconnect()
 
+    #def SynVer(self):
+    #    print("SynVer")
+    #    self.read_char.read_value()
+
+    #def GetSynConfig(self):
+    #    print("GetSynConfig")
+    #    #15, 5, 16, 0, 0, 0, 17, -1, -1
+    #    self.write_char.write_value(bytearray(b'\x0f\x05\x10\x00\x00\x00\x11\xff\xff'))
+
     #def ______RESET(self):
     #    #15, 5, 16, 0, 0, 0, 17, -1, -1  ??? maybe reset?
     #    pass
 
+    #def GetSN(self):
+    #    print("GetSN")
+    #    #15, 5, 17, 0, 0, 0, 18, -1, -1
+    #    self.write_char.write_value(bytearray(b'\x0f\x05\x11\x00\x00\x00\x12\xff\xff'))
+
+    #    self.SynVer()
+    #    self.notify_char.enable_notifications()
+    #    self.Login("1337")
+    #    self.GetSynConfig()
+    #    #self.GetSN()
+
     class NotConnectedException(Exception):
         pass
 
-    class SendMessageFailed(Exception):
-        pass
-
-    class NotLoggedIn(Exception):
-        pass
-
     class BTLEMessage():
         MAGIC_START = bytearray([0x0f])
         MAGIC_END = bytearray([0xff, 0xff])
@@ -248,7 +176,7 @@ class SEMSocket():
                 self.__btle_device.reconnect()
 
             self.__btle_device._write_char.write(self.__data, True)
-            return self.__btle_device._btle_device.waitForNotifications(5)
+            self.__btle_device._btle_device.waitForNotifications(5)
 
 
     class BTLEHandler(btle.DefaultDelegate):
@@ -257,28 +185,16 @@ class SEMSocket():
             self.__btle_device = btle_device
 
         def handleNotification(self, cHandle, data):
-            if len(data) <= 3:
-                print("Notification data seems invalid or incomplete. Could not parse: ", end="")
-                print(data)
-                return
-
             message_type = data[2]
             if message_type == 0x00:
                 if data[4] == 0x01:
                     print("Checksum error!")
                 else:
                     print("Unknown error:", data)
-            elif message_type == 0x01: #sync time response
-                if not data[3:] == b'\x00\x00\x02\xff\xff':
-                    print("Time synced failed with unknown data: ", end="")
-                    print(data)
-            elif message_type == 0x02: #set name response
-                if not data[3:] == b'\x00\x00\x03\xff\xff':
-                    print("Set name failed with unknown data: ", end="")
-                    print(data)
-            elif message_type == 0x03: #switch toggle response
+            elif message_type == 0x03: #switch toggle
+                print("Switch toggled")
                 self.__btle_device.getStatus()
-            elif message_type == 0x04: #status related response
+            elif message_type == 0x04: #status related data
                 voltage     = data[8]
                 current     = (data[9] << 8 | data[10]) / 1000
                 power       = (data[5] << 16 | data[6] << 8 | data[7]) / 1000
@@ -296,36 +212,11 @@ class SEMSocket():
                     self.__btle_device.power_factor = power / (voltage * current)
                 except ZeroDivisionError:
                     self.__btle_device.power_factor = None
-            elif message_type == 0x10: #plug settings response
-                self.__btle_device.default_charge = (data[5] / 100)
-                self.__btle_device.night_charge = (data[6] / 100)
-                night_charge_start_time = int((data[7] << 8 | data[8]) / 60)
-                night_charge_end_time = int((data[9] << 8 | data[10]) / 60)
-                self.__btle_device.night_charge_start_time = time.strptime(str(night_charge_start_time), "%H")
-                self.__btle_device.night_charge_end_time = time.strptime(str(night_charge_end_time), "%H")
-                self.__btle_device.night_mode = not bool(data[11])
-                self.__btle_device.icon_idx = data[12]
-                self.__btle_device.power_protect = (data[13] << 8 | data[14])
-            elif message_type == 0x11: #serial number response
-                self.__btle_device._serial = data[-16:].decode("UTF-8")
-            elif message_type == 0x17: #authentication related response
+            elif message_type == 0x17:
                 if data[5] == 0x00 or data[5] == 0x01:
-                    # in theory the fifth byte indicates a login attempt response (0) or a response to a password change (1)
-                    # but since a password change requires a valid login and a successful password changes logs you in,
-                    # we can just ignore this bit and set the authenticated flag accordingly for both responses
-                    self.__btle_device.authenticated = not data[4]
+                    if data[4]:
+                        print("Login failed")
                 else:
                     print("5th byte of login-response is > 1:", data)
-            elif message_type == 0x0f: #set icon response
-                if data[3:6] == b'\x00\x03\x00':
-                    # LED set successfully
-                    pass
-                elif data[3:6] == b'\x00\x05\x00':
-                    # Icon set successfully
-                    pass
-                else:
-                    print("Unknown response for setting icon: ", end="")
-                    print(data[3:])
             else:
                 print ("Unknown message from Handle: 0x" + format(cHandle,'02X') + " Value: "+ format(data))
-