Browse Source

Corrected region serializers

Jeremy Stretch 8 years ago
parent
commit
90fe556e5f
2 changed files with 7 additions and 6 deletions
  1. 4 4
      netbox/dcim/api/serializers.py
  2. 3 2
      netbox/dcim/api/views.py

+ 4 - 4
netbox/dcim/api/serializers.py

@@ -30,7 +30,7 @@ class RegionSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = Region
-        fields = ['id', 'url', 'name', 'slug', 'parent']
+        fields = ['id', 'name', 'slug', 'parent']
 
 
 class WritableRegionSerializer(serializers.ModelSerializer):
@@ -51,9 +51,9 @@ class SiteSerializer(CustomFieldModelSerializer):
     class Meta:
         model = Site
         fields = [
-            'id', 'name', 'slug', 'tenant', 'facility', 'asn', 'physical_address', 'shipping_address', 'contact_name',
-            'contact_phone', 'contact_email', 'comments', 'custom_fields', 'count_prefixes', 'count_vlans',
-            'count_racks', 'count_devices', 'count_circuits',
+            'id', 'name', 'slug', 'region', 'tenant', 'facility', 'asn', 'physical_address', 'shipping_address',
+            'contact_name', 'contact_phone', 'contact_email', 'comments', 'custom_fields', 'count_prefixes',
+            'count_vlans', 'count_racks', 'count_devices', 'count_circuits',
         ]
 
 

+ 3 - 2
netbox/dcim/api/views.py

@@ -29,7 +29,7 @@ from . import serializers
 # Regions
 #
 
-class RegionViewSet(WritableSerializerMixin, CustomFieldModelViewSet):
+class RegionViewSet(WritableSerializerMixin, ModelViewSet):
     queryset = Region.objects.all()
     serializer_class = serializers.RegionSerializer
     write_serializer_class = serializers.WritableRegionSerializer
@@ -40,8 +40,9 @@ class RegionViewSet(WritableSerializerMixin, CustomFieldModelViewSet):
 #
 
 class SiteViewSet(WritableSerializerMixin, CustomFieldModelViewSet):
-    queryset = Site.objects.select_related('tenant')
+    queryset = Site.objects.select_related('region', 'tenant')
     serializer_class = serializers.SiteSerializer
+    filter_class = filters.SiteFilter
     write_serializer_class = serializers.WritableSiteSerializer
 
     @detail_route()