mirror of https://github.com/zulip/zulip.git
98 lines
4.8 KiB
Python
98 lines
4.8 KiB
Python
import datetime
|
|
import pytz
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from zephyr.models import UserProfile, Realm, Stream, Message, Recipient, StreamColor, UserActivity
|
|
|
|
class Command(BaseCommand):
|
|
help = "Generate statistics on realm activity."
|
|
|
|
def active_users(self, realm):
|
|
# Has been active (on the website, for now) in the last 7 days.
|
|
activity_cutoff = datetime.datetime.now(tz=pytz.utc) - datetime.timedelta(days=7)
|
|
return [activity.user_profile for activity in \
|
|
UserActivity.objects.filter(user_profile__realm=realm,
|
|
last_visit__gt=activity_cutoff,
|
|
query="/json/update_pointer",
|
|
client__name="website")]
|
|
|
|
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 report_percentage(self, numerator, denominator, text):
|
|
if not denominator:
|
|
fraction = 0.0
|
|
else:
|
|
fraction = numerator / float(denominator)
|
|
print "%.2f%% of" % (fraction * 100,), text
|
|
|
|
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)
|
|
active_users = self.active_users(realm)
|
|
num_active = len(active_users)
|
|
|
|
print "%d active users (%d total)" % (num_active, 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),)
|
|
|
|
num_notifications_enabled = len(filter(lambda x: x.enable_desktop_notifications == True,
|
|
active_users))
|
|
self.report_percentage(num_notifications_enabled, num_active,
|
|
"active users have desktop notifications enabled")
|
|
|
|
colorizers = 0
|
|
for profile in active_users:
|
|
if StreamColor.objects.filter(subscription__user_profile=profile).count() > 0:
|
|
colorizers += 1
|
|
self.report_percentage(colorizers, num_active,
|
|
"active users have colorized streams")
|
|
|
|
num_enter_sends = len(filter(lambda x: x.enter_sends, active_users))
|
|
self.report_percentage(num_enter_sends, num_active,
|
|
"active users have enter-sends")
|
|
|
|
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()
|
|
self.report_percentage(multi_paragraph_message_count, all_message_count,
|
|
"all messages are multi-paragraph")
|
|
print ""
|