from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import *
from Alch import User;
from flask import *


app = Flask(__name__)

Base = declarative_base()
engine = create_engine('mysql+pymysql://root:root@localhost:3306/ES')

# set the secret key. keep this really secret:
app.secret_key = 'A0Zr98j/3yX R~XHH!jmN]LWX/,?RT'

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

@app.route('/registo', methods=['POST', 'GET'])
def registo():

    if request.method == 'POST':

        if request.form['action'] == "Cancel":
            return render_template('login.html')

        elif regist(request.form['email'],request.form['password'],request.form['name']):
            return render_template('list.html', name=request.form['email'])

        return render_template('registo.html')

    else:
        return render_template('registo.html')

def regist(mail,passw,name):
    Session = sessionmaker(bind=engine)
    session = Session()
    type= request.form['type']

    user = User(name=name, email=mail, password=passw)

    try:
        if(type=="owner"):
            user.owner=True;
        session.add(user)
        session.commit();
    except:
        return False;

    return True;

@app.route('/do_login', methods=['POST', 'GET'])
def login():

    if request.method == 'POST':
        if valid_login(request.form['email'],request.form['password']):
            return render_template('PersonalInfo.html', email=session['email'],name=session['name'])

        else:
            name = 'Invalid username/password'

    return render_template('login.html', name=name)

def valid_login(mail,passw):
    Session = sessionmaker(bind=engine)
    sessions = Session()
    query = sessions.query(User).filter_by(email=mail,password=passw)

    if query.first():
        email=query.first().email
        name=query.first().name;
        session['name'] = name;
        session['email'] = email;
        return True
    else:
        return False

@app.route('/personal',methods=['POST'])
def info():
    if request.form['action'] == "Change":

        email=request.form['email']
        password=request.form['password']
        name=request.form['name']

        if password !="" or name !="" or email!="":
            Session = sessionmaker(bind=engine)
            sessions = Session()
            query = sessions.query(User).filter_by(email=session['email']).first()

            if  password !="":
                query.password = password
            if  name !="":
                query.name = name
            if  email !="":
                query.email = email
            try:
                sessions.commit()
            except:
                return render_template('PersonalInfo.html',email=session['email'],name=session['name'])


        return render_template('login2.html')

    return render_template('list.html')

@app.route('/logout')
def logout():
    session.pop('name', None)
    session.pop('email', None)
    return render_template('login.html')


if __name__ == "__main__":
    app.debug=True
    app.run()