Webserver.py

from flask import Flask, jsonify, render_template, flash, request, Response, abort
import ConfigManager as config
import VideoManager as video
import EmailManager as email
import Updater as updater
import json
import os, sys, time, signal
import threading

app = Flask(__name__, static_folder="web/static", template_folder="web/templates")
app.secret_key = b'SDFKHW$%^8wTsoigt098'

self = sys.modules[__name__]
self.shutdownState = None

def shutdown_server():
    func = request.environ.get('werkzeug.server.shutdown')
    if func is None:
        return False

    func()
    return True

def Shutdown():
    self.shutdownState = "shutdown"
    return shutdown_server()

def Restart():
    self.shutdownState = "restart"
    return shutdown_server()

def UpdateServer():
    self.shutdownState = "update"
    return shutdown_server()

def GetShutdownState():
    return self.shutdownState

#   Disable cache
@app.after_request
def add_header(r):
    """
    Add headers to both force latest IE rendering engine or Chrome Frame,
    and also to cache the rendered page for 10 minutes.
    """
    r.headers["Cache-Control"] = "no-cache, no-store, must-revalidate"
    r.headers["Pragma"] = "no-cache"
    r.headers["Expires"] = "0"
    r.headers['Cache-Control'] = 'public, max-age=0'
    return r

@app.route('/')
def indexPage():
    return render_template("index.html")

@app.route('/cameras')
def camerasPage():
    camInfo = video.GetCamInfo()
    return render_template("cameras.html", cam_info=camInfo)

@app.route('/config', methods=["GET", "POST", "DELETE"])
def configPage():
    if request.method == "GET":
        config_data = config.AsString()
        return render_template("config.html", config_data=config_data)
    elif request.method == "POST":
        data = request.form.get("data")
        data = json.loads(data)
        config.SaveConfig(data)
        return Response("Config Saved! Restart server for changes to take effect", 200)
    elif request.method == "DELETE":
        config.RestoreConfig()
        return Response("Config Restored", 200)
    else:
        abort(400)

@app.route('/api/config')
def api_config():
    configData = config.AsString()
    return Response(configData, 200)

@app.route('/api/messages')
def get_messages():
    state = config.GetValue("email", "enabled")
    response = json.dumps({
        "enabled": state
    })
    return Response(response, 200)

@app.route('/api/messages/<state>')
def set_messages(state):
    enabled = (state == "true" or state == "on")
    config.SetValue(enabled, "email", "enabled")
    response = json.dumps({
        "enabled": enabled
    })
    return Response(response, 200)

@app.route('/api/motion')
def get_motion():
    state = config.GetValue("motionEnabled")
    response = json.dumps({
        "enabled": state
    })
    return Response(response, 200)

@app.route('/api/motion/<state>')
def set_motion(state):
    enabled = (state == "true" or state == "on")
    config.SetValue(enabled, "motionEnabled")

    return Response(enabled, 200)

@app.route('/api/restart')
def restart():
    if Restart():
        print("success")
        return Response('Restarting...', 200)
    else:
        print("fail")
        return Response('Error restarting server', 400)

@app.route('/api/shutdown')
def shutdown():
    if Shutdown():
        return Response('Shutting down...', 200)
    else:
        return Response('Error shutting down server', 400)

@app.route('/api/update_server')
def api_update_server():
    if UpdateServer():
        return Response("Updating...", 200)
    else:
        return Response('Error shutting down server', 400)

@app.route('/api/cam_info')
def cameras_status():
    cam_info = video.GetCamInfo()
    return Response(json.dumps(cam_info))

@app.route('/api/versions')
def api_versions():
    versions = {
        "latest": updater.GetLatestVersion(),
        "current": config.GetValue("version")
    }
    return Response(json.dumps(versions))

@app.route('/api/ping')
def api_ping():
    return Response("pong!", 200)


def Run(debug=False):
    port = config.GetValue("webserver", "port")
    app.run(host="0.0.0.0", port=port, debug=debug)

if __name__=="__main__":
    Run(True)