app.py

from flask import Flask, render_template, request, redirect, make_response
from siwel_files import siwel

app = Flask(__name__)

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

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

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

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

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






# lewis custom library stuff:
# purchase a specific membership type
@app.route('/membership/purchases/')
def purchase():
    type = request.args.get("type")
    if type not in ["standard", "premium", "family"]: # get query from url to show contents
        return redirect("/membership/", code=302)
    else:
        return siwel.return_purchase_html(type)
    



# allow user to view events/classes
@app.route('/classes/event-view/', methods=['POST']) # post form which gets the date from user input
def event_view():
    day = request.form.get("day")
    month = request.form.get("month")
    year = request.form.get("year")
    return siwel.return_event_html(day, month, year) # display event html page



# admin stuff
# add event/class page
@app.route('/admin/')
def admin():
    try:
        admin = request.cookies.get("admin") # if admin cookie exists then allow user onto the page
        if bool(int(admin)):
            return siwel.return_admin_html()
    except:
        pass
    return redirect("/", code=302)

# handle event add form
@app.route('/event-add/', methods=['POST']) # get all the inputs the admin put in for adding an event
def event_add():
    class_name = request.form.get("class-name")
    day = request.form.get("day")
    month = request.form.get("month")
    year = request.form.get("year")
    start_time = request.form.get("start-time")
    end_time = request.form.get("end-time")
    trainer = request.form.get("trainers")


    siwel.db_event_add(class_name, day, month, year, start_time, end_time, trainer) # pass to add event route
    return redirect("/admin/", code=302)

# manage user type
@app.route('/manage-user/', methods=['POST'])
def manage_user():
    usern = request.form.get("usern")
    usert = request.form.get("user-type")

    siwel.db_user_update(usern, usert) # pass to manage user route
    return redirect("/admin/", code=302)







# handling user stuff
# profile page
@app.route('/profile/')
def profile():
    user = request.cookies.get("user")
    return siwel.return_profile_html(user) # load page content based off if user cookie exists or not

# login page
@app.route('/profile/login/')
def login():
    return render_template('login.html')

# handle the login form
@app.route('/login-event/', methods=['POST']) # get username and password from login page form
def login_event():
    usern = request.form.get("username")
    passw = request.form.get("password")
    login = siwel.log_in_user(usern, passw) # pass to login route

    if login["login"]: # if login returned true meaning the details are correct
        res = make_response(redirect("/profile/", code=302))
        if login["data"][1]: # extra step to check if user is admin
            res.set_cookie("admin", "1") # set admin cookie
        res.set_cookie("user", login["data"][0]) # set user logged in cookie
        return res

    return redirect("/profile/login/", code=302) # else just redirect them back to login page

# create a user account
@app.route('/profile/create-account/')
def create_account():
    return render_template('create-account.html')

# handle the create account form
@app.route('/create-user/', methods=['POST']) # get first name, last name, and password from create account form 
def create_event():
    firstn = request.form.get("firstname")
    lastn = request.form.get("lastname")
    passw = request.form.get("password")
    create = siwel.create_user(firstn, lastn, passw) # pass to create account route

    res = make_response(redirect("/profile/", code=302))
    res.set_cookie("user", create) # set logged in user cookie
    return res

# logout and delete cookies
@app.route('/logout/')
def logout_event():
    res = make_response(redirect("/profile/", code=302))
    res.set_cookie("user", max_age=0) # as there is no delete cookie function then make the cookie die with max age 0
    res.set_cookie("admin", max_age=0) # same for admin cookie if they were admin
    return res



# host the website
if __name__ == "__main__":
    app.run(debug=True) # debug true for testing purposes