import pynetbox
import yaml

with open('/etc/netbox_agent.yaml', 'r') as ymlfile:
    # FIXME: validate configuration file
    config = yaml.load(ymlfile)

netbox_instance = pynetbox.api(
    url=config['netbox']['url'],
    token=config['netbox']['token']
)

DATACENTER_LOCATION_DRIVER_FILE = None
DATACENTER_LOCATION = None
DATACENTER_LOCATION_REGEX = None
RACK_LOCATION_DRIVER_FILE = None
RACK_LOCATION = None
RACK_LOCATION_REGEX = None

if config.get('datacenter_location'):
    dc_loc = config.get('datacenter_location')
    DATACENTER_LOCATION_DRIVER_FILE = dc_loc.get('driver_file')
    DATACENTER_LOCATION = dc_loc.get('driver')
    DATACENTER_LOCATION_REGEX = dc_loc.get('regex')

if config.get('rack_location'):
    rack_location = config['rack_location']
    RACK_LOCATION_DRIVER_FILE = rack_location.get('driver_file')
    RACK_LOCATION = rack_location.get('driver')
    RACK_LOCATION_REGEX = rack_location.get('regex')