view ygroup/management/commands/sync_ygroup_threads.py @ 629:f4c043cf55ac

Wiki integration. Requests don't always have sessions. In particular this occurs when a request is made without a trailing slash. The Common middleware redirects when this happens, and the middleware process_request() processing stops before a session can get added. So just set an attribute on the request object for each operation. This seemed weird to me at first, but there are plenty of examples of this in the Django code base already.
author Brian Neal <bgneal@gmail.com>
date Tue, 13 Nov 2012 13:50:06 -0600
parents ee87ea74d46b
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')