apps/explorer/views.py

   1 # -*- coding: utf-8 -*-
   2 import urllib2
   3 import hg
   4 from librarian import html, parser, dcparser, ParseError, ValidationError
   5
   6 from django.conf import settings
   7 from django.contrib.auth.decorators import login_required, permission_required
   8
   9 from django.core.urlresolvers import reverse
  10 from django.http import HttpResponseRedirect, HttpResponse
  11 from django.utils import simplejson as json
  12 from django.views.generic.simple import direct_to_template
  13
  14 from explorer import forms, models
  15 from toolbar import models as toolbar_models
  16
  17 #
  18 # Some useful decorators
  19 #
  20 def with_repo(view):
  21     """Open a repository for this view"""
  22     def view_with_repo(request, *args, **kwargs):
  23         kwargs['repo'] = hg.Repository(settings.REPOSITORY_PATH)
  24         return view(request, *args, **kwargs)
  25     return view_with_repo
  26
  27 #
  28 def ajax_login_required(view):
  29     """Similar ro @login_required, but instead of redirect,
  30     just return some JSON stuff with error."""
  31     def view_with_auth(request, *args, **kwargs):
  32         if request.user.is_authenticated():
  33             return view(request, *args, **kwargs)
  34         # not authenticated
  35         return HttpResponse( json.dumps({'result': 'access_denied'}) );
  36     return view_with_auth
  37
  38 #
  39 # View all files
  40 #
  41 @with_repo
  42 def file_list(request, repo):
  43     latest_default = repo.repo.branchtags()['default']
  44     files = list( repo.repo[latest_default] )
  45     bookform = forms.BookUploadForm()
  46
  47     return direct_to_template(request, 'explorer/file_list.html', extra_context={
  48         'files': files, 'bookform': bookform,
  49     })
  50
  51 @permission_required('explorer.can_add_files')
  52 @with_repo
  53 def file_upload(request, repo):
  54     other_errors = []
  55     if request.method == 'POST':
  56         form = forms.BookUploadForm(request.POST, request.FILES)
  57         if form.is_valid():
  58             try:
  59                 # prepare the data
  60                 f = request.FILES['file']
  61                 decoded = f.read().decode('utf-8')
  62
  63                 def upload_action():
  64                     print 'Adding file: %s' % f.name
  65                     repo._add_file(f.name, decoded.encode('utf-8') )
  66                     repo._commit(
  67                         message="File %s uploaded from platform by %s" %\
  68                             (f.name, request.user.username), \
  69                         user=request.user.username \
  70                     )
  71
  72                     # end of upload
  73
  74                 repo.in_branch(upload_action, 'default')
  75
  76                 # if everything is ok, redirect to the editor
  77                 return HttpResponseRedirect( reverse('editor_view',
  78                         kwargs={'path': f.name}) )
  79
  80             except hg.RepositoryException, e:
  81                 other_errors.append(u'Błąd repozytorium: ' + unicode(e) )
  82             except UnicodeDecodeError, e:
  83                 other_errors.append(u'Niepoprawne kodowanie pliku: ' + e.reason \
  84                  + u'. Żądane kodowanie: ' + e.encoding)
  85         # invalid form
  86
  87     # get
  88     form = forms.BookUploadForm()
  89     return direct_to_template(request, 'explorer/file_upload.html',
  90         extra_context = {'form' : form, 'other_errors': other_errors})
  91
  92 #
  93 # Edit the file
  94 #
  95
  96 @ajax_login_required
  97 @with_repo
  98 def file_xml(request, repo, path):
  99     if request.method == 'POST':
 100         errors = None
 101         form = forms.BookForm(request.POST)
 102         if form.is_valid():
 103             print 'Saving whole text.', request.user.username
 104             def save_action():
 105                 print 'In branch: ' + repo.repo[None].branch()
 106                 repo._add_file(path, form.cleaned_data['content'])
 107                 repo._commit(message=(form.cleaned_data['commit_message'] or 'Lokalny zapis platformy.'),\
 108                      user=request.user.username)
 109             try:
 110                 # wczytaj dokument z ciągu znaków -> weryfikacja
 111                 document = parser.WLDocument.from_string(form.cleaned_data['content'])
 112
 113                 #  save to user's branch
 114                 repo.in_branch(save_action, models.user_branch(request.user) );
 115             except (ParseError, ValidationError), e:
 116                 errors = [e.message]
 117
 118         if not errors:
 119             errors = dict( (field[0], field[1].as_text()) for field in form.errors.iteritems() )
 120
 121         return HttpResponse(json.dumps({'result': errors and 'error' or 'ok', 'errors': errors}));
 122
 123     form = forms.BookForm()
 124     data = repo.get_file(path, models.user_branch(request.user))
 125     form.fields['content'].initial = data
 126     return HttpResponse( json.dumps({'result': 'ok', 'content': data}) )
 127
 128 @ajax_login_required
 129 @with_repo
 130 def file_dc(request, path, repo):
 131     errors = None
 132
 133     if request.method == 'POST':
 134         form = forms.DublinCoreForm(request.POST)
 135
 136         if form.is_valid():
 137             def save_action():
 138                 file_contents = repo._get_file(path)
 139
 140                 # wczytaj dokument z repozytorium
 141                 document = parser.WLDocument.from_string(file_contents)
 142                 document.book_info.update(form.cleaned_data)
 143
 144                 print "SAVING DC"
 145
 146                 # zapisz
 147                 repo._write_file(path, document.serialize())
 148                 repo._commit( \
 149                     message=(form.cleaned_data['commit_message'] or 'Lokalny zapis platformy.'), \
 150                     user=request.user.username )
 151
 152             try:
 153                 repo.in_branch(save_action, models.user_branch(request.user) )
 154             except (ParseError, ValidationError), e:
 155                 errors = [e.message]
 156
 157         if errors is None:
 158             errors = ["Pole '%s': %s\n" % (field[0], field[1].as_text()) for field in form.errors.iteritems()]
 159
 160         return HttpResponse( json.dumps({'result': errors and 'error' or 'ok', 'errors': errors}) );
 161
 162     # this is unused currently, but may come in handy
 163     content = []
 164
 165     try:
 166         fulltext = repo.get_file(path, models.user_branch(request.user))
 167         bookinfo = dcparser.BookInfo.from_string(fulltext)
 168         content = bookinfo.to_dict()
 169     except (ParseError, ValidationError), e:
 170         errors = [e.message]
 171
 172     return HttpResponse( json.dumps({'result': errors and 'error' or 'ok',
 173         'errors': errors, 'content': content }) )
 174
 175 # Display the main editor view
 176
 177 @login_required
 178 @with_repo
 179 def display_editor(request, path, repo):
 180
 181     if not repo.file_exists(path, models.user_branch(request.user)):
 182         try:
 183             data = repo.get_file(path, 'default')
 184             print type(data)
 185
 186             def new_file():
 187                 repo._add_file(path, data)
 188                 repo._commit(message='File import from default branch',
 189                     user=request.user.username)
 190
 191             repo.in_branch(new_file, models.user_branch(request.user) )
 192         except hg.RepositoryException, e:
 193             return direct_to_template(request, 'explorer/file_unavailble.html',\
 194                 extra_context = { 'path': path, 'error': e })
 195
 196     return direct_to_template(request, 'explorer/editor.html', extra_context={
 197         'hash': path,
 198         'panel_list': ['lewy', 'prawy'],
 199         'scriptlets': toolbar_models.Scriptlet.objects.all()
 200     })
 201
 202 # ===============
 203 # = Panel views =
 204 # ===============
 205
 206 @ajax_login_required
 207 @with_repo
 208 def xmleditor_panel(request, path, repo):
 209     text = repo.get_file(path, models.user_branch(request.user))
 210
 211     return direct_to_template(request, 'explorer/panels/xmleditor.html', extra_context={
 212         'fpath': path,
 213         'text': text,
 214     })
 215
 216
 217 @ajax_login_required
 218 def gallery_panel(request, path):
 219     return direct_to_template(request, 'explorer/panels/gallery.html', extra_context={
 220         'fpath': path,
 221         'form': forms.ImageFoldersForm(),
 222     })
 223
 224 @ajax_login_required
 225 @with_repo
 226 def htmleditor_panel(request, path, repo):
 227     user_branch = models.user_branch(request.user)
 228     try:
 229         return direct_to_template(request, 'explorer/panels/htmleditor.html', extra_context={
 230             'fpath': path,
 231             'html': html.transform(repo.get_file(path, user_branch), is_file=False),
 232         })
 233     except (ParseError, ValidationError), e:
 234         return direct_to_template(request, 'explorer/panels/parse_error.html', extra_context={
 235             'fpath': path, 'exception_type': type(e).__name__, 'exception': e, 'panel_name': 'Edytor HTML'})
 236
 237 @ajax_login_required
 238 @with_repo
 239 def dceditor_panel(request, path, repo):
 240     user_branch = models.user_branch(request.user)
 241
 242     try:
 243         doc_text = repo.get_file(path, user_branch)
 244         document = parser.WLDocument.from_string(doc_text)
 245         form = forms.DublinCoreForm(info=document.book_info)
 246         return direct_to_template(request, 'explorer/panels/dceditor.html', extra_context={
 247             'fpath': path,
 248             'form': form,
 249         })
 250     except (ParseError, ValidationError), e:
 251         return direct_to_template(request, 'explorer/panels/parse_error.html', extra_context={
 252             'fpath': path, 'exception_type': type(e).__name__, 'exception': e,
 253             'panel_name': 'Edytor DublinCore'})
 254
 255 # =================
 256 # = Utility views =
 257 # =================
 258 @ajax_login_required
 259 def folder_images(request, folder):
 260     return direct_to_template(request, 'explorer/folder_images.html', extra_context={
 261         'images': models.get_images_from_folder(folder),
 262     })
 263
 264
 265 def _add_references(message, issues):
 266     return message + " - " + ", ".join(map(lambda issue: "Refs #%d" % issue['id'], issues))
 267
 268 def _get_issues_for_file(path):
 269     if not path.endswith('.xml'):
 270         raise ValueError('Path must end with .xml')
 271
 272     book_id = path[:-4]
 273     uf = None
 274
 275     try:
 276         uf = urllib2.urlopen(settings.REDMINE_URL + 'publications/issues/%s.json' % book_id)
 277         return json.loads(uf.read())
 278     except urllib2.HTTPError:
 279         return []
 280     finally:
 281         if uf: uf.close()
 282
 283
 284 # =================
 285 # = Pull requests =
 286 # =================
 287 def pull_requests(request):
 288     return direct_to_template(request, 'manager/pull_request.html', extra_context = {
 289         'objects': models.PullRequest.objects.all()} )