from flask import Flask
from flask import Flask, flash, redirect, render_template, request, session, abort
import os
from flask import g
from sqlalchemy.orm import sessionmaker
from tabledef import *
from sqlalchemy import create_engine

engine = create_engine('mysql://root:damian123@localhost/sklep')

app = Flask(__name__)


@app.route('/')
def home():
    if not session.get('logged_in'):
        return render_template('login.html')
    else:
        return "Witaj %s ! <a href='/logout'>Logout</a>"


@app.route('/login', methods=['POST'])
def do_admin_login():
    POST_USERNAME = str(request.form['username'])
    POST_PASSWORD = str(request.form['password'])

    Session = sessionmaker(bind=engine)
    s = Session()
    query = s.query(User).filter(User.username.in_([POST_USERNAME]), User.password.in_([POST_PASSWORD]))
    result = query.first()
    if result:
        session['logged_in'] = True
        user = query.filter_by(username=session["userid"]).first()
        g.user = user
        session['userid'] = user.id
    else:
        flash('wrong password!')
    return home()


@app.route("/logout")
def logout():
    session['logged_in'] = False
    return home()


if __name__ == "__main__":
    app.secret_key = os.urandom(12)
    app.run()