Merge pull request #1 from sterapps/changeable-ip-list-settings

Make ip list setting changeable
anexia · Mar 28, 2023 · baaba56 · baaba56
2 parents 1a73d6c + dc74de5
commit baaba56
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/drf_ip_restrictions/permissions.py b/drf_ip_restrictions/permissions.py
@@ -7,12 +7,14 @@ class AllowedIpList(object):
     Ensure the request's IP address is on the ip white list configured in Django settings.
     """
 
+    setting_ip_list = "ALLOWED_IP_LIST"
+
     def has_permission(self, request, view):
         client_ip, is_routable = get_client_ip(request)
 
         if client_ip:
             settings = ip_restriction_settings()
-            allowed_ips = settings.ALLOWED_IP_LIST
+            allowed_ips = getattr(settings, self.setting_ip_list)
             for allowed_ip in allowed_ips:
                 if client_ip == allowed_ip or client_ip.startswith(allowed_ip):
                     return True

diff --git a/tests/testapp/migrations/0001_initial.py b/tests/testapp/migrations/0001_initial.py
@@ -4,7 +4,6 @@
 
 
 class Migration(migrations.Migration):
-
     initial = True
 
     dependencies = []
Original file line number	Diff line number	Diff line change
Expand Up		@@ -4,7 +4,6 @@


		class Migration(migrations.Migration):

		initial = True

		dependencies = []
Expand Down