apps/wiki/views.py

   1 import os
   2
   3 from django.conf import settings
   4 from django.views.generic.simple import direct_to_template
   5 from django.http import HttpResponse, Http404
   6 from django.utils import simplejson as json
   7
   8 from wiki.models import storage, Document, DocumentNotFound
   9 from wiki.forms import DocumentForm
  10 from datetime import datetime
  11
  12 # import google_diff
  13 # import difflib
  14 import nice_diff
  15 import operator
  16
  17 MAX_LAST_DOCS = 10
  18
  19 class DateTimeEncoder(json.JSONEncoder):
  20      def default(self, obj):
  21          if isinstance(obj, datetime):
  22              return datetime.ctime(obj) + " " + (datetime.tzname(obj) or 'GMT')
  23          return json.JSONEncoder.default(self, obj)
  24
  25 def document_list(request, template_name = 'wiki/document_list.html'):
  26     # TODO: find a way to cache "Storage All"
  27     return direct_to_template(request, template_name, extra_context = {
  28         'document_list': storage.all(),
  29         'last_docs': sorted(request.session.get("wiki_last_docs", {}).items(),
  30                         key=operator.itemgetter(1), reverse = True)
  31     })
  32
  33
  34 def document_detail(request, name, template_name = 'wiki/document_details.html'):
  35     try:
  36         document = storage.get(name)
  37     except DocumentNotFound:
  38         raise Http404
  39
  40     access_time = datetime.now()
  41     last_documents = request.session.get("wiki_last_docs", {})
  42     last_documents[name] = access_time
  43
  44     if len(last_documents) > MAX_LAST_DOCS:
  45         oldest_key = min(last_documents, key = last_documents.__getitem__)
  46         del last_documents[oldest_key]
  47     request.session['wiki_last_docs'] = last_documents
  48
  49     if request.method == 'POST':
  50
  51         form = DocumentForm(request.POST, instance = document)
  52         if form.is_valid():
  53             document = form.save()
  54             return HttpResponse(json.dumps({'text': document.plain_text, 'meta': document.meta(), 'revision': document.revision()}))
  55         else:
  56             return HttpResponse(json.dumps({'errors': form.errors}))
  57     else:
  58         form = DocumentForm(instance = document)
  59
  60     return direct_to_template(request, template_name, extra_context = {
  61         'document': document,
  62         'form': form,
  63     })
  64
  65
  66 def document_gallery(request, directory):
  67     try:
  68         base_dir = os.path.join(settings.MEDIA_ROOT, settings.FILEBROWSER_DIRECTORY, directory)
  69         images = [u'%s%s%s/%s' % (settings.MEDIA_URL, settings.FILEBROWSER_DIRECTORY, directory, f) for f in os.listdir(base_dir) if os.path.splitext(f)[1].lower() in (u'.jpg', u'.jpeg', u'.png')]
  70         images.sort()
  71         return HttpResponse(json.dumps(images))
  72     except (IndexError, OSError), exc:
  73         import traceback
  74         traceback.print_exc()
  75
  76         raise Http404
  77
  78 def document_diff(request, name, revA, revB):
  79     docA = storage.get(name, int(revA))
  80     docB = storage.get(name, int(revB))
  81
  82
  83     return HttpResponse(nice_diff.html_diff_table(docA.plain_text.splitlines(),
  84                                          docB.plain_text.splitlines()) )
  85
  86
  87 def document_history(reuqest, name):
  88     return HttpResponse( json.dumps(storage.history(name), cls=DateTimeEncoder), mimetype='application/json')