Add a realm_stats management command.

(imported from commit e23c9c521dff5e551615977ec5cd575bb5899ec6)

zephyr/management/commands/realm_stats.py

@@ -0,0 +1,67 @@
+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 ""