hakkoso/app.py

137 lines
3.6 KiB
Python

import json
from flask import Flask, Response
from flask import render_template, render_template_string, escape
from flask import request, send_file
from flask_socketio import SocketIO, join_room
from random import randint
from urllib.parse import urlparse, urlunparse
from os.path import exists
from phasectrl import State, ManualIntervention
from sensors import sensors
import phasectrl
from time import perf_counter, sleep
def create_app():
app = Flask('hakkoso')
app.config['DATABASE'] = 'server.db'
return app
app = create_app()
socketio = SocketIO(app, async_mode='threading')
def updateState():
socketio.emit('state', make_state())
print('SENT STATE UPDATE', make_state())
statemachine = State('recipes.json', updateState)
def make_state():
return {
'manual': statemachine.manual.getState(),
'recipes': tuple(r.getState() for r in statemachine.recipes),
'recipe': (statemachine.recipe.getState(statemachine.phase)
if statemachine.recipe else None),
'state': statemachine.getState(),
'sensors': {
'units': {
'Temperature': '°C',
'Acidity': 'pH',
'Humidity': 'RH',
'Durezza Acqua': 'BOH',
'Switch': 'On/Off',
},
'found': sensors.list()
}
}
@app.route('/')
def index():
return render_template('index.html')
@app.route('/edit-recipe/<id>')
def recipe_editor(id):
return render_template('recipe-editor.html',
recipe=statemachine.recipeById(id))
@app.route('/edit-recipe/')
def recipes_list():
return render_template('recipes-list.html', recipes=statemachine.recipes)
@app.route('/status')
def status():
return make_state()
@app.route('/dist/<filename>')
def dist(filename):
# TODO: check file exists and path is under src
filename = 'dist/' + filename
if exists(filename):
return send_file(filename)
print('Missing', filename)
return ''
@app.route('/recipe')
def recipe():
return render_template('recipe.html', recipe=phasectrl.recipe)
@socketio.on('new client')
def handle_new_client():
updateState()
socketio.emit('sensor history', sensors.get_history())
@socketio.on('manual response')
def handle_manual_response(response):
if statemachine.recipe is not None:
statemachine.manual.set(response)
updateState()
@socketio.on('load recipe idx')
def load_recipe_idx(idx):
statemachine.loadByIdx(idx)
@socketio.on('update recipe phase')
def update_recipe_phase(recipeid, phaseid, content):
recipe = statemachine.recipeById(recipeid)
recipe.phases[phaseid] = phasectrl.load_phase(content)
statemachine.recipes[recipeid] = recipe
phasectrl.store_recipes("recipes.json",
[r.getState() for r in statemachine.recipes])
@socketio.on('stop recipe')
def stop_recipe():
statemachine.stop()
@socketio.on('next phase')
def next_phase():
done = statemachine.next() is None
if done:
statemachine.done()
def run_recipes():
while True:
while statemachine.recipe is not None:
done = statemachine.run_step()
if done:
statemachine.done()
sleep(1)
def read_sensors():
while True:
sensors.read()
socketio.emit('sensors', (sensors.get(),))
sleep(0.5)
import threading
if __name__ == '__main__':
thread = threading.Thread(target=run_recipes, daemon=True)
thread.start()
sensors_thread = threading.Thread(target=read_sensors, daemon=True)
sensors_thread.start()
print('RUN APP')
socketio.run(app, host='0.0.0.0')