Adds queryset filtering to all models with published attribute

armada_sde_rest/__init__.py

@@ -1 +1 @@
-__version__ = '0.1.1'
+__version__ = '0.1.2'

armada_sde_rest/management/commands/generate_viewsets.py

@@ -9,7 +9,12 @@
 
 
 QUERYSET_OVERRIDE = {
-    'InvType': 'filter(published=True)'
+    'DgmAttributeType': 'filter(published=True)',
+    'DgmEffect': 'filter(published=True)',
+    'InvCategory': 'filter(published=True)',
+    'InvGroup': 'filter(published=True)',
+    'InvType': 'filter(published=True)',
+    'RamActivity': 'filter(published=True)',
 }
 
 VIEWSETS_HEADER = '''# Generated by v{} for sde v{}

armada_sde_rest/serializers.py

@@ -1,6 +1,6 @@
 
 
-# Generated by  v0.1.0 for sde v0.2.2
+# Generated by  v0.1.2 for sde v0.3.0
 from rest_framework import serializers
 from armada_sde import models
 

armada_sde_rest/urls.py

@@ -1,4 +1,4 @@
-# Generated by v0.1.0 for sde v0.2.2
+# Generated by v0.1.2 for sde v0.3.0
 from rest_framework.routers import DefaultRouter
 from . import views
 

armada_sde_rest/views.py

@@ -1,4 +1,4 @@
-# Generated by v0.1.0 for sde v0.2.2
+# Generated by v0.1.2 for sde v0.3.0
 from armada_sde import models
 from . import serializers
 from .viewsets import SDEViewSet
@@ -115,13 +115,13 @@ class DgmAttributeCategoryViewSet(SDEViewSet):
 class DgmAttributeTypeViewSet(SDEViewSet):
     list_serializer = serializers.DgmAttributeTypeListSerializer
     details_serializer = serializers.DgmAttributeTypeDetailsSerializer
-    queryset = models.DgmAttributeType.objects.all()
+    queryset = models.DgmAttributeType.objects.filter(published=True)
 
 
 class DgmEffectViewSet(SDEViewSet):
     list_serializer = serializers.DgmEffectListSerializer
     details_serializer = serializers.DgmEffectDetailsSerializer
-    queryset = models.DgmEffect.objects.all()
+    queryset = models.DgmEffect.objects.filter(published=True)
 
 
 class DgmExpressionViewSet(SDEViewSet):
@@ -205,7 +205,7 @@ class IndustryBlueprintViewSet(SDEViewSet):
 class InvCategoryViewSet(SDEViewSet):
     list_serializer = serializers.InvCategoryListSerializer
     details_serializer = serializers.InvCategoryDetailsSerializer
-    queryset = models.InvCategory.objects.all()
+    queryset = models.InvCategory.objects.filter(published=True)
 
 
 class InvContrabandTypeViewSet(SDEViewSet):
@@ -235,7 +235,7 @@ class InvFlagViewSet(SDEViewSet):
 class InvGroupViewSet(SDEViewSet):
     list_serializer = serializers.InvGroupListSerializer
     details_serializer = serializers.InvGroupDetailsSerializer
-    queryset = models.InvGroup.objects.all()
+    queryset = models.InvGroup.objects.filter(published=True)
 
 
 class InvItemViewSet(SDEViewSet):
@@ -409,7 +409,7 @@ class PlanetSchematicsTypeMapViewSet(SDEViewSet):
 class RamActivityViewSet(SDEViewSet):
     list_serializer = serializers.RamActivityListSerializer
     details_serializer = serializers.RamActivityDetailsSerializer
-    queryset = models.RamActivity.objects.all()
+    queryset = models.RamActivity.objects.filter(published=True)
 
 
 class RamAssemblyLineStationViewSet(SDEViewSet):

requirements.txt

@@ -1,4 +1,4 @@
 Django>=2.0
 djangorestframework
-armada-sde>=0.2.2
+armada-sde>=0.3.0
 django-environ