Misago/misago/datamover/management/commands/runmigration.py

from django.core.management import call_command

from misago.datamover.management.base import BaseCommand


MOVE_COMMANDS = [
    'movesettings',
    'moveusers',
    'movecategories',
    'movethreads',
    'buildmovesindex',
    'synchronizethreads',
    'synchronizecategories',
    'rebuildpostssearch',
    'invalidatebans',
    'populateonlinetracker',
    'synchronizeusers',
]


class Command(BaseCommand):
    help = ("Executes complete migration from Misago 0.5 together with cleanups.")

    def handle(self, *args, **options):
        self.stdout.write("Running complete migration...")

        self.start_timer()

        for command_to_call in MOVE_COMMANDS:
            call_command(command_to_call)

        summary = "Migration was completed in %s" % self.stop_timer()
        self.stdout.write(self.style.SUCCESS(summary))