from bluepy import btle import struct import time import pdb import uuid class SEMSocket(): password = "0000" powered = False voltage = 0 current = 0 power = 0 frequency = 0 mac_address = "" custom_service = None read_char = None write_char = None notify_char = None def __init__(self, mac): self.mac_address = mac self.btle_device = btle.Peripheral(self.mac_address,addrType=btle.ADDR_TYPE_PUBLIC,iface=0) self.btle_handler = self.BTLEHandler(self) self.custom_service = self.btle_device.getServiceByUUID(0xfff0) self.read_char = self.custom_service.getCharacteristics("0000fff1-0000-1000-8000-00805f9b34fb")[0] print(self.read_char.read()) 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) class BTLEMessage(): MAGIC_START = bytearray([0x0f]) MAGIC_END = bytearray([0xff, 0xff]) __data = bytearray() __cmd = bytearray(1) # cmd cannot be empty __payload = bytearray() def __init__(self, btle_device, cmd=bytearray(), payload=bytearray()): self.__btle_device = btle_device self.cmd = cmd self.payload = payload @property def cmd(self): return self.__cmd @cmd.setter def cmd(self, cmd): self.__data = self.MAGIC_START + bytearray(1) + cmd + self.payload + bytearray(1) + self.MAGIC_END self.__cmd = cmd self.__calc_length() self.__calc_checksum() @property def payload(self): return self.__payload @payload.setter def payload(self, payload): self.__data = self.MAGIC_START + bytearray(1) + self.cmd + payload + bytearray(1) + self.MAGIC_END self.__payload = payload self.__calc_length() self.__calc_checksum() def __calc_checksum(self): checksum = 1 for i in range(2, self.__data[1] + 2): checksum += self.__data[i] self.__data[-3] = checksum & 0xff def __calc_length(self): self.__data[1] = 1 + len(self.__payload) + 1 # cmd + payload + checksum def send(self): self.__btle_device.write_char.write(self.__data, True) self.__btle_device.btle_device.waitForNotifications(5) class BTLEHandler(btle.DefaultDelegate): def __init__(self, btle_device): btle.DefaultDelegate.__init__(self) self.btle_device = btle_device def handleNotification(self, cHandle, data): message_type = data[2] if message_type == 0x00: if data[4] == 0x01: print("Checksum error!") else: print("Unknown error:", data) elif message_type == 0x03: #switch toggle print("Switch toggled") self.btle_device.getStatus() elif message_type == 0x04: #status related data self.btle_device.voltage = data[8] self.btle_device.current = (data[9] << 8 | data[10]) / 1000 self.btle_device.power = (data[5] << 16 | data[6] << 8 | data[7]) / 1000 self.btle_device.frequency = data[11] self.btle_device.powered = bool(data[4]) elif message_type == 0x17: if data[5] == 0x00 or data[5] == 0x01: if not data[4]: print("Login successful") else: print("Login failed") else: print("5th byte of login-response is > 1:", data) else: print ("Unknown message from Handle: 0x" + format(cHandle,'02X') + " Value: "+ format(data)) def getStatus(self): print("GetStatus") #15, 5, 4, 0, 0, 0, 5, -1, -1 cmd = bytearray([0x04]) payload = bytearray([0x00, 0x00, 0x00]) msg = self.BTLEMessage(self, cmd, payload) msg.send() def setStatus(self, status): print("SetStatus:", status) # 0f 06 03 00 01 00 00 05 ff ff -> on # 0f 06 03 00 00 00 00 04 ff ff -> off cmd = bytearray([0x03]) payload = bytearray([0x00, status, 0x00, 0x00]) msg = self.BTLEMessage(self, cmd, payload) msg.send() def login(self, password): print("Login") self.password = password cmd = bytearray([0x17]) payload = bytearray() payload.append(0x00) payload.append(0x00) payload.append(int(self.password[0])) payload.append(int(self.password[1])) payload.append(int(self.password[2])) payload.append(int(self.password[3])) payload.append(0x00) payload.append(0x00) payload.append(0x00) payload.append(0x00) msg = self.BTLEMessage(self, cmd, payload) msg.send() def changePassword(self, newPassword): print("Change Password") cmd = bytearray([0x17]) payload = bytearray() payload.append(0x00) payload.append(0x01) payload.append(int(newPassword[0])) payload.append(int(newPassword[1])) payload.append(int(newPassword[2])) payload.append(int(newPassword[3])) payload.append(int(self.password[0])) payload.append(int(self.password[1])) payload.append(int(self.password[2])) payload.append(int(self.password[3])) self.password = newPassword msg = self.BTLEMessage(self, cmd, payload) msg.send() #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()