# -*- coding:utf8 -*-
# Copyright (C) 2012 FinalsClub Foundation
+import os
+
+from django.conf import settings
+from django.contrib.sites.models import Site
from django.http import HttpResponse
from django.views.generic import DetailView
from django.views.generic import FormView
from django.views.generic import View
+from django.views.generic import TemplateView
from django.views.generic.detail import SingleObjectMixin
from django.shortcuts import get_object_or_404
from karmaworld.apps.notes.forms import NoteForm
+
+def is_pdf(self):
+ _path = self.object.note_file.name
+ _, _extension = os.path.splitext(_path)
+ if _extension.lower() == '.pdf':
+ return True
+ return False
+
+def is_ppt(self):
+ if self.object.file_type == 'ppt':
+ return True
+ return False
+
+
class NoteDetailView(DetailView):
""" Class-based view for the note html page """
model = Note
context_object_name = u"note" # name passed to template
+ def get_context_data(self, **kwargs):
+ """ Generate custom context for the page rendering a Note
+ + if pdf, set the `pdf` flag
+ """
+ # not current using these
+ #kwargs['hostname'] = Site.objects.get_current()
+
+ kwargs['pdf'] = is_pdf(self)
+ kwargs['ppt'] = is_ppt(self)
+
+ return super(NoteDetailView, self).get_context_data(**kwargs)
+
class NoteSaveView(FormView, SingleObjectMixin):
""" Save a Note and then view the page,
behaves the same as NoteDetailView, except for saving the
NoteForm ModelForm
"""
- # TODO: implement def form_invalid for returning a form with input and error
form_class = NoteForm
model = Note
template_name = 'notes/note_detail.html'
context = {
'object': self.get_object(),
}
+ print "get context for NoteSaveView"
return super(NoteSaveView, self).get_context_data(**context)
def get_success_url(self):
namely, saving the new data to the existing note object
"""
self.object = self.get_object()
- self.object.name = form.cleaned_data['name']
- self.object.desc = form.cleaned_data['desc']
+ if len(form.cleaned_data['name'].strip()) > 0:
+ self.object.name = form.cleaned_data['name']
+ self.object.year = form.cleaned_data['year']
# use *arg expansion to pass tags a list of tags
self.object.tags.add(*form.cleaned_data['tags'])
+ # User has submitted this form, so set the SHOW flag
+ self.object.is_hidden = False
self.object.save()
return super(NoteSaveView, self).form_valid(form)
+ def form_invalid(self, form):
+ """ Do stuff when the form is invalid !!! TODO """
+ # TODO: implement def form_invalid for returning a form with input and error
+ print "running form_invalid"
+ print form
+ print form.errors
+
class NoteView(View):
""" Notes superclass that wraps http methods """
model = Note
-def raw_file(request, pk):
- """ Display the raw html from a Note object for embedding in an iframe """
- note = get_object_or_404(Note, pk=pk)
- return HttpResponse(note.html)
+class PDFView(DetailView):
+ """ Render PDF files in an iframe based on ID"""
+ template_name = u'partial/pdfembed.html'
+ model = Note
+
+ def get_context_data(self, **kwargs):
+ """ Generate a path to the pdf file associated with this note
+ by generating a path to the MEDIA_URL by hand """
+
+ if is_ppt(self):
+ kwargs['pdf_path'] = "{0}{1}".format(settings.MEDIA_URL,
+ os.path.basename(self.object.pdf_file.name))
+ elif is_pdf(self):
+ kwargs['pdf_path'] = "{0}{1}".format(settings.MEDIA_URL,
+ os.path.basename(self.object.note_file.name))
+
+ return super(PDFView, self).get_context_data(**kwargs)