From bc65fae4b2b9b884c9d37a6ec4734456508bfa1b Mon Sep 17 00:00:00 2001 From: Jens Langhammer Date: Wed, 28 Nov 2018 14:01:18 +0100 Subject: [PATCH] Admin: remove .cast() calls and replace with select_subclasses() --- passbook/admin/views/sources.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/passbook/admin/views/sources.py b/passbook/admin/views/sources.py index 777289ff1..3040b8323 100644 --- a/passbook/admin/views/sources.py +++ b/passbook/admin/views/sources.py @@ -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()