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 # Retrieve piwik information
  29 try:
  30     _host = urlparse.urlsplit(settings.PIWIK_URL).netloc
  31 except AttributeError:
  32     logger.debug("PIWIK_URL not configured.")
  33     _host = None
  34 try:
  35     _id_piwik = PiwikSite.objects.get(site=Site.objects.get_current().id).id_site
  36 except PiwikSite.DoesNotExist:
  37     logger.debug("No PiwikSite is configured.")
  38     _id_piwik = None
  39
  40 def piwik_track(klass_or_method):
  41     """Track decorated class or method using Piwik (according to configuration in settings and django-piwik)
  42     Works for handler classes (executed by __call__) or handler methods. Expects request to be the first parameter
  43     """
  44     if _id_piwik is None:
  45         return klass_or_method
  46
  47     # get target method
  48     if isclass(klass_or_method):
  49         klass = klass_or_method
  50         call_func = klass.__call__
  51     else:
  52         call_func = klass_or_method
  53
  54     def wrap(self, request, *args, **kw):
  55         conn = httplib.HTTPConnection(_host)
  56         conn.request('GET', piwik_url(
  57             rec=1,
  58             apiv=PIWIK_API_VERSION,
  59             rand=int(random() * 0x10000),
  60             token_auth=urllib.quote(settings.PIWIK_TOKEN),
  61             cip=urllib.quote(request.META['REMOTE_ADDR']),
  62             url=urllib.quote('http://' + request.META['HTTP_HOST'] + request.path),
  63             urlref=urllib.quote(request.META['HTTP_REFERER']) if 'HTTP_REFERER' in request.META else '',
  64             idsite=_id_piwik))
  65
  66         conn.close()
  67         return call_func(self, request, *args, **kw)
  68
  69     # and wrap it
  70     update_wrapper(wrap, call_func)
  71
  72     if isclass(klass_or_method):
  73         klass.__call__ = wrap
  74         return klass
  75     else:
  76         return wrap