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

make inventory compatible

Browse source
This commit is contained in:
Solvik Blum 2021-05-11 21:45:00 +02:00
commit 576eb07dd7
1 changed files with 11 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

19
netbox_agent/inventory.py
View file

@ -58,6 +58,7 @@ class Inventory():
self.lshw = LSHW() self.lshw = LSHW()
def create_netbox_tags(self): def create_netbox_tags(self):
ret = []
for key, tag in INVENTORY_TAG.items(): for key, tag in INVENTORY_TAG.items():
nb_tag = nb.extras.tags.get( nb_tag = nb.extras.tags.get(
name=tag['name'] name=tag['name']
@ -68,6 +69,8 @@ class Inventory():
slug=tag['slug'], slug=tag['slug'],
comments=tag['name'], comments=tag['name'],
) )
ret.append(nb_tag)
return ret
def find_or_create_manufacturer(self, name): def find_or_create_manufacturer(self, name):
if name is None: if name is None:
@ -97,7 +100,7 @@ class Inventory():
logging.info('Tag {tag} is missing, returning empty array.'.format(tag=tag)) logging.info('Tag {tag} is missing, returning empty array.'.format(tag=tag))
items = [] items = []
return items return list(items)
def create_netbox_inventory_item(self, device_id, tags, vendor, name, serial, description): def create_netbox_inventory_item(self, device_id, tags, vendor, name, serial, description):
manufacturer = self.find_or_create_manufacturer(vendor) manufacturer = self.find_or_create_manufacturer(vendor)
@ -152,7 +155,7 @@ class Inventory():
if motherboard.get('serial') not in [x.serial for x in nb_motherboards]: if motherboard.get('serial') not in [x.serial for x in nb_motherboards]:
self.create_netbox_inventory_item( self.create_netbox_inventory_item(
device_id=self.device_id, device_id=self.device_id,
tags=[INVENTORY_TAG['motherboard']['name']], tags=[{'name': INVENTORY_TAG['motherboard']['name']}],
vendor='{}'.format(motherboard.get('vendor', 'N/A')), vendor='{}'.format(motherboard.get('vendor', 'N/A')),
serial='{}'.format(motherboard.get('serial', 'No SN')), serial='{}'.format(motherboard.get('serial', 'No SN')),
name='{}'.format(motherboard.get('name')), name='{}'.format(motherboard.get('name')),
@ -165,7 +168,7 @@ class Inventory():
device=self.device_id, device=self.device_id,
manufacturer=manufacturer.id, manufacturer=manufacturer.id,
discovered=True, discovered=True,
tags=[INVENTORY_TAG['interface']['name']], tags=[{'name': INVENTORY_TAG['interface']['name']}],
name="{}".format(iface['product']), name="{}".format(iface['product']),
serial='{}'.format(iface['serial']), serial='{}'.format(iface['serial']),
description='{} {}'.format(iface['description'], iface['name']) description='{} {}'.format(iface['description'], iface['name'])
@ -198,7 +201,7 @@ class Inventory():
device=self.device_id, device=self.device_id,
manufacturer=manufacturer.id, manufacturer=manufacturer.id,
discovered=True, discovered=True,
tags=[INVENTORY_TAG['cpu']['name']], tags=[{'name': INVENTORY_TAG['cpu']['name']}],
name=cpu['product'], name=cpu['product'],
description='CPU {}'.format(cpu['location']), description='CPU {}'.format(cpu['location']),
# asset_tag=cpu['location'] # asset_tag=cpu['location']
@ -250,7 +253,7 @@ class Inventory():
device=self.device_id, device=self.device_id,
discovered=True, discovered=True,
manufacturer=manufacturer.id if manufacturer else None, manufacturer=manufacturer.id if manufacturer else None,
tags=[INVENTORY_TAG['raid_card']['name']], tags=[{'name': INVENTORY_TAG['raid_card']['name']}],
name='{}'.format(name), name='{}'.format(name),
serial='{}'.format(serial), serial='{}'.format(serial),
description='RAID Card', description='RAID Card',
@ -367,7 +370,7 @@ class Inventory():
_ = nb.dcim.inventory_items.create( _ = nb.dcim.inventory_items.create(
device=self.device_id, device=self.device_id,
discovered=True, discovered=True,
tags=[INVENTORY_TAG['disk']['name']], tags=[{'name': INVENTORY_TAG['disk']['name']}],
name=name, name=name,
serial=disk['SN'], serial=disk['SN'],
part_id=disk['Model'], part_id=disk['Model'],
@ -407,7 +410,7 @@ class Inventory():
device=self.device_id, device=self.device_id,
discovered=True, discovered=True,
manufacturer=manufacturer.id, manufacturer=manufacturer.id,
tags=[INVENTORY_TAG['memory']['name']], tags=[{'name': INVENTORY_TAG['memory']['name']}],
name=name, name=name,
part_id=memory['product'], part_id=memory['product'],
serial=memory['serial'], serial=memory['serial'],
@ -449,7 +452,7 @@ class Inventory():
device=self.device_id, device=self.device_id,
manufacturer=manufacturer.id, manufacturer=manufacturer.id,
discovered=True, discovered=True,
tags=[INVENTORY_TAG['gpu']['name']], tags=[{'name': INVENTORY_TAG['gpu']['name']}],
name=gpu['product'], name=gpu['product'],
description='GPU {}'.format(gpu['product']), description='GPU {}'.format(gpu['product']),
) )