mirror of
https://github.com/zulip/zulip.git
synced 2025-10-26 09:34:02 +00:00
64fb17f9c2
Move commands related to stats collection and reporting from zilencer to analytics. To do this, we had to make "analytics" officially an app. (imported from commit 63ef6c68d1b1ebb5043ee4aca999aa209e7f494d)
58 lines
2.8 KiB
Python
58 lines
2.8 KiB
Python
from __future__ import absolute_import
|
|
|
|
from django.core.management.base import BaseCommand
|
|
|
|
from zerver.models import UserPresence, UserActivity
|
|
from zerver.lib.utils import statsd, statsd_key
|
|
|
|
from datetime import datetime, timedelta
|
|
from collections import defaultdict
|
|
|
|
class Command(BaseCommand):
|
|
help = """Sends active user statistics to statsd.
|
|
|
|
Run as a cron job that runs every 10 minutes."""
|
|
|
|
def handle(self, *args, **options):
|
|
# Get list of all active users in the last 1 week
|
|
cutoff = datetime.now() - timedelta(minutes=30, hours=168)
|
|
|
|
users = UserPresence.objects.select_related().filter(timestamp__gt=cutoff)
|
|
|
|
# Calculate 10min, 2hrs, 12hrs, 1day, 2 business days (TODO business days), 1 week bucket of stats
|
|
hour_buckets = [0.16, 2, 12, 24, 48, 168]
|
|
user_info = defaultdict(dict)
|
|
|
|
for last_presence in users:
|
|
if last_presence.status == UserPresence.IDLE:
|
|
known_active = last_presence.timestamp - timedelta(minutes=30)
|
|
else:
|
|
known_active = last_presence.timestamp
|
|
|
|
for bucket in hour_buckets:
|
|
if not bucket in user_info[last_presence.user_profile.realm.domain]:
|
|
user_info[last_presence.user_profile.realm.domain][bucket] = []
|
|
if datetime.now(known_active.tzinfo) - known_active < timedelta(hours=bucket):
|
|
user_info[last_presence.user_profile.realm.domain][bucket].append(last_presence.user_profile.email)
|
|
|
|
for realm, buckets in user_info.items():
|
|
print("Realm %s" % realm)
|
|
for hr, users in sorted(buckets.items()):
|
|
print("\tUsers for %s: %s" % (hr, len(users)))
|
|
statsd.gauge("users.active.%s.%shr" % (statsd_key(realm, True), statsd_key(hr, True)), len(users))
|
|
|
|
# Also do stats for how many users have been reading the app.
|
|
users_reading = UserActivity.objects.select_related().filter(query="/json/update_message_flags")
|
|
user_info = defaultdict(dict)
|
|
for activity in users_reading:
|
|
for bucket in hour_buckets:
|
|
if not bucket in user_info[activity.user_profile.realm.domain]:
|
|
user_info[activity.user_profile.realm.domain][bucket] = []
|
|
if datetime.now(activity.last_visit.tzinfo) - activity.last_visit < timedelta(hours=bucket):
|
|
user_info[activity.user_profile.realm.domain][bucket].append(activity.user_profile.email)
|
|
for realm, buckets in user_info.items():
|
|
print("Realm %s" % realm)
|
|
for hr, users in sorted(buckets.items()):
|
|
print("\tUsers reading for %s: %s" % (hr, len(users)))
|
|
statsd.gauge("users.reading.%s.%shr" % (statsd_key(realm, True), statsd_key(hr, True)), len(users))
|