panikdb/views.py

   1 import datetime
   2
   3 from django.contrib.auth.decorators import login_required
   4 from django.core.exceptions import PermissionDenied
   5 from django.core.urlresolvers import reverse
   6 from django.template import loader, Context
   7 from django.views.generic.base import TemplateView, RedirectView
   8 from django.views.generic.list import ListView
   9
  10 from emissions.models import *
  11 from emissions.utils import period_program
  12
  13 import emissions.views
  14 import newsletter.views
  15
  16 class Home(TemplateView):
  17     template_name = 'home.html'
  18
  19     def get_context_data(self, **kwargs):
  20         context = super(Home, self).get_context_data(**kwargs)
  21         context['emissions'] = self.request.user.emissions.all().order_by('title')
  22         context['news_categories'] = self.request.user.news_categories.all().order_by('title')
  23         if self.request.user.has_perm('emissions.add_focus'):
  24             context['focused_items'] = Focus.objects.select_related()
  25
  26         return context
  27
  28 home = login_required(Home.as_view())
  29
  30
  31 class EmissionListView(emissions.views.EmissionListView):
  32     pass
  33
  34 emissions = login_required(EmissionListView.as_view())
  35
  36
  37
  38 class NewsListView(ListView):
  39     model = NewsCategory
  40
  41     def get_queryset(self):
  42         return NewsCategory.objects.order_by('title')
  43
  44     def get_context_data(self, **kwargs):
  45         context = super(NewsListView, self).get_context_data(**kwargs)
  46         context['unsorted_newsitems'] = NewsItem.objects.filter(category__isnull=True).order_by('title')
  47         return context
  48
  49 news = login_required(NewsListView.as_view())
  50
  51
  52 class NewsletterCreateView(newsletter.views.NewsletterCreateView):
  53     def get_initial(self):
  54         initial = super(NewsletterCreateView, self).get_initial()
  55
  56         newsletter_body = loader.get_template('newsletter/body.html')
  57         context = Context()
  58
  59         date = datetime.datetime.today()
  60         if date.weekday() > 0:
  61             date = date + datetime.timedelta(days=7-date.weekday())
  62
  63         date = datetime.datetime(*date.timetuple()[:3])
  64
  65         context['date'] = date
  66
  67         context['soundfiles'] = SoundFile.objects.filter(
  68                 podcastable=True, fragment=False).select_related().filter(
  69                         creation_timestamp__gte=date-datetime.timedelta(days=7))
  70
  71         context['diffusions'] = list(Diffusion.objects.select_related().filter(
  72                 datetime__gte=date,
  73                 datetime__lte=date+datetime.timedelta(days=7)).order_by('datetime'))
  74
  75         # remove rebroadcasts made during the same week
  76         seen_episodes = {}
  77         for diff in context['diffusions'][:]:
  78             if diff.episode.id in seen_episodes:
  79                 context['diffusions'].remove(diff)
  80             seen_episodes[diff.episode.id] = True
  81
  82         initial['text'] = newsletter_body.render(context)
  83         return initial
  84
  85 newsletter_create = NewsletterCreateView.as_view()
  86
  87
  88 class FocusSetView(RedirectView):
  89     permanent = False
  90
  91     def get_redirect_url(self, object_type, object_id):
  92         if not self.request.user.has_perm('emissions.add_focus'):
  93             raise PermissionDenied()
  94         if object_type == 'emission':
  95             emission = Emission.objects.get(id=object_id)
  96             f = Focus()
  97             f.emission = emission
  98             f.save()
  99             return reverse('emission-view', kwargs={'slug': emission.slug})
 100         if object_type == 'episode':
 101             episode = Episode.objects.get(id=object_id)
 102             f = Focus()
 103             f.episode = episode
 104             f.save()
 105             return reverse('episode-view', kwargs={
 106                 'emission_slug': episode.emission.slug, 'slug': episode.slug})
 107         if object_type == 'newsitem':
 108             newsitem = NewsItem.objects.get(id=object_id)
 109             f = Focus()
 110             f.newsitem = newsitem
 111             f.save()
 112             return reverse('newsitem-view', kwargs={'slug': newsitem.slug})
 113         if object_type == 'soundfile':
 114             soundfile = SoundFile.objects.get(id=object_id)
 115             f = Focus()
 116             f.soundfile = soundfile
 117             f.save()
 118             return reverse('episode-view', kwargs={
 119                 'slug': soundfile.episode.slug,
 120                 'emission_slug': soundfile.episode.emission.slug,
 121                 })
 122
 123 focus_set = FocusSetView.as_view()
 124
 125
 126 class FocusUnsetView(RedirectView):
 127     permanent = False
 128
 129     def get_redirect_url(self, object_type, object_id):
 130         if not self.request.user.has_perm('emissions.delete_focus'):
 131             raise PermissionDenied()
 132         if object_type == 'emission':
 133             emission = Emission.objects.get(id=object_id)
 134             Focus.objects.get(emission=emission).delete()
 135             return reverse('emission-view', kwargs={'slug': emission.slug})
 136         if object_type == 'episode':
 137             episode = Episode.objects.get(id=object_id)
 138             Focus.objects.get(episode=episode).delete()
 139             return reverse('episode-view', kwargs={
 140                 'emission_slug': episode.emission.slug, 'slug': episode.slug})
 141         if object_type == 'newsitem':
 142             newsitem = NewsItem.objects.get(id=object_id)
 143             Focus.objects.get(newsitem=newsitem).delete()
 144             return reverse('newsitem-view', kwargs={'slug': newsitem.slug})
 145         if object_type == 'soundfile':
 146             soundfile = SoundFile.objects.get(id=object_id)
 147             Focus.objects.get(soundfile=soundfile).delete()
 148             return reverse('episode-view', kwargs={
 149                 'slug': soundfile.episode.slug,
 150                 'emission_slug': soundfile.episode.emission.slug,
 151                 })
 152
 153 focus_unset = FocusUnsetView.as_view()