mirror of
https://github.com/zulip/zulip.git
synced 2025-10-24 16:43:57 +00:00
123bcea518
Using sys.exit in a management command makes it impossible
to unit test the code in question. The correct approach to do the same
thing in Django management commands is to raise CommandError.
Followup of b570c0dafa
57 lines
2.3 KiB
Python
57 lines
2.3 KiB
Python
from argparse import ArgumentParser
|
|
from typing import Any
|
|
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from django.db.models import Q
|
|
|
|
from zerver.models import Message, Realm, \
|
|
Recipient, Stream, Subscription, get_realm
|
|
|
|
class Command(BaseCommand):
|
|
help = "Generate statistics on the streams for a realm."
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
parser.add_argument('realms', metavar='<realm>', type=str, nargs='*',
|
|
help="realm to generate statistics for")
|
|
|
|
def handle(self, *args: Any, **options: str) -> None:
|
|
if options['realms']:
|
|
try:
|
|
realms = [get_realm(string_id) for string_id in options['realms']]
|
|
except Realm.DoesNotExist as e:
|
|
raise CommandError(e)
|
|
else:
|
|
realms = Realm.objects.all()
|
|
|
|
for realm in realms:
|
|
streams = Stream.objects.filter(realm=realm).exclude(Q(name__istartswith="tutorial-"))
|
|
# private stream count
|
|
private_count = 0
|
|
# public stream count
|
|
public_count = 0
|
|
for stream in streams:
|
|
if stream.invite_only:
|
|
private_count += 1
|
|
else:
|
|
public_count += 1
|
|
print("------------")
|
|
print(realm.string_id, end=' ')
|
|
print("%10s %d public streams and" % ("(", public_count), end=' ')
|
|
print("%d private streams )" % (private_count,))
|
|
print("------------")
|
|
print("%25s %15s %10s %12s" % ("stream", "subscribers", "messages", "type"))
|
|
|
|
for stream in streams:
|
|
if stream.invite_only:
|
|
stream_type = 'private'
|
|
else:
|
|
stream_type = 'public'
|
|
print("%25s" % (stream.name,), end=' ')
|
|
recipient = Recipient.objects.filter(type=Recipient.STREAM, type_id=stream.id)
|
|
print("%10d" % (len(Subscription.objects.filter(recipient=recipient,
|
|
active=True)),), end=' ')
|
|
num_messages = len(Message.objects.filter(recipient=recipient))
|
|
print("%12d" % (num_messages,), end=' ')
|
|
print("%15s" % (stream_type,))
|
|
print("")
|