netbox_agent.yaml.example

@@ -3,8 +3,10 @@ netbox:
  token: supersecrettoken
 
 datacenter_location:
- # driver_file: /opt/netbox_driver_dc.py
+ driver: "cmd:cat /etc/qualification | tr [a-z] [A-Z]"
- driver: file:/etc/qualification
+ regex: "DATACENTER: (?P<datacenter>[A-Za-z0-9]+)"
- regex: "datacenter: (?P<datacenter>[A-Za-z0-9]+)"
 # driver: 'cmd:lldpctl'
-# regex = 'SysName: .*\.(?P<datacenter>[A-Za-z0-9]+)'
+# regex: 'SysName: .*\.([A-Za-z0-9]+)'
+#
+# driver: "file:/tmp/datacenter"
+# regex: "(.*)"

netbox_agent/datacenter.py

@@ -10,7 +10,7 @@ class Datacenter():
     This class is used to guess the datacenter in order to push the information
     in Netbox for a `Device`
 
-    A driver takes a `value` and evaluates a regex with a `named group`: `datacenter`.
+    A driver takes a `value` and evaluates a regex with a `capture group`.
 
     There's embeded drivers such as `file` or `cmd` which read a file or return the
     output of a file.
@@ -20,7 +20,7 @@ class Datacenter():
     """
     def __init__(self, *args, **kwargs):
         self.driver = DATACENTER_LOCATION.split(':')[0]
-        self.driver_value = DATACENTER_LOCATION.split(':')[1]
+        self.driver_value = ':'.join(DATACENTER_LOCATION.split(':')[1:])
         self.driver_file = DATACENTER_LOCATION_DRIVER_FILE
 
         if self.driver_file:

netbox_agent/drivers/datacenter_cmd.py

@@ -5,7 +5,6 @@ import subprocess
 def get(value, regex):
     output = subprocess.getoutput(value)
     r = re.search(regex, output)
-    if r:
+    if r and len(r.groups()) > 0:
-        result = r.group('datacenter')
+        return r.groups()[0]
-        return result
     return None

netbox_agent/drivers/datacenter_file.py

@@ -4,6 +4,6 @@ import re
 def get(value, regex):
     for line in open(value, 'r'):
         r = re.search(regex, line)
-        if r:
+        if r and len(r.groups()) > 0:
-            return r.group('datacenter')
+            return r.groups()[0]
     return None