Bugfix: Add rule in admin was throwing an exception

sztamas · Aug 30, 2020 · 914e6de · 914e6de
1 parent df90b18
commit 914e6de
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/iprestrict/models.py b/iprestrict/models.py
@@ -207,8 +207,12 @@ class Meta:
     rank = models.IntegerField(blank=True)
 
     def __init__(self, *args, **kwargs):
-        super(Rule, self).__init__(*args, **kwargs)
-        self.ip_group = typed_ip_group(self.ip_group)
+        super().__init__(*args, **kwargs)
+        # TODO review this code, it fails without the try except when using Add rule in admin
+        try:
+            self.ip_group = typed_ip_group(self.ip_group)
+        except IPGroup.DoesNotExist:
+            pass
 
     @property
     def regex(self):

diff --git a/testsites/testsite_31/testsite_31/settings.py b/testsites/testsite_31/testsite_31/settings.py
@@ -32,6 +32,7 @@
 
 INSTALLED_APPS = [
     'iprestrict',
+    'django_extensions',
     'polls.apps.PollsConfig',
     'django.contrib.admin',
     'django.contrib.auth',