CGM_Public/pretix_original
2
0
Fork 1
mirror of https://github.com/pretix/pretix.git synced 2026-05-05 15:14:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix StaticDevice.MultipleObjectsReturned

Browse Source 
PRETIXEU-1YX
This commit is contained in:
Raphael Michel
2020-03-16 11:41:33 +01:00
parent 8d23d75dfd
commit 1d3ac5f02f
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/pretix/control/views/user.py
View File

@@ -242,6 +242,10 @@ class User2FAMainView(RecentAuthenticationRequiredMixin, TemplateView):
try:
ctx['static_tokens'] = StaticDevice.objects.get(user=self.request.user, name='emergency').token_set.all()
except StaticDevice.MultipleObjectsReturned:
ctx['static_tokens'] = StaticDevice.objects.filter(
user=self.request.user, name='emergency'
).first().token_set.all()
except StaticDevice.DoesNotExist:
d = StaticDevice.objects.create(user=self.request.user, name='emergency')
for i in range(10):
@@ -543,8 +547,8 @@ class User2FARegenerateEmergencyView(RecentAuthenticationRequiredMixin, Template
template_name = 'pretixcontrol/user/2fa_regenemergency.html'
def post(self, request, *args, **kwargs):
d = StaticDevice.objects.get(user=self.request.user, name='emergency')
d.token_set.all().delete()
StaticDevice.objects.filter(user=self.request.user, name='emergency').delete()
d = StaticDevice.objects.create(user=self.request.user, name='emergency')
for i in range(10):
d.token_set.create(token=get_random_string(length=12, allowed_chars='1234567890'))
self.request.user.log_action('pretix.user.settings.2fa.regenemergency', user=self.request.user)