apps/wiki/views.py

   1 import os
   2 import functools
   3 import logging
   4 logger = logging.getLogger("fnp.wiki")
   5
   6 from django.conf import settings
   7
   8 from django.views.generic.simple import direct_to_template
   9 from django.views.decorators.http import require_POST, require_GET
  10 from django.core.urlresolvers import reverse
  11 from wiki.helpers import (JSONResponse, JSONFormInvalid, JSONServerError,
  12                 ajax_require_permission, recursive_groupby)
  13 from django import http
  14
  15 from django.contrib.auth.decorators import login_required
  16
  17 from wiki.models import getstorage, DocumentNotFound, normalize_name, split_name, join_name
  18 from wiki.forms import DocumentTextSaveForm, DocumentTagForm, DocumentCreateForm
  19 from datetime import datetime
  20 from django.utils.encoding import smart_unicode
  21 from django.utils.translation import ugettext_lazy as _
  22
  23
  24 #
  25 # Quick hack around caching problems, TODO: use ETags
  26 #
  27 from django.views.decorators.cache import never_cache
  28
  29 import wlapi
  30 import nice_diff
  31 import operator
  32
  33 MAX_LAST_DOCS = 10
  34
  35
  36 def normalized_name(view):
  37
  38     @functools.wraps(view)
  39     def decorated(request, name, *args):
  40         normalized = normalize_name(name)
  41         logger.debug('View check %r -> %r', name, normalized)
  42
  43         if normalized != name:
  44             return http.HttpResponseRedirect(
  45                         reverse('wiki_' + view.__name__, kwargs={'name': normalized}))
  46
  47         return view(request, name, *args)
  48
  49     return decorated
  50
  51
  52
  53
  54 @require_GET
  55 @normalized_name
  56 def editor_readonly(request, name, template_name='wiki/document_details_readonly.html'):
  57     name = normalize_name(name)
  58     storage = getstorage()
  59
  60     try:
  61         revision = request.GET['revision']
  62         document = storage.get(name, revision)
  63     except (KeyError, DocumentNotFound) as e:
  64         raise http.Http404
  65
  66     access_time = datetime.now()
  67     last_documents = request.session.get("wiki_last_docs", {})
  68     last_documents[name] = access_time
  69
  70     if len(last_documents) > MAX_LAST_DOCS:
  71         oldest_key = min(last_documents, key=last_documents.__getitem__)
  72         del last_documents[oldest_key]
  73     request.session['wiki_last_docs'] = last_documents
  74
  75     return direct_to_template(request, template_name, extra_context={
  76         'document': document,
  77         'document_name': document.name,
  78         'document_info': dict(document.info(), readonly=True),
  79         'document_meta': document.meta,
  80     })
  81
  82
  83 @normalized_name
  84 def create_missing(request, name):
  85     storage = getstorage()
  86
  87     if request.method == "POST":
  88         form = DocumentCreateForm(request.POST, request.FILES)
  89         if form.is_valid():
  90             doc = storage.create_document(
  91                 name=form.cleaned_data['id'],
  92                 text=form.cleaned_data['text'],
  93             )
  94
  95             return http.HttpResponseRedirect(reverse("wiki_editor", args=[doc.name]))
  96     else:
  97         form = DocumentCreateForm(initial={
  98                 "id": name.replace(" ", "_"),
  99                 "title": name.title(),
 100         })
 101
 102     return direct_to_template(request, "wiki/document_create_missing.html", extra_context={
 103         "document_name": name,
 104         "form": form,
 105     })
 106
 107
 108 @never_cache
 109 @normalized_name
 110 def text(request, name):
 111     storage = getstorage()
 112
 113     if request.method == 'POST':
 114         form = DocumentTextSaveForm(request.POST, prefix="textsave")
 115
 116         if form.is_valid():
 117             revision = form.cleaned_data['parent_revision']
 118
 119             document = storage.get_or_404(name, revision)
 120             document.text = form.cleaned_data['text']
 121
 122             comment = form.cleaned_data['comment']
 123
 124             if form.cleaned_data['stage_completed']:
 125                 comment += '\n#stage-finished: %s\n' % form.cleaned_data['stage_completed']
 126
 127             author = "%s <%s>" % (form.cleaned_data['author_name'], form.cleaned_data['author_email'])
 128
 129             storage.put(document, author=author, comment=comment, parent=revision)
 130             document = storage.get(name)
 131
 132             return JSONResponse({
 133                 'text': document.plain_text if revision != document.revision else None,
 134                 'meta': document.meta(),
 135                 'revision': document.revision,
 136             })
 137         else:
 138             return JSONFormInvalid(form)
 139     else:
 140         revision = request.GET.get("revision", None)
 141
 142         try:
 143             try:
 144                 revision = revision and int(revision)
 145                 logger.info("Fetching %s", revision)
 146                 document = storage.get(name, revision)
 147             except ValueError:
 148                 # treat as a tag
 149                 logger.info("Fetching tag %s", revision)
 150                 document = storage.get_by_tag(name, revision)
 151         except DocumentNotFound:
 152             raise http.Http404
 153
 154         return JSONResponse({
 155             'text': document.plain_text,
 156             'meta': document.meta(),
 157             'revision': document.revision,
 158         })
 159
 160
 161 @never_cache
 162 @normalized_name
 163 @require_POST
 164 def revert(request, name):
 165     storage = getstorage()
 166     revision = request.POST['target_revision']
 167
 168     try:
 169         document = storage.revert(name, revision)
 170
 171         return JSONResponse({
 172             'text': document.plain_text if revision != document.revision else None,
 173             'meta': document.meta(),
 174             'revision': document.revision,
 175         })
 176     except DocumentNotFound:
 177         raise http.Http404
 178
 179 @never_cache
 180 def gallery(request, directory):
 181     try:
 182         base_url = ''.join((
 183                         smart_unicode(settings.MEDIA_URL),
 184                         smart_unicode(settings.FILEBROWSER_DIRECTORY),
 185                         smart_unicode(directory)))
 186
 187         base_dir = os.path.join(
 188                     smart_unicode(settings.MEDIA_ROOT),
 189                     smart_unicode(settings.FILEBROWSER_DIRECTORY),
 190                     smart_unicode(directory))
 191
 192         def map_to_url(filename):
 193             return "%s/%s" % (base_url, smart_unicode(filename))
 194
 195         def is_image(filename):
 196             return os.path.splitext(f)[1].lower() in (u'.jpg', u'.jpeg', u'.png')
 197
 198         images = [map_to_url(f) for f in map(smart_unicode, os.listdir(base_dir)) if is_image(f)]
 199         images.sort()
 200         return JSONResponse(images)
 201     except (IndexError, OSError) as e:
 202         logger.exception("Unable to fetch gallery")
 203         raise http.Http404
 204
 205
 206 @never_cache
 207 @normalized_name
 208 def diff(request, name):
 209     storage = getstorage()
 210
 211     revA = int(request.GET.get('from', 0))
 212     revB = int(request.GET.get('to', 0))
 213
 214     if revA > revB:
 215         revA, revB = revB, revA
 216
 217     if revB == 0:
 218         revB = None
 219
 220     docA = storage.get_or_404(name, int(revA))
 221     docB = storage.get_or_404(name, int(revB))
 222
 223     return http.HttpResponse(nice_diff.html_diff_table(docA.plain_text.splitlines(),
 224                                          docB.plain_text.splitlines(), context=3))
 225
 226
 227 @never_cache
 228 @normalized_name
 229 def history(request, name):
 230     storage = getstorage()
 231
 232     # TODO: pagination
 233     changesets = list(storage.history(name))
 234
 235     return JSONResponse(changesets)
 236
 237
 238 @require_POST
 239 @ajax_require_permission('wiki.can_change_tags')
 240 def add_tag(request, name):
 241     name = normalize_name(name)
 242     storage = getstorage()
 243
 244     form = DocumentTagForm(request.POST, prefix="addtag")
 245     if form.is_valid():
 246         doc = storage.get_or_404(form.cleaned_data['id'])
 247         doc.add_tag(tag=form.cleaned_data['tag'],
 248                     revision=form.cleaned_data['revision'],
 249                     author=request.user.username)
 250         return JSONResponse({"message": _("Tag added")})
 251     else:
 252         return JSONFormInvalid(form)
 253
 254
 255 @require_POST
 256 @ajax_require_permission('wiki.can_publish')
 257 def publish(request, name):
 258     name = normalize_name(name)
 259
 260     storage = getstorage()
 261     document = storage.get_by_tag(name, "ready_to_publish")
 262
 263     api = wlapi.WLAPI(**settings.WL_API_CONFIG)
 264
 265     try:
 266         return JSONResponse({"result": api.publish_book(document)})
 267     except wlapi.APICallException, e:
 268         return JSONServerError({"message": str(e)})
 269
 270 @login_required
 271 def status_report(request):
 272     pass
 273