Catch pynetbox.RequestError

salt/modules/netbox.py

@@ -31,7 +31,6 @@
 
 try:
     import pynetbox
-    from pynetbox.core.query import RequestError
     HAS_PYNETBOX = True
 except ImportError:
     HAS_PYNETBOX = False
@@ -102,7 +101,7 @@ def _add(app, endpoint, payload):
     nb = _nb_obj(auth_required=True)
     try:
         return getattr(getattr(nb, app), endpoint).create(**payload)
-    except RequestError as e:
+    except pynetbox.RequestError as e:
         log.error("{}, {}, {}".format(e.req.request.headers, e.request_body, e.error))
         return False
 
@@ -378,7 +377,7 @@ def create_device(name,
             return False
 
         status = {'label': "Active", 'value': 1}
-    except RequestError as e:
+    except pynetbox.RequestError as e:
         log.error("{}, {}, {}".format(e.req.request.headers, e.request_body, e.error))
         return False
 
@@ -415,7 +414,7 @@ def update_device(name, **kwargs):
     try:
         nb_device.save()
         return {'dcim': {'devices': kwargs}}
-    except RequestError as e:
+    except pynetbox.RequestError as e:
         log.error("{}, {}, {}".format(e.req.request.headers, e.request_body, e.error))
         return False
 
@@ -797,7 +796,7 @@ def update_interface(device_name, interface_name, **kwargs):
         try:
             nb_interface.save()
             return {'dcim': {'interfaces': {nb_interface.id: dict(nb_interface)}}}
-        except RequestError as e:
+        except pynetbox.RequestError as e:
             log.error("{}, {}, {}".format(e.req.request.headers, e.request_body, e.error))
             return False