Merge pull request #34 from eschava/master

allow to specify network interface for discover method
This commit is contained in:
Matthew Garrett 2016-12-25 11:07:54 +00:00 committed by GitHub
commit 877a7b7460

11
broadlink/__init__.py Executable file → Normal file
View file

@ -51,15 +51,16 @@ def gendevice(devtype, host, mac):
else: else:
return device(host=host, mac=mac) return device(host=host, mac=mac)
def discover(timeout=None): def discover(timeout=None, local_ip_address=None):
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) if local_ip_address is None:
s.connect(('8.8.8.8', 53)) # connecting to a UDP address doesn't send packets s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
local_ip_address = s.getsockname()[0] s.connect(('8.8.8.8', 53)) # connecting to a UDP address doesn't send packets
local_ip_address = s.getsockname()[0]
address = local_ip_address.split('.') address = local_ip_address.split('.')
cs = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) cs = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
cs.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) cs.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
cs.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1) cs.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
cs.bind(('',0)) cs.bind((local_ip_address,0))
port = cs.getsockname()[1] port = cs.getsockname()[1]
starttime = time.time() starttime = time.time()