diff --git a/netbox/virtualization/models/virtualmachines.py b/netbox/virtualization/models/virtualmachines.py index 970167bf3..aa913cc9e 100644 --- a/netbox/virtualization/models/virtualmachines.py +++ b/netbox/virtualization/models/virtualmachines.py @@ -242,6 +242,10 @@ class VirtualMachine(ContactsMixin, RenderConfigMixin, ConfigContextModel, Prima else: return None + @property + def disk_size(self): + return sum(self.virtualdisks.all().values_list('size', flat=True)) + class VMInterface(NetBoxModel, BaseInterface, TrackingModelMixin): virtual_machine = models.ForeignKey( diff --git a/netbox/virtualization/views.py b/netbox/virtualization/views.py index 486d69057..313df3ee8 100644 --- a/netbox/virtualization/views.py +++ b/netbox/virtualization/views.py @@ -352,8 +352,7 @@ class VirtualMachineListView(generic.ObjectListView): @register_model_view(VirtualMachine) class VirtualMachineView(generic.ObjectView): - queryset = VirtualMachine.objects.all().annotate( - disk_size=Sum('virtualdisks__size')) + queryset = VirtualMachine.objects.all() @register_model_view(VirtualMachine, 'interfaces')