|
1 | | -from django.apps.registry import apps |
2 | 1 | from django.conf import settings |
3 | 2 | from django.contrib.auth.decorators import login_required |
4 | 3 | from django.shortcuts import redirect, resolve_url |
@@ -30,24 +29,22 @@ class ProfileView(TemplateView): |
30 | 29 | template_name = 'two_factor/profile/profile.html' |
31 | 30 |
|
32 | 31 | def get_context_data(self, **kwargs): |
| 32 | + user = self.request.user |
| 33 | + |
33 | 34 | try: |
34 | | - backup_tokens = self.request.user.staticdevice_set.all()[0].token_set.count() |
| 35 | + backup_tokens = user.staticdevice_set.all()[0].token_set.count() |
35 | 36 |
|
36 | 37 | except Exception: |
37 | 38 | backup_tokens = 0 |
38 | 39 |
|
39 | 40 | context = { |
40 | | - 'default_device': default_device(self.request.user), |
41 | | - 'default_device_type': default_device(self.request.user).__class__.__name__, |
| 41 | + 'default_device': default_device(user), |
| 42 | + 'default_device_type': default_device(user).__class__.__name__, |
42 | 43 | 'backup_tokens': backup_tokens, |
| 44 | + 'backup_phones': backup_phones(user), |
| 45 | + 'available_phone_methods': get_available_phone_methods(), |
43 | 46 | } |
44 | 47 |
|
45 | | - if (apps.is_installed('two_factor.plugins.phonenumber')): |
46 | | - context.update({ |
47 | | - 'backup_phones': backup_phones(self.request.user), |
48 | | - 'available_phone_methods': get_available_phone_methods(), |
49 | | - }) |
50 | | - |
51 | 48 | return context |
52 | 49 |
|
53 | 50 |
|
|
0 commit comments