from flask import Flask, Response
from flask import Flask, flash, render_template, request, session, Markup
import os
from sqlalchemy.orm import sessionmaker
from tabledef import *
from sqlalchemy import create_engine
from flask.ext.login import LoginManager

engine = create_engine('mysql://root:damian123@localhost/sklep')
login_manager = LoginManager
app = Flask(__name__)
sesja = sessionmaker(bind=engine)
s = sesja()

@app.route('/')
def home():
    if not session.get('logged_in'):
        return render_template('login.html')
    else:
        return render_template('index.html')

@app.route('/register', methods=['POST','GET'])
def register():
    if request.method == 'GET':
        return render_template('register.html')
    elif request.method == 'POST':
        username = str(request.form['username'])
        password = str(request.form['password'])
        user = User(username=username, password=password)
        s.add(user)
        s.commit()
        flash('You have registered the username {0}. Please login'.format(username))
        return render_template('login.html')

##Logowanie
@app.route('/login', methods=['POST'])
def login():
    username = str(request.form['username'])
    password = str(request.form['password'])
    query = s.query(User).filter(User.username.in_([username]), User.password.in_([password]))
    result = query.first()
    if result:
        session['logged_in'] = True
    else:
        flash(Markup('Zly login lub haslo.<br>Chcialbys sie <a href="/register">zarejestrowac?</a>'))
    return home()

###Logout##
@app.route("/logout")
def logout():
    session['logged_in'] = False
    return render_template('login.html')


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