main.py

from flask import Flask, Response, render_template, request, session
import pyrebase
import firebase
from camera import VideoCamera
import os
from threading import Thread
import datetime

app = Flask(__name__, template_folder="templates")
app.secret_key = os.urandom(24)

config = firebase.get_credentials()
firebase = pyrebase.initialize_app(config)
auth = firebase.auth()
video = VideoCamera()
threads = []

current_hour = 100
kill = False


def record():
    global video
    while True:
        try:
            if not kill:
                video.record()
            else:
                video.save()
                video = VideoCamera()
                break
        except Exception as e:
            print(str(e))


def gen(feed):
    while True:
        try:
            frame = feed.get_frame()
            yield (b'--frame\r\n'
                   b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n\r\n')
        except Exception as e:
            print(str(e))


@app.route('/video_feed')
def video_feed():
    try:
        print(session['usr'])
        return Response(gen(video),
                        mimetype='multipart/x-mixed-replace; boundary=frame')
    except KeyError:
        print("Wrong id")
        return render_template('login.html', us="Unauthorised")


@app.route('/', methods=['GET', 'POST'])
def index():
    unsuccessful = 'Please check your credentials'
    if request.method == 'POST':
        email = request.form['name']
        password = request.form['pass']
        try:
            user = auth.sign_in_with_email_and_password(email, password)
            session['usr'] = user.get('idToken')
            return render_template('index.html')
        except Exception as e:
            print(str(e))
            return render_template('login.html', us=unsuccessful)

    return render_template('login.html')


@app.route('/start', methods=['GET'])
def start():
    global threads, kill
    if len(threads) == 0:
        kill = False
        t1 = Thread(target=record)
        threads.append(t1)
        t1.start()
        print("Starting ...")
    else:
        kill = True
        del threads[:]

    return "started"


if __name__ == "__main__":
    app.run(host='0.0.0.0', port=2000, debug=True, use_reloader=False)