mirror of
https://github.com/mjg59/python-broadlink.git
synced 2024-11-11 02:10:12 +01:00
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
import json
|
|
from typing import Union
|
|
|
|
from .device import device
|
|
from .exceptions import check_error
|
|
|
|
|
|
class lb1(device):
|
|
"""Controls a Broadlink LB1."""
|
|
|
|
state_dict = []
|
|
effect_map_dict = {
|
|
'lovely color': 0,
|
|
'flashlight': 1,
|
|
'lightning': 2,
|
|
'color fading': 3,
|
|
'color breathing': 4,
|
|
'multicolor breathing': 5,
|
|
'color jumping': 6,
|
|
'multicolor jumping': 7,
|
|
}
|
|
|
|
def __init__(self, *args, **kwargs) -> None:
|
|
"""Initialize the controller."""
|
|
device.__init__(self, *args, **kwargs)
|
|
self.type = "SmartBulb"
|
|
|
|
def send_command(self, command: str, type: str = 'set') -> None:
|
|
"""Send a command to the device."""
|
|
packet = bytearray(16+(int(len(command)/16) + 1)*16)
|
|
packet[0x00] = 0x0c + len(command) & 0xff
|
|
packet[0x02] = 0xa5
|
|
packet[0x03] = 0xa5
|
|
packet[0x04] = 0x5a
|
|
packet[0x05] = 0x5a
|
|
packet[0x08] = 0x02 if type == "set" else 0x01 # 0x01 => query, # 0x02 => set
|
|
packet[0x09] = 0x0b
|
|
packet[0x0a] = len(command)
|
|
packet[0x0e:] = map(ord, command)
|
|
|
|
checksum = sum(packet, 0xbeaf) & 0xffff
|
|
packet[0x06] = checksum & 0xff # Checksum 1 position
|
|
packet[0x07] = checksum >> 8 # Checksum 2 position
|
|
|
|
response = self.send_packet(0x6a, packet)
|
|
check_error(response[0x36:0x38])
|
|
payload = self.decrypt(bytes(response[0x38:]))
|
|
|
|
responseLength = int(payload[0x0a]) | (int(payload[0x0b]) << 8)
|
|
if responseLength > 0:
|
|
self.state_dict = json.loads(payload[0x0e:0x0e+responseLength])
|
|
|
|
def set_json(self, jsonstr: str) -> str:
|
|
"""Send a command to the device and return state."""
|
|
reconvert = json.loads(jsonstr)
|
|
if 'bulb_sceneidx' in reconvert.keys():
|
|
reconvert['bulb_sceneidx'] = self.effect_map_dict.get(reconvert['bulb_sceneidx'], 255)
|
|
|
|
self.send_command(json.dumps(reconvert))
|
|
return json.dumps(self.state_dict)
|
|
|
|
def set_state(self, state: Union[str, int]) -> None:
|
|
"""Set the state of the device."""
|
|
cmd = '{"pwr":%d}' % (1 if state == "ON" or state == 1 else 0)
|
|
self.send_command(cmd)
|
|
|
|
def get_state(self) -> dict:
|
|
"""Return the state of the device."""
|
|
cmd = "{}"
|
|
self.send_command(cmd)
|
|
return self.state_dict
|