kamp/python-broadlink
1
0
Fork 0
mirror of https://github.com/mjg59/python-broadlink.git synced 2024-11-22 07:00:12 +01:00
Code

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)