Source code for akvo.rsr.management.commands.switch_indicators_to_cumulative_reporting

import argparse
from io import TextIOBase
from django.core.management.base import BaseCommand
from django.db import transaction
from akvo.rsr.models import Project, Indicator
from akvo.rsr.models.result.utils import QUANTITATIVE


class Command(BaseCommand):
    help = "Switch all quantitative indicators of the given project and its descendants to cumulative reporting"

    def add_arguments(self, parser: argparse.ArgumentParser):
        parser.add_argument("project_id", type=int)
        parser.add_argument(
            "--dry-run", action="store_true", help="Don not actually apply the changes"
        )

    def handle(self, *args, **options):
        project = Project.objects.get(id=int(options["project_id"]))
        runner = CommandRunner(self.stdout, project, options["dry_run"])

        try:
            runner.run()
        except InterruptedError:
            self.stdout.write("Changes not applied\n")
        else:
            self.stdout.write("Changes applied\n")
        self.stdout.write("DONE!\n")


class CommandRunner:
    def __init__(self, stdout: TextIOBase, project: Project, dry_run=False):
        self.stdout = stdout
        self.project = project
        self.dry_run = dry_run

    @transaction.atomic
    def run(self):
        indicators = Indicator.objects.filter(
            result__project__in=self.project.descendants(),
            type=QUANTITATIVE,
            parent_indicator__isnull=True,
        )
        self.stdout.write(
            f"Switching {indicators.count()} indicators to cumulative reporting\n"
        )
        for indicator in indicators:
            indicator.cumulative = True
            indicator.save()
        if self.dry_run:
            raise InterruptedError()