import hashlib
import sys
from flask import Blueprint, render_template, request, json, redirect, url_for
from . import db
from .models import ftpuser

main = Blueprint('main', __name__)


@main.route('/')
def index():
    userzy = ftpuser.query.all()
    return render_template('index.html', userzy=userzy)


@main.route('/sign')
def sign():
    return render_template('sign.html')


@main.route('/sign', methods=['POST'])
def sign_post():
    user = request.form.get('user')
    password = request.form.get('password')
    h = hashlib.sha256(password.encode())
    print(h.hexdigest())
    hashed = (h.hexdigest())
    new_user = ftpuser(userid=user,  passwd=hashed)
    db.session.add(new_user)
    db.session.commit()
    return render_template("user.html", password=password, user=user)


@main.route('/delete')
def delete():
    userzy = ftpuser.query.all()
    return render_template('delete.html', userzy=userzy)


@main.route('/delete', methods=['POST'])
def delete_post():
    delete = request.form.get('delete')
    ftpuser.query.filter(ftpuser.userid == delete).delete()
    db.session.commit()
    return redirect(url_for('main.index'))