From 89ee93afd188fe23010666b236109d0feb1eac8e Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@digitalocean.com>
Date: Fri, 19 Jan 2018 09:25:16 -0500
Subject: [PATCH] Closes #1824: Add virtual machine count to platforms list

---
 netbox/dcim/tables.py | 10 +++++++---
 netbox/dcim/views.py  |  5 ++++-
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/netbox/dcim/tables.py b/netbox/dcim/tables.py
index e95277704..cb3b1ff3b 100644
--- a/netbox/dcim/tables.py
+++ b/netbox/dcim/tables.py
@@ -381,13 +381,17 @@ class PlatformTable(BaseTable):
     pk = ToggleColumn()
     name = tables.LinkColumn(verbose_name='Name')
     device_count = tables.Column(verbose_name='Devices')
+    vm_count = tables.Column(verbose_name='VMs')
     slug = tables.Column(verbose_name='Slug')
-    actions = tables.TemplateColumn(template_code=PLATFORM_ACTIONS, attrs={'td': {'class': 'text-right'}},
-                                    verbose_name='')
+    actions = tables.TemplateColumn(
+        template_code=PLATFORM_ACTIONS,
+        attrs={'td': {'class': 'text-right'}},
+        verbose_name=''
+    )
 
     class Meta(BaseTable.Meta):
         model = Platform
-        fields = ('pk', 'name', 'device_count', 'slug', 'napalm_driver', 'actions')
+        fields = ('pk', 'name', 'device_count', 'vm_count', 'slug', 'napalm_driver', 'actions')
 
 
 #
diff --git a/netbox/dcim/views.py b/netbox/dcim/views.py
index 9b681e4a7..0dc393cfb 100644
--- a/netbox/dcim/views.py
+++ b/netbox/dcim/views.py
@@ -754,7 +754,10 @@ class DeviceRoleBulkDeleteView(PermissionRequiredMixin, BulkDeleteView):
 #
 
 class PlatformListView(ObjectListView):
-    queryset = Platform.objects.annotate(device_count=Count('devices'))
+    queryset = Platform.objects.annotate(
+        device_count=Count('devices', distinct=True),
+        vm_count=Count('virtual_machines', distinct=True)
+    )
     table = tables.PlatformTable
     template_name = 'dcim/platform_list.html'