Update dep.

[wolnelektury.git] / src / stats / utils.py

diff --git a/src/stats/utils.py b/src/stats/utils.py
index fc473a6..eba7202 100644 (file)
--- a/src/stats/utils.py
+++ b/src/stats/utils.py
@@ -10,7 +10,7 @@ import urllib
 from random import random
 from inspect import isclass
 
-from django.utils.encoding import force_str
+from django.utils.encoding import force_bytes
 
 from .tasks import track_request
 
@@ -21,10 +21,10 @@ def piwik_url(request):
     return urllib.urlencode(dict(
         idsite=getattr(settings, 'PIWIK_SITE_ID', '0'),
         rec=1,
-        url=force_str('http://%s%s' % (request.META['HTTP_HOST'], request.path)),
+        url=force_bytes('http://%s%s' % (request.META['HTTP_HOST'], request.path)),
         rand=int(random() * 0x10000),
         apiv=PIWIK_API_VERSION,
-        urlref=force_str(request.META.get('HTTP_REFERER', '')),
+        urlref=force_bytes(request.META.get('HTTP_REFERER', '')),
         ua=request.META.get('HTTP_USER_AGENT', ''),
         lang=request.META.get('HTTP_ACCEPT_LANGUAGE', ''),
         token_auth=getattr(settings, 'PIWIK_TOKEN', ''),
@@ -62,3 +62,16 @@ def piwik_track(klass_or_method):
         return klass
     else:
         return wrap
+
+
+def piwik_track_view(view):
+    if not getattr(settings, 'PIWIK_SITE_ID', 0):
+        return view
+
+    def wrap(request, *args, **kwargs):
+        if getattr(request, 'piwik_track', True):
+            track_request.delay(piwik_url(request))
+        return view(request, *args, **kwargs)
+
+    update_wrapper(wrap, view)
+    return wrap