annotate gpp/core/management/commands/max_users.py @ 285:8fd4984d5c3b

This is a first rough commit for #95, adding the ability to embed YouTube videos in forum posts. Some more polish and testing needs to happen at this point. I wanted to get all these changes off my hard drive and into the repository.
author Brian Neal <bgneal@gmail.com>
date Thu, 14 Oct 2010 02:39:35 +0000
parents dcc929973bba
children 3fe60148f75c
rev   line source
bgneal@239 1 """
bgneal@239 2 max_users is a custom manage.py command.
bgneal@239 3 It is intended to be called from a cron job to calculate the maximum
bgneal@239 4 number of users online statistic.
bgneal@239 5 """
bgneal@239 6 import datetime
bgneal@239 7
bgneal@239 8 from django.core.management.base import NoArgsCommand
bgneal@239 9
bgneal@239 10 from core.models import UserLastVisit, AnonLastVisit, Statistic
bgneal@239 11
bgneal@239 12
bgneal@239 13 class Command(NoArgsCommand):
bgneal@239 14 help = "Run periodically to compute the max users online statistic."
bgneal@239 15
bgneal@239 16 def handle_noargs(self, **options):
bgneal@239 17
bgneal@239 18 now = datetime.datetime.now()
bgneal@239 19 cut_off = now - datetime.timedelta(minutes=15)
bgneal@239 20
bgneal@239 21 users = UserLastVisit.objects.filter(last_visit__gte=cut_off).count()
bgneal@239 22 guests = AnonLastVisit.objects.filter(last_visit__gte=cut_off).count()
bgneal@239 23
bgneal@239 24 updated = False
bgneal@239 25 try:
bgneal@239 26 stat = Statistic.objects.get(pk=1)
bgneal@239 27 except Statistic.DoesNotExist:
bgneal@239 28 stat = Statistic(max_users=users,
bgneal@239 29 max_users_date=now,
bgneal@239 30 max_anon_users=guests,
bgneal@239 31 max_anon_users_date=now)
bgneal@239 32 updated=True
bgneal@239 33 else:
bgneal@239 34 if users > stat.max_users:
bgneal@239 35 stat.max_users = users
bgneal@239 36 stat.max_users_date = now
bgneal@239 37 updated=True
bgneal@239 38 if guests > stat.max_anon_users:
bgneal@239 39 stat.max_anon_users = guests
bgneal@239 40 stat.max_anon_users_date = now
bgneal@239 41 updated=True
bgneal@239 42
bgneal@239 43 if updated:
bgneal@239 44 stat.save()