mirror of https://github.com/zulip/zulip.git
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("{:>10} {} public streams and".format("(", public_count), end=' ')
|
|
print(f"{private_count} private streams )")
|
|
print("------------")
|
|
print("{:>25} {:>15} {:>10} {:>12}".format("stream", "subscribers", "messages", "type"))
|
|
|
|
for stream in streams:
|
|
if stream.invite_only:
|
|
stream_type = 'private'
|
|
else:
|
|
stream_type = 'public'
|
|
print(f"{stream.name:>25}", end=' ')
|
|
recipient = Recipient.objects.filter(type=Recipient.STREAM, type_id=stream.id)
|
|
print("{:10}".format(len(Subscription.objects.filter(recipient=recipient,
|
|
active=True))), end=' ')
|
|
num_messages = len(Message.objects.filter(recipient=recipient))
|
|
print(f"{num_messages:12}", end=' ')
|
|
print(f"{stream_type:>15}")
|
|
print("")
|