Parcourir la source

Closes #1141: Include VRF name and RD in form selections

Jeremy Stretch il y a 7 ans
Parent
commit
8bcd8c404d
2 fichiers modifiés avec 8 ajouts et 2 suppressions
  1. 1 1
      netbox/ipam/api/serializers.py
  2. 7 1
      netbox/ipam/models.py

+ 1 - 1
netbox/ipam/api/serializers.py

@@ -22,7 +22,7 @@ class VRFSerializer(CustomFieldModelSerializer):
 
     class Meta:
         model = VRF
-        fields = ['id', 'name', 'rd', 'tenant', 'enforce_unique', 'description', 'custom_fields']
+        fields = ['id', 'name', 'rd', 'tenant', 'enforce_unique', 'description', 'display_name', 'custom_fields']
 
 
 class NestedVRFSerializer(serializers.ModelSerializer):

+ 7 - 1
netbox/ipam/models.py

@@ -97,7 +97,7 @@ class VRF(CreatedUpdatedModel, CustomFieldModel):
         verbose_name_plural = 'VRFs'
 
     def __str__(self):
-        return self.name
+        return self.display_name or super(VRF, self).__str__()
 
     def get_absolute_url(self):
         return reverse('ipam:vrf', args=[self.pk])
@@ -111,6 +111,12 @@ class VRF(CreatedUpdatedModel, CustomFieldModel):
             self.description,
         ])
 
+    @property
+    def display_name(self):
+        if self.name and self.rd:
+            return "{} ({})".format(self.name, self.rd)
+        return None
+
 
 @python_2_unicode_compatible
 class RIR(models.Model):