Implement more advanced 2FA library

core/views/trades.py

@@ -1,6 +1,7 @@
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.shortcuts import render
 from django.views import View
+from two_factor.views.mixins import OTPRequiredMixin
 
 from core.forms import TradeForm
 from core.models import Trade
@@ -16,7 +17,7 @@ from core.views import (
 log = logs.get_logger(__name__)
 
 
-class TradeList(LoginRequiredMixin, ObjectList):
+class TradeList(LoginRequiredMixin, OTPRequiredMixin, ObjectList):
     list_template = "partials/trade-list.html"
     model = Trade
     page_title = (
@@ -32,7 +33,7 @@ class TradeList(LoginRequiredMixin, ObjectList):
     delete_all_url_name = "trade_delete_all"
 
 
-class TradeCreate(LoginRequiredMixin, ObjectCreate):
+class TradeCreate(LoginRequiredMixin, OTPRequiredMixin, ObjectCreate):
     model = Trade
     form_class = TradeForm
     list_url_name = "trades"
@@ -45,7 +46,7 @@ class TradeCreate(LoginRequiredMixin, ObjectCreate):
         log.debug(f"Posting trade {obj}")
 
 
-class TradeUpdate(LoginRequiredMixin, ObjectUpdate):
+class TradeUpdate(LoginRequiredMixin, OTPRequiredMixin, ObjectUpdate):
     model = Trade
     form_class = TradeForm
     list_url_name = "trades"
@@ -54,14 +55,14 @@ class TradeUpdate(LoginRequiredMixin, ObjectUpdate):
     submit_url_name = "trade_update"
 
 
-class TradeDelete(LoginRequiredMixin, ObjectDelete):
+class TradeDelete(LoginRequiredMixin, OTPRequiredMixin, ObjectDelete):
     model = Trade
 
     list_url_name = "trades"
     list_url_args = ["type"]
 
 
-class TradeDeleteAll(LoginRequiredMixin, ObjectNameMixin, View):
+class TradeDeleteAll(LoginRequiredMixin, OTPRequiredMixin, ObjectNameMixin, View):
     template_name = "partials/notify.html"
     model = Trade