Update netbox_connection.py

Refactor Netbox connection handling and improve error logging

Author: LoH-lu

netbox_connection.py

@@ -11,15 +11,28 @@ def connect_to_netbox(url, token):
 
     Returns:
     - netbox (pynetbox.core.api.Api): The Netbox API object configured to use the provided URL and token.
+
+    Raises:
+    - Exception: If the connection to Netbox fails.
     """
     # Create a custom requests session with SSL verification disabled
     session = requests.Session()
     session.verify = False  # Disabling SSL verification for the session
 
-    # Create a Netbox API object without specifying any session
-    netbox = pynetbox.api(url, token)
-
-    # Set the custom session for the Netbox API object's requests session
-    netbox.http_session = session
+    # Test the connection by making a direct request to /api/status/
+    try:
+        headers = {"Authorization": f"Token {token}"}
+        response = session.get(f"{url}/api/status/", headers=headers)
+        response.raise_for_status()  # Raise an error for HTTP status codes 4xx/5xx
 
-    return netbox
+        # Check if the response contains the "netbox-version" key
+        status_data = response.json()
+        if "netbox-version" in status_data:
+            # Create and return the Netbox API object
+            netbox = pynetbox.api(url, token)
+            netbox.http_session = session
+            return netbox
+        else:
+            raise Exception("Unexpected response from Netbox /api/status/ endpoint.")
+    except Exception as e:
+        raise Exception(f"Failed to connect to Netbox: {e}")