zulip/zerver/management/commands/analyze_user_activity.py

from __future__ import absolute_import

from optparse import make_option
from django.core.management.base import BaseCommand
from zerver.models import get_user_profile_by_email, UserActivityInterval, \
    UserProfile
import sys
import datetime
from django.utils.timezone import utc, is_naive

def analyze_activity(options):
    day_start = datetime.datetime.strptime(options["date"], "%Y-%m-%d").replace(tzinfo=utc)
    day_end = day_start + datetime.timedelta(days=options["duration"])

    user_profile_query = UserProfile.objects.all()
    if options["realm"]:
        user_profile_query = user_profile_query.filter(realm__domain=options["realm"])

    print "Per-user online duration:\n"
    total_duration = datetime.timedelta(0)
    for user_profile in user_profile_query:
        intervals = UserActivityInterval.objects.filter(user_profile=user_profile,
                                                        end__gte=day_start, start__lte=day_end)
        if len(intervals) == 0:
            continue

        duration = datetime.timedelta(0)
        for interval in intervals:
            start = max(day_start, interval.start)
            end = min(day_end, interval.end)
            duration += end - start

        total_duration += duration
        print "%-*s%s" % (37, user_profile.email, duration, )

    print "\nTotal Duration:                      %s" % (total_duration,)
    print "\nTotal Duration in minutes:           %s" % (total_duration.total_seconds() / 60.,)
    print "Total Duration amortized to a month: %s" % (total_duration.total_seconds() * 30. / 60.,)

class Command(BaseCommand):
    help = """Report analytics of user activity on a per-user and realm basis.

This command aggregates user activity data that is collected by each user using Zulip. It attempts
to approximate how much each user has been using Zulip per day, measured by recording each 15 minute
period where some activity has occurred (mouse move or keyboard activity).

It will correctly not count server-initiated reloads in the activity statistics.

The duration flag can be used to control how many days to show usage duration for

Usage: python manage.py analyze_user_activity [--realm=zulip.com] [--date=2013-09-10] [--duration=1]

By default, if no date is selected 2013-09-10 is used. If no realm is provided, information
is shown for all realms"""

    option_list = BaseCommand.option_list + (
        make_option('--realm', action='store'),
        make_option('--date', action='store', default="2013-09-06"),
        make_option('--duration', action='store', default=1, type=int, help="How many days to show usage information for"),
        )

    def handle(self, *args, **options):
        analyze_activity(options)