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