pyledriver/webInterface.py

68 lines
1.9 KiB
Python
Raw Normal View History

import logging
from subprocess import check_output, CalledProcessError, run, PIPE
2016-12-30 02:51:56 -05:00
from flask import Flask, render_template, Response, Blueprint, redirect, url_for
from flask_wtf import FlaskForm
from wtforms.fields import StringField, SubmitField
2016-12-30 02:51:56 -05:00
from wtforms.validators import InputRequired
from exceptionThreading import async
2017-05-25 02:40:36 -04:00
logger = logging.getLogger(__name__)
# gag the flask logger unless it has something useful to say
werkzeug = logging.getLogger('werkzeug')
werkzeug.setLevel(logging.ERROR)
2016-12-30 02:51:56 -05:00
class TTSForm(FlaskForm):
tts = StringField(validators=[InputRequired()])
submitTTS = SubmitField('Speak')
class JanusRestart(FlaskForm):
submitRestart = SubmitField('Restart Janus')
2016-12-30 02:51:56 -05:00
# TODO: fix random connection fails (might be an nginx thing)
def janusRunning():
try:
check_output(['pidof', 'janus'])
except CalledProcessError:
logger.warning('Janus not running')
return False
else:
return True
@async(daemon=True)
def startWebInterface(stateMachine):
siteRoot = Blueprint('siteRoot', __name__, static_folder='static', static_url_path='')
2016-12-30 02:51:56 -05:00
@siteRoot.route('/', methods=['GET', 'POST'])
@siteRoot.route('/index', methods=['GET', 'POST'])
def index():
ttsForm = TTSForm()
janusRestart = JanusRestart()
2016-12-30 02:51:56 -05:00
if ttsForm.validate_on_submit() and ttsForm.submitTTS.data:
stateMachine.soundLib.speak(ttsForm.tts.data)
return redirect(url_for('siteRoot.index'))
elif janusRestart.validate_on_submit():
logger.info('Restarting Janus')
run(['systemctl', 'restart', 'janus'], stdout=PIPE, stderr=PIPE)
return redirect(url_for('siteRoot.index'))
return render_template(
'index.html',
ttsForm=ttsForm,
state=stateMachine.currentState,
janusRunning=janusRunning(),
janusRestart=janusRestart
)
janusRunning()
2017-05-25 02:40:36 -04:00
app = Flask(__name__)
app.secret_key = '3276d68dac56985bea352325125641ff'
app.register_blueprint(siteRoot, url_prefix='/pyledriver')
2017-05-25 02:40:36 -04:00
2017-06-03 01:41:27 -04:00
logger.info('Starting web interface')
app.run(debug=False, threaded=True)