Add the Oauth check

This commit is contained in:
2025-09-27 20:59:08 +02:00
parent aaca71afc0
commit 7a5e387665
2 changed files with 16 additions and 4 deletions

10
main.py
View File

@@ -16,7 +16,10 @@ from routes.friends import friends_bp
from routes.notifications import noti_bp from routes.notifications import noti_bp
from routes.credits import credits_bp from routes.credits import credits_bp
from models import db, User, Reward, Event, UserShopItem, ShopItem, SHOPITEM_ID_PREMIUM, SHOPITEM_ID_GOLDRAHMEN, SHOPITEM_ID_EXTRA_TYPES, SHOPITEM_ID_EXTRA_UPLOAD from models import db, User, Reward, Event, UserShopItem, ShopItem, SHOPITEM_ID_PREMIUM, SHOPITEM_ID_GOLDRAHMEN, SHOPITEM_ID_EXTRA_TYPES, SHOPITEM_ID_EXTRA_UPLOAD
from routes.oauth import oauth try:
from routes.oauth import oauth
except ImportError:
pass
import re import re
import os import os
@@ -34,7 +37,10 @@ app_login.login_view = 'log.login'
app_login.login_message = _('Please log in to access this page.') app_login.login_message = _('Please log in to access this page.')
app_login.login_message_category = 'info' app_login.login_message_category = 'info'
babel = Babel(app) babel = Babel(app)
oauth.init_app(app) try:
oauth.init_app(app)
except NameError:
pass
if not os.path.exists('instance/site.db'): if not os.path.exists('instance/site.db'):
with app.app_context(): with app.app_context():

View File

@@ -1,11 +1,17 @@
from flask import Blueprint, render_template, redirect, url_for, flash, request from flask import Blueprint, render_template, redirect, url_for, flash, request, abort
from flask_login import login_required, current_user from flask_login import login_required, current_user
from models import db, User from models import db, User
from werkzeug.security import generate_password_hash from werkzeug.security import generate_password_hash
from flask_babel import gettext as _ from flask_babel import gettext as _
from routes.oauth import discord try:
from routes.oauth import discord
except ImportError:
discord = None
from routes.login import login_user from routes.login import login_user
if discord is None:
abort("OAuth not configured. Please set up OAuth in routes/oauth.py")
discord_bp = Blueprint('discord', __name__) discord_bp = Blueprint('discord', __name__)
@discord_bp.route('/login/discord') @discord_bp.route('/login/discord')