mirror of
https://github.com/netbox-community/netbox.git
synced 2025-12-25 13:46:26 +00:00
26 lines
855 B
Python
26 lines
855 B
Python
from django.db.models import Sum
|
|
from django.db.models.signals import post_delete, post_save
|
|
from django.dispatch import receiver
|
|
|
|
from .models import Cluster, VirtualDisk, VirtualMachine
|
|
|
|
|
|
@receiver((post_delete, post_save), sender=VirtualDisk)
|
|
def update_virtualmachine_disk(instance, **kwargs):
|
|
"""
|
|
When a VirtualDisk has been modified, update the aggregate disk_size value of its VM.
|
|
"""
|
|
vm = instance.virtual_machine
|
|
VirtualMachine.objects.filter(pk=vm.pk).update(
|
|
disk=vm.virtualdisks.aggregate(Sum('size'))['size__sum']
|
|
)
|
|
|
|
|
|
@receiver(post_save, sender=Cluster)
|
|
def update_virtualmachine_site(instance, **kwargs):
|
|
"""
|
|
Update the assigned site for all VMs to match that of the Cluster (if any).
|
|
"""
|
|
if instance._site:
|
|
VirtualMachine.objects.filter(cluster=instance).update(site=instance._site)
|