ENH: Collections DRF (#25)

ccdb/collections_ccdb/serializers.py

@@ -0,0 +1,43 @@
+from rest_framework import serializers
+
+from .models import Collection, CollectionMethod, CollectionType, Flaw
+
+
+class CollectionSerializer(serializers.ModelSerializer):
+    included_serializers = {
+        'project': 'ccdb.projects.serializers.ProjectSerializer',
+        'study_location': 'ccdb.locations.serializers.StudyLocationSerializer',
+        'collection_method':
+            'ccdb.collections_ccdb.serializers.CollectionMethodSerializer',
+        'collection_type':
+            'ccdb.collections_ccdb.serializers.CollectionTypeSerializer',
+        'flaw': 'ccdb.collections_ccdb.serializers.FlawSerializer',
+    }
+
+    class Meta:
+        model = Collection
+        fields = ('id', 'project', 'study_location', 'collection_type',
+                  'collection_method', 'number_of_traps',
+                  'collection_start_date', 'collection_start_time',
+                  'collection_end_date', 'collection_end_time',
+                  'storage_location', 'specimen_state', 'process_type',
+                  'reagent', 'adfg_permit', 'flaw', 'display_name')
+
+
+class CollectionMethodSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = CollectionMethod
+        fields = ('id', 'name', 'code', 'collection_method_class',
+                  'sort_order')
+
+
+class CollectionTypeSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = CollectionType
+        fields = ('id', 'name', 'code', 'sort_order')
+
+
+class FlawSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Flaw
+        fields = ('id', 'name', 'description', 'sort_order')

ccdb/collections_ccdb/viewsets.py

@@ -0,0 +1,25 @@
+from rest_framework import viewsets
+
+from .models import Collection, CollectionMethod, CollectionType, Flaw
+from .serializers import (CollectionSerializer, CollectionMethodSerializer,
+                          CollectionTypeSerializer, FlawSerializer)
+
+
+class CollectionViewSet(viewsets.ModelViewSet):
+    queryset = Collection.objects.all()
+    serializer_class = CollectionSerializer
+
+
+class CollectionMethodViewSet(viewsets.ModelViewSet):
+    queryset = CollectionMethod.objects.all()
+    serializer_class = CollectionMethodSerializer
+
+
+class CollectionTypeViewSet(viewsets.ModelViewSet):
+    queryset = CollectionType.objects.all()
+    serializer_class = CollectionTypeSerializer
+
+
+class FlawViewSet(viewsets.ModelViewSet):
+    queryset = Flaw.objects.all()
+    serializer_class = FlawSerializer