app.py

from flask import Flask, render_template, make_response, redirect
import os
import time
import sys
from werkzeug.routing import BaseConverter
import random
from flask_socketio import SocketIO, send, emit
import json
from profanityfilter import ProfanityFilter
from dotenv import load_dotenv
import string

load_dotenv()

keyfile = os.environ.get('KEYFILE')
certfile = os.environ.get('CERTFILE')
domain = os.environ.get('DOMAIN')

app = Flask(__name__)
if domain != None:
  app.config["SERVER_NAME"] = domain

sessions = {}
num = 0
socketio = SocketIO(app)

pf = ProfanityFilter()


class RegexConverter(BaseConverter):
    def __init__(self, url_map, *items):
        super(RegexConverter, self).__init__(url_map)
        self.regex = items[0]


app.url_map.converters['regex'] = RegexConverter


@app.route('/admin/<id>:<password>')
def admin(id, password):
    print(id, file=sys.stdout)
    if not id in sessions:
        return redirect('/404')
    if sessions[id]['password'] != password:
        return redirect('/view/'+id)
    return render_template('draw.html', id=id, password=password)

@app.route('/404')
def four ():
    return render_template('404.html')

@app.route('/view/<id>')
def view(id):
    print(id, file=sys.stdout)
    if not id in sessions:
        return redirect('/404')
    return render_template('view.html', id=id)


def format_server_time():
    server_time = time.localtime()
    return time.strftime("%I:%M:%S %p", server_time)


@app.route('/')
def index():
    context = {'server_time': format_server_time()}
    return render_template('index.html', context=context)


@app.route('/about')
def about():
    context = {'server_time': format_server_time()}
    return render_template('about.html', context=context)

@app.errorhandler(404)
def page_not_found(e):
    # note that we set the 404 status explicitly
    return redirect('/404')

@app.route('/create')
def create():
    global num
    num += 1
    num2 = ''.join(random.choices(string.ascii_uppercase + string.digits, k=8))
    password = ''.join(random.choices(string.ascii_uppercase + string.digits, k=13))
    sessionid = str(num)+num2
    sessions[sessionid] = {"session": sessionid, "password": password, "data": [], "messages": []}
    @socketio.on('drawing_'+sessionid+':'+password)
    def drawing(data):
        global sessions
        sessions[sessionid]["data"].append(data)
        emit('drawing_'+sessionid, data, broadcast=True)

    @socketio.on('clear_'+sessionid+':'+password)
    def onclear():
        emit('clear_'+sessionid,broadcast=True)
        global sessions
        sessions[sessionid]["data"] = []

    @socketio.on('chat_'+sessionid)
    def chat(data):
        data2 = pf.censor(data["content"])
        data["content"] = data2
        global sessions
        if data["person"] == "Instructor":
            if data["password"] != sessions[sessionid]["password"]:
                return
        data["password"] = ""
        emit('chat_'+sessionid,data,broadcast=True)
        sessions[sessionid]["messages"].append(data)
    @socketio.on("background_"+sessionid+"_"+password)
    def background(data):
        emit("background_"+sessionid,data,broadcast=True)
        sessions[sessionid]["background"] = data

    return redirect('/admin/'+sessionid+':'+password)

@socketio.on('connecteda')
def onconnection(data):
    global sessions
    for i in sessions[data]["data"]:
        emit('drawing_'+data, i)
    for i in sessions[data]["messages"]:
        emit("chat_"+data, i)
    if "background" in sessions[data]:
        emit("background_"+data,sessions[data]["background"])

if __name__ == '__main__':
    if keyfile != None and certfile != None:
      socketio.run(app, host='0.0.0.0', port=443, debug=True, keyfile=keyfile, certfile=certfile)
    else:
      socketio.run(app, host='0.0.0.0', debug=True, port=80)