habsi/authentik
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Admin: remove .cast() calls and replace with select_subclasses()

Browse Source
This commit is contained in:
Jens Langhammer
2018-11-28 14:01:18 +01:00
parent 005d0f1c8f
commit bc65fae4b2
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
passbook/admin/views/sources.py
View File

@ -54,8 +54,7 @@ class SourceUpdateView(SuccessMessageMixin, AdminRequiredMixin, UpdateView):
return form_class
def get_object(self, queryset=None):
obj = Source.objects.get(pk=self.kwargs.get('pk'))
return obj.cast()
return Source.objects.filter(pk=self.kwargs.get('pk')).select_subclasses().first()
class SourceDeleteView(SuccessMessageMixin, AdminRequiredMixin, DeleteView):
@ -67,5 +66,4 @@ class SourceDeleteView(SuccessMessageMixin, AdminRequiredMixin, DeleteView):
success_message = _('Successfully updated Source')
def get_object(self, queryset=None):
obj = Source.objects.get(pk=self.kwargs.get('pk'))
return obj.cast()
return Source.objects.filter(pk=self.kwargs.get('pk')).select_subclasses().first()