zulip/zephyr/management/commands/realm_stats.py

68 lines
3.5 KiB
Python

import datetime
import pytz
from django.core.management.base import BaseCommand
from zephyr.models import UserProfile, Realm, Stream, Message, Recipient, StreamColor
class Command(BaseCommand):
help = "Generate statistics on realm activity."
def messages_sent_by(self, user, days_ago):
sent_time_cutoff = datetime.datetime.now(tz=pytz.utc) - datetime.timedelta(days=days_ago)
return Message.objects.filter(sender=user, pub_date__gt=sent_time_cutoff).count()
def stream_messages(self, realm, days_ago):
sent_time_cutoff = datetime.datetime.now(tz=pytz.utc) - datetime.timedelta(days=days_ago)
return Message.objects.filter(sender__realm=realm, pub_date__gt=sent_time_cutoff,
recipient__type=Recipient.STREAM).count()
def private_messages(self, realm, days_ago):
sent_time_cutoff = datetime.datetime.now(tz=pytz.utc) - datetime.timedelta(days=days_ago)
return Message.objects.filter(sender__realm=realm, pub_date__gt=sent_time_cutoff).exclude(
recipient__type=Recipient.STREAM).exclude(recipient__type=Recipient.HUDDLE).count()
def group_private_messages(self, realm, days_ago):
sent_time_cutoff = datetime.datetime.now(tz=pytz.utc) - datetime.timedelta(days=days_ago)
return Message.objects.filter(sender__realm=realm, pub_date__gt=sent_time_cutoff).exclude(
recipient__type=Recipient.STREAM).exclude(recipient__type=Recipient.PERSONAL).count()
def handle(self, *args, **options):
if args:
try:
realms = [Realm.objects.get(domain=domain) for domain in args]
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)
print "%d users" % (len(user_profiles),)
print "%d streams" % (Stream.objects.filter(realm=realm).count(),)
for days_ago in (1, 7, 30):
print "In last %d days, users sent:" % (days_ago,)
sender_quantities = [self.messages_sent_by(user, days_ago) for user in user_profiles]
for quantity in sorted(sender_quantities, reverse=True):
print quantity,
print ""
print "%d stream messages" % (self.stream_messages(realm, days_ago),)
print "%d one-on-one private messages" % (self.private_messages(realm, days_ago),)
print "%d group private messages" % (self.group_private_messages(realm, days_ago),)
print "%.2f%% have desktop notifications enabled" % (float(len(user_profiles.filter(
enable_desktop_notifications=True))) * 100 /len(user_profiles),)
colorizers = 0
for profile in user_profiles:
if StreamColor.objects.filter(subscription__user_profile=profile).count() > 0:
colorizers += 1
print "%.2f%% have colorized streams" % (float(colorizers) * 100/len(user_profiles),)
all_message_count = Message.objects.filter(sender__realm=realm).count()
multi_paragraph_message_count = Message.objects.filter(sender__realm=realm,
content__contains="\n\n").count()
print "%.2f%% of all messages are multi-paragraph" % (
float(multi_paragraph_message_count) * 100 / all_message_count)
print ""