32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
import importlib
|
|
import importlib.machinery
|
|
|
|
from netbox_agent.config import DATACENTER_LOCATION, DATACENTER_LOCATION_DRIVER_FILE, \
|
|
DATACENTER_LOCATION_REGEX
|
|
|
|
|
|
class Datacenter():
|
|
"""
|
|
"""
|
|
def __init__(self, *args, **kwargs):
|
|
self.driver = DATACENTER_LOCATION.split(':')[0]
|
|
self.driver_value = DATACENTER_LOCATION.split(':')[1]
|
|
self.driver_file = DATACENTER_LOCATION_DRIVER_FILE
|
|
|
|
if self.driver_file:
|
|
try:
|
|
# FIXME: Works with Python 3.3+, support older version?
|
|
loader = importlib.machinery.SourceFileLoader('driver_file', self.driver_file)
|
|
self.driver = loader.load_module()
|
|
except ImportError:
|
|
raise ImportError("Couldn't import {} as a module".format(self.driver_file))
|
|
else:
|
|
try:
|
|
self.driver = importlib.import_module(
|
|
'netbox_agent.drivers.datacenter_{}'.format(self.driver)
|
|
)
|
|
except ImportError:
|
|
raise ImportError("Driver {} doesn't exists".format(self.driver))
|
|
|
|
def get(self):
|
|
return getattr(self.driver, 'get')(self.driver_value, DATACENTER_LOCATION_REGEX)
|