From e663ef15fce85b000733bd96bf26b252483bb303 Mon Sep 17 00:00:00 2001
From: Dirk Doesburg <dirk.doesburg@gmail.com>
Date: Fri, 21 Jun 2024 20:50:11 +0200
Subject: [PATCH] Fix tests failing due to 2FA requirement

---
 website/thaliawebsite/admin.py    | 2 +-
 website/thaliawebsite/settings.py | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/website/thaliawebsite/admin.py b/website/thaliawebsite/admin.py
index 122fdaa73..ae24f03b0 100644
--- a/website/thaliawebsite/admin.py
+++ b/website/thaliawebsite/admin.py
@@ -13,5 +13,5 @@ class ThaliaAdminSite(admin.AdminSite):
 
     def has_permission(self, request):
         return super().has_permission(request) and (
-            settings.DEBUG or user_has_device(request.user)
+            not settings.ADMIN_REQUIRE_2FA or user_has_device(request.user)
         )
diff --git a/website/thaliawebsite/settings.py b/website/thaliawebsite/settings.py
index 32b7e7111..45c835489 100644
--- a/website/thaliawebsite/settings.py
+++ b/website/thaliawebsite/settings.py
@@ -1206,3 +1206,5 @@ def show_toolbar(request):
     if os.environ.get("MONEYBIRD_ZERO_TAX_RATE_ID")
     else None
 )
+
+ADMIN_REQUIRE_2FA = setting(development=False, production=True)