Refactor Note > Document, an Abstract Base Class

[oweals/karmaworld.git] / karmaworld / apps / notes / views.py

diff --git a/karmaworld/apps/notes/views.py b/karmaworld/apps/notes/views.py
index bc1583f67ca52d3b332926a819a5b73c9b93460e..f0f9a2983035bed6e3f47d56578d4a9df9731ca2 100644 (file)
--- a/karmaworld/apps/notes/views.py
+++ b/karmaworld/apps/notes/views.py
@@ -2,10 +2,15 @@
 # -*- 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
 
@@ -13,11 +18,36 @@ from karmaworld.apps.notes.models import Note
 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 """
-    # name passed to template
-    context_object_name = u"note"
     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):
@@ -33,15 +63,17 @@ class NoteSaveView(FormView, SingleObjectMixin):
         context = {
             'object': self.get_object(),
         }
+        print "get context for NoteSaveView"
         return super(NoteSaveView, self).get_context_data(**context)
 
     def get_success_url(self):
         """ On form submission success, redirect to what url """
         #TODO: redirect to note slug if possible (auto-slugify)
-        return u'/{school_slug}/{course_slug}/{pk}'.format(
+        return u'/{school_slug}/{course_slug}?url=/{school_slug}/{course_slug}/{pk}&name={name}&thankyou'.format(
                 school_slug=self.object.course.school.slug,
                 course_slug=self.object.course.slug,
-                pk=self.object.pk
+                pk=self.object.pk,
+                name=self.object.name
             )
 
     def form_valid(self, form):
@@ -49,13 +81,23 @@ class NoteSaveView(FormView, SingleObjectMixin):
             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.draft = 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 """
@@ -69,14 +111,27 @@ class NoteView(View):
         return view(request, *args, **kwargs)
 
 
-
 class RawNoteDetailView(DetailView):
     """ Class-based view for the raw note html for iframes """
     template_name = u'notes/note_raw.html'
     context_object_name = u"note"
     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)