src/stats/models.py

   1 import re
   2 from django.conf import settings
   3 from django.db import models
   4
   5
   6 class Visits(models.Model):
   7     book = models.ForeignKey('catalogue.Book', models.CASCADE)
   8     year = models.PositiveSmallIntegerField()
   9     month = models.PositiveSmallIntegerField()
  10     views = models.IntegerField()
  11     unique_views = models.IntegerField()
  12
  13     @classmethod
  14     def build_month(cls, year, month):
  15         ### TODO: Delete existing?
  16
  17         date = f'{year}-{month:02d}'
  18         url = f'{settings.PIWIK_URL}?date={date}&filter_limit=-1&format=CSV&idSite={settings.PIWIK_SITE_ID}&language=pl&method=Actions.getPageUrls&module=API&period=month&segment=&token_auth={settings.PIWIK_TOKEN}&flat=1'
  19         data = urlopen(url).read().decode('utf-16')
  20         lines = data.split('\n')[1:]
  21         for line in lines:
  22             m = re.match('^/katalog/lektura/([^,]+)\.html,', line)
  23             if m is not None:
  24                 slug = m.group(1)
  25                 _url, uviews, views = line.split(',', 3)
  26                 uviews = int(uviews)
  27                 views = int(views)
  28                 try:
  29                     book = Book.objects.get(slug=slug)
  30                 except Book.DoesNotExists:
  31                     continue
  32                 else:
  33                     cls.objects.create(
  34                         book=book, year=year, month=month,
  35                         views=views, uviews=uviews
  36                     )