mirror of
https://github.com/zulip/zulip.git
synced 2026-07-18 21:04:19 +08: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
42 lines
1.7 KiB
Python
42 lines
1.7 KiB
Python
import datetime
|
|
from argparse import ArgumentParser
|
|
from typing import Any
|
|
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from django.utils.timezone import now as timezone_now
|
|
|
|
from zerver.models import Message, Realm, Stream, UserProfile, get_realm
|
|
|
|
class Command(BaseCommand):
|
|
help = "Generate statistics on user activity."
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
parser.add_argument('realms', metavar='<realm>', type=str, nargs='*',
|
|
help="realm to generate statistics for")
|
|
|
|
def messages_sent_by(self, user: UserProfile, week: int) -> int:
|
|
start = timezone_now() - datetime.timedelta(days=(week + 1)*7)
|
|
end = timezone_now() - datetime.timedelta(days=week*7)
|
|
return Message.objects.filter(sender=user, pub_date__gt=start, pub_date__lte=end).count()
|
|
|
|
def handle(self, *args: Any, **options: Any) -> 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:
|
|
print(realm.string_id)
|
|
user_profiles = UserProfile.objects.filter(realm=realm, is_active=True)
|
|
print("%d users" % (len(user_profiles),))
|
|
print("%d streams" % (len(Stream.objects.filter(realm=realm)),))
|
|
|
|
for user_profile in user_profiles:
|
|
print("%35s" % (user_profile.email,), end=' ')
|
|
for week in range(10):
|
|
print("%5d" % (self.messages_sent_by(user_profile, week),), end=' ')
|
|
print("")
|