jplayer
[wolnelektury.git] / 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