109 lines
3.7 KiB
Python
109 lines
3.7 KiB
Python
#!/usr/bin/env python
|
|
#
|
|
# Copyright (C) 2017 pilino1234
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
#
|
|
|
|
"""Example Python 3 client for pilight daemon socket API
|
|
|
|
Runs a SSDP discovery for pilight daemons on the network, and connects to it.
|
|
Identifies as a "receiver" type client, and prints received messages until
|
|
terminated.
|
|
|
|
"""
|
|
|
|
import socket
|
|
import struct
|
|
import re
|
|
|
|
|
|
def discover(service, retries=1):
|
|
"""SSDP discovery function
|
|
|
|
:param service: The name of the SSDP service to run discovery for
|
|
:type service: str
|
|
:param retries: The number of discovery retries. Defaults to 1.
|
|
:type retries: int
|
|
|
|
"""
|
|
|
|
group = ("239.255.255.250", 1900)
|
|
discover_message = "\r\n".join([
|
|
'M-SEARCH * HTTP/1.1',
|
|
'HOST: {0}:{1}'.format(*group),
|
|
'MAN: "ssdp:discover"',
|
|
'ST: {st}', 'MX: 3', '', '']).format(st=service)
|
|
|
|
ssdp_responses = {}
|
|
|
|
for i in range(retries):
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM,
|
|
socket.IPPROTO_UDP)
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_RCVTIMEO,
|
|
struct.pack('LL', 0, 10000))
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|
sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, 2)
|
|
sock.sendto(bytes(discover_message, "utf-8"), group)
|
|
|
|
while True:
|
|
try:
|
|
ssdp_responses[i] = sock.recv(1024).decode("utf-8")
|
|
break
|
|
except socket.timeout:
|
|
break
|
|
except IOError:
|
|
print("no pilight ssdp connections found")
|
|
break
|
|
|
|
sock.close()
|
|
return ssdp_responses
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
RESPONSES = discover("urn:schemas-upnp-org:service:pilight:1")
|
|
|
|
if len(RESPONSES) > 0:
|
|
for resp in RESPONSES: # try all responses until one works
|
|
locationsrc = re.search("Location:([0-9.]+):(.*)",
|
|
str(RESPONSES[resp]), re.IGNORECASE)
|
|
|
|
if locationsrc:
|
|
location = locationsrc.group(1)
|
|
port = locationsrc.group(2)
|
|
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
|
try:
|
|
s.connect((location, int(port)))
|
|
except ConnectionRefusedError:
|
|
# Connection was refused, try next SSDP response
|
|
print("connection refused")
|
|
continue
|
|
|
|
s.send(bytes('{"action": "identify", \
|
|
"options": {"receiver": 1}}', "utf-8"))
|
|
response = ""
|
|
while True:
|
|
line = s.recv(1024).decode("utf-8")
|
|
response += line
|
|
if "\n\n" in line[-2:]:
|
|
response = response[:-2]
|
|
break
|
|
|
|
if response == '{"status":"success"}':
|
|
message = ""
|
|
while True:
|
|
line = s.recv(1024).decode("utf-8")
|
|
message += line
|
|
if "\n\n" in line[-2:]:
|
|
message = message[:-2]
|
|
for f in iter(message.splitlines()):
|
|
print(f)
|
|
message = ""
|
|
elif response == '{"status":"failure"}':
|
|
# connection failed, try next SSDP response
|
|
print("connection failed")
|
|
continue
|