fix defect: views's filter doesn't work in django 2.0+

shadowsocks_manager/domain/views.py

@@ -4,14 +4,14 @@
 from __future__ import unicode_literals
 from __future__ import absolute_import
 
-from rest_framework import viewsets
+from shadowsocks_manager.utils.viewsets import CompatModelViewSet
 
 from . import models, serializers
 
 
 # Create your views here.
 
-class NameServerViewSet(viewsets.ModelViewSet):
+class NameServerViewSet(CompatModelViewSet):
     """
     This viewset automatically provides `list` and `detail` actions.
     """
@@ -20,7 +20,7 @@ class NameServerViewSet(viewsets.ModelViewSet):
     filter_fields = ['name', 'api_cls_name', 'user']
 
 
-class DomainViewSet(viewsets.ModelViewSet):
+class DomainViewSet(CompatModelViewSet):
     """
     This viewset automatically provides `list` and `detail` actions.
     """
@@ -29,7 +29,7 @@ class DomainViewSet(viewsets.ModelViewSet):
     filter_fields = ['name', 'nameserver', 'nameserver__name']
 
 
-class RecordViewSet(viewsets.ModelViewSet):
+class RecordViewSet(CompatModelViewSet):
     """
     This viewset automatically provides `list` and `detail` actions.
     """

shadowsocks_manager/notification/views.py

@@ -4,14 +4,14 @@
 from __future__ import unicode_literals
 from __future__ import absolute_import
 
-from rest_framework import viewsets
+from shadowsocks_manager.utils.viewsets import CompatModelViewSet
 
 from . import models, serializers
 
 
 # Create your views here.
 
-class TemplateViewSet(viewsets.ModelViewSet):
+class TemplateViewSet(CompatModelViewSet):
     """
     This viewset automatically provides `list` and `detail` actions.
     """

shadowsocks_manager/shadowsocks/views.py

@@ -4,30 +4,30 @@
 from __future__ import unicode_literals
 from __future__ import absolute_import
 
-from rest_framework import viewsets
+from shadowsocks_manager.utils.viewsets import CompatModelViewSet
 
 from . import models, serializers
 
 
 # Create your views here.
 
-class ConfigViewSet(viewsets.ModelViewSet):
+class ConfigViewSet(CompatModelViewSet):
     """
     This viewset automatically provides `list` and `detail` actions.
     """
     queryset = models.Config.objects.all()
     serializer_class = serializers.ConfigSerializer
 
 
-class AccountViewSet(viewsets.ModelViewSet):
+class AccountViewSet(CompatModelViewSet):
     """
     This viewset automatically provides `list` and `detail` actions.
     """
     queryset = models.Account.objects.all()
     serializer_class = serializers.AccountSerializer
 
 
-class NodeViewSet(viewsets.ModelViewSet):
+class NodeViewSet(CompatModelViewSet):
     """
     This viewset automatically provides `list` and `detail` actions.
     """
@@ -36,15 +36,15 @@ class NodeViewSet(viewsets.ModelViewSet):
     filter_fields = ['name']
 
 
-class NodeAccountViewSet(viewsets.ModelViewSet):
+class NodeAccountViewSet(CompatModelViewSet):
     """
     This viewset automatically provides `list` and `detail` actions.
     """
     queryset = models.NodeAccount.objects.all()
     serializer_class = serializers.NodeAccountSerializer
 
 
-class SSManagerViewSet(viewsets.ModelViewSet):
+class SSManagerViewSet(CompatModelViewSet):
     """
     This viewset automatically provides `list` and `detail` actions.
     """

shadowsocks_manager/statistic/views.py

@@ -3,22 +3,22 @@
 # py2.7 and py3 compatibility imports
 from __future__ import unicode_literals
 
-from rest_framework import viewsets
+from shadowsocks_manager.utils.viewsets import CompatModelViewSet
 
 from . import models, serializers
 
 
 # Create your views here.
 
-class PeriodViewSet(viewsets.ModelViewSet):
+class PeriodViewSet(CompatModelViewSet):
     """
     This viewset automatically provides `list` and `detail` actions.
     """
     queryset = models.Period.objects.all()
     serializer_class = serializers.PeriodSerializer
 
 
-class StatisticViewSet(viewsets.ModelViewSet):
+class StatisticViewSet(CompatModelViewSet):
     """
     This viewset automatically provides `list` and `detail` actions.
     """

shadowsocks_manager/utils/viewsets.py

@@ -0,0 +1,30 @@
+from django import VERSION as DJANGO_VERSION
+from rest_framework.viewsets import ModelViewSet
+
+
+class CompatModelViewSet(ModelViewSet):
+    """
+    A compatibility class that extends the ModelViewSet class from Django REST Framework.
+    This class provides compatibility for different versions of Django.
+
+    Attributes:
+        filter_fields (list): 
+            A list of fields to be used for filtering the queryset.
+            Used in Django 1.x.
+        filterset_fields (list): 
+            Used since Django 2.x.
+            A list of fields to be used for filtering the queryset.
+
+    Methods:
+        __init__(self, *args, **kwargs): Initializes the CompatModelViewSet instance.
+
+    """
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        if DJANGO_VERSION < (2, 0):
+            if not hasattr(self, 'filter_fields') and hasattr(self, 'filterset_fields'):
+                self.filter_fields = getattr(self, 'filterset_fields', [])
+        else:
+            if not hasattr(self, 'filterset_fields') and hasattr(self, 'filter_fields'):
+                self.filterset_fields = getattr(self, 'filter_fields', [])