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

Merge pull request #16 from PeWu/smartplug

Browse Source 
Updated SmartPlug commands - set_power() and check_power()
This commit is contained in:
Matthew Garrett 2016-11-19 16:47:28 -08:00 committed by GitHub
commit 7ac243b838
2 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
README.md
View File

@ -44,7 +44,12 @@ Obtain sensor data from an A1:
data = devices[0].check_sensors() data = devices[0].check_sensors()
``` ```
Set power state on an SP2/SP3 (0 for off, 1 for on): Set power state on a SmartPlug SP2/SP3:
```
devices[0].set_power(True)
```
Check power state on a SmartPlug:
```
state = devices[0].check_power()
``` ```
devices[0].set_power(1)
```

15
broadlink/__init__.py
View File

@ -241,11 +241,22 @@ class sp2(device):
device.__init__(self, host, mac) device.__init__(self, host, mac)
def set_power(self, state): def set_power(self, state):
packet = bytearray(8) """Sets the power state of the smart plug."""
packet = bytearray(16)
packet[0] = 2 packet[0] = 2
packet[4] = state packet[4] = 1 if state else 0
self.send_packet(0x6a, packet) self.send_packet(0x6a, packet)
def check_power(self):
"""Returns the power state of the smart plug."""
packet = bytearray(16)
packet[0] = 1
response = self.send_packet(0x6a, packet)
err = ord(response[0x22]) | (ord(response[0x23]) << 8)
if err == 0:
aes = AES.new(str(self.key), AES.MODE_CBC, str(self.iv))
payload = aes.decrypt(str(response[0x38:]))
return bool(ord(payload[0x4]))
class a1(device): class a1(device):
def __init__ (self, host, mac): def __init__ (self, host, mac):