forked from sqozz/sem6000
Add timeout and reconnect support
This commit is contained in:
parent
21433a3d20
commit
5c06fc131a
29
example.py
29
example.py
|
@ -1,16 +1,23 @@
|
|||
from sem6000 import SEMSocket
|
||||
import time
|
||||
socket = SEMSocket('f0:c7:7f:0d:e7:17')
|
||||
from sem6000 import SEMSocket
|
||||
|
||||
socket.login("1337")
|
||||
socket.changePassword("1234")
|
||||
socket.login("1234")
|
||||
#socket.changePassword("1337")
|
||||
# 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', auto_reconnect_timeout=None)
|
||||
|
||||
#socket.login("1337")
|
||||
#socket.changePassword("1234")
|
||||
#socket.login("1234")
|
||||
|
||||
while True:
|
||||
break;
|
||||
time.sleep(1)
|
||||
socket.getStatus()
|
||||
socket.setStatus(True)
|
||||
print("=== {} ({}) ===".format(socket.mac_address, "on" if socket.powered else "off"))
|
||||
print("\t{}V {}A → {}W@{}Hz".format(socket.voltage, socket.current, socket.power, socket.frequency))
|
||||
try:
|
||||
socket.getStatus()
|
||||
socket.setStatus(True)
|
||||
print("=== {} ({}) ===".format(socket.mac_address, "on" if socket.powered else "off"))
|
||||
print("\t{}V {}A → {}W@{}Hz".format(socket.voltage, socket.current, socket.power, socket.frequency))
|
||||
except SEMSocket.NotConnectedException:
|
||||
socket.reconnect(-1) #infinite reconnect attempts
|
||||
|
|
54
sem6000.py
54
sem6000.py
|
@ -1,11 +1,11 @@
|
|||
from bluepy import btle
|
||||
import struct
|
||||
import time
|
||||
import pdb
|
||||
import uuid
|
||||
|
||||
class SEMSocket():
|
||||
password = "0000"
|
||||
auto_reconnect_timeout = None
|
||||
powered = False
|
||||
voltage = 0
|
||||
current = 0
|
||||
|
@ -16,18 +16,19 @@ class SEMSocket():
|
|||
read_char = None
|
||||
write_char = None
|
||||
notify_char = None
|
||||
connected = False
|
||||
|
||||
def __init__(self, mac):
|
||||
def __init__(self, mac, auto_reconnect_timeout = None):
|
||||
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.auto_reconnect_timeout = auto_reconnect_timeout
|
||||
try:
|
||||
self.reconnect()
|
||||
except self.NotConnectedException:
|
||||
# initial connection may fail. It is up to the code what to do
|
||||
pass
|
||||
|
||||
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 NotConnectedException(Exception):
|
||||
pass
|
||||
|
||||
class BTLEMessage():
|
||||
MAGIC_START = bytearray([0x0f])
|
||||
|
@ -73,6 +74,9 @@ class SEMSocket():
|
|||
self.__data[1] = 1 + len(self.__payload) + 1 # cmd + payload + checksum
|
||||
|
||||
def send(self):
|
||||
if not self.__btle_device.connected:
|
||||
self.__btle_device.reconnect(self.__btle_device.auto_reconnect_timeout)
|
||||
|
||||
self.__btle_device.write_char.write(self.__data, True)
|
||||
self.__btle_device.btle_device.waitForNotifications(5)
|
||||
|
||||
|
@ -110,7 +114,6 @@ class SEMSocket():
|
|||
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])
|
||||
|
@ -118,7 +121,6 @@ class SEMSocket():
|
|||
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])
|
||||
|
@ -127,7 +129,6 @@ class SEMSocket():
|
|||
msg.send()
|
||||
|
||||
def login(self, password):
|
||||
print("Login")
|
||||
self.password = password
|
||||
cmd = bytearray([0x17])
|
||||
payload = bytearray()
|
||||
|
@ -145,7 +146,6 @@ class SEMSocket():
|
|||
msg.send()
|
||||
|
||||
def changePassword(self, newPassword):
|
||||
print("Change Password")
|
||||
cmd = bytearray([0x17])
|
||||
payload = bytearray()
|
||||
payload.append(0x00)
|
||||
|
@ -162,6 +162,32 @@ class SEMSocket():
|
|||
msg = self.BTLEMessage(self, cmd, payload)
|
||||
msg.send()
|
||||
|
||||
def __reconnect(self):
|
||||
try:
|
||||
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]
|
||||
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)
|
||||
except btle.BTLEException as e:
|
||||
self.connected = False
|
||||
else:
|
||||
self.connected = True
|
||||
|
||||
def reconnect(self, timeout = None):
|
||||
if timeout == None:
|
||||
self.__reconnect()
|
||||
else:
|
||||
reconnect_start = time.time()
|
||||
while abs(reconnect_start - time.time()) < timeout or timeout == -1:
|
||||
self.__reconnect()
|
||||
|
||||
if not self.connected:
|
||||
raise self.NotConnectedException
|
||||
|
||||
#def SynVer(self):
|
||||
# print("SynVer")
|
||||
# self.read_char.read_value()
|
||||
|
|
Loading…
Reference in a new issue