diff --git a/app/pages/admin/websitefeaturesdefinition.py b/app/pages/admin/websitefeaturesdefinition.py index fd5e635..965df22 100644 --- a/app/pages/admin/websitefeaturesdefinition.py +++ b/app/pages/admin/websitefeaturesdefinition.py @@ -79,5 +79,8 @@ WebsiteFeaturesDefinition = [ }, { "name": "CurrencyExchange" + }, + { + "name": "InviteKeys" } ] \ No newline at end of file diff --git a/app/pages/invitekeys/handler.py b/app/pages/invitekeys/handler.py index e3e8d7c..4571f1b 100644 --- a/app/pages/invitekeys/handler.py +++ b/app/pages/invitekeys/handler.py @@ -1,7 +1,7 @@ from flask import Blueprint, render_template, request, redirect, url_for, jsonify, make_response, flash from app.models.user import User from app.models.invite_key import InviteKey -from app.util import auth, transactions +from app.util import auth, transactions, websiteFeatures from app.services import invitekeys, economy from app.extensions import db, limiter, csrf from datetime import datetime, timedelta @@ -11,6 +11,9 @@ inviteKeyRoute = Blueprint("invitekey", __name__, url_prefix="/") @inviteKeyRoute.route("/invite-keys", methods=["GET"]) @auth.authenticated_required def inviteKeysPage(): + if not websiteFeatures.GetWebsiteFeature("InviteKeys"): + flash("Invite keys are temporarily disabled", "error") + return redirect("/home") AuthenticatedUser : User = auth.GetCurrentUser() pageNumber = request.args.get("page", 1, int) if pageNumber < 1: @@ -23,6 +26,9 @@ def inviteKeysPage(): @auth.authenticated_required @csrf.exempt def createInviteKey(): + if not websiteFeatures.GetWebsiteFeature("InviteKeys"): + flash("Invite keys are temporarily disabled", "error") + return redirect("/home") AuthenticatedUser : User = auth.GetCurrentUser() if AuthenticatedUser is None: return jsonify({"success": False, "message": "Unauthorized"}),401 diff --git a/app/pages/signup/signup.html b/app/pages/signup/signup.html index 4c174c7..d52925d 100644 --- a/app/pages/signup/signup.html +++ b/app/pages/signup/signup.html @@ -24,7 +24,9 @@
+ {% if InviteKeysEnabled %} + {% endif %}