Python 多个类似方法类

Question

我正在尝试重构我的 Django 应用程序。这是我觉得烦人的重复代码：

class EducationInfoViewSet(viewsets.ModelViewSet):

    queryset = pf.education_info.objects.all()
    serializer_class = EducationInfoSerializer
    permission_classes = (VisionUserPermissions, )

    def get_queryset(self):

        model = pf.education_info

        identity = self.request.query_params.get('username',None)
        queryset = model.objects.all()
        user = User.objects.none()
        if identity is not None:
            user = User.objects.get(username=identity)
            student = get_object_or_404(dbd.user_type, username=user)        
            queryset = model.objects.filter(username=student)
        if not permission_check(self.request,user):
            return User.objects.none()
        return queryset


class FamilyInfoViewSet(viewsets.ModelViewSet):

    queryset = pf.family_info.objects.all()
    serializer_class = FamilyInfoSerializer
    permission_classes = (VisionUserPermissions, )

    def get_queryset(self):

        model = pf.family_info

        identity = self.request.query_params.get('username',None)
        queryset = model.objects.all()
        user = User.objects.none()
        if identity is not None:
            user = User.objects.get(username=identity)
            student = get_object_or_404(dbd.user_type, username=user)        
            queryset = model.objects.filter(username=student)
        if not permission_check(self.request,user):
            return User.objects.none()
        return queryset

所以我的 "get_queryset" 函数除了一个变量外是相同的，而且我有多个类像这两个。我应该如何实现而不重复自己？

提前致谢！

Answer 1

创建混音class

class QuerysetMixin(viewsets.ModelViewSet):

    def get_queryset(self):

        model = getattr(pf, self.field_name)

        identity = self.request.query_params.get('username',None)
        queryset = model.objects.all()
        user = User.objects.none()
        if identity is not None:
            user = User.objects.get(username=identity)
            student = get_object_or_404(dbd.user_type, username=user)        
            queryset = model.objects.filter(username=student)
        if not permission_check(self.request,user):
            return User.objects.none()
        return queryset

并继承

class FamilyInfoViewSet(viewsets.ModelViewSet, QuerysetMixin):

    queryset = pf.family_info.objects.all()
    serializer_class = FamilyInfoSerializer
    permission_classes = (VisionUserPermissions, )
    field_name = 'family_info'

Python 多个类似方法类

Python similar method in multiple classes

python

django

class

multiple-inheritance

django-rest-framework

Python 多个类似方法 类

Python similar method in multiple classes

python

django

class

multiple-inheritance

django-rest-framework

Python 多个类似方法类