from flask import Blueprint, render_template, request, redirect
from users.models import User


blueprint = Blueprint('users', __name__, url_prefix='/users')
User.create_table()

# /users -> list of users
# /users/create -> create user form
# /users/<user>/edit -> update user form
# /users/<user>/delete -> delete user


#localhost:5000/users
@blueprint.route('')
def view():
    users = User.select()
    return render_template('users/view.html', users=users)

#localhost:5000/users/create
@blueprint.route('/create', methods=['GET', 'POST'])
def create():

    if request.method == 'POST':
        form = request.form
        User.create(
            username=form.get('username'),
            email=form.get('email'),
            first_name=form.get('first_name'),
            last_name=form.get('last_name'),
        )

    return render_template('users/create.html')

#localhost:5000/users/<user.id>/edit
@blueprint.route('/<int:user_id>/edit', methods=['GET', 'POST'])
def edit(user_id):
    user = User.get_by_id(user_id)

    if request.method == 'POST':
        form = request.form
        user.username = form.get('username')
        user.email = form.get('email')
        user.first_name = form.get('first_name')
        user.last_name = form.get('last_name')
        user.save()


    return render_template('users/edit.html', user=user)

#localhost:5000/users/<user.id>/delete
@blueprint.route('/<int:user_id>/delete', methods=['GET', 'POST'])
def delete(user_id):
    user = User.get_by_id(user_id)
    user.delete_instance()

    return redirect('/users')