apps/wiki/views.py

   1 # -*- coding: utf-8 -*-
   2 from datetime import datetime
   3 import os
   4 import logging
   5 import urllib
   6 import json
   7
   8 from django.conf import settings
   9 from django.core.urlresolvers import reverse
  10 from django import http
  11 from django.http import Http404, HttpResponseForbidden
  12 from django.middleware.gzip import GZipMiddleware
  13 from django.utils.decorators import decorator_from_middleware
  14 from django.utils.encoding import smart_unicode
  15 from django.utils.formats import localize
  16 from django.utils.translation import ugettext as _
  17 from django.views.decorators.http import require_POST, require_GET
  18 from django.shortcuts import get_object_or_404, render
  19
  20 from catalogue.models import Book, Chunk, Template
  21 import nice_diff
  22 from wiki import forms
  23 from wiki.helpers import JSONResponse, JSONFormInvalid, ajax_require_permission
  24 from wiki.models import Theme
  25
  26 #
  27 # Quick hack around caching problems, TODO: use ETags
  28 #
  29 from django.views.decorators.cache import never_cache
  30
  31 logger = logging.getLogger("fnp.wiki")
  32
  33 MAX_LAST_DOCS = 10
  34
  35
  36 def get_history(chunk):
  37     changes = []
  38     for change in chunk.history():
  39         changes.append({
  40                 "version": change.revision,
  41                 "description": change.description,
  42                 "author": change.author_str(),
  43                 "date": localize(change.created_at),
  44                 "publishable": _("Publishable") + "\n" if change.publishable else "",
  45                 "tag": ',\n'.join(unicode(tag) for tag in change.tags.all()),
  46                 "published": (
  47                     _("Published") + ": " +
  48                     localize(change.publish_log.order_by('-book_record__timestamp')[0].book_record.timestamp)
  49                     if change.publish_log.exists() else ""),
  50             })
  51     return changes
  52
  53
  54 @never_cache
  55 def editor(request, slug, chunk=None, template_name='wiki/bootstrap.html'):
  56     try:
  57         chunk = Chunk.get(slug, chunk)
  58     except Chunk.MultipleObjectsReturned:
  59         raise Http404
  60     except Chunk.DoesNotExist:
  61         if chunk is None:
  62             try:
  63                 Book.objects.get(slug=slug)
  64             except Book.DoesNotExist:
  65                 return http.HttpResponseRedirect(reverse("catalogue_create_missing", args=[slug]))
  66             else:
  67                 raise Http404
  68         else:
  69             raise Http404
  70     if not chunk.book.accessible(request):
  71         return HttpResponseForbidden("Not authorized.")
  72
  73     access_time = datetime.now()
  74     last_books = request.session.get("wiki_last_books", {})
  75     last_books[slug, chunk.slug] = {
  76         'time': access_time,
  77         'title': chunk.pretty_name(),
  78     }
  79
  80     if len(last_books) > MAX_LAST_DOCS:
  81         oldest_key = min(last_books, key=lambda x: last_books[x]['time'])
  82         del last_books[oldest_key]
  83     request.session['wiki_last_books'] = last_books
  84
  85     save_form = forms.DocumentTextSaveForm(user=request.user, chunk=chunk, prefix="textsave")
  86     try:
  87         version = int(request.GET.get('version', ''))
  88     except ValueError:
  89         version = None
  90     if version:
  91         text = chunk.at_revision(version).materialize()
  92     else:
  93         text = chunk.materialize()
  94     materials = set(os.listdir(os.path.join(settings.MEDIA_ROOT, settings.IMAGE_DIR, chunk.book.gallery)))
  95     materials.update([n.rsplit('.', 1)[0] for n in materials])
  96     return render(request, template_name, {
  97         'serialized_document_data': json.dumps({
  98             'document': text,
  99             'document_id': chunk.id,
 100             'title': chunk.book.title,
 101             'history': get_history(chunk),
 102             'version': version or chunk.revision(),
 103             'stage': chunk.stage.name if chunk.stage else None,
 104             'assignment': chunk.user.username if chunk.user else None
 105         }),
 106         'serialized_templates': json.dumps([
 107             {'id': t.id, 'name': t.name, 'content': t.content} for t in Template.objects.filter(is_partial=True)
 108         ]),
 109         'forms': {
 110             "text_save": save_form,
 111             "text_revert": forms.DocumentTextRevertForm(prefix="textrevert")
 112         },
 113         'tags': list(save_form.fields['stage_completed'].choices),
 114         'can_pubmark': request.user.has_perm('catalogue.can_pubmark'),
 115         'slug': chunk.book.slug,
 116         'gallery': chunk.book.gallery,
 117         'materials': json.dumps(list(materials)),
 118     })
 119
 120
 121 @require_GET
 122 def editor_readonly(request, slug, chunk=None, template_name='wiki/document_details_readonly.html'):
 123     try:
 124         chunk = Chunk.get(slug, chunk)
 125         revision = request.GET['revision']
 126     except (Chunk.MultipleObjectsReturned, Chunk.DoesNotExist, KeyError):
 127         raise Http404
 128     if not chunk.book.accessible(request):
 129         return HttpResponseForbidden("Not authorized.")
 130
 131     access_time = datetime.now()
 132     last_books = request.session.get("wiki_last_books", {})
 133     last_books[slug, chunk.slug] = {
 134         'time': access_time,
 135         'title': chunk.book.title,
 136         }
 137
 138     if len(last_books) > MAX_LAST_DOCS:
 139         oldest_key = min(last_books, key=lambda x: last_books[x]['time'])
 140         del last_books[oldest_key]
 141     request.session['wiki_last_books'] = last_books
 142
 143     return render(request, template_name, {
 144         'chunk': chunk,
 145         'revision': revision,
 146         'readonly': True,
 147         'REDMINE_URL': settings.REDMINE_URL,
 148     })
 149
 150
 151 @never_cache
 152 @decorator_from_middleware(GZipMiddleware)
 153 def text(request, chunk_id):
 154     doc = get_object_or_404(Chunk, pk=chunk_id)
 155     if not doc.book.accessible(request):
 156         return HttpResponseForbidden("Not authorized.")
 157
 158     if request.method == 'POST':
 159         form = forms.DocumentTextSaveForm(request.POST, user=request.user, chunk=doc, prefix="textsave")
 160         if form.is_valid():
 161             form.save()
 162             parent_revision = form.cleaned_data['parent_revision']
 163             revision = doc.revision()
 164             return JSONResponse({
 165                 'text': doc.materialize() if parent_revision != revision else None,
 166                 'version': revision,
 167                 'stage': doc.stage.name if doc.stage else None,
 168                 'assignment': doc.user.username if doc.user else None
 169             })
 170         else:
 171             return JSONFormInvalid(form)
 172     else:
 173         revision = request.GET.get("revision", None)
 174
 175         try:
 176             revision = int(revision)
 177         except (ValueError, TypeError):
 178             revision = doc.revision()
 179
 180         if revision is not None:
 181             text = doc.at_revision(revision).materialize()
 182         else:
 183             text = ''
 184
 185         return JSONResponse({
 186             'text': text,
 187             'meta': {},
 188             'revision': revision,
 189         })
 190
 191
 192 @never_cache
 193 @require_POST
 194 def revert(request, chunk_id):
 195     form = forms.DocumentTextRevertForm(request.POST, prefix="textrevert")
 196     if form.is_valid():
 197         doc = get_object_or_404(Chunk, pk=chunk_id)
 198         if not doc.book.accessible(request):
 199             return HttpResponseForbidden("Not authorized.")
 200
 201         revision = form.cleaned_data['revision']
 202
 203         comment = form.cleaned_data['comment']
 204         comment += "\n#revert to %s" % revision
 205
 206         if request.user.is_authenticated():
 207             author = request.user
 208         else:
 209             author = None
 210
 211         before = doc.revision()
 212         logger.info("Reverting %s to %s", chunk_id, revision)
 213         doc.at_revision(revision).revert(author=author, description=comment)
 214
 215         return JSONResponse({
 216             'document': doc.materialize() if before != doc.revision() else None,
 217             'version': doc.revision(),
 218         })
 219     else:
 220         return JSONFormInvalid(form)
 221
 222
 223 @never_cache
 224 def gallery(request, directory):
 225     try:
 226         base_url = ''.join((
 227                         smart_unicode(settings.MEDIA_URL),
 228                         smart_unicode(settings.IMAGE_DIR),
 229                         smart_unicode(directory)))
 230
 231         base_dir = os.path.join(
 232                     smart_unicode(settings.MEDIA_ROOT),
 233                     smart_unicode(settings.IMAGE_DIR),
 234                     smart_unicode(directory))
 235
 236         def map_to_url(filename):
 237             return urllib.quote("%s/%s" % (base_url, smart_unicode(filename)))
 238
 239         def is_image(filename):
 240             return os.path.splitext(filename)[1].lower() in (u'.jpg', u'.jpeg', u'.png')
 241
 242         images = [map_to_url(f) for f in map(smart_unicode, os.listdir(base_dir)) if is_image(f)]
 243         images.sort()
 244
 245         if not request.user.is_authenticated():
 246             return HttpResponseForbidden("Not authorized.")
 247
 248         return JSONResponse(images)
 249     except (IndexError, OSError):
 250         logger.exception("Unable to fetch gallery")
 251         raise http.Http404
 252
 253
 254 @never_cache
 255 def diff(request, chunk_id):
 256     revA = int(request.GET.get('from', 0))
 257     revB = int(request.GET.get('to', 0))
 258
 259     if revA > revB:
 260         revA, revB = revB, revA
 261
 262     if revB == 0:
 263         revB = None
 264
 265     doc = get_object_or_404(Chunk, pk=chunk_id)
 266     if not doc.book.accessible(request):
 267         return HttpResponseForbidden("Not authorized.")
 268
 269     # allow diff from the beginning
 270     if revA:
 271         docA = doc.at_revision(revA).materialize()
 272     else:
 273         docA = ""
 274     docB = doc.at_revision(revB).materialize()
 275
 276     return http.HttpResponse(nice_diff.html_diff_table(docA.splitlines(), docB.splitlines(), context=3))
 277
 278
 279 @never_cache
 280 def revision(request, chunk_id):
 281     doc = get_object_or_404(Chunk, pk=chunk_id)
 282     if not doc.book.accessible(request):
 283         return HttpResponseForbidden("Not authorized.")
 284     return http.HttpResponse(str(doc.revision()))
 285
 286
 287 @never_cache
 288 def history(request, chunk_id):
 289     # TODO: pagination
 290     doc = get_object_or_404(Chunk, pk=chunk_id)
 291     if not doc.book.accessible(request):
 292         return HttpResponseForbidden("Not authorized.")
 293
 294     return JSONResponse(get_history(doc))
 295
 296
 297 @require_POST
 298 @ajax_require_permission('catalogue.can_pubmark')
 299 def pubmark(request, chunk_id):
 300     form = forms.DocumentPubmarkForm(request.POST, prefix="pubmark")
 301     if form.is_valid():
 302         doc = get_object_or_404(Chunk, pk=chunk_id)
 303         if not doc.book.accessible(request):
 304             return HttpResponseForbidden("Not authorized.")
 305
 306         revision = form.cleaned_data['revision']
 307         publishable = form.cleaned_data['publishable']
 308         change = doc.at_revision(revision)
 309         if publishable != change.publishable:
 310             change.set_publishable(publishable)
 311             return JSONResponse({"message": _("Revision marked")})
 312         else:
 313             return JSONResponse({"message": _("Nothing changed")})
 314     else:
 315         return JSONFormInvalid(form)
 316
 317
 318 def themes(request):
 319     prefix = request.GET.get('q', '')
 320     return http.HttpResponse('\n'.join([str(t) for t in Theme.objects.filter(name__istartswith=prefix)]))