mygpo/data/historic.py

   1 from mygpo.api.models import Podcast, SubscriptionAction
   2 from mygpo.data.models import HistoricPodcastData
   3 from mygpo.utils import daterange
   4 from datetime import date, timedelta
   5
   6
   7 def all_podcasts():
   8     for p in Podcast.objects.all().iterator():
   9         calc_podcast(p)
  10
  11
  12 def calc_podcast(podcast):
  13
  14     latest_historic = HistoricPodcastData.objects.filter(podcast=podcast).order_by('-date')
  15     if latest_historic.count() > 0:
  16         start = latest_historic[0].date + timedelta(days=1)
  17         prev = latest_historic[0].subscriber_count
  18
  19     else:
  20         sas = SubscriptionAction.objects.filter(podcast=podcast).order_by('timestamp')
  21         if sas.exists():
  22             first = sas[0]
  23             start = first.timestamp.date()
  24         else:
  25             start = date.today()
  26         prev = 0
  27
  28     for day in daterange(start, date.today() - timedelta(days=1)):
  29         nextday = day + timedelta(days=1)
  30         act = [x['action'] for x in SubscriptionAction.objects.filter(podcast=podcast, timestamp__range=(day, nextday)).values('action')]
  31         prev += sum(act)
  32
  33         rec = HistoricPodcastData.objects.get_or_create(podcast=podcast, date=day, defaults={'subscriber_count': prev})
  34
  35