80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
from flask import Flask, render_template, request, redirect, url_for, flash
|
|
|
|
import hashlib
|
|
|
|
def hash_password(password):
|
|
return hashlib.sha512(password.strip().encode('utf-8')).hexdigest()
|
|
|
|
app = Flask(__name__)
|
|
app.secret_key = 'your_secret_key'
|
|
|
|
@app.route('/')
|
|
def home():
|
|
return redirect(url_for('login'))
|
|
|
|
@app.route('/register', methods=['GET', 'POST'])
|
|
def register():
|
|
import standard.getter as st_getter
|
|
import sqlLite.set as setter
|
|
|
|
if request.method == 'POST':
|
|
username = request.form['username']
|
|
email = request.form.get('email')
|
|
password = request.form.get('password')
|
|
pwd_confirm = request.form.get('confirm_password')
|
|
|
|
if not email or not password or not pwd_confirm or not username:
|
|
flash('Please fill out all fields', 'error')
|
|
return redirect(url_for('register'))
|
|
|
|
if password != pwd_confirm:
|
|
flash('Passwords do not match', 'error')
|
|
return redirect(url_for('register'))
|
|
|
|
try:
|
|
if st_getter.get_validate_email(email):
|
|
setter.set_login(username, email, password)
|
|
flash('Registration successful! Please log in.', 'success')
|
|
return redirect(url_for('login'))
|
|
else:
|
|
flash('Invalid email format', 'error')
|
|
return redirect(url_for('register'))
|
|
except Exception as e:
|
|
flash(f'Error: {str(e)}', 'error')
|
|
return redirect(url_for('register'))
|
|
# For GET-requests:
|
|
return render_template('register.html')
|
|
|
|
|
|
@app.route('/login', methods=['GET', 'POST'])
|
|
def login():
|
|
if request.method == 'POST':
|
|
enter_email = request.form.get('email')
|
|
enter_password = request.form.get('password')
|
|
import hashlib
|
|
import sqlLite.get as getter
|
|
import standard.getter as st_getter
|
|
|
|
stored_hash = getter.get_password_by_email(enter_email) # use email here
|
|
|
|
if stored_hash is None:
|
|
flash("User not found!")
|
|
return redirect(url_for("login"))
|
|
|
|
hash_entered = st_getter.get_password_hash(enter_password)
|
|
|
|
if hash_entered == stored_hash:
|
|
return redirect(url_for('dashboard'))
|
|
else:
|
|
flash('Invalid email or password', 'error')
|
|
print("Stored hash:", stored_hash)
|
|
print("Entered hash:", hash_entered)
|
|
return redirect(url_for('login'))
|
|
|
|
return render_template('login.html')
|
|
|
|
@app.route('/dashboard')
|
|
def dashboard():
|
|
#return "Welcome to the dashboard! Login successful."
|
|
return render_template('dashboard.html')
|