karmaworld/apps/notes/views.py

   1 #!/usr/bin/env python
   2 # -*- coding:utf8 -*-
   3 # Copyright (C) 2012  FinalsClub Foundation
   4
   5 import os
   6
   7 from django.conf import settings
   8 from django.contrib.sites.models import Site
   9 from django.http import HttpResponse
  10 from django.views.generic import DetailView
  11 from django.views.generic import FormView
  12 from django.views.generic import View
  13 from django.views.generic import TemplateView
  14 from django.views.generic.detail import SingleObjectMixin
  15 from django.shortcuts import get_object_or_404
  16
  17 from karmaworld.apps.notes.models import Note
  18 from karmaworld.apps.notes.forms import NoteForm
  19
  20
  21
  22 def is_pdf(self):
  23     _path = self.object.note_file.name
  24     _, _extension = os.path.splitext(_path)
  25     if _extension.lower() == '.pdf':
  26         return True
  27     return False
  28
  29 def is_ppt(self):
  30     if self.object.file_type == 'ppt':
  31         return True
  32     return False
  33
  34
  35 class NoteDetailView(DetailView):
  36     """ Class-based view for the note html page """
  37     model = Note
  38     context_object_name = u"note" # name passed to template
  39
  40     def get_context_data(self, **kwargs):
  41         """ Generate custom context for the page rendering a Note
  42             + if pdf, set the `pdf` flag
  43         """
  44         # not current using these
  45         #kwargs['hostname'] = Site.objects.get_current()
  46
  47         kwargs['pdf'] = is_pdf(self)
  48         kwargs['ppt'] = is_ppt(self)
  49
  50         return super(NoteDetailView, self).get_context_data(**kwargs)
  51
  52
  53 class NoteSaveView(FormView, SingleObjectMixin):
  54     """ Save a Note and then view the page,
  55         behaves the same as NoteDetailView, except for saving the
  56         NoteForm ModelForm
  57     """
  58     form_class = NoteForm
  59     model = Note
  60     template_name = 'notes/note_detail.html'
  61
  62     def get_context_data(self, **kwargs):
  63         context = {
  64             'object': self.get_object(),
  65         }
  66         print "get context for NoteSaveView"
  67         return super(NoteSaveView, self).get_context_data(**context)
  68
  69     def get_success_url(self):
  70         """ On form submission success, redirect to what url """
  71         #TODO: redirect to note slug if possible (auto-slugify)
  72         return u'/{school_slug}/{course_slug}?url=/{school_slug}/{course_slug}/{pk}&name={name}&thankyou'.format(
  73                 school_slug=self.object.course.school.slug,
  74                 course_slug=self.object.course.slug,
  75                 pk=self.object.pk,
  76                 name=self.object.name
  77             )
  78
  79     def form_valid(self, form):
  80         """ Actions to take if the submitted form is valid
  81             namely, saving the new data to the existing note object
  82         """
  83         self.object = self.get_object()
  84         if len(form.cleaned_data['name'].strip()) > 0:
  85             self.object.name = form.cleaned_data['name']
  86         self.object.year = form.cleaned_data['year']
  87         # use *arg expansion to pass tags a list of tags
  88         self.object.tags.add(*form.cleaned_data['tags'])
  89         self.object.save()
  90         return super(NoteSaveView, self).form_valid(form)
  91
  92     def form_invalid(self, form):
  93         """ Do stuff when the form is invalid !!! TODO """
  94         # TODO: implement def form_invalid for returning a form with input and error
  95         print "running form_invalid"
  96         print form
  97         print form.errors
  98
  99
 100 class NoteView(View):
 101     """ Notes superclass that wraps http methods """
 102
 103     def get(self, request, *args, **kwargs):
 104         view = NoteDetailView.as_view()
 105         return view(request, *args, **kwargs)
 106
 107     def post(self, request, *args, **kwargs):
 108         view = NoteSaveView.as_view()
 109         return view(request, *args, **kwargs)
 110
 111
 112 class RawNoteDetailView(DetailView):
 113     """ Class-based view for the raw note html for iframes """
 114     template_name = u'notes/note_raw.html'
 115     context_object_name = u"note"
 116     model = Note
 117
 118
 119 class PDFView(DetailView):
 120     """ Render PDF files in an iframe based on ID"""
 121     template_name = u'partial/pdfembed.html'
 122     model = Note
 123
 124     def get_context_data(self, **kwargs):
 125         """ Generate a path to the pdf file associated with this note
 126             by generating a path to the MEDIA_URL by hand """
 127
 128         if is_ppt:
 129             kwargs['pdf_path'] = "{0}{1}".format(settings.MEDIA_URL,
 130                 os.path.basename(self.object.pdf_file.name))
 131         elif is_pdf:
 132             kwargs['pdf_path'] = "{0}{1}".format(settings.MEDIA_URL,
 133                 os.path.basename(self.object.note_file.name))
 134
 135         return super(PDFView, self).get_context_data(**kwargs)