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

exit upon error if no datacenter is found

Browse source
This commit is contained in:
Solvik Blum 2020-07-01 18:16:03 +02:00
commit a940e725cd
1 changed files with 15 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

18
netbox_agent/server.py
View file

@ -1,6 +1,7 @@
import logging import logging
import socket import socket
import subprocess import subprocess
import sys
from pprint import pprint from pprint import pprint
import netbox_agent.dmidecode as dmidecode import netbox_agent.dmidecode as dmidecode
@ -50,11 +51,18 @@ class ServerBase():
def get_netbox_datacenter(self): def get_netbox_datacenter(self):
dc = self.get_datacenter() dc = self.get_datacenter()
if dc: if dc is None:
return nb.dcim.sites.get( logging.error("Specificing a datacenter (Site) is mandatory in Netbox")
sys.exit(1)
nb_dc = nb.dcim.sites.get(
slug=dc, slug=dc,
) )
return None if nb_dc is None:
logging.error("Site (slug: {}) has not been found".format(dc))
sys.exit(1)
return nb_dc
def update_netbox_location(self, server): def update_netbox_location(self, server):
dc = self.get_datacenter() dc = self.get_datacenter()
@ -93,8 +101,8 @@ class ServerBase():
if not rack: if not rack:
return None return None
if rack and not datacenter: if rack and not datacenter:
logging.warning("Can't get rack if no datacenter is configured or found") logging.error("Can't get rack if no datacenter is configured or found")
return None sys.exit(1)
return nb.dcim.racks.get( return nb.dcim.racks.get(
name=rack, name=rack,