Mercurial > public > sg101
view gpp/ygroup/management/commands/sync_ygroup_threads.py @ 515:ae89ba801e8b
For #194, convert the POTD management command to a celery task.
Refactored to put the logic for the command into a function, and the command simply calls this function. The task can also just call this function. Added some basic tests for the new function.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Wed, 14 Dec 2011 02:41:15 +0000 |
parents | 0c18dfb1da1c |
children |
line wrap: on
line source
""" sync_ygroup_threads.py - A management command to synchronize the yahoo group archives by recomputing the de-normalized fields in the thread objects. """ import optparse from django.core.management.base import NoArgsCommand, CommandError from ygroup.models import Thread, Post import ygroup.views class Command(NoArgsCommand): help = """\ This command synchronizes the ygroup application's thread objects by updating their de-normalized fields. """ option_list = NoArgsCommand.option_list + ( optparse.make_option("-p", "--progress", action="store_true", help="Output a . after every 50 threads to show progress"), ) def handle_noargs(self, **opts): show_progress = opts.get('progress', False) or False n = 0 for thread in Thread.objects.iterator(): thread.post_count = Post.objects.filter(thread=thread).count() thread.page = n / ygroup.views.THREADS_PER_PAGE + 1 thread.save() n += 1 if n % 50 == 0: self.stdout.write('.') self.stdout.flush() self.stdout.write('\n')