uhh

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