app.py

#!/usr/bin/env python3
from importlib import import_module
import os
from flask import Flask, render_template, Response, request, redirect, jsonify
import time
import json
# import camera driver
if os.environ.get('CAMERA'):
    Camera = import_module('camera_' + os.environ['CAMERA']).Camera
else:
    raise RuntimeError('No environment variable CAMERA defined!')

Led = import_module('led').Led
# Raspberry Pi camera module (requires picamera package)

with open('config.json') as f:
    config = json.load(f)
app = Flask(__name__)
led = Led(config['pin'])
cam = Camera(led)


@app.route('/_update', methods = ['GET'])
def stuff():
    return jsonify(led_state=led.status())

@app.route('/')
def index():
    """Index page."""
    return render_template("index.html", pin=str(config['pin']), shapshot_delay=str(config['shapshot_delay']), led_state=str(led.status()))

def gen(camera):
    """Video streaming generator function."""
    while True:
        frame = camera.get_frame()
        yield (b'--frame\r\n'
               b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')


def gen_snapshot(camera):
    frame = camera.get_frame(True)
    return frame


@app.route('/video_feed')
def video_feed():
    """Video streaming route. Put this in the src attribute of an img tag."""
    if(not led.leds_on()):
        led.switch_leds(True)
    return Response(gen(cam),
                    mimetype='multipart/x-mixed-replace; boundary=frame')


@app.route('/snapshot')
def snapshot():
    """Video streaming route. Put this in the src attribute of an img tag."""
    # if(not cam.leds_on):
    #     led.switch_leds(True)
    if(not led.leds_on()):
        led.switch_leds(True)
        time.sleep(config['shapshot_delay'])
    return Response(gen_snapshot(cam),
                    mimetype='image/jpeg')

@app.route('/led_on')
def led_on():
    led.switch_leds(True)
    return redirect('/')

@app.route('/led_off')
def led_off():
    led.switch_leds(False)
    return redirect('/')

@app.route('/control_leds', methods=['POST'])
def control_leds():
    """Video streaming route. Put this in the src attribute of an img tag."""
    json = request.json
    led_state = json["state"]
    if(led_state == 'on'):
        led.switch_leds(True)
        resp = 'true'
    elif(led_state == 'off'):
        led.switch_leds(False)
        resp = 'false'
    else:
        return ('', 504)
    return Response(f'{{"led_status": {resp}}}', mimetype="application/json")


@app.route('/leds_on')
def led_status():
    status = led.status()
    if(status == 1):
        resp = 'true'
    else:
        resp = 'false'
    return Response(f'{{"led_status": {resp}}}', mimetype="application/json")

# if __name__ == '__main__':
#     app.run(host='0.0.0.0', threaded=True, port=3000)