apps/stats/utils.py

   1 # -*- coding: utf-8 -*-
   2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   4 #
   5 from django.contrib.sites.models import Site
   6 from piwik.django.models import PiwikSite
   7 from django.conf import settings
   8 from datetime import datetime
   9 import logging
  10 from functools import update_wrapper
  11 import urllib
  12 from random import random
  13 from inspect import isclass
  14 from .tasks import track_request
  15
  16 logger = logging.getLogger(__name__)
  17
  18
  19 def piwik_url(request):
  20     return urllib.urlencode(dict(
  21         idsite=_id_piwik,
  22         rec=1,
  23         url='http://%s%s' % (request.META['HTTP_HOST'], request.path),
  24         rand=int(random() * 0x10000),
  25         apiv=PIWIK_API_VERSION,
  26         urlref=request.META.get('HTTP_REFERER', ''),
  27         ua=request.META.get('HTTP_USER_AGENT', ''),
  28         lang=request.META.get('HTTP_ACCEPT_LANGUAGE', ''),
  29         token_auth=getattr(settings, 'PIWIK_TOKEN', ''),
  30         cip=request.META['REMOTE_ADDR'],
  31         cdt=datetime.utcnow().strftime("%Y-%m-%d %H:%M:%S")
  32     ))
  33
  34 PIWIK_API_VERSION = 1
  35
  36
  37 # Retrieve piwik information
  38 try:
  39     _id_piwik = PiwikSite.objects.get(site=Site.objects.get_current().id).id_site
  40 except PiwikSite.DoesNotExist:
  41     logger.debug("No PiwikSite is configured.")
  42     _id_piwik = None
  43
  44 def piwik_track(klass_or_method):
  45     """Track decorated class or method using Piwik (according to configuration in settings and django-piwik)
  46     Works for handler classes (executed by __call__) or handler methods. Expects request to be the first parameter
  47     """
  48     if _id_piwik is None:
  49         return klass_or_method
  50
  51     # get target method
  52     if isclass(klass_or_method):
  53         klass = klass_or_method
  54         call_func = klass.__call__
  55     else:
  56         call_func = klass_or_method
  57
  58     def wrap(self, request, *args, **kw):
  59         track_request.delay(piwik_url(request))
  60         return call_func(self, request, *args, **kw)
  61
  62     # and wrap it
  63     update_wrapper(wrap, call_func)
  64
  65     if isclass(klass_or_method):
  66         klass.__call__ = wrap
  67         return klass
  68     else:
  69         return wrap