mirror of https://github.com/zulip/zulip.git
149 lines
7.5 KiB
Python
149 lines
7.5 KiB
Python
from __future__ import absolute_import
|
|
|
|
import datetime
|
|
import pytz
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from django.db.models import Count
|
|
from zephyr.models import UserProfile, Realm, Stream, Message, Recipient, UserActivity, \
|
|
Subscription, UserMessage
|
|
|
|
MOBILE_CLIENT_LIST = ["Android", "iPhone"]
|
|
HUMAN_CLIENT_LIST = MOBILE_CLIENT_LIST + ["website"]
|
|
|
|
human_messages = Message.objects.filter(sending_client__name__in=HUMAN_CLIENT_LIST)
|
|
|
|
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,
|
|
user_profile__is_active=True,
|
|
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 human_messages.filter(sender=user, pub_date__gt=sent_time_cutoff).count()
|
|
|
|
def total_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).count()
|
|
|
|
def human_messages(self, realm, days_ago):
|
|
sent_time_cutoff = datetime.datetime.now(tz=pytz.utc) - datetime.timedelta(days=days_ago)
|
|
return human_messages.filter(sender__realm=realm, pub_date__gt=sent_time_cutoff).count()
|
|
|
|
def api_messages(self, realm, days_ago):
|
|
return (self.total_messages(realm, days_ago) - self.human_messages(realm, days_ago))
|
|
|
|
def stream_messages(self, realm, days_ago):
|
|
sent_time_cutoff = datetime.datetime.now(tz=pytz.utc) - datetime.timedelta(days=days_ago)
|
|
return human_messages.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 human_messages.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 human_messages.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, is_active=True)
|
|
active_users = self.active_users(realm)
|
|
num_active = len(active_users)
|
|
|
|
print "%d active users (%d total)" % (num_active, len(user_profiles))
|
|
streams = Stream.objects.filter(realm=realm).extra(
|
|
tables=['zephyr_subscription', 'zephyr_recipient'],
|
|
where=['zephyr_subscription.recipient_id = zephyr_recipient.id',
|
|
'zephyr_recipient.type = 2',
|
|
'zephyr_recipient.type_id = zephyr_stream.id',
|
|
'zephyr_subscription.active = true']).annotate(count=Count("name"))
|
|
print "%d streams" % (streams.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 messages sent via the API" % (self.api_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")
|
|
|
|
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 = human_messages.filter(sender__realm=realm).count()
|
|
multi_paragraph_message_count = human_messages.filter(
|
|
sender__realm=realm, content__contains="\n\n").count()
|
|
self.report_percentage(multi_paragraph_message_count, all_message_count,
|
|
"all messages are multi-paragraph")
|
|
|
|
# Starred messages
|
|
starrers = UserMessage.objects.filter(user_profile__in=user_profiles,
|
|
flags=UserMessage.flags.starred).values(
|
|
"user_profile").annotate(count=Count("user_profile"))
|
|
print "%d users have starred %d messages" % (
|
|
len(starrers), sum([elt["count"] for elt in starrers]))
|
|
|
|
active_user_subs = Subscription.objects.filter(
|
|
user_profile__in=user_profiles, active=True)
|
|
|
|
# Streams not in home view
|
|
non_home_view = active_user_subs.filter(in_home_view=False).values(
|
|
"user_profile").annotate(count=Count("user_profile"))
|
|
print "%d users have %d streams not in home view" % (
|
|
len(non_home_view), sum([elt["count"] for elt in non_home_view]))
|
|
|
|
# Code block markup
|
|
markup_messages = human_messages.filter(
|
|
sender__realm=realm, content__contains="~~~").values(
|
|
"sender").annotate(count=Count("sender"))
|
|
print "%d users have used code block markup on %s messages" % (
|
|
len(markup_messages), sum([elt["count"] for elt in markup_messages]))
|
|
|
|
# Notifications for stream messages
|
|
notifications = active_user_subs.filter(notifications=True).values(
|
|
"user_profile").annotate(count=Count("user_profile"))
|
|
print "%d users receive desktop notifications for %d streams" % (
|
|
len(notifications), sum([elt["count"] for elt in notifications]))
|
|
|
|
print ""
|