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

Allow setting custom discover ip port

Browse Source
This commit is contained in:
Daan Klijn 2020-08-10 21:28:09 +02:00 committed by Matthew Garrett
parent fe4aa5b991
commit 8167736588
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
broadlink/__init__.py
View File

@ -109,7 +109,12 @@ def gendevice(dev_type, host, mac, name=None, is_locked=None):
) )
def discover(timeout=None, local_ip_address=None, discover_ip_address='255.255.255.255'): def discover(
timeout=None,
local_ip_address=None,
discover_ip_address='255.255.255.255',
discover_ip_port=80
):
if local_ip_address is None: if local_ip_address is None:
with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s: with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
s.connect(('8.8.8.8', 53)) # connecting to a UDP address doesn't send packets s.connect(('8.8.8.8', 53)) # connecting to a UDP address doesn't send packets
@ -156,7 +161,7 @@ def discover(timeout=None, local_ip_address=None, discover_ip_address='255.255.2
packet[0x1c] = port & 0xff packet[0x1c] = port & 0xff
packet[0x1d] = port >> 8 packet[0x1d] = port >> 8
packet[0x26] = 6 packet[0x26] = 6
checksum = 0xbeaf checksum = 0xbeaf
for b in packet: for b in packet:
checksum = (checksum + b) & 0xffff checksum = (checksum + b) & 0xffff
@ -164,7 +169,7 @@ def discover(timeout=None, local_ip_address=None, discover_ip_address='255.255.2
packet[0x20] = checksum & 0xff packet[0x20] = checksum & 0xff
packet[0x21] = checksum >> 8 packet[0x21] = checksum >> 8
cs.sendto(packet, (discover_ip_address, 80)) cs.sendto(packet, (discover_ip_address, discover_ip_port))
if timeout is None: if timeout is None:
response = cs.recvfrom(1024) response = cs.recvfrom(1024)
responsepacket = bytearray(response[0]) responsepacket = bytearray(response[0])
@ -285,7 +290,7 @@ class device:
response = self.send_packet(0x6a, packet) response = self.send_packet(0x6a, packet)
check_error(response[0x22:0x24]) check_error(response[0x22:0x24])
payload = self.decrypt(response[0x38:]) payload = self.decrypt(response[0x38:])
return payload[0x4] | payload[0x5] << 8 return payload[0x4] | payload[0x5] << 8
def set_name(self, name): def set_name(self, name):
packet = bytearray(4) packet = bytearray(4)
@ -446,7 +451,7 @@ class bg1(device):
def get_state(self): def get_state(self):
"""Get state of device. """Get state of device.
Returns: Returns:
dict: Dictionary of current state dict: Dictionary of current state
eg. `{"pwr":1,"pwr1":1,"pwr2":0,"maxworktime":60,"maxworktime1":60,"maxworktime2":0,"idcbrightness":50}`""" eg. `{"pwr":1,"pwr1":1,"pwr2":0,"maxworktime":60,"maxworktime1":60,"maxworktime2":0,"idcbrightness":50}`"""