from flask import Flask, Response
from flask import Flask, flash, render_template, request, session, Markup
import os
from flask_sqlalchemy import SQLAlchemy


app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URL'] = 'mysql://root:damian123@localhost/sklep'
db = SQLAlchemy(app)

class User(db.model):
    """"""
    __tablename__ = "users"

    id = Column(db.Integer, primary_key=True, autoincrement=True)
    username = db.Column(db.String)
    password = db.Column(db.String)



##Sesja przydzielona jako s

@app.route('/')
def home():
    if not session.get('logged_in'):
        return render_template('login.html')
    else:
        username = str(request.form['username'])
        halo = db.query(User).filter(User.username.in_([username]))
        hey = halo.first() ##Pierwsza wartosc wyswietlana

        return render_template('index.html',user_show = hey )

@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)
        if len(username) >=4 and len(username) <8:
            s.add(user)
            s.commit()
            flash('You have registered the username {0}. Please login'.format(username))
            return render_template('login.html')
        else:
            flash('Login musi zawierac wiecej niz 4 i mniej niz 8 znakow')
            return render_template('register.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()