Heizberry/beamerctl.py

170 lines
4.5 KiB
Python
Raw Normal View History

2019-02-20 17:59:07 +00:00
#!/usr/bin/env python3
import argparse
import re
import serial
import time
import signal
import logging
import sys
import schedule
import paho.mqtt.client as mqtt
from beamer import acer as beamercodes
code = beamercodes.clubbeamer()
def on_connect(client, userdata, flags, rc):
global beamer_state
global input_source
log.debug("Connected with result code " + str(rc))
beamer_state = readBeamerState()
client.publish("foobar/oben/cantina/beamer/status", beamer_state, qos=1, retain=True)
input_source = readInputSource()
client.publish("foobar/oben/cantina/beamer/source", input_source, qos=1, retain=True)
client.subscribe("foobar/oben/cantina/beamer/action")
def on_publish(client, userdata, mid):
log.debug("mid: "+str(mid))
def on_message(client, userdata, message):
msg = message.payload.decode("utf-8")
log.debug('received message: {}'.format(msg))
if (msg == "on"):
ser.write(code.set['on'])
if (msg == "off"):
ser.write(code.set['off'])
if (msg == "vga"):
ser.write(code.set['vga'])
if (msg == "hdmi1"):
ser.write(code.set['hdmi1'][0])
time.sleep(1)
ser.write(code.set['hdmi1'][1])
if (msg == "hdmi2"):
ser.write(code.set['hdmi2'][0])
time.sleep(0.5)
ser.write(code.set['hdmi2'][1])
time.sleep(0.5)
ser.write(code.set['hdmi2'][2])
def readBeamerState():
ser.write(b"* 0 Lamp ?\r")
read_val = ser.read(size=64)
if(re.match(b'.*Lamp.1.*', read_val)):
status = "on"
else:
status = "off"
return status
def beamerStatusChanged():
global beamer_state
if (beamer_state != readBeamerState()):
log.debug('Beamer state has changed')
beamer_state = readBeamerState()
return True
else:
return False
def readInputSource():
ser.write(b"* 0 Src ?\r")
read_val = ser.read(size=64)
if(re.match(b'.*Src.0.*', read_val)):
return "no_input"
elif (re.match(b'.*Src.1.*', read_val)):
return "vga"
elif (re.match(b'.*Src.8.*', read_val)):
return "hdmi"
elif (readBeamerState()):
return "off"
def inputSourceChanged():
global input_source
if (input_source != readInputSource()):
log.debug('Input source has changed to {}'.format(readInputSource() ))
input_source = readInputSource()
return True
else:
return False
def terminate(signum, frame):
log.warn('SIGTERM received. Shutting down!')
log.info('Closing serial connection')
ser.close()
log.info('stopping mqtt client')
client.loop_stop()
log.info('disconnecting mqtt client')
client.disconnect()
log.info('beamerctl stopped all functions; exit')
sys.exit(0)
def getArgs():
parser = argparse.ArgumentParser()
parser.add_argument("-v", "--verbose", action="store_true",
help="increase output verbosity")
return parser.parse_args()
def periodicUpdate():
log.debug('run scheduled statusupdates')
client.publish("foobar/oben/cantina/beamer/status", readBeamerState(), qos=1, retain=True)
client.publish("foobar/oben/cantina/beamer/source", readInputSource(), qos=1, retain=True)
if __name__ == '__main__':
beamer_state = False
input_source = False
signal.signal(signal.SIGINT, terminate)
args = getArgs()
logging.basicConfig(
level=logging.DEBUG, format='%(asctime)s %(name)-12s %(levelname)-8s %(message)s')
log = logging.getLogger('beamerctl')
if args.verbose:
log.setLevel(logging.DEBUG)
log.info('Loglevel set to DEBUG')
else:
log.setLevel(logging.WARN)
serialport = '/dev/ttyUSB0'
try:
ser = serial.Serial(port=serialport, baudrate=9600, timeout=1)
except serial.serialutil.SerialException as e:
log.critical(e)
sys.exit(1)
client = mqtt.Client()
client.on_connect = on_connect
client.on_message = on_message
client.connect("10.42.0.244", 1883, 60)
client.loop_start()
log.debug('schedule periodic statusupdates')
schedule.every(1).minutes.do(periodicUpdate)
time.sleep(5)
while True:
schedule.run_pending()
if (inputSourceChanged()):
client.publish("foobar/oben/cantina/beamer/source",
input_source, qos=1, retain=True)
if (beamerStatusChanged()):
client.publish("foobar/oben/cantina/beamer/status",
beamer_state, qos=1, retain=True)
time.sleep(1)