# coding: utf-8
from flask import Flask, render_template, session, request, redirect, flash
import os

app = Flask(__name__)

@app.route('/')
def home():
    if not session.get('logged_in'):
        return render_template('login.html')
    else:
        if session.get('user_type') == 'lecturer':
            return redirect('/lecturer')
        else:
            return redirect('/student')

@app.route('/student',methods=['GET', 'POST'])
def student():
    if request.method == 'POST':
        question = request.get_data().decode("utf-8")
        question = question.split('=')
        print(question[1])
        return '200'
    else:
        return render_template('student.html')

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

@app.route('/login', methods=['POST'])
def do_login():
    if request.form['password'] == 'lecturer' and request.form['name'] == 'lecturer':
        session['logged_in'] = True
        session['user_type'] = 'lecturer'
    elif request.form['password'] == 'student' and request.form['name'] == 'student':
        session['logged_in'] = True
        session['user_type'] = 'student'
    else:
        flash('Niepoprawne hasło/nazwa użytkownika!'.decode('utf-8'))
    return redirect('/')

@app.route('/logout', methods=['GET', 'POST'])
def logout():
    session['logged_in'] = False
    return redirect('/')

if __name__ == '__main__':
    app.secret_key = os.urandom(12)
    app.run(host='0.0.0.0', debug = True)