Corrected region serializers

netbox-community · Feb 28, 2017 · 90fe556 · 90fe556
1 parent c015294
commit 90fe556
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/netbox/dcim/api/serializers.py b/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',
         ]
 
 

diff --git a/netbox/dcim/api/views.py b/netbox/dcim/api/views.py
@@ -29,7 +29,7 @@
 # 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()