shimatta_kenkyusho/api/serializers.py

@@ -24,22 +24,43 @@ class PackageSerializer(serializers.HyperlinkedModelSerializer):
         model = parts_models.Package
         fields = '__all__'
 
-class StorageSerializer(serializers.HyperlinkedModelSerializer):
-    full_path = serializers.ReadOnlyField(source='get_full_path')
-
+class ComponentParameterSerializer(serializers.HyperlinkedModelSerializer):
+    id = serializers.ReadOnlyField()
+    ro_parameter_type = serializers.ReadOnlyField(source='parameter_type.parameter_name')
     class Meta:
-        model = parts_models.Storage
-        fields = ['url', 'id', 'name', 'verbose_name', 'parent_storage', 'responsible', 'template', 'full_path']
+        model = parts_models.ComponentParameter
+        fields = '__all__'
+
+class ComponentDistributorNumSerializer(serializers.HyperlinkedModelSerializer):
+    id = serializers.ReadOnlyField()
+    class Meta:
+        model = parts_models.DistributorNum
+        fields = '__all__'
 
 class ComponentSerializer(serializers.HyperlinkedModelSerializer):
     package_data = PackageSerializerNoLink(source='package', read_only=True)
     ro_manufacturer_name = serializers.ReadOnlyField(source='manufacturer.name')
     ro_image = serializers.ReadOnlyField(source='get_resolved_image')
+    ro_component_type = serializers.ReadOnlyField(source='component_type.class_name')
+    ro_parameters = ComponentParameterSerializer(many=True, source='componentparameter_set', read_only=True)
+    ro_distributor_numbers = ComponentDistributorNumSerializer(many=True, source='distributornum_set', read_only=True)
 
     class Meta:
         model = parts_models.Component
-        fields = ['url', 'id', 'name', 'package_data', 'package', 'pref_distri', 'image', 'manufacturer', 'ro_manufacturer_name', 'ro_image']
-        
+        fields = ['url',
+                  'id',
+                  'name',
+                  'package_data',
+                  'package',
+                  'pref_distri',
+                  'image',
+                  'manufacturer',
+                  'component_type',
+                  'ro_manufacturer_name',
+                  'ro_image',
+                  'ro_component_type',
+                  'ro_parameters',
+                  'ro_distributor_numbers']
 
 class StockSerializer(serializers.HyperlinkedModelSerializer):
     id = serializers.ReadOnlyField()
@@ -51,6 +72,22 @@ class StockSerializer(serializers.HyperlinkedModelSerializer):
         model = parts_models.Stock
         fields = '__all__'
 
+class StockSerializerExpandComponent(StockSerializer):
+    ro_component = ComponentSerializer(read_only=True, source='component')
+
+class StorageSerializer(serializers.HyperlinkedModelSerializer):
+    full_path = serializers.ReadOnlyField(source='get_full_path')
+
+    class Meta:
+        model = parts_models.Storage
+        fields = ['url', 'id', 'name', 'verbose_name', 'parent_storage', 'responsible', 'template', 'full_path']
+
+class StorageSerializerStocksExpanded(StorageSerializer):
+    ro_stocks = StockSerializerExpandComponent(many=True, read_only=True, source='stock_set')
+
+    class Meta(StorageSerializer.Meta):
+        fields = StorageSerializer.Meta.fields + ['ro_stocks']
+
 class DistributorSerializer(serializers.HyperlinkedModelSerializer):
     id = serializers.ReadOnlyField()
     class Meta:

shimatta_kenkyusho/api/urls.py

@@ -9,6 +9,8 @@ router.register(r'groups', GroupViewSet)
 router.register(r'parts/storages', PartsStorageViewSet)
 router.register(r'parts/storage_templates', PartsStorageTemplatesViewSet, basename='storage-template')
 router.register(r'parts/components', PartsComponentViewSet)
+router.register(r'parts/component-parameters', PartsComponentParameterViewSet)
+router.register(r'parts/component-distributor-numbers', PartsComponentDistributorNumViewSet)
 router.register(r'parts/stocks', PartsStockViewSet)
 router.register(r'parts/packages', PartsPackageViewSet)
 router.register(r'parts/distributors', PartsDistributorviewSet)

shimatta_kenkyusho/api/views.py

@@ -42,12 +42,16 @@ class GroupViewSet(viewsets.ReadOnlyModelViewSet):
 
 class PartsStorageViewSet(viewsets.ModelViewSet):
     queryset = parts_models.Storage.objects.all()
-    serializer_class = StorageSerializer
     permission_classes = [permissions.DjangoModelPermissions]
     filter_backends = [django_filters.rest_framework.DjangoFilterBackend]
     search_fields = ['id', 'name', 'parent_storage']
     filterset_fields = ['id', 'name', 'parent_storage']
 
+    def get_serializer_class(self):
+        if self.request.GET.get('expand_stocks'):
+            return StorageSerializerStocksExpanded
+        return StorageSerializer
+
 class PartsStorageTemplatesViewSet(viewsets.ReadOnlyModelViewSet):
     queryset = parts_models.Storage.objects.filter(is_template=True)
     serializer_class = StorageSerializer
@@ -64,6 +68,22 @@ class PartsComponentViewSet(viewsets.ModelViewSet):
     search_fields = ['id', 'name', 'package__name', 'manufacturer__name']
     filterset_fields = ['id', 'name']
 
+class PartsComponentParameterViewSet(viewsets.ModelViewSet):
+    queryset = parts_models.ComponentParameter.objects.all()
+    serializer_class = ComponentParameterSerializer
+    permission_classes = [permissions.DjangoModelPermissions]
+    filter_backends = [filters.SearchFilter, django_filters.rest_framework.DjangoFilterBackend]
+    search_fields = ['id', 'parameter_type__parameter_name']
+    filterset_fields = ['id']
+
+class PartsComponentDistributorNumViewSet(viewsets.ModelViewSet):
+    queryset = parts_models.DistributorNum.objects.all()
+    serializer_class = ComponentDistributorNumSerializer
+    permission_classes = [permissions.DjangoModelPermissions]
+    filter_backends = [filters.SearchFilter, django_filters.rest_framework.DjangoFilterBackend]
+    search_fields = ['id', 'distributor', 'distributor_part_number']
+    filterset_fields = ['id']
+
 class PartsComponentTypeViewSet(viewsets.ModelViewSet):
     queryset = parts_models.ComponentType.objects.all()
     serializer_class = ComponentTypeSerializer