Mercurial > public > sg101
annotate gpp/ygroup/management/commands/sync_ygroup_threads.py @ 334:6805d15cda13
Adding a script I had to write on the fly to filter out posts from the posts csv file that had no parent topics. MyISAM let me get away with that, but InnoDB won't.
| author | Brian Neal <bgneal@gmail.com> |
|---|---|
| date | Sat, 26 Feb 2011 01:28:22 +0000 |
| parents | 0c18dfb1da1c |
| children |
| rev | line source |
|---|---|
| bgneal@323 | 1 """ |
| bgneal@323 | 2 sync_ygroup_threads.py - A management command to synchronize the yahoo group |
| bgneal@323 | 3 archives by recomputing the de-normalized fields in the thread objects. |
| bgneal@323 | 4 |
| bgneal@323 | 5 """ |
| bgneal@323 | 6 import optparse |
| bgneal@323 | 7 |
| bgneal@323 | 8 from django.core.management.base import NoArgsCommand, CommandError |
| bgneal@323 | 9 |
| bgneal@323 | 10 from ygroup.models import Thread, Post |
| bgneal@323 | 11 import ygroup.views |
| bgneal@323 | 12 |
| bgneal@323 | 13 |
| bgneal@323 | 14 class Command(NoArgsCommand): |
| bgneal@323 | 15 help = """\ |
| bgneal@323 | 16 This command synchronizes the ygroup application's thread objects |
| bgneal@323 | 17 by updating their de-normalized fields. |
| bgneal@323 | 18 """ |
| bgneal@323 | 19 option_list = NoArgsCommand.option_list + ( |
| bgneal@323 | 20 optparse.make_option("-p", "--progress", action="store_true", |
| bgneal@323 | 21 help="Output a . after every 50 threads to show progress"), |
| bgneal@323 | 22 ) |
| bgneal@323 | 23 |
| bgneal@323 | 24 def handle_noargs(self, **opts): |
| bgneal@323 | 25 |
| bgneal@323 | 26 show_progress = opts.get('progress', False) or False |
| bgneal@323 | 27 |
| bgneal@323 | 28 n = 0 |
| bgneal@323 | 29 for thread in Thread.objects.iterator(): |
| bgneal@323 | 30 thread.post_count = Post.objects.filter(thread=thread).count() |
| bgneal@323 | 31 thread.page = n / ygroup.views.THREADS_PER_PAGE + 1 |
| bgneal@323 | 32 thread.save() |
| bgneal@323 | 33 n += 1 |
| bgneal@323 | 34 if n % 50 == 0: |
| bgneal@323 | 35 self.stdout.write('.') |
| bgneal@323 | 36 self.stdout.flush() |
| bgneal@323 | 37 |
| bgneal@323 | 38 self.stdout.write('\n') |
| bgneal@323 | 39 |