Misago/misago/onlines.py

from datetime import timedelta
from django.core.cache import cache
from django.utils import timezone
from misago.models import Session

class MembersOnline(object):
    def __init__(self, monitor, frequency=180):
        self.recounted = False
        self.monitor = monitor
        self.frequency = frequency
        self._members = int(monitor['online_members'])
        self._all = int(monitor['online_all'])
        self._om = self._members
        self._oa = self._all
        if (monitor.expired('online_all', frequency) or
                monitor.expired('online_members', frequency)):
            self.count_sessions()
            self.recounted = False

    def count_sessions(self):
        if not self.recounted:
            self.recounted = True
            queryset = Session.objects.filter(matched=True).filter(crawler__isnull=True).filter(last__gte=timezone.now() - timedelta(seconds=self.frequency))
            self._all = queryset.count()
            self._members = queryset.filter(user__isnull=False).count()
            cache.delete_many(['team_users_online', 'ranks_online'])

    def new_session(self):
        self._all += 1

    def sign_in(self):
        self._members += 1

    def sign_out(self):
        if self._members:
            self._members -= 1

    def all(self):
        return self._all

    def members(self):
        return self._members

    def sync(self):
        if self._members != self._om:
            self.monitor['online_members'] = self._members
        if self._all != self._oa:
            self.monitor['online_all'] = self._all

    def stats(self):
        return {
                'members': self.members(),
                'all': self.all(),
               }