DGNum/netbox-agent
6
0
Fork 0
Code Issues 28 Pull requests 9 Projects Releases 14 Packages Wiki Activity Actions

check for get() function

Browse source
This commit is contained in:
Solvik Blum 2019-08-04 19:18:11 +02:00
parent 0e0ceb7811
commit 87939d3e35
No known key found for this signature in database
GPG key ID: CC12B3DC262B6C47
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
netbox_agent/datacenter.py
View file

@ -39,4 +39,6 @@ class Datacenter():
raise ImportError("Driver {} doesn't exists".format(self.driver)) raise ImportError("Driver {} doesn't exists".format(self.driver))
def get(self): def get(self):
if not hasattr(self.driver, 'get'):
raise Exception("Your driver {} doesn't have a get() function, please fix it".format(self.driver))
return getattr(self.driver, 'get')(self.driver_value, DATACENTER_LOCATION_REGEX) return getattr(self.driver, 'get')(self.driver_value, DATACENTER_LOCATION_REGEX)