Implement more advanced 2FA library

core/views/strategies.py

@@ -1,17 +1,17 @@
 from django.contrib.auth.mixins import LoginRequiredMixin
 
+# from django.urls import reverse
+from two_factor.views.mixins import OTPRequiredMixin
+
 from core.forms import StrategyForm
 from core.models import Strategy
 from core.util import logs
 from core.views import ObjectCreate, ObjectDelete, ObjectList, ObjectUpdate
 
-# from django.urls import reverse
-
-
 log = logs.get_logger(__name__)
 
 
-class StrategyList(LoginRequiredMixin, ObjectList):
+class StrategyList(LoginRequiredMixin, OTPRequiredMixin, ObjectList):
     list_template = "partials/strategy-list.html"
     model = Strategy
     page_title = "List of strategies"
@@ -22,7 +22,7 @@ class StrategyList(LoginRequiredMixin, ObjectList):
     submit_url_name = "strategy_create"
 
 
-class StrategyCreate(LoginRequiredMixin, ObjectCreate):
+class StrategyCreate(LoginRequiredMixin, OTPRequiredMixin, ObjectCreate):
     model = Strategy
     form_class = StrategyForm
     list_url_name = "strategies"
@@ -31,7 +31,7 @@ class StrategyCreate(LoginRequiredMixin, ObjectCreate):
     submit_url_name = "strategy_create"
 
 
-class StrategyUpdate(LoginRequiredMixin, ObjectUpdate):
+class StrategyUpdate(LoginRequiredMixin, OTPRequiredMixin, ObjectUpdate):
     model = Strategy
     form_class = StrategyForm
 
@@ -41,7 +41,7 @@ class StrategyUpdate(LoginRequiredMixin, ObjectUpdate):
     submit_url_name = "strategy_update"
 
 
-class StrategyDelete(LoginRequiredMixin, ObjectDelete):
+class StrategyDelete(LoginRequiredMixin, OTPRequiredMixin, ObjectDelete):
     model = Strategy
 
     list_url_name = "strategies"