mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Expose the type of the discovered device

Browse source 
Allow clients to identify what kind of device has been discovered
This commit is contained in:
Matthew Garrett 2016-12-03 14:22:20 -08:00
parent c157860f0d
commit 31cd077165
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
broadlink/__init__.py
View file

@ -123,6 +123,7 @@ def discover(timeout=None):
dev = gendevice(devtype, host, mac) dev = gendevice(devtype, host, mac)
devices.append(dev) devices.append(dev)
class device: class device:
def __init__(self, host, mac, timeout=10): def __init__(self, host, mac, timeout=10):
self.host = host self.host = host
@ -136,6 +137,7 @@ class device:
self.cs.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) self.cs.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.cs.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1) self.cs.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
self.cs.bind(('',0)) self.cs.bind(('',0))
self.type = "Unknown"
def auth(self): def auth(self):
payload = bytearray(0x50) payload = bytearray(0x50)
@ -174,6 +176,9 @@ class device:
self.id = payload[0x00:0x04] self.id = payload[0x00:0x04]
self.key = payload[0x04:0x14] self.key = payload[0x04:0x14]
def get_type(self):
return self.type
def send_packet(self, command, payload): def send_packet(self, command, payload):
self.count = (self.count + 1) & 0xffff self.count = (self.count + 1) & 0xffff
packet = bytearray(0x38) packet = bytearray(0x38)
@ -239,6 +244,7 @@ class device:
class sp1(device): class sp1(device):
def __init__ (self, host, mac): def __init__ (self, host, mac):
device.__init__(self, host, mac) device.__init__(self, host, mac)
self.type = "SP1"
def set_power(self, state): def set_power(self, state):
packet = bytearray(4) packet = bytearray(4)
@ -249,6 +255,7 @@ class sp1(device):
class sp2(device): class sp2(device):
def __init__ (self, host, mac): def __init__ (self, host, mac):
device.__init__(self, host, mac) device.__init__(self, host, mac)
self.type = "SP2"
def set_power(self, state): def set_power(self, state):
"""Sets the power state of the smart plug.""" """Sets the power state of the smart plug."""
@ -271,6 +278,7 @@ class sp2(device):
class a1(device): class a1(device):
def __init__ (self, host, mac): def __init__ (self, host, mac):
device.__init__(self, host, mac) device.__init__(self, host, mac)
self.type = "A1"
def check_sensors(self): def check_sensors(self):
packet = bytearray(16) packet = bytearray(16)
@ -320,6 +328,7 @@ class a1(device):
class rm(device): class rm(device):
def __init__ (self, host, mac): def __init__ (self, host, mac):
device.__init__(self, host, mac) device.__init__(self, host, mac)
self.type = "RM2"
def check_data(self): def check_data(self):
packet = bytearray(16) packet = bytearray(16)