mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 22:13:26 +00:00
2b2be8120f
datetime.now with a timezone set is equivalent to timezone.now() if it's never being printed out, but the latter is cleaner and more idiomatic.
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
|
|
from argparse import ArgumentParser
|
|
import datetime
|
|
import pytz
|
|
from typing import Any
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from django.utils import timezone
|
|
|
|
from zerver.models import UserProfile, Realm, Stream, Message, get_realm
|
|
from six.moves import range
|
|
|
|
class Command(BaseCommand):
|
|
help = "Generate statistics on user activity."
|
|
|
|
def add_arguments(self, parser):
|
|
# type: (ArgumentParser) -> None
|
|
parser.add_argument('realms', metavar='<realm>', type=str, nargs='*',
|
|
help="realm to generate statistics for")
|
|
|
|
def messages_sent_by(self, user, week):
|
|
# type: (UserProfile, 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, **options):
|
|
# type: (*Any, **Any) -> None
|
|
if options['realms']:
|
|
try:
|
|
realms = [get_realm(string_id) for string_id in options['realms']]
|
|
except Realm.DoesNotExist as e:
|
|
print(e)
|
|
exit(1)
|
|
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("")
|