mygpo/admin/views.py

   1 import re
   2 from itertools import count
   3
   4 from django.shortcuts import render
   5 from django.contrib import messages
   6 from django.core.urlresolvers import reverse
   7 from django.http import HttpResponseRedirect
   8 from django.utils.translation import ugettext as _
   9 from django.views.generic import TemplateView
  10 from django.utils.decorators import method_decorator
  11
  12 from mygpo.admin.auth import require_staff
  13 from mygpo.admin.group import PodcastGrouper
  14 from mygpo.core.models import Podcast
  15 from mygpo.counter import Counter
  16 from mygpo.maintenance.merge import PodcastMerger, IncorrectMergeException
  17 from mygpo.users.models import User
  18
  19
  20 class AdminView(TemplateView):
  21
  22     @method_decorator(require_staff)
  23     def dispatch(self, *args, **kwargs):
  24         return super(AdminView, self).dispatch(*args, **kwargs)
  25
  26
  27 class Overview(AdminView):
  28     template_name = 'admin/overview.html'
  29
  30
  31 class MergeSelect(AdminView):
  32     template_name = 'admin/merge-select.html'
  33
  34     def get(self, request):
  35         num = int(request.GET.get('podcasts', 2))
  36         urls = [''] * num
  37
  38         return self.render_to_response({
  39                 'urls': urls,
  40             })
  41
  42
  43 class MergeBase(AdminView):
  44
  45     def _get_podcasts(self, request):
  46         podcasts = []
  47         for n in count():
  48             podcast_url = request.POST.get('feed%d' % n, None)
  49             if podcast_url is None:
  50                 break
  51
  52             if not podcast_url:
  53                 continue
  54
  55             podcast = Podcast.for_url(podcast_url)
  56
  57             if not podcast:
  58                 raise InvalidPodcast(podcast_url)
  59
  60             podcasts.append(Podcast.for_url(podcast_url))
  61
  62         return podcasts
  63
  64
  65 class MergeVerify(MergeBase):
  66
  67     template_name = 'admin/merge-grouping.html'
  68
  69     def post(self, request):
  70
  71         try:
  72             podcasts = self._get_podcasts(request)
  73
  74         except InvalidPodcast as ip:
  75             messages.error(request,
  76                     _('No podcast with URL {url}').format(url=str(ip)))
  77
  78         grouper = PodcastGrouper(podcasts)
  79
  80         get_features = lambda (e_id, e): ((e.url, e.title), e_id)
  81
  82         num_groups = grouper.group(get_features)
  83
  84         return self.render_to_response({
  85                 'podcasts': podcasts,
  86                 'groups': num_groups,
  87             })
  88
  89
  90 class MergeProcess(MergeBase):
  91
  92     RE_EPISODE = re.compile(r'episode_([0-9a-fA-F]{32})')
  93
  94     def post(self, request):
  95
  96         try:
  97             podcasts = self._get_podcasts(request)
  98
  99         except InvalidPodcast as ip:
 100             messages.error(request,
 101                     _('No podcast with URL {url}').format(url=str(ip)))
 102
 103         grouper = PodcastGrouper(podcasts)
 104
 105         features = {}
 106         for key, feature in request.POST.items():
 107             m = self.RE_EPISODE.match(key)
 108             if m:
 109                 episode_id = m.group(1)
 110                 features[episode_id] = feature
 111
 112         get_features = lambda (e_id, e): (features[e_id], e_id)
 113
 114         num_groups = grouper.group(get_features)
 115
 116         if 'renew' in request.POST:
 117             return render(request, 'admin/merge-grouping.html', {
 118                     'podcasts': podcasts,
 119                     'groups': num_groups,
 120                 })
 121
 122
 123         elif 'merge' in request.POST:
 124
 125             actions = Counter()
 126
 127             try:
 128                 # merge podcast, reassign episodes
 129                 pm = PodcastMerger(podcasts, actions, num_groups)
 130                 pm.merge()
 131
 132             except IncorrectMergeException as ime:
 133                 messages.error(request, str(ime))
 134                 return HttpResponseRedirect(reverse('admin-merge'))
 135
 136             return render(request, 'admin/merge-finished.html', {
 137                     'actions': actions.items(),
 138                     'podcast': podcasts[0],
 139                 })
 140
 141
 142
 143 class UserAgentStats(AdminView):
 144     template_name = 'admin/useragents.html'
 145
 146     def get(self, request):
 147
 148         db = User.get_db()
 149         res = db.view('clients/by_ua_string',
 150                 group_level = 1,
 151             )
 152
 153         useragents = Counter(dict((r['key'], r['value']) for r in res))
 154
 155         if useragents:
 156             max_users = useragents.most_common(1)[0][1]
 157             total = sum(useragents.values())
 158         else:
 159             max_users = 0
 160             total = 0
 161
 162         return self.render_to_response({
 163                 'useragents': useragents.most_common(),
 164                 'max_users': max_users,
 165                 'total': total,
 166             })