zulip/zilencer/management/commands/compare_messages.py

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}')