import logging

from core.redis import start_job_async_or_sync
from django.core.management.base import BaseCommand
from projects.models import Project
from tasks.functions import update_tasks_counters

logger = logging.getLogger(__name__)


class Command(BaseCommand):
    help = 'Recalculate organization project stats (total_annotations, etc)'

    def add_arguments(self, parser):
        parser.add_argument('organization', type=int, help='organization id')

    def handle(self, *args, **options):
        logger.debug(f"Start recalculating for Organization {options['organization']}.")
        projects = Project.objects.filter(organization_id=options['organization'])

        for project in projects:
            logger.debug(f'Start processing project {project.id}.')
            start_job_async_or_sync(update_tasks_counters, project.tasks.all())
            logger.debug(f'End processing project {project.id}.')

        logger.debug(f"Organization {options['organization']} stats were recalculated.")