@@ -201,9 +201,7 @@ def get_form_kwargs(self):
201201 return kwargs
202202
203203 def get_context_data (self , ** kwargs ):
204- context = super ().get_context_data (** kwargs )
205-
206- return context | {
204+ context = {
207205 "title" : _ ("Set up two-factor authentication" ),
208206 "device" : self .device ,
209207 "backup_device" : self .backup_device ,
@@ -213,7 +211,9 @@ def get_context_data(self, **kwargs):
213211 self .backup_device .token_set .values_list ("token" , flat = True )
214212 ),
215213 "back_url" : reverse ("hidp_otp_management:manage" ),
214+ "logout_url" : reverse ("hidp_accounts:logout" ),
216215 }
216+ return super ().get_context_data () | context | kwargs
217217
218218 def form_valid (self , form ):
219219 form .save ()
@@ -280,9 +280,11 @@ def get_recovery_code_url(self, request): # noqa: PLR6301
280280 return base_url
281281
282282 def get_context_data (self , ** kwargs ):
283- context = super ().get_context_data (** kwargs )
284- context ["recovery_code_url" ] = self .get_recovery_code_url (self .request )
285- return context
283+ context = {
284+ "recovery_code_url" : self .get_recovery_code_url (self .request ),
285+ "logout_url" : reverse ("hidp_accounts:logout" ),
286+ }
287+ return super ().get_context_data () | context | kwargs
286288
287289
288290class VerifyRecoveryCodeView (VerifyOTPBase ):
@@ -296,6 +298,10 @@ def form_valid(self, form):
296298
297299 return result
298300
301+ def get_context_data (self , ** kwargs ):
302+ context = {"logout_url" : reverse ("hidp_accounts:logout" )}
303+ return super ().get_context_data () | context | kwargs
304+
299305 def send_mail (self ):
300306 """Notify the user that a recovery code was used."""
301307 base_url = self .request .build_absolute_uri ("/" )
0 commit comments