add scripts/netbox-dump

scripts/netbox-dump

@@ -0,0 +1,152 @@
+#!/usr/bin/env python3
+
+from json import dump
+from os import environ
+from os.path import dirname, join
+from sys import exit
+
+import bwpass
+from requests import post
+
+from bundlewrap.utils.text import validate_name
+
+TOKEN = environ.get("NETBOX_AUTH_TOKEN")
+
+# editorconfig-checker-disable
+QUERY = """{
+  device_list(tag: "bundlewrap") {
+    name
+    site {
+      id
+    }
+    interfaces {
+      id
+      name
+      enabled
+      description
+      mode
+      untagged_vlan {
+        name
+      }
+      tagged_vlans {
+        name
+      }
+      link_peers {
+        ... on InterfaceType {
+          name
+          device {
+            name
+          }
+        }
+        ... on FrontPortType {
+          name
+          device {
+            name
+          }
+        }
+      }
+      connected_endpoints {
+        ... on InterfaceType {
+          name
+          device {
+            name
+          }
+        }
+      }
+    }
+  }
+  site_list {
+    id
+    vlans {
+      name
+      vid
+    }
+  }
+}"""
+# editorconfig-checker-enable
+
+if not TOKEN:
+    try:
+        TOKEN = bwpass.attr("netbox.franzi.business/kunsi", "token")
+    except Exception:
+        print("NETBOX_AUTH_TOKEN is missing")
+        exit(1)
+
+r = post(
+    "https://netbox.franzi.business/graphql/",
+    headers={
+        "Accept": "application/json",
+        "Authorization": f"Token {TOKEN}",
+    },
+    json={
+        "query": QUERY,
+    },
+)
+r.raise_for_status()
+
+data = r.json()["data"]
+
+site_vlans = {site["id"]: site["vlans"] for site in data["site_list"]}
+
+for device in data["device_list"]:
+    if not device["name"] or not validate_name(device["name"]):
+        # invalid node name, ignore
+        continue
+
+    result = {
+        "interfaces": {},
+        "vlans": site_vlans[device["site"]["id"]],
+    }
+
+    for interface in device["interfaces"]:
+        description = ""
+        peers = None
+
+        if interface["connected_endpoints"]:
+            peers = interface["connected_endpoints"]
+        elif interface["link_peers"]:
+            peers = interface["link_peers"]
+
+        if interface["description"]:
+            description = interface["description"]
+        elif peers:
+            peer_list = set()
+
+            for i in peers:
+                peer_list.add(
+                    "{} ({})".format(
+                        i["device"]["name"],
+                        i["name"],
+                    )
+                )
+
+            description = "; ".join(sorted(peer_list))
+        else:
+            description = ""
+
+        assert description.isascii()
+
+        result["interfaces"][interface["name"]] = {
+            "description": description,
+            "enabled": interface["enabled"],
+            "mode": interface["mode"],
+            "untagged_vlan": interface["untagged_vlan"]["name"]
+            if interface["untagged_vlan"]
+            else None,
+            "tagged_vlans": sorted({v["name"] for v in interface["tagged_vlans"]}),
+        }
+
+    with open(
+        join(
+            dirname(dirname(__file__)),
+            "configs",
+            "netbox_device_{}.json".format(device["name"]),
+        ),
+        "w+",
+    ) as f:
+        dump(
+            result,
+            f,
+            indent=4,
+            sort_keys=True,
+        )