Add optional external rate limit file.

mysociety · Mar 15, 2024 · ddaa49e · ddaa49e
1 parent 43abf6f
commit ddaa49e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/mapit/ratelimitcache.py b/mapit/ratelimitcache.py
@@ -14,6 +14,11 @@
     }
 _sentinel = object()
 
+CONFIG_FILE = getattr(settings, 'MAPIT_RATE_LIMIT_FILE', {})
+if isinstance(CONFIG_FILE, dict) and CONFIG_FILE.get('internal_ips'):
+    CONFIG['ips'] += CONFIG_FILE['internal_ips']['ipv4']
+    CONFIG['ips'] += CONFIG_FILE['internal_ips']['ipv6']
+
 
 def ratelimit(_f=_sentinel, **kwargs):
     if _f is _sentinel:

diff --git a/project/settings.py b/project/settings.py
@@ -37,6 +37,14 @@
 # excluded from rate limiting. Optional.
 MAPIT_RATE_LIMIT = config.get('RATE_LIMIT', {})
 
+MAPIT_RATE_LIMIT_FILE = config.get('RATE_LIMIT_FILE')
+if MAPIT_RATE_LIMIT_FILE:
+    try:
+        with open(MAPIT_RATE_LIMIT_FILE, 'r') as fp:
+            MAPIT_RATE_LIMIT_FILE = yaml.load(fp, Loader=yaml.SafeLoader)
+    except:
+        MAPIT_RATE_LIMIT_FILE = None
+
 # A GA code for analytics
 GOOGLE_ANALYTICS = config.get('GOOGLE_ANALYTICS', '')