mirror of
https://github.com/zulip/zulip.git
synced 2025-10-27 10:03:56 +00:00
42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
from __future__ import absolute_import
|
|
|
|
import datetime
|
|
import pytz
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from zerver.models import UserProfile, Realm, Stream, Message
|
|
|
|
class Command(BaseCommand):
|
|
help = "Generate statistics on user activity."
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('realms', metavar='<realm>', type=str, nargs='*',
|
|
help="realm to generate statistics for")
|
|
|
|
def messages_sent_by(self, user, week):
|
|
start = datetime.datetime.now(tz=pytz.utc) - datetime.timedelta(days=(week + 1)*7)
|
|
end = datetime.datetime.now(tz=pytz.utc) - 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):
|
|
if options['realms']:
|
|
try:
|
|
realms = [Realm.objects.get(domain=domain) for domain in options['realms']]
|
|
except Realm.DoesNotExist, e:
|
|
print e
|
|
exit(1)
|
|
else:
|
|
realms = Realm.objects.all()
|
|
|
|
for realm in realms:
|
|
print realm.domain
|
|
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,),
|
|
for week in range(10):
|
|
print "%5d" % (self.messages_sent_by(user_profile, week)),
|
|
print ""
|