mirror of
https://github.com/Michatec/MiniFaceBook.git
synced 2026-03-31 23:46:30 +02:00
90 lines
3.6 KiB
Python
90 lines
3.6 KiB
Python
from flask import Blueprint, render_template, redirect, url_for, flash, request, abort
|
|
from flask_login import login_required, current_user
|
|
from models import db, User
|
|
from werkzeug.security import generate_password_hash
|
|
from flask_babel import gettext as _
|
|
try:
|
|
from routes.oauth import discord
|
|
except ImportError:
|
|
pass
|
|
from routes.login import login_user
|
|
|
|
discord_bp = Blueprint('discord', __name__)
|
|
|
|
@discord_bp.route('/login/discord')
|
|
def login_discord():
|
|
redirect_uri = url_for('discord.discord_login_callback', _external=True)
|
|
return discord.authorize_redirect(redirect_uri)
|
|
|
|
@discord_bp.route('/login/discord/callback', methods=['GET', 'POST'])
|
|
def discord_login_callback():
|
|
if request.method == 'GET':
|
|
token = discord.authorize_access_token()
|
|
user_data = discord.get('users/@me').json()
|
|
user = User.query.filter_by(discord_id=user_data['id']).first()
|
|
if user:
|
|
login_user(user)
|
|
flash(_('Logged in with Discord.'), 'success')
|
|
return redirect(url_for('post.feed'))
|
|
else:
|
|
flash(_('No account linked with this Discord. Please register.'), 'info')
|
|
return render_template(
|
|
'discord_register.html',
|
|
username=user_data['username'],
|
|
email=user_data.get('email', ''),
|
|
discord_id=user_data['id']
|
|
)
|
|
else:
|
|
username = request.form.get('username')
|
|
email = request.form.get('email')
|
|
discord_id = request.form.get('discord_id')
|
|
password = request.form.get('password')
|
|
confirm_password = request.form.get('confirm_password')
|
|
if not password or len(password) < 8:
|
|
flash(_('Password must be at least 8 characters long.'), 'danger')
|
|
return render_template('discord_register.html', username=username, email=email, discord_id=discord_id)
|
|
if password != confirm_password:
|
|
flash(_('Passwords do not match.'), 'danger')
|
|
return render_template('discord_register.html', username=username, email=email, discord_id=discord_id)
|
|
if db.session.query(User).filter_by(username=username).first():
|
|
flash(_('Username already exists. Please Report It.'), 'danger')
|
|
return render_template('discord_register.html', username="", email=email, discord_id=discord_id)
|
|
hashed_password = generate_password_hash(password, method='pbkdf2:sha256')
|
|
new_user = User(
|
|
username=username,
|
|
email=email,
|
|
password=hashed_password,
|
|
discord_id=discord_id,
|
|
discord_linked=True
|
|
)
|
|
db.session.add(new_user)
|
|
db.session.commit()
|
|
login_user(new_user)
|
|
flash(_('Account created and logged in with Discord.'), 'success')
|
|
return redirect(url_for('post.feed'))
|
|
|
|
@discord_bp.route('/link_discord')
|
|
@login_required
|
|
def link_discord():
|
|
redirect_uri = url_for('discord.authorize_discord', _external=True)
|
|
return discord.authorize_redirect(redirect_uri)
|
|
|
|
@discord_bp.route('/authorize/discord')
|
|
@login_required
|
|
def authorize_discord():
|
|
token = discord.authorize_access_token()
|
|
user_data = discord.get('users/@me').json()
|
|
current_user.discord_id = user_data['id']
|
|
current_user.discord_linked = True
|
|
db.session.commit()
|
|
flash(_('Discord account linked!'), 'success')
|
|
return redirect(url_for('profil.profile'))
|
|
|
|
@discord_bp.route('/unlink_discord', methods=['POST'])
|
|
@login_required
|
|
def unlink_discord():
|
|
current_user.discord_id = None
|
|
current_user.discord_linked = False
|
|
db.session.commit()
|
|
flash(_('Discord account unlinked!'), 'success')
|
|
return redirect(url_for('profil.profile')) |