Skip to content

Commit 3ea487f

Browse files
make phonenumber plugin really a plugin
1 parent d6d8b63 commit 3ea487f

File tree

1 file changed

+7
-10
lines changed

1 file changed

+7
-10
lines changed

two_factor/views/profile.py

+7-10
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
from django.apps.registry import apps
21
from django.conf import settings
32
from django.contrib.auth.decorators import login_required
43
from django.shortcuts import redirect, resolve_url
@@ -30,24 +29,22 @@ class ProfileView(TemplateView):
3029
template_name = 'two_factor/profile/profile.html'
3130

3231
def get_context_data(self, **kwargs):
32+
user = self.request.user
33+
3334
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()
3536

3637
except Exception:
3738
backup_tokens = 0
3839

3940
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__,
4243
'backup_tokens': backup_tokens,
44+
'backup_phones': backup_phones(user),
45+
'available_phone_methods': get_available_phone_methods(),
4346
}
4447

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-
5148
return context
5249

5350

0 commit comments

Comments
 (0)