Created a Class that checks if objects exist in netbox. Renamed netbox_dcim.py to netbox_create.py as that class now only creates netbox objects.

Author: khalford

Netbox_CSV_Read/Netbox_Api/netbox_dcim.py renamed to Netbox_CSV_Read/Netbox_Api/netbox_create.py

@@ -4,7 +4,7 @@
 
 class NetboxDCIM:
     """
-    This class contains methods that will interact with the Netbox Api.
+    This class contains methods that will interact create objects in Netbox.
     """
 
     def __init__(self, url: str, token: str, api: Optional = None):
@@ -13,24 +13,6 @@ def __init__(self, url: str, token: str, api: Optional = None):
         else:
             self.netbox = api
 
-    def check_device_exists(self, device_name: str) -> bool:
-        """
-        This method will check if a device exists in Netbox.
-        :param device_name: The name of the device.
-        :return: Returns bool.
-        """
-        device = self.netbox.dcim.devices.get(name=device_name)
-        return bool(device)
-
-    def check_device_type_exists(self, device_type: str) -> bool:
-        """
-        This method will check if a device exists in Netbox.
-        :param device_type: The name of the device.
-        :return: Returns bool.
-        """
-        device_type = self.netbox.dcim.device_types.get(slug=device_type)
-        return bool(device_type)
-
     def create_device(self, data: dict | list) -> bool:
         """
         This method uses the pynetbox Api to create a device in Netbox.

Netbox_CSV_Read/Netbox_Api/netbox_existence.py

@@ -0,0 +1,32 @@
+from typing import Optional
+from Netbox_Api.netbox_connect import NetboxConnect
+
+
+class NetboxExistence:
+    """
+    This class contains methods that check if an object exists in Netbox.
+    """
+
+    def __init__(self, url: str, token: str, api: Optional = None):
+        if not api:
+            self.netbox = NetboxConnect(url, token).api_object()
+        else:
+            self.netbox = api
+
+    def check_device_exists(self, device_name: str) -> bool:
+        """
+        This method will check if a device exists in Netbox.
+        :param device_name: The name of the device.
+        :return: Returns bool.
+        """
+        device = self.netbox.dcim.devices.get(name=device_name)
+        return bool(device)
+
+    def check_device_type_exists(self, device_type: str) -> bool:
+        """
+        This method will check if a device exists in Netbox.
+        :param device_type: The name of the device.
+        :return: Returns bool.
+        """
+        device_type = self.netbox.dcim.device_types.get(slug=device_type)
+        return bool(device_type)

Netbox_CSV_Read/Tests/test_netbox_dcim.py renamed to Netbox_CSV_Read/Tests/test_netbox_create.py

@@ -1,5 +1,5 @@
 from unittest.mock import MagicMock, NonCallableMock
-from Netbox_Api.netbox_dcim import NetboxDCIM
+from Netbox_Api.netbox_create import NetboxDCIM
 import pytest
 
 
@@ -15,28 +15,6 @@ def instance_fixture(api_mock):
     return NetboxDCIM(url, token, api_mock)
 
 
-def test_check_device_exists(instance):
-    """
-    This test ensures the .get method is called once with the correct argument.
-    """
-    mock_device = MagicMock()
-    device_name = NonCallableMock()
-    instance.netbox.dcim.devices = mock_device
-    instance.check_device_exists(device_name)
-    mock_device.get.assert_called_once_with(name=device_name)
-
-
-def test_check_device_type_exists(instance):
-    """
-    This test ensures the .get method is called once with the correct argument.
-    """
-    mock_device_types = MagicMock()
-    device_type = NonCallableMock()
-    instance.netbox.dcim.device_types = mock_device_types
-    instance.check_device_type_exists(device_type)
-    mock_device_types.get.assert_called_once_with(slug=device_type)
-
-
 def test_create_device(instance):
     """
     This test ensures the .create method is called once with the correct argument.

Netbox_CSV_Read/Tests/test_netbox_existence.py

@@ -0,0 +1,37 @@
+from unittest.mock import MagicMock, NonCallableMock
+from Netbox_Api.netbox_existence import NetboxExistence
+import pytest
+
+
+@pytest.fixture(name="api_mock", scope="function")
+def instance_api_fixture():
+    return MagicMock()
+
+
+@pytest.fixture(name="instance")
+def instance_fixture(api_mock):
+    url = "not real url"
+    token = "not real token"
+    return NetboxExistence(url, token, api_mock)
+
+
+def test_check_device_exists(instance):
+    """
+    This test ensures the .get method is called once with the correct argument.
+    """
+    mock_device = MagicMock()
+    device_name = NonCallableMock()
+    instance.netbox.dcim.devices = mock_device
+    instance.check_device_exists(device_name)
+    mock_device.get.assert_called_once_with(name=device_name)
+
+
+def test_check_device_type_exists(instance):
+    """
+    This test ensures the .get method is called once with the correct argument.
+    """
+    mock_device_types = MagicMock()
+    device_type = NonCallableMock()
+    instance.netbox.dcim.device_types = mock_device_types
+    instance.check_device_type_exists(device_type)
+    mock_device_types.get.assert_called_once_with(slug=device_type)