Hosterra/netbox-agent
Hosterra/netbox-agent
1
0
Fork 0
Code Issues Pull requests Activity

chore: cleanup virtual_machines get_device_platform

Browse source 
Signed-off-by: Cyril Levis <git@levis.name>
This commit is contained in:
Cyril Levis 2022-03-31 21:04:32 +02:00
commit 3dbeb5b9de
No known key found for this signature in database
GPG key ID: 6DB88737C11F5A48
1 changed files with 6 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

12
netbox_agent/virtualmachine.py
View file

@ -31,6 +31,7 @@ class VirtualMachine(object):
else: else:
self.dmi = dmidecode.parse() self.dmi = dmidecode.parse()
self.network = None self.network = None
self.device_platform = get_device_platform(config.device.platform)
self.tags = list(set(config.device.tags.split(','))) if config.device.tags else [] self.tags = list(set(config.device.tags.split(','))) if config.device.tags else []
if self.tags and len(self.tags): if self.tags and len(self.tags):
@ -90,13 +91,12 @@ class VirtualMachine(object):
if not vm: if not vm:
logging.debug('Creating Virtual machine..') logging.debug('Creating Virtual machine..')
cluster = self.get_netbox_cluster(config.virtual.cluster_name) cluster = self.get_netbox_cluster(config.virtual.cluster_name)
device_platform = get_device_platform(config)
vm = nb.virtualization.virtual_machines.create( vm = nb.virtualization.virtual_machines.create(
name=hostname, name=hostname,
cluster=cluster.id, cluster=cluster.id,
platform=device_platform.id if device_platform is not None else None, platform=self.device_platform,
device_platform=device_platform, device_platform=self.device_platform,
vcpus=vcpus, vcpus=vcpus,
memory=memory, memory=memory,
tenant=tenant.id if tenant else None, tenant=tenant.id if tenant else None,
@ -117,11 +117,9 @@ class VirtualMachine(object):
if sorted(set(vm.tags)) != sorted(set(self.tags)): if sorted(set(vm.tags)) != sorted(set(self.tags)):
vm.tags = self.tags vm.tags = self.tags
updated += 1 updated += 1
if get_device_platform(config) is not None: if vm.platform != self.device_platform:
if vm.platform != get_device_platform(config).name: vm.platform = self.device_platform
updated += 1 updated += 1
vm.platform = get_device_platform(config).id
logging.debug('Finished updating Platform!')
if updated: if updated:
vm.save() vm.save()