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

handle CPU's first..

Browse source
This commit is contained in:
Thomas Davis 2019-08-29 13:06:48 -07:00
commit c56e6a0bbb
2 changed files with 27 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

57
netbox_agent/inventory.py
View file

@ -59,33 +59,40 @@ class Inventory():
self.lshw = LSHW() self.lshw = LSHW()
def get_cpus(self): def find_or_create_manufacturer(self, name):
model = None if name is None:
nb = None return None
output = subprocess.getoutput('lscpu') manufacturer = nb.dcim.manufacturers.get(
model_re = re.search(r'Model name: (.*)', output) name=name,
if len(model_re.groups()) > 0: )
model = model_re.groups()[0].strip()
socket_re = re.search(r'Socket\(s\): (.*)', output) if not manufacturer:
if len(socket_re.groups()) > 0: manufacturer = nb.dcim.manufacturers.create(
nb = int(socket_re.groups()[0].strip()) name=name,
return nb, model slug=name.lower(),
)
logging.info('Creating missing manufacturer {name}'.format(name=name))
return manufacturer
def create_netbox_cpus(self): def create_netbox_cpus(self):
nb_cpus, model = self.get_cpus() for i in self.get_hw_linux("cpu")
for i in range(nb_cpus): manufacturer = self.find_or_create_manufacturer(self, cpu["vendor"])
_ = nb.dcim.inventory_items.create( _ = nb.dcim.inventory_items.create(
device=self.device_id, device=self.device_id,
tags=[INVENTORY_TAG['cpu']['name']], manufacturer = manufacturer.id
name=model,
discovered=True, discovered=True,
tags=[INVENTORY_TAG['cpu']['name']],
name=cpu['product'],
description='CPU', description='CPU',
asset_tag=cpu['location']
) )
logging.info('Creating CPU model {model}'.format(model=model))
logging.info('Creating CPU model {}'.format(cpu['product']))
def update_netbox_cpus(self): def update_netbox_cpus(self):
cpus_number, model = self.get_cpus() cpus_number, model = self.get_hw_linux(cpu))
nb_cpus = nb.dcim.inventory_items.filter( nb_cpus = nb.dcim.inventory_items.filter(
device_id=self.device_id, device_id=self.device_id,
tag=INVENTORY_TAG['cpu']['slug'], tag=INVENTORY_TAG['cpu']['slug'],
@ -119,20 +126,6 @@ class Inventory():
) )
return raid_cards return raid_cards
def find_or_create_manufacturer(self, name):
if name is None:
return None
manufacturer = nb.dcim.manufacturers.get(
name=name,
)
if not manufacturer:
manufacturer = nb.dcim.manufacturers.create(
name=name,
slug=name.lower(),
)
logging.info('Creating missing manufacturer {name}'.format(name=name))
return manufacturer
def create_netbox_raid_card(self, raid_card): def create_netbox_raid_card(self, raid_card):
manufacturer = self.find_or_create_manufacturer( manufacturer = self.find_or_create_manufacturer(
raid_card.get_manufacturer() raid_card.get_manufacturer()
@ -299,7 +292,7 @@ class Inventory():
for memory in self.get_memory(): for memory in self.get_memory():
self.create_netbox_memory(memory) self.create_netbox_memory(memory)
def update_netbox_memory(self): def update_netbox_memories(self):
memories = self.get_memory() memories = self.get_memory()
nb_memories = self.get_netbox_memory() nb_memories = self.get_netbox_memory()

1
netbox_agent/lshw.py
View file

@ -94,6 +94,7 @@ class LSHW():
c["product"] = obj["product"] c["product"] = obj["product"]
c["vendor"] = obj["vendor"] c["vendor"] = obj["vendor"]
c["description"] = obj["description"] c["description"] = obj["description"]
c["location"] = obj["slot"]
self.cpus.append(c) self.cpus.append(c)