Mercurial > public > sg101
annotate accounts/management/commands/rate_limit_clear.py @ 661:15dbe0ccda95
Prevent exceptions when viewing downloads in the admin when the file
doesn't exist on the filesystem. This is usually seen in development
but can also happen in production if the file is missing.
| author | Brian Neal <bgneal@gmail.com> |
|---|---|
| date | Tue, 14 May 2013 21:02:47 -0500 |
| 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) |