Merge branch 'feature/name_icon' of sqozz/sem6000 into master

This commit is contained in:
sqozz 2020-05-07 21:13:03 +02:00 committed by Gitea
commit be6528f074
1 changed files with 56 additions and 5 deletions

View File

@ -15,9 +15,11 @@ class SEMSocket():
mac_address = ""
custom_service = None
authenticated = False
_name = None
_read_char = None
_write_char = None
_notify_char = None
_name_char = None
_btle_device = None
def __init__(self, mac):
@ -87,6 +89,32 @@ class SEMSocket():
self.authenticated = self.authenticated and success
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 connected(self):
try:
@ -109,6 +137,8 @@ class SEMSocket():
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._name_char = self._custom_service.getCharacteristics("0000fff6-0000-1000-8000-00805f9b34fb")[0]
self._btle_device.setDelegate(self._btle_handler)
def disconnect(self):
@ -142,6 +172,12 @@ class SEMSocket():
class NotConnectedException(Exception):
pass
class SendMessageFailed(Exception):
pass
class NotLoggedIn(Exception):
pass
class BTLEMessage():
MAGIC_START = bytearray([0x0f])
MAGIC_END = bytearray([0xff, 0xff])
@ -199,17 +235,28 @@ 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:
print("Time synced")
elif message_type == 0x03: #switch toggle
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
self.__btle_device.getStatus()
elif message_type == 0x04: #status related data
elif message_type == 0x04: #status related response
voltage = data[8]
current = (data[9] << 8 | data[10]) / 1000
power = (data[5] << 16 | data[6] << 8 | data[7]) / 1000
@ -227,7 +274,7 @@ class SEMSocket():
self.__btle_device.power_factor = power / (voltage * current)
except ZeroDivisionError:
self.__btle_device.power_factor = None
elif message_type == 0x17:
elif message_type == 0x17: #authentication related response
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,
@ -235,6 +282,10 @@ class SEMSocket():
self.__btle_device.authenticated = not data[4]
else:
print("5th byte of login-response is > 1:", data)
elif message_type == 0x0f: #set icon response
if not data[3:] == b'\x00\x03\x00\x13\xff\xff':
print("Unknown response for setting icon: ", end="")
print(data[3:])
else:
print ("Unknown message from Handle: 0x" + format(cHandle,'02X') + " Value: "+ format(data))