115 lines
3.7 KiB
Python
115 lines
3.7 KiB
Python
#!/usr/bin/env python3
|
|
|
|
""" The dashboard client """
|
|
|
|
import os
|
|
import logging
|
|
import subprocess
|
|
|
|
from dotenv import find_dotenv, load_dotenv
|
|
import paho.mqtt.client as mqtt
|
|
|
|
# Configuration
|
|
load_dotenv(find_dotenv())
|
|
MQTT_SERVER = os.getenv("MQTT_SERVER")
|
|
MQTT_USER = os.getenv("MQTT_USER")
|
|
MQTT_PASSWORD = os.getenv("MQTT_PASSWORD")
|
|
MQTT_BASE_TOPIC = os.getenv("MQTT_BASE_TOPIC", "pylokid")
|
|
CEC_ENABLED = os.getenv("CEC_ENABLED", "yes")
|
|
TMP_DIR = os.getenv("TMP_DIR", "/tmp")
|
|
|
|
# Initialization
|
|
logging.basicConfig(
|
|
level=logging.DEBUG,
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|
)
|
|
LOGGER = logging.getLogger('dashboard')
|
|
|
|
PIDS = {}
|
|
|
|
def on_connect(client, userdata, flags, rc):
|
|
LOGGER.info("Connected to MQTT with result code %s", str(rc))
|
|
|
|
# Subscribing in on_connect() means that if we lose the connection and
|
|
# reconnect then subscriptions will be renewed.
|
|
client.subscribe("pylokid/#")
|
|
|
|
def on_message(client, userdata, msg):
|
|
topic_detail = msg.topic.split("/")
|
|
f_id = topic_detail[2]
|
|
if topic_detail[1] == 'Einsatzausdruck_FW' and topic_detail[3] == 'pdf':
|
|
LOGGER.info("[%s] New Einsatzausdruck received", f_id)
|
|
file_name = TMP_DIR + "/dashboard_" + f_id + ".pdf"
|
|
file = open(file_name, "wb")
|
|
file.write(msg.payload)
|
|
file.close()
|
|
|
|
if f_id in PIDS:
|
|
LOGGER.info(
|
|
"[%s] Einsatzausdruck already being displayed with PID %s",
|
|
f_id,
|
|
str(PIDS[f_id])
|
|
)
|
|
else:
|
|
LOGGER.info("[%s] Displaying Einsatzausdruck with xpdf", f_id)
|
|
# TODO turn on TV with cec-client
|
|
process = subprocess.Popen(
|
|
["/usr/bin/xpdf", "-z", "width", "-fullscreen", file_name],
|
|
env=dict(os.environ, DISPLAY=":0")
|
|
)
|
|
PIDS[f_id] = process.pid
|
|
|
|
if CEC_ENABLED == "yes":
|
|
# Check power state of TV
|
|
status = subprocess.run(
|
|
["/usr/bin/cec-client", "-s", "-d", "1"],
|
|
stdout=subprocess.PIPE,
|
|
input=b'pow 0').stdout
|
|
if status.splitlines()[1] == b'power status: standby':
|
|
LOGGER.info("[%s] CEC power status: standby. Powering TV on", f_id)
|
|
subprocess.run(
|
|
["/usr/bin/cec-client", "-s", "-d", "1"],
|
|
stdout=subprocess.PIPE,
|
|
input=b'on 0'
|
|
)
|
|
else:
|
|
LOGGER.info("[%s] CEC power status: probably on", f_id)
|
|
elif topic_detail[1] == 'Einsatzprotokoll':
|
|
LOGGER.info("[%s] New Einsatzprotokoll received", f_id)
|
|
if f_id in PIDS:
|
|
LOGGER.info("[%s] Killing xpdf PID %s", f_id, str(PIDS[f_id]))
|
|
os.kill(PIDS[f_id], 9)
|
|
PIDS.pop(f_id)
|
|
else:
|
|
LOGGER.info("[%s] No xpdf PID found", f_id)
|
|
|
|
if CEC_ENABLED == "yes":
|
|
# Turn off TV if no xpdf running anymore
|
|
if not PIDS:
|
|
LOGGER.info("[%s] No xpdf running anymore. Powering TV off", f_id)
|
|
subprocess.run(
|
|
["/usr/bin/cec-client", "-s", "-d", "1"],
|
|
stdout=subprocess.PIPE,
|
|
input=b'standby 0'
|
|
)
|
|
else:
|
|
LOGGER.info("[%s] Unknown", topic_detail[1])
|
|
|
|
def main():
|
|
""" main """
|
|
|
|
mqtt_client = mqtt.Client()
|
|
mqtt_client.on_connect = on_connect
|
|
mqtt_client.on_message = on_message
|
|
|
|
mqtt_client.username_pw_set(MQTT_USER, password=MQTT_PASSWORD)
|
|
mqtt_client.tls_set()
|
|
mqtt_client.connect(MQTT_SERVER, 8883, 60)
|
|
mqtt_client.loop_forever()
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
print("Byebye")
|