Add a user_stats management command.

(imported from commit 1b13fd70dc271509cf0b5524878300d17c20f50e)

zephyr/management/commands/user_stats.py

@@ -0,0 +1,35 @@
+import datetime
+import pytz
+
+from django.core.management.base import BaseCommand
+from zephyr.models import UserProfile, Realm, Stream, Message
+
+class Command(BaseCommand):
+    help = "Generate statistics on user activity."
+
+    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 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, user__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.user.email,),
+                for week in range(10):
+                    print "%5d" % (self.messages_sent_by(user_profile, week)),
+                print ""