diff accounts/management/commands/rate_limit_clear.py @ 581:ee87ea74d46b

For Django 1.4, rearranged project structure for new manage.py.
author Brian Neal <bgneal@gmail.com>
date Sat, 05 May 2012 17:10:48 -0500
parents gpp/accounts/management/commands/rate_limit_clear.py@6f5fff924877
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/accounts/management/commands/rate_limit_clear.py	Sat May 05 17:10:48 2012 -0500
@@ -0,0 +1,54 @@
+"""
+The rate_limit_clear command is used to clear IP addresses out from our rate
+limit protection database.
+
+"""
+from optparse import make_option
+import re
+
+from django.core.management.base import BaseCommand
+import redis
+
+from core.services import get_redis_connection
+
+
+IP_RE = re.compile(r'^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$')
+
+
+class Command(BaseCommand):
+    help = """Remove IP addresses from the rate limit protection datastore."""
+    option_list = list(BaseCommand.option_list) + [
+        make_option("--purge", action="store_true",
+            help="Purge all IP addresses"),
+    ]
+
+    def handle(self, *args, **kwargs):
+        try:
+            con = get_redis_connection()
+
+            # get all rate-limit keys
+            keys = con.keys('rate-limit-*')
+
+            # if purging, delete them all...
+            if kwargs['purge']:
+                if keys:
+                    con.delete(*keys)
+                return
+
+            # otherwise delete the ones the user asked for
+            ips = []
+            for ip in args:
+                if IP_RE.match(ip):
+                    key = 'rate-limit-%s' % ip
+                    if key in keys:
+                        ips.append(key)
+                    else:
+                        self.stdout.write('%s not found\n' % ip)
+                else:
+                    self.stderr.write('invalid IP address %s\n' % ip)
+
+            if ips:
+                con.delete(*ips)
+
+        except redis.RedisError, e:
+            self.stderr.write('%s\n' % e)