app.py

#!/usr/bin/env python
import os, sqlite3
from flask import Flask, render_template, redirect, url_for, request, session, jsonify, json, make_response
from flask_socketio import SocketIO, emit, send, join_room, leave_room, close_room, rooms, disconnect
from flask_login import LoginManager, UserMixin, login_user, login_required, logout_user, current_user
from classes.socket_module import GameLobbyNs, GameRoomNs
from classes.database import add_user, find_user, find_scores
import classes.settings as config
from classes.utility import alphanum
#from gevent import monkey
#monkey.patch_all()

app = Flask(__name__)
app.secret_key = 'thisissecret' # os.getenv("SABOTEUR_SECRET_KEY")
DATABASE = 'database.db'
db = sqlite3.connect(DATABASE)
c = db.cursor()

socketio = SocketIO(app, ping_timeout=30000)

game_rooms = {'roomId1': ["Jhon","Alex","Alice"],
            'roomId2': ["Bob"],
            'roomId3': ["Ted","Max"]}


socketio.on_namespace(GameLobbyNs('/lobby', appCtx=app, sio=socketio))

login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = 'login'

class User(UserMixin):
    def __init__(self, username):
        self.username = username
    @property
    def id(self):
        return self.username

@login_manager.user_loader
def load_user(id):
    return User(id)
 
@app.route('/')
def index():
    if current_user.is_authenticated:
        return redirect('dashboard')
    return render_template('index.html')

@app.route('/signup', methods=['GET','POST'])
def signup(text=''):
    if current_user.is_authenticated:
        return redirect('dashboard')
    if request.method == 'POST':
        username = request.form.get('username')
        if not alphanum(username):
            return render_template('signup.html', text="Username must be alphanumeric")
        password = request.form.get('password')
        text = add_user(username,password)
        if text == 'Signed up successfully!':
            user = find_user(username, password)
            if user is not None:
                login_user(User(username),remember=request.form.get('remember_me'))
                session['username'] = username
                return redirect('dashboard')
    return render_template('signup.html', text=text)

@app.route('/login', methods=['GET','POST'])
def login(text=''):    
    if current_user.is_authenticated:
        return redirect('dashboard')
    if request.method == 'POST':
        username = request.form.get('username')
        password = request.form.get('password')
        user = find_user(username, password)
        if user is not None:
            login_user(User(username),remember=request.form.get('remember_me'))
            session['username'] = username
            return redirect('dashboard')
        else:
            text = 'Wrong username or password'
    return render_template('login.html', text=text)

@app.route('/logout')
@login_required
def logout():
    logout_user()
    resp = make_response(render_template('index.html', 
    user=session['username']))
    resp.set_cookie('endpoint', '', expires=0)
    return resp

@app.route('/dashboard')
@login_required
def dashboard():
    resp = make_response(render_template('dashboard.html', 
    user=session['username']))
    resp.set_cookie('user-cookie=', 'room_list')
    return resp

@app.route('/game/<gamename>')
@login_required
def game(gamename):
    if(isinstance(gamename, type(None))):
        return redirect('dashboard')
    print(gamename)
    #gamename = request.cookies.get('endpoint')
    print(gamename)
    socketio.on_namespace(GameRoomNs('/'+gamename, appCtx=app, sio=socketio))
    return make_response(render_template('game.html', gamename=gamename, user=session['username']))

@app.route('/scores/<gameid>')
@login_required
def scores(gameid):
    if(isinstance(gameid, type(None)) or not alphanum(gameid)):
        return redirect('dashboard')
    scores = find_scores(gameid)
    if scores is not None:
        print(scores)
        return render_template('scores.html', data = build_score(scores) , gamename=gameid[:-14])
    return render_template('scores.html')

def build_score(scores):
    users=scores[0].split(", ")
    roles=scores[1].split(", ")
    score=scores[2].split(", ")
    lista = []
    for i in range(len(users)):
        dicto = {}
        dicto['name'] = users[i]
        dicto['role'] = roles[i]
        dicto['score']= score[i]
        lista.append(dicto)
    return lista

if __name__ == "__main__":
    app.debug = False
    port = int(os.environ.get("PORT", 33507))
    app.run(host='0.0.0.0', port=port)
    #app.run()