apps/stats/utils.py

   1
   2 from django.contrib.sites.models import Site
   3 from piwik.django.models import PiwikSite
   4 from django.conf import settings
   5 import logging
   6 from functools import update_wrapper
   7 import httplib
   8 import urlparse
   9 import urllib
  10 from random import random
  11 from inspect import isclass
  12
  13 logger = logging.getLogger(__name__)
  14
  15
  16 def piwik_url(**kw):
  17     url = settings.PIWIK_URL + u"/piwik.php?"
  18     url += u'&'.join([k + u"=" + str(v) for k, v in kw.items()])
  19     logger.info("piwik url: %s" % url)
  20     return url
  21
  22 PIWIK_API_VERSION = 1
  23
  24
  25 def piwik_track(klass_or_method):
  26     """Track decorated class or method using Piwik (according to configuration in settings and django-piwik)
  27     Works for handler classes (executed by __call__) or handler methods. Expects request to be the first parameter
  28     """
  29     # Retrieve piwik information
  30     current_site = Site.objects.get_current()
  31     piwik_site = PiwikSite.objects.filter(site=current_site.id)
  32
  33     if len(piwik_site) == 0:
  34         logger.debug("No PiwikSite is configured for Site " + current_site.name)
  35         return klass_or_method
  36
  37     id_piwik = piwik_site[0].id_site
  38     host = urlparse.urlsplit(settings.PIWIK_URL).netloc
  39
  40     # get target method
  41     if isclass(klass_or_method):
  42         klass = klass_or_method
  43         call_func = klass.__call__
  44     else:
  45         call_func = klass_or_method
  46
  47     def wrap(self, request, *args, **kw):
  48         conn = httplib.HTTPConnection(host)
  49         conn.request('GET', piwik_url(
  50             rec=1,
  51             apiv=PIWIK_API_VERSION,
  52             rand=int(random() * 0x10000),
  53             token_auth=urllib.quote(settings.PIWIK_TOKEN),
  54             cip=urllib.quote(request.META['REMOTE_ADDR']),
  55             url=urllib.quote('http://' + request.META['HTTP_HOST'] + request.path),
  56             urlref=urllib.quote(request.META['HTTP_REFERER']) if 'HTTP_REFERER' in request.META else '',
  57             idsite=id_piwik))
  58
  59         conn.close()
  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         print klass_or_method
  70         return wrap