ENH: Collection filterings (#30)

Fixes #29

ccdb/locations/serializers.py

@@ -1,10 +1,30 @@
 from rest_framework import serializers
 
-from .models import StudyLocation
+from .models import Region, Site, StudyLocation
+
+
+class RegionSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Region
+        fields = ('name', 'code', 'sort_order')
+
+
+class SiteSerializer(serializers.ModelSerializer):
+    included_serializers = {
+        'region': 'ccdb.locations.serializers.RegionSerializer',
+    }
+
+    class Meta:
+        model = Site
+        fields = ('name', 'code', 'description', 'sort_order', 'region')
 
 
 class StudyLocationSerializer(serializers.ModelSerializer):
+    included_serializers = {
+        'site': 'ccdb.locations.serializers.SiteSerializer',
+    }
+
     class Meta:
         model = StudyLocation
         fields = ('name', 'code', 'study_location_type', 'treatment_type',
-                  'collecting_location', 'description', 'sort_order')
+                  'collecting_location', 'description', 'sort_order', 'site')

ccdb/locations/viewsets.py

@@ -1,7 +1,18 @@
 from rest_framework import viewsets
 
-from .models import StudyLocation
-from .serializers import StudyLocationSerializer
+from .models import Region, Site, StudyLocation
+from .serializers import (
+    RegionSerializer, SiteSerializer, StudyLocationSerializer)
+
+
+class RegionViewSet(viewsets.ModelViewSet):
+    queryset = Region.objects.all()
+    serializer_class = RegionSerializer
+
+
+class SiteViewSet(viewsets.ModelViewSet):
+    queryset = Site.objects.all()
+    serializer_class = SiteSerializer
 
 
 class StudyLocationViewSet(viewsets.ModelViewSet):