Mercurial > public > sg101
annotate forums/management/commands/sync_forums.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@294 | 1 """ |
| bgneal@294 | 2 sync_forums.py - A management command to synchronize the forums by recomputing |
| bgneal@294 | 3 the de-normalized fields in the forum and topic objects. |
| bgneal@294 | 4 |
| bgneal@294 | 5 """ |
| bgneal@294 | 6 import optparse |
| bgneal@294 | 7 |
| bgneal@294 | 8 from django.core.management.base import NoArgsCommand, CommandError |
| bgneal@294 | 9 |
| bgneal@294 | 10 from forums.models import Forum |
| bgneal@294 | 11 from forums.models import Topic |
| bgneal@294 | 12 |
| bgneal@294 | 13 |
| bgneal@294 | 14 class Command(NoArgsCommand): |
| bgneal@294 | 15 help = """\ |
| bgneal@294 | 16 This command synchronizes the forum application's forums and topic objects |
| bgneal@294 | 17 by updating their de-normalized fields. |
| bgneal@294 | 18 """ |
| bgneal@294 | 19 option_list = NoArgsCommand.option_list + ( |
| bgneal@294 | 20 optparse.make_option("-p", "--progress", action="store_true", |
| bgneal@294 | 21 help="Output a . after every 50 topics to show progress"), |
| bgneal@294 | 22 ) |
| bgneal@294 | 23 |
| bgneal@294 | 24 def handle_noargs(self, **opts): |
| bgneal@294 | 25 |
| bgneal@294 | 26 show_progress = opts.get('progress', False) or False |
| bgneal@294 | 27 |
| bgneal@294 | 28 n = 0 |
| bgneal@294 | 29 for topic in Topic.objects.iterator(): |
| bgneal@294 | 30 topic.post_count_update() |
| bgneal@294 | 31 topic.save() |
| bgneal@294 | 32 n += 1 |
| bgneal@294 | 33 if n % 50 == 0: |
| bgneal@294 | 34 self.stdout.write('.') |
| bgneal@294 | 35 self.stdout.flush() |
| bgneal@294 | 36 |
| bgneal@294 | 37 for forum in Forum.objects.all(): |
| bgneal@294 | 38 forum.sync() |
| bgneal@294 | 39 forum.save() |
| bgneal@294 | 40 |
| bgneal@294 | 41 self.stdout.write('\n') |
| bgneal@294 | 42 |