python - 多余的 Django 代码

Question

我有一些我写的代码，它可以很好地用于它的目的。

from django.shortcuts import get_object_or_404
from django.views.generic import ListView
from cab.models import Language

class LanguageDetail(ListView):


    def get_queryset(self):
        slug = self.kwargs['slug']
        language = get_object_or_404(Language, slug=slug)
        return language.snippet_set.all()

    paginate_by = 20
    template_name = 'cab/language_detail.html'

    def get_context_data(self, **kwargs):
        slug = self.kwargs['slug']
        language = get_object_or_404(Language, slug=slug)
        context = super(LanguageDetail, self).get_context_data(**kwargs)
        context['language'] = language
        return context

我的问题是我在两个函数中重复了两行代码，我不喜欢这样。

slug = self.kwargs['slug']
language = get_object_or_404(Language, slug=slug)

我怎么能避免呢？提前致谢。

Answer 1

你的意思是这样的吗？

def get_slug(self, name):   
    slug = self.kwargs[name]
    return get_object_or_404(Language, slug=slug)

可以用作language = self.get_slug('slug')

请记住，如果kwargs不包含name，则会引发错误。为了防止这种情况，您可以执行以下操作：

slug = self.kwargs.get(name, None)
if slug is not None:
    ...