119 lines
4 KiB
Python
119 lines
4 KiB
Python
from django.http import HttpResponseBadRequest, HttpResponse, HttpRequest
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
from rest_framework import viewsets
|
|
from rest_framework.response import Response
|
|
|
|
from right_tree.api.models import Habitat, HabitatImage, Plant, EcologicalDistrictLayer, SoilOrder
|
|
from right_tree.api.serializers import HabitatImageSerializer, HabitatSerializer, PlantSerializer, SoilOrderSerializer, EcologicalDistrictLayerSerializer, AddressSerializer
|
|
|
|
from .filters import *
|
|
from .utils import get_address_from_coordinates
|
|
from .csv_utils import create_plant_csv_file, get_plant_csv_filepath
|
|
|
|
|
|
class PlantViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
Filtered viewset for plants.
|
|
"""
|
|
queryset = Plant.objects.all()
|
|
serializer_class = PlantSerializer
|
|
|
|
def get_queryset(self):
|
|
""" Filtering plant query set by query parameters in the URL.
|
|
(May want to eventually use django filters to break up the logic...)
|
|
"""
|
|
queryset = Plant.objects.all()
|
|
queryset = coordinate_filter(self.request, queryset)
|
|
queryset = soil_variant_filter(self.request, queryset)
|
|
queryset = zone_filter(self.request, queryset)
|
|
|
|
return queryset
|
|
|
|
|
|
class SoilOrderViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
Filtered viewset for soil details.
|
|
"""
|
|
serializer_class = SoilOrderSerializer
|
|
|
|
def get_queryset(self):
|
|
""" Filtering soil order query set by coordinate parameters in the URL.
|
|
"""
|
|
coordinates = self.request.query_params.get('coordinates')
|
|
if coordinates is not None:
|
|
return soil_order_coordinate_filter(coordinates)
|
|
|
|
return SoilOrder.objects.all()
|
|
|
|
|
|
class EcologicalDistrictViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
Filtered viewset for ecological district/region details.
|
|
"""
|
|
serializer_class = EcologicalDistrictLayerSerializer
|
|
|
|
def get_queryset(self):
|
|
""" Filtering ecological district/region query set by coordinate parameters in the URL.
|
|
"""
|
|
coordinates = self.request.query_params.get('coordinates')
|
|
if coordinates is not None:
|
|
return ecological_district_coordinate_filter(coordinates)
|
|
|
|
return EcologicalDistrictLayer.objects.all()
|
|
|
|
|
|
class LINZPropertyViewSet(viewsets.ViewSet):
|
|
"""
|
|
Filtered viewset for ecological district/region details.
|
|
"""
|
|
|
|
def list(self, request):
|
|
coordinates = self.request.query_params.get('coordinates')
|
|
if coordinates is not None:
|
|
address_data = get_address_from_coordinates(coordinates)
|
|
serializer = AddressSerializer(address_data)
|
|
return Response(serializer.data)
|
|
else:
|
|
return HttpResponseBadRequest("No coordinate given.")
|
|
|
|
|
|
class HabitatViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
Viewset for all habitats.
|
|
"""
|
|
serializer_class = HabitatSerializer
|
|
queryset = Habitat.objects.all()
|
|
|
|
|
|
class HabitatImageViewSet(viewsets.ViewSet):
|
|
"""
|
|
Viewset for a habitat image.
|
|
"""
|
|
|
|
def list(self, request):
|
|
queryset = HabitatImage.objects.all()
|
|
serializer = HabitatImageSerializer(queryset, many=True)
|
|
return Response(serializer.data)
|
|
|
|
def retrieve(self, request, pk=None):
|
|
queryset = HabitatImage.objects.all()
|
|
habitat_image = get_object_or_404(queryset, pk=pk)
|
|
serializer = HabitatImageSerializer(habitat_image)
|
|
return Response(serializer.data)
|
|
|
|
|
|
class CSVDownloadView(viewsets.ViewSet):
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
filtered_plants = Plant.objects.all()
|
|
filtered_plants = coordinate_filter(request, filtered_plants)
|
|
filtered_plants = soil_variant_filter(request, filtered_plants)
|
|
filtered_plants = zone_filter(request, filtered_plants)
|
|
|
|
create_plant_csv_file(request, filtered_plants)
|
|
|
|
csv_file = open(get_plant_csv_filepath(), 'rb')
|
|
response = HttpResponse(csv_file, content_type='text/csv')
|
|
response['Content-Disposition'] = 'attachment; filename="test.csv"'
|
|
return response
|