Mercurial > public > sg101
annotate accounts/management/commands/rate_limit_clear.py @ 631:f36d1a168be7
For issue 27, disable login dialog button during POST.
This seems to prevent multiple logins most of the time. You can
still bang on the enter key and sometimes get more through.
| author | Brian Neal <bgneal@gmail.com> |
|---|---|
| date | Wed, 14 Nov 2012 20:57:05 -0600 |
| parents | ee87ea74d46b |
| children |
| rev | line source |
|---|---|
| bgneal@506 | 1 """ |
| bgneal@506 | 2 The rate_limit_clear command is used to clear IP addresses out from our rate |
| bgneal@506 | 3 limit protection database. |
| bgneal@506 | 4 |
| bgneal@506 | 5 """ |
| bgneal@506 | 6 from optparse import make_option |
| bgneal@506 | 7 import re |
| bgneal@506 | 8 |
| bgneal@506 | 9 from django.core.management.base import BaseCommand |
| bgneal@506 | 10 import redis |
| bgneal@506 | 11 |
| bgneal@508 | 12 from core.services import get_redis_connection |
| bgneal@508 | 13 |
| bgneal@508 | 14 |
| bgneal@506 | 15 IP_RE = re.compile(r'^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$') |
| bgneal@506 | 16 |
| bgneal@506 | 17 |
| bgneal@506 | 18 class Command(BaseCommand): |
| bgneal@506 | 19 help = """Remove IP addresses from the rate limit protection datastore.""" |
| bgneal@506 | 20 option_list = list(BaseCommand.option_list) + [ |
| bgneal@506 | 21 make_option("--purge", action="store_true", |
| bgneal@506 | 22 help="Purge all IP addresses"), |
| bgneal@506 | 23 ] |
| bgneal@506 | 24 |
| bgneal@506 | 25 def handle(self, *args, **kwargs): |
| bgneal@506 | 26 try: |
| bgneal@508 | 27 con = get_redis_connection() |
| bgneal@506 | 28 |
| bgneal@506 | 29 # get all rate-limit keys |
| bgneal@506 | 30 keys = con.keys('rate-limit-*') |
| bgneal@506 | 31 |
| bgneal@506 | 32 # if purging, delete them all... |
| bgneal@506 | 33 if kwargs['purge']: |
| bgneal@506 | 34 if keys: |
| bgneal@506 | 35 con.delete(*keys) |
| bgneal@506 | 36 return |
| bgneal@506 | 37 |
| bgneal@506 | 38 # otherwise delete the ones the user asked for |
| bgneal@506 | 39 ips = [] |
| bgneal@506 | 40 for ip in args: |
| bgneal@506 | 41 if IP_RE.match(ip): |
| bgneal@506 | 42 key = 'rate-limit-%s' % ip |
| bgneal@506 | 43 if key in keys: |
| bgneal@506 | 44 ips.append(key) |
| bgneal@506 | 45 else: |
| bgneal@506 | 46 self.stdout.write('%s not found\n' % ip) |
| bgneal@506 | 47 else: |
| bgneal@506 | 48 self.stderr.write('invalid IP address %s\n' % ip) |
| bgneal@506 | 49 |
| bgneal@506 | 50 if ips: |
| bgneal@506 | 51 con.delete(*ips) |
| bgneal@506 | 52 |
| bgneal@506 | 53 except redis.RedisError, e: |
| bgneal@506 | 54 self.stderr.write('%s\n' % e) |