3 # Copyright (C) 2012 FinalsClub Foundation
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
17 from karmaworld.apps.notes.models import Note
18 from karmaworld.apps.notes.forms import NoteForm
23 _path = self.object.note_file.name
24 _, _extension = os.path.splitext(_path)
25 if _extension.lower() == '.pdf':
30 if self.object.file_type == 'ppt':
35 class NoteDetailView(DetailView):
36 """ Class-based view for the note html page """
38 context_object_name = u"note" # name passed to template
40 def get_context_data(self, **kwargs):
41 """ Generate custom context for the page rendering a Note
42 + if pdf, set the `pdf` flag
44 # not current using these
45 #kwargs['hostname'] = Site.objects.get_current()
47 kwargs['pdf'] = is_pdf(self)
48 kwargs['ppt'] = is_ppt(self)
50 return super(NoteDetailView, self).get_context_data(**kwargs)
53 class NoteSaveView(FormView, SingleObjectMixin):
54 """ Save a Note and then view the page,
55 behaves the same as NoteDetailView, except for saving the
60 template_name = 'notes/note_detail.html'
62 def get_context_data(self, **kwargs):
64 'object': self.get_object(),
66 print "get context for NoteSaveView"
67 return super(NoteSaveView, self).get_context_data(**context)
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,
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
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 # User has submitted this form, so set the SHOW flag
90 self.object.draft = False
92 return super(NoteSaveView, self).form_valid(form)
94 def form_invalid(self, form):
95 """ Do stuff when the form is invalid !!! TODO """
96 # TODO: implement def form_invalid for returning a form with input and error
97 print "running form_invalid"
102 class NoteView(View):
103 """ Notes superclass that wraps http methods """
105 def get(self, request, *args, **kwargs):
106 view = NoteDetailView.as_view()
107 return view(request, *args, **kwargs)
109 def post(self, request, *args, **kwargs):
110 view = NoteSaveView.as_view()
111 return view(request, *args, **kwargs)
114 class RawNoteDetailView(DetailView):
115 """ Class-based view for the raw note html for iframes """
116 template_name = u'notes/note_raw.html'
117 context_object_name = u"note"
121 class PDFView(DetailView):
122 """ Render PDF files in an iframe based on ID"""
123 template_name = u'partial/pdfembed.html'
126 def get_context_data(self, **kwargs):
127 """ Generate a path to the pdf file associated with this note
128 by generating a path to the MEDIA_URL by hand """
131 kwargs['pdf_path'] = "{0}{1}".format(settings.MEDIA_URL,
132 os.path.basename(self.object.pdf_file.name))
134 kwargs['pdf_path'] = "{0}{1}".format(settings.MEDIA_URL,
135 os.path.basename(self.object.note_file.name))
137 return super(PDFView, self).get_context_data(**kwargs)