Просмотр исходного кода

Add segment_count property to CablePath

Jeremy Stretch 5 лет назад
Родитель
Сommit
eb5f6167a5
1 измененных файлов с 5 добавлено и 0 удалено
  1. 5 0
      netbox/dcim/models/cables.py

+ 5 - 0
netbox/dcim/models/cables.py

@@ -379,6 +379,11 @@ class CablePath(models.Model):
         model = self.origin._meta.model
         model.objects.filter(pk=self.origin.pk).update(_path=self.pk)
 
+    @property
+    def segment_count(self):
+        total_length = 1 + len(self.path) + (1 if self.destination else 0)
+        return int(total_length / 3)
+
     def get_path(self):
         """
         Return the path as a list of prefetched objects.