Mercurial > public > sg101
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)