|
-
-
-
-
-
-
-
-
- from flask import Blueprint, render_template, redirect, url_for, request, flash
- from werkzeug.security import generate_password_hash, check_password_hash
- from flask_login import login_user, logout_user, login_required
- from .models import User
- from . import db
- import os
-
- auth = Blueprint('auth', __name__)
-
-
- @auth.route('/login')
- def login():
- return render_template('login.html')
-
- @auth.route('/login', methods=['POST'])
- def login_post():
- email = request.form.get('email')
- password = request.form.get('password')
- remember = True if request.form.get('remember') else False
-
-
- user = User.query.filter_by(email=email).first()
-
-
-
- if not user or not check_password_hash(user.password, password):
- flash('Please check your login details and try again.')
- return redirect(url_for('auth.login',_external=True,_scheme=os.environ.get('SSL_SCHEME')))
-
-
- login_user(user, remember=remember)
- return redirect(url_for('main.profile',_external=True,_scheme=os.environ.get('SSL_SCHEME')))
-
-
- @auth.route('/signup')
- def signup():
- return render_template('signup.html')
-
- @auth.route('/signup', methods=['POST'])
- def signup_post():
- email = request.form.get('email')
- name = request.form.get('name')
- password = request.form.get('password')
-
- user = User.query.filter_by(email=email).first()
-
- if user:
- flash('Email address already exists')
- return redirect(url_for('auth.signup',_external=True,_scheme=os.environ.get('SSL_SCHEME')))
-
-
- new_user = User(email=email, name=name, password=generate_password_hash(password, method='sha256'))
-
-
- db.session.add(new_user)
- db.session.commit()
-
- return redirect(url_for('auth.login',_external=True,_scheme=os.environ.get('SSL_SCHEME')))
-
-
- @auth.route('/logout')
- @login_required
- def logout():
- logout_user()
- return redirect(url_for('main.index',_external=True,_scheme=os.environ.get('SSL_SCHEME')))
|