failback if gateway doesnt exist

ITJamie · Apr 11, 2022 · 817902c · 817902c
1 parent 66d5539
commit 817902c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/netbox_gateways/__init__.py b/netbox_gateways/__init__.py
@@ -5,7 +5,7 @@ class NetBoxGatewayConfig(PluginConfig):
     name = "netbox_gateways"
     verbose_name = " Netbox Gateway"
     description = "Manage simple prefix default gateways"
-    version = "0.2"
+    version = "0.3"
     base_url = "nb_gateways"
 
 

diff --git a/netbox_gateways/template_content.py b/netbox_gateways/template_content.py
@@ -30,7 +30,10 @@ def x_page(self):
             prefix = gateways_obj.prefix
 
         if not gateways_obj:
-            gateways_obj = Gateway.objects.get(prefix=prefix)
+            try:
+                gateways_obj = Gateway.objects.get(prefix=prefix)
+            except: 
+                gateways_obj = None
 
         return self.render(
             "netbox_gateways/ip_card.html",

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="netbox-gateways",
-    version="0.2",
+    version="0.3",
     description="Manage simple prefix default gateways",
     install_requires=[],
     packages=find_packages(),