mirror of https://github.com/zulip/zulip.git
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
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 <destination> <--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}")
|