Enable filtering on DataView.
parent
55642dda34
commit
3055a9ef7c
|
@ -1,3 +1,4 @@
|
||||||
|
import django_filters
|
||||||
from rest_framework import viewsets, mixins
|
from rest_framework import viewsets, mixins
|
||||||
|
|
||||||
from network.api.perms import StationOwnerCanEditPermission
|
from network.api.perms import StationOwnerCanEditPermission
|
||||||
|
@ -31,10 +32,22 @@ class ObservationView(viewsets.ModelViewSet):
|
||||||
serializer_class = serializers.ObservationSerializer
|
serializer_class = serializers.ObservationSerializer
|
||||||
|
|
||||||
|
|
||||||
class DataView(viewsets.ReadOnlyModelViewSet,
|
class DataFilter(django_filters.FilterSet):
|
||||||
mixins.UpdateModelMixin):
|
class Meta:
|
||||||
|
model = Data
|
||||||
|
fields = ['start', 'end', 'ground_station']
|
||||||
|
|
||||||
|
|
||||||
|
class DataView(viewsets.ReadOnlyModelViewSet, mixins.UpdateModelMixin):
|
||||||
queryset = Data.objects.all()
|
queryset = Data.objects.all()
|
||||||
serializer_class = serializers.DataSerializer
|
serializer_class = serializers.DataSerializer
|
||||||
|
filter_class = DataFilter
|
||||||
permission_classes = [
|
permission_classes = [
|
||||||
StationOwnerCanEditPermission
|
StationOwnerCanEditPermission
|
||||||
]
|
]
|
||||||
|
|
||||||
|
def get_queryset(self):
|
||||||
|
payload = self.request.QUERY_PARAMS.get('payload', None)
|
||||||
|
if payload == '':
|
||||||
|
return self.queryset.filter(payload='')
|
||||||
|
return super(DataView, self).get_queryset()
|
||||||
|
|
|
@ -156,3 +156,7 @@ SECRET_KEY = getenv('DJANGO_SECRET_KEY', 'changeme')
|
||||||
import dj_database_url
|
import dj_database_url
|
||||||
DATABASE_URL = getenv('DJANGO_DATABASE_URL', 'sqlite:///db.sqlite3')
|
DATABASE_URL = getenv('DJANGO_DATABASE_URL', 'sqlite:///db.sqlite3')
|
||||||
DATABASES = {'default': dj_database_url.parse(DATABASE_URL)}
|
DATABASES = {'default': dj_database_url.parse(DATABASE_URL)}
|
||||||
|
|
||||||
|
REST_FRAMEWORK = {
|
||||||
|
'DEFAULT_FILTER_BACKENDS': ('rest_framework.filters.DjangoFilterBackend',)
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue