mirror of
https://github.com/zulip/zulip.git
synced 2025-10-30 11:33:51 +00:00
8e7ce7cc79
This is a preparatory commit for using isort for sorting all of our imports, merging changes to files where we can easily review the changes as something we're happy with. These are also files with relatively little active development, which means we don't expect much merge conflict risk from these changes.
58 lines
2.3 KiB
Python
58 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("")
|