from typing import Any import ijson from django.core.management.base import BaseCommand, CommandParser class Command(BaseCommand): help = """ Render messages to a file. Usage: ./manage.py render_messages <--amount> """ def add_arguments(self, parser: CommandParser) -> None: parser.add_argument("dump1", help="First file to compare") parser.add_argument("dump2", help="Second file to compare") def handle(self, *args: Any, **options: Any) -> None: total_count = 0 changed_count = 0 with open(options["dump1"]) as dump1, open(options["dump2"]) as dump2: for m1, m2 in zip(ijson.items(dump1, "item"), ijson.items(dump2, "item")): total_count += 1 if m1["id"] != m2["id"]: self.stderr.write("Inconsistent messages dump") break if m1["content"] != m2["content"]: changed_count += 1 self.stdout.write("Changed message id: {id}".format(id=m1["id"])) self.stdout.write(f"Total messages: {total_count}") self.stdout.write(f"Changed messages: {changed_count}")