__init__.py 2.49 KB
Newer Older
Mitchell Moore's avatar
Mitchell Moore committed
1
# app/__init__.py
2

3
4
# local imports
from __future__ import print_function
5
import vars
6

7
# third-party imports
8
import uuid
9
from flask import Flask, redirect, url_for, request, render_template, flash, session
10
from flask_bootstrap import Bootstrap
11
import random
12
13
import os
import json
14
15
16
17

def create_app(config_name):
    app = Flask(__name__) # initialization of the flask app
    Bootstrap(app) # allowing app to use bootstrap
18

19
    def get_authorized_user():
20
21
22
        username_key = list(filter(lambda key: (request.headers.get(key)  is not None), vars.username_key))
        fullname_key = list(filter(lambda key: (request.headers.get(key) is not None), vars.fullname_key))
        email_key = list(filter(lambda key: (request.headers.get(key) is not None), vars.email_key))
23
24

        user = {
25
26
27
            "username": (request.headers.get(username_key[0]) if len(username_key) > 0 else None),
            "fullname": (request.headers.get(fullname_key[0]) if len(fullname_key) > 0 else None),
            "email": (request.headers.get(email_key[0]) if len(email_key) > 0 else None),
28
29
30
31
        }

        return user

32
33
    @app.route('/', methods=['GET', 'POST']) # initial route to display the reg page
    def index():
34

35
36
37
        if 'uid' not in session:
            session['uid']=str(uuid.uuid4())

38
39
40
        if 'user' not in session:
            session["user"] = get_authorized_user()

41
        if "redir" in request.args and 'return_url' not in session: # check for redir arg in url
42
            session['return_url'] = request.args.get("redir")
43

44
45
46
47
        elif "redir" not in request.args and 'return_url' not in session:
            session['return_url'] = vars.default_referrer
        else:
            session['return_url'] = request.referrer
48

49
50
51
52
53
        return render_template('auth/SignUp.html', room_id=session['uid'],
                               username=session['user'].get('username'),
                               fullname=session['user'].get('fullname'), email=session['user'].get('email'),
                               referrer=session['return_url'], cancel_url=vars.default_referrer)

54

55
    # misc page error catching
56
57
58
59
60
61
62
63
64
65
66
67
68
    @app.errorhandler(403)
    def forbidden(error):
        return render_template('errors/403.html', title='Forbidden'), 403

    @app.errorhandler(404)
    def page_not_found(error):
        return render_template('errors/404.html', title='Page Not Found'), 404

    @app.errorhandler(500)
    def internal_server_error(error):
        return render_template('errors/500.html', title='Server Error'), 500

    return app