"""fix_answer_counts management command
to run type (on the command line:)

python manage.py fix_answer_counts
"""
from django.core.management import BaseCommand
from django.db.models import signals
from askbot import models
from askbot.utils.console import ProgressBar

class Command(BaseCommand):
    """Command class for "fix_answer_counts"
    """

    def remove_save_signals(self):
        """removes signals on model pre-save and
        post-save, so that there are no side-effects
        besides actually updating the answer counts
        """
        signals.pre_save.receivers = []
        signals.post_save.receivers = []

    def handle(self, *arguments, **options):
        """function that handles the command job
        """
        self.remove_save_signals()
        threads = models.Thread.objects.all()
        count = threads.count()
        message = 'Fixing answer counts'
        for thread in ProgressBar(threads.iterator(), count, message):
            thread.update_answer_count()
