from django.core.management import BaseCommand
from books.categorize_modules import categorize_modules


class Command(BaseCommand):
    def handle(self, *args, **options):
        self.stdout.write("Migrating Wagtail documents to Custommyskillbox documents")
        categorize_modules()
        self.stdout.write("Finish migration")