app.py

import os
import random
import string
import time

from werkzeug.utils import secure_filename, escape
from flask import Flask, render_template, request, json, session, abort, redirect, url_for, flash
from utility import RepeatedTimer, getVideoFolderPath, get_next_possible_filename, allowed_file

app = Flask(__name__)
# Secret Key for Sessions !!!Important: Please Change!!!
app.secret_key = "iguzuzcsd89sd80f9z08whwucf03q2cd08hcwkjkjsdqq"
# how many seconds the video rewinds to the correct position
maxDelaySeconds = 1
# Admin Login
adminUsername = "admin"
adminPassword = "hugo"
# Dictonary of All Avaible Rooms
Rooms = {}

app.config['ALLOWED_EXTENSIONS'] = {'mp4', 'avi', 'webm'}


# Class to manage Rooms
class Raum:
    def __init__(self, raumCode, status, video):
        self.raumCode = raumCode
        self.status = status
        self.video = video
        self.time = float(0)
        self.lastPlay = 0
        self.type = "video"


@app.route('/')
def hello_world():
    return render_template("enterCode.html")


@app.errorhandler(404)
def err404(e):
    return render_template("404.html"), 404


@app.errorhandler(403)
def err403(e):
    return render_template("403.html"), 403


@app.route('/watch-<string:code>')
def watch(code):
    try:
        return render_template("watch.html", code=code, video=Rooms[code].video, type=Rooms[code].type)
    except:
        return "Falscher Code"
        # abort(403)


@app.route('/create', methods=['POST'])
def createwatch():
    if (request.method == "POST"):
        vid = request.form.get("video")
        vid = secure_filename(vid)
        code = ''.join(random.choices(string.ascii_uppercase + string.digits, k=6))
        v = Raum(code, "pause", vid)
        Rooms[code] = v
        # Store a link to a Video in a Txt File
        if vid.endswith('txt'):
            # get Path to Video Folder
            path = getVideoFolderPath(filename=vid)
            f = open(path, "r")
            # Read the Link to a video
            videolocation = f.read()
            Rooms[code].video = videolocation
            Rooms[code].type = "txt"
        return redirect("/watch-" + code, code=302)
    return


@app.route('/watch', methods=['POST'])
def joinwatch():
    if (request.method == "POST"):
        code = request.form.get("codeIn")
        return redirect("/watch-" + code, code=302)


# Update Status if Play or Pause pressed or if the Video Seeked
@app.route('/videoStatus', methods=['POST'])
def videostatus():
    # Get Params from POST request
    videoEvent = request.form['event']
    videotime = request.form['time']
    videoid = request.form['id']

    # Update the Status in the current Room
    r = Rooms[videoid]

    # If Video Already playing send current Play Time, else set current Position to Global position
    if videoEvent == 'play':
        r.time = videotime
        r.lastPlay = time.time()
        if r.status == 'play':
            return json.dumps({'time': r.time, "updated": True, 'status': r.status})
        else:
            r.status = videoEvent
    if videoEvent == 'seeked':
        # update the Time for the Video
        r.time = float(videotime)
        r.lastPlay = time.time()
    elif videoEvent == "pause":
        r.status = videoEvent
        r.time = videotime
    return json.dumps({'status': 'OK', 'user': "Test"});


# Method for request to read Video state and update the client
@app.route('/updateVideo', methods=['POST'])
def updatetime():
    videotime = request.form['time']
    videoid = request.form['id']
    try:
        r = Rooms[videoid]
    except:
        # The Room Code was not found
        return "Bad Room-Code"

    # if abs(r.lastPlay + float(videotime) - time.time()) > 1:
    time_pos_of_video = abs(calc_the_time_where_the_video_should_be(videoid))
    # print("Video should be at:" + str(time_pos_of_video))
    if abs(time_pos_of_video - float(videotime)) > maxDelaySeconds and r.status == "play":
        print("Video Delayed:" + str(abs(time_pos_of_video - float(videotime))))
        return json.dumps({'time': time_pos_of_video, "updated": True, 'status': r.status})
    return json.dumps({"updated": False, 'status': r.status})


@app.route("/startwatch")
def startwatch():
    if session.get('user') == "admin":
        # Get Path to Video Folder
        path = os.getcwd()
        path = os.path.join(path, "static")
        path = os.path.join(path, "video")
        list_of_files = {}
        # Put every File in Video Folder in the list
        for filename in os.listdir(path):
            list_of_files[filename] = "/static/video" + filename
        # return the Template with the List, of all files in the Video folder
        return render_template("startwatch.html", videos=list_of_files)
    else:
        abort(403)


@app.route("/logout")
def logout():
    session['user'] = None
    return redirect("/")


@app.route("/add", methods=["POST"])
def addFile():
    fileEnding = ".txt"
    if (request.method == "POST" and session.get("user") == "admin"):
        content = request.form.get("txt")
        name = request.form.get("txtTitle")
        content = escape(content)  # Secure User Inpt
        name = secure_filename(escape(name))

        path = getVideoFolderPath()  # In this Folder we want to create the new TXT File
        path = os.path.join(path, name)
        path = get_next_possible_filename(path, file_ending=fileEnding)
        try:
            f = open(path, "x")
            f.write(content)
        except:
            abort(406)
        return redirect("/startwatch", 302)


def calc_the_time_where_the_video_should_be(raumID):
    currentTime = time.time()
    r = Rooms[raumID]
    timepos = float(r.lastPlay) - float(r.time)
    timeWhereVideoShouldBe = abs(timepos - currentTime)
    return timeWhereVideoShouldBe


# Simple Login Method. Only One Username Exist.
@app.route("/login", methods=["POST", "GET"])
def login():
    if request.method == "POST":
        user = request.form.get("user").lower()
        passw = request.form.get("password")
        if user == adminUsername and passw == adminPassword:
            session["user"] = "admin"
        return redirect("/startwatch")
    else:
        return render_template("login.html")


@app.route("/upload", methods=['POST'])
def upload():
    if request.method == 'POST':
        # check if the post request has the file part
        if 'file' not in request.files:
            flash('No file part')
            return redirect(request.url)
        file = request.files['file']
        # if user does not select file, browser also
        # submit an empty part without filename
        if file.filename == '':
            flash('No selected file')
            return redirect(request.url)
        if file and allowed_file(file.filename, app.config['ALLOWED_EXTENSIONS']):
            filename = secure_filename(file.filename)
            save_location = os.path.join(getVideoFolderPath(), filename)
            save_location = get_next_possible_filename(save_location)
            file.save(save_location)
            return redirect("/startwatch", 302)
    return redirect("/startwatch", 302)


@app.route("/delete", methods=["POST"])
def delete():
    if request.method == "POST" and session["user"] == "admin":
        delFile = secure_filename(request.form.get("delBtn"))
        os.remove(getVideoFolderPath(filename=delFile))
        return redirect("/startwatch", 302)
    return ""


if __name__ == '__main__':
    app.run(debug=True, host='0.0.0.0', port=1337)