apps/filebrowser/views.py

   1 # coding: utf-8
   2
   3 # general imports
   4 import re
   5
   6 # django imports
   7 from django.shortcuts import render_to_response, HttpResponse
   8 from django.template import RequestContext as Context
   9 from django.http import HttpResponseRedirect
  10 from django.contrib.admin.views.decorators import staff_member_required
  11 from django.views.decorators.cache import never_cache
  12 from django.utils.translation import ugettext as _
  13 from django.conf import settings
  14 from django import forms
  15 from django.core.urlresolvers import reverse
  16 from django.core.exceptions import ImproperlyConfigured
  17 from django.dispatch import Signal
  18 from django.views.decorators.csrf import csrf_exempt
  19
  20 from django.utils.encoding import smart_unicode, smart_str
  21
  22 # filebrowser imports
  23 from filebrowser.fb_settings import *
  24
  25 from filebrowser.functions import (
  26         _url_to_path, _path_to_url, _get_path, _get_file, _get_version_path,
  27         _get_breadcrumbs, _get_filterdate, _get_settings_var, _handle_file_upload,
  28         _get_file_type, _url_join, _convert_filename)
  29
  30 from filebrowser.templatetags.fb_tags import query_helper
  31 from filebrowser.base import FileObject
  32 from filebrowser.decorators import flash_login_required
  33
  34 # Precompile regular expressions
  35 filter_re = []
  36
  37 for exp in EXCLUDE:
  38     filter_re.append(re.compile(exp))
  39 for k, v in VERSIONS.iteritems():
  40     exp = (r'_%s.(%s)') % (k, '|'.join(EXTENSION_LIST))
  41     filter_re.append(re.compile(exp))
  42
  43
  44 def browse(request):
  45     """
  46     Browse Files/Directories.
  47     """
  48
  49     # QUERY / PATH CHECK
  50     query = request.GET
  51     path = _get_path(query.get('dir', ''))
  52     directory = _get_path('')
  53
  54     if path is None:
  55         msg = _('Directory/File does not exist.')
  56         request.user.message_set.create(message=msg)
  57         if directory is None:
  58             # The DIRECTORY does not exist, raise an error to prevent eternal redirecting.
  59             raise ImproperlyConfigured(_("Error finding upload directory: %s. Maybe it does not exist?" % os.path.join(MEDIA_ROOT, DIRECTORY)))
  60         redirect_url = reverse("fb_browse") + query_helper(query, "", "dir")
  61         return HttpResponseRedirect(redirect_url)
  62     abs_path = os.path.join(MEDIA_ROOT, DIRECTORY, path)
  63
  64     # INITIAL VARIABLES
  65     results_var = {'results_total': 0, 'results_current': 0, 'delete_total': 0, 'images_total': 0, 'select_total': 0}
  66     counter = {}
  67     for k, v in EXTENSIONS.iteritems():
  68         counter[k] = 0
  69
  70     dir_list = os.listdir(abs_path)
  71     files = []
  72     # print "LISTING FILES: ", dir_list
  73     for file in dir_list:
  74         # print repr(file)
  75         # EXCLUDE FILES MATCHING VERSIONS_PREFIX OR ANY OF THE EXCLUDE PATTERNS
  76         filtered = file.startswith('.')
  77         for re_prefix in filter_re:
  78             if re_prefix.search(file):
  79                 filtered = True
  80         if filtered:
  81             continue
  82         results_var['results_total'] += 1
  83
  84         # CREATE FILEOBJECT
  85         fileobject = FileObject(os.path.join(smart_str(DIRECTORY), smart_str(path), smart_str(file)))
  86
  87         # FILTER / SEARCH
  88         append = False
  89         if fileobject.filetype == request.GET.get('filter_type', fileobject.filetype) and _get_filterdate(request.GET.get('filter_date', ''), fileobject.date):
  90             append = True
  91         if request.GET.get('q') and not re.compile(request.GET.get('q').lower(), re.M).search(file.lower()):
  92             append = False
  93
  94         # APPEND FILE_LIST
  95         if append:
  96             files.append(fileobject)
  97             results_var['results_current'] += 1
  98             # COUNTER/RESULTS
  99             if fileobject.filetype == 'Image':
 100                 results_var['images_total'] += 1
 101             if fileobject.filetype != 'Folder':
 102                 results_var['delete_total'] += 1
 103             elif fileobject.filetype == 'Folder' and fileobject.is_empty:
 104                 results_var['delete_total'] += 1
 105             if query.get('type') and query.get('type') in SELECT_FORMATS and fileobject.filetype in SELECT_FORMATS[query.get('type')]:
 106                 results_var['select_total'] += 1
 107             elif not query.get('type'):
 108                 results_var['select_total'] += 1
 109
 110         # COUNTER/RESULTS
 111         if fileobject.filetype:
 112             counter[fileobject.filetype] += 1
 113
 114     # SORTING
 115     files.sort(key=lambda e: getattr(e, request.GET.get('o', DEFAULT_ORDER)))
 116     if request.GET.get('ot') == "desc":
 117         files.reverse()
 118
 119     return render_to_response('filebrowser/index.html', {
 120         'dir': path,
 121         'files': files,
 122         'results_var': results_var,
 123         'counter': counter,
 124         'query': query,
 125         'title': _(u'FileBrowser'),
 126         'settings_var': _get_settings_var(),
 127         'breadcrumbs': _get_breadcrumbs(query, path, ''),
 128     }, context_instance=Context(request))
 129 browse = staff_member_required(never_cache(browse))
 130
 131
 132 # mkdir signals
 133 filebrowser_pre_createdir = Signal(providing_args=["path", "dirname"])
 134 filebrowser_post_createdir = Signal(providing_args=["path", "dirname"])
 135
 136
 137 def mkdir(request):
 138     """
 139     Make Directory.
 140     """
 141
 142     from filebrowser.forms import MakeDirForm
 143
 144     # QUERY / PATH CHECK
 145     query = request.GET
 146     path = _get_path(query.get('dir', ''))
 147     if path is None:
 148         msg = _('Directory/File does not exist.')
 149         request.user.message_set.create(message=msg)
 150         return HttpResponseRedirect(reverse("fb_browse"))
 151     abs_path = os.path.join(MEDIA_ROOT, DIRECTORY, path)
 152
 153     if request.method == 'POST':
 154         form = MakeDirForm(abs_path, request.POST)
 155         if form.is_valid():
 156             server_path = os.path.join(abs_path, form.cleaned_data['dir_name'])
 157             try:
 158                 # PRE CREATE SIGNAL
 159                 filebrowser_pre_createdir.send(sender=request, path=path, dirname=form.cleaned_data['dir_name'])
 160                 # CREATE FOLDER
 161                 os.mkdir(server_path)
 162                 os.chmod(server_path, 0775)
 163                 # POST CREATE SIGNAL
 164                 filebrowser_post_createdir.send(sender=request, path=path, dirname=form.cleaned_data['dir_name'])
 165                 # MESSAGE & REDIRECT
 166                 msg = _('The Folder %s was successfully created.') % (form.cleaned_data['dir_name'])
 167                 request.user.message_set.create(message=msg)
 168                 # on redirect, sort by date desc to see the new directory on top of the list
 169                 # remove filter in order to actually _see_ the new folder
 170                 redirect_url = reverse("fb_browse") + query_helper(query, "ot=desc,o=date", "ot,o,filter_type,filter_date,q")
 171                 return HttpResponseRedirect(redirect_url)
 172             except OSError, (errno, strerror):
 173                 if errno == 13:
 174                     form.errors['dir_name'] = forms.util.ErrorList([_('Permission denied.')])
 175                 else:
 176                     form.errors['dir_name'] = forms.util.ErrorList([_('Error creating directory.')])
 177     else:
 178         form = MakeDirForm(abs_path)
 179
 180     return render_to_response('filebrowser/makedir.html', {
 181         'form': form,
 182         'query': query,
 183         'title': _(u'New Folder'),
 184         'settings_var': _get_settings_var(),
 185         'breadcrumbs': _get_breadcrumbs(query, path, _(u'New Folder')),
 186     }, context_instance=Context(request))
 187 mkdir = staff_member_required(never_cache(mkdir))
 188
 189
 190 @csrf_exempt
 191 def upload(request):
 192     """
 193     Multipe File Upload.
 194     """
 195
 196     from django.http import parse_cookie
 197
 198     # QUERY / PATH CHECK
 199     query = request.GET
 200     path = _get_path(query.get('dir', ''))
 201     if path is None:
 202         msg = _('Directory/File does not exist.')
 203         request.user.message_set.create(message=msg)
 204         return HttpResponseRedirect(reverse("fb_browse"))
 205     abs_path = os.path.join(MEDIA_ROOT, DIRECTORY, path)
 206
 207     # SESSION (used for flash-uploading)
 208     cookie_dict = parse_cookie(request.META.get('HTTP_COOKIE', ''))
 209     engine = __import__(settings.SESSION_ENGINE, {}, {}, [''])
 210     session_key = cookie_dict.get(settings.SESSION_COOKIE_NAME, None)
 211
 212     return render_to_response('filebrowser/upload.html', {
 213         'query': query,
 214         'title': _(u'Select files to upload'),
 215         'settings_var': _get_settings_var(),
 216         'breadcrumbs': _get_breadcrumbs(query, path, _(u'Upload')),
 217         'session_key': session_key,
 218     }, context_instance=Context(request))
 219 upload = staff_member_required(never_cache(upload))
 220
 221
 222 @csrf_exempt
 223 def _check_file(request):
 224     """
 225     Check if file already exists on the server.
 226     """
 227
 228     from django.utils import simplejson
 229
 230     folder = request.POST.get('folder')
 231     fb_uploadurl_re = re.compile(r'^(%s)' % reverse("fb_upload"))
 232     folder = fb_uploadurl_re.sub('', folder)
 233
 234     fileArray = {}
 235     if request.method == 'POST':
 236         for k, v in request.POST.items():
 237             if k != "folder":
 238                 v = _convert_filename(v)
 239                 if os.path.isfile(os.path.join(MEDIA_ROOT, DIRECTORY, folder, v)):
 240                     fileArray[k] = v
 241
 242     return HttpResponse(simplejson.dumps(fileArray))
 243
 244
 245 # upload signals
 246 filebrowser_pre_upload = Signal(providing_args=["path", "file"])
 247 filebrowser_post_upload = Signal(providing_args=["path", "file"])
 248
 249
 250 def _upload_file(request):
 251     """
 252     Upload file to the server.
 253     """
 254
 255     from django.core.files.move import file_move_safe
 256
 257     if request.method == 'POST':
 258         folder = request.POST.get('folder')
 259         fb_uploadurl_re = re.compile(r'^(%s)' % reverse("fb_upload"))
 260         folder = fb_uploadurl_re.sub('', folder)
 261         abs_path = os.path.join(MEDIA_ROOT, DIRECTORY, folder)
 262         if request.FILES:
 263             filedata = request.FILES['Filedata']
 264             filedata.name = _convert_filename(filedata.name)
 265             # PRE UPLOAD SIGNAL
 266             filebrowser_pre_upload.send(sender=request, path=request.POST.get('folder'), file=filedata)
 267             # HANDLE UPLOAD
 268             uploadedfile = _handle_file_upload(abs_path, filedata)
 269             # MOVE UPLOADED FILE
 270             # if file already exists
 271             if os.path.isfile(os.path.join(MEDIA_ROOT, DIRECTORY, folder, filedata.name)):
 272                 old_file = os.path.join(abs_path, filedata.name)
 273                 new_file = os.path.join(abs_path, uploadedfile)
 274                 file_move_safe(new_file, old_file)
 275             # POST UPLOAD SIGNAL
 276             filebrowser_post_upload.send(sender=request, path=request.POST.get('folder'), file=FileObject(os.path.join(DIRECTORY, folder, filedata.name)))
 277     return HttpResponse('True')
 278 _upload_file = csrf_exempt(flash_login_required(_upload_file))
 279
 280
 281 # delete signals
 282 filebrowser_pre_delete = Signal(providing_args=["path", "filename"])
 283 filebrowser_post_delete = Signal(providing_args=["path", "filename"])
 284
 285
 286 def delete(request):
 287     """
 288     Delete existing File/Directory.
 289
 290     When trying to delete a Directory, the Directory has to be empty.
 291     """
 292
 293     # QUERY / PATH CHECK
 294     query = request.GET
 295     path = _get_path(query.get('dir', ''))
 296     filename = _get_file(query.get('dir', ''), query.get('filename', ''))
 297     if path is None or filename is None:
 298         msg = _('Directory/File does not exist.')
 299         request.user.message_set.create(message=msg)
 300         return HttpResponseRedirect(reverse("fb_browse"))
 301     abs_path = os.path.join(MEDIA_ROOT, DIRECTORY, path)
 302
 303     msg = ""
 304     if request.GET:
 305         if request.GET.get('filetype') != "Folder":
 306             relative_server_path = os.path.join(DIRECTORY, path, filename)
 307             try:
 308                 # PRE DELETE SIGNAL
 309                 filebrowser_pre_delete.send(sender=request, path=path, filename=filename)
 310                 # DELETE IMAGE VERSIONS/THUMBNAILS
 311                 for version in VERSIONS:
 312                     try:
 313                         os.unlink(os.path.join(MEDIA_ROOT, _get_version_path(relative_server_path, version)))
 314                     except:
 315                         pass
 316                 # DELETE FILE
 317                 os.unlink(os.path.join(abs_path, filename))
 318                 # POST DELETE SIGNAL
 319                 filebrowser_post_delete.send(sender=request, path=path, filename=filename)
 320                 # MESSAGE & REDIRECT
 321                 msg = _('The file %s was successfully deleted.') % (filename.lower())
 322                 request.user.message_set.create(message=msg)
 323                 redirect_url = reverse("fb_browse") + query_helper(query, "", "filename,filetype")
 324                 return HttpResponseRedirect(redirect_url)
 325             except OSError:
 326                 # todo: define error message
 327                 msg = OSError
 328         else:
 329             try:
 330                 # PRE DELETE SIGNAL
 331                 filebrowser_pre_delete.send(sender=request, path=path, filename=filename)
 332                 # DELETE FOLDER
 333                 os.rmdir(os.path.join(abs_path, filename))
 334                 # POST DELETE SIGNAL
 335                 filebrowser_post_delete.send(sender=request, path=path, filename=filename)
 336                 # MESSAGE & REDIRECT
 337                 msg = _('The directory %s was successfully deleted.') % (filename.lower())
 338                 request.user.message_set.create(message=msg)
 339                 redirect_url = reverse("fb_browse") + query_helper(query, "", "filename,filetype")
 340                 return HttpResponseRedirect(redirect_url)
 341             except OSError:
 342                 # todo: define error message
 343                 msg = OSError
 344
 345     if msg:
 346         request.user.message_set.create(message=msg)
 347
 348     return render_to_response('filebrowser/index.html', {
 349         'dir': dir_name,
 350         'file': request.GET.get('filename', ''),
 351         'query': query,
 352         'settings_var': _get_settings_var(),
 353         'breadcrumbs': _get_breadcrumbs(query, dir_name, ''),
 354     }, context_instance=Context(request))
 355 delete = staff_member_required(never_cache(delete))
 356
 357
 358 # delete signals
 359 filebrowser_pre_rename = Signal(providing_args=["path", "filename"])
 360 filebrowser_post_rename = Signal(providing_args=["path", "filename"])
 361
 362
 363 def rename(request):
 364     """
 365     Rename existing File/Directory.
 366
 367     Includes renaming existing Image Versions/Thumbnails.
 368     """
 369
 370     from filebrowser.forms import RenameForm
 371
 372     # QUERY / PATH CHECK
 373     query = request.GET
 374     path = _get_path(query.get('dir', ''))
 375     filename = _get_file(query.get('dir', ''), query.get('filename', ''))
 376     if path is None or filename is None:
 377         msg = _('Directory/File does not exist.')
 378         request.user.message_set.create(message=msg)
 379         return HttpResponseRedirect(reverse("fb_browse"))
 380     abs_path = os.path.join(MEDIA_ROOT, DIRECTORY, path)
 381     file_extension = os.path.splitext(filename)[1].lower()
 382
 383     if request.method == 'POST':
 384         form = RenameForm(abs_path, file_extension, request.POST)
 385         if form.is_valid():
 386             relative_server_path = os.path.join(DIRECTORY, path, filename)
 387             new_relative_server_path = os.path.join(DIRECTORY, path, form.cleaned_data['name'] + file_extension)
 388             try:
 389                 # PRE RENAME SIGNAL
 390                 filebrowser_pre_delete.send(sender=request, path=path, filename=filename)
 391                 # DELETE IMAGE VERSIONS/THUMBNAILS
 392                 # regenerating versions/thumbs will be done automatically
 393                 for version in VERSIONS:
 394                     try:
 395                         os.unlink(os.path.join(MEDIA_ROOT, _get_version_path(relative_server_path, version)))
 396                     except:
 397                         pass
 398                 # RENAME ORIGINAL
 399                 os.rename(os.path.join(MEDIA_ROOT, relative_server_path), os.path.join(MEDIA_ROOT, new_relative_server_path))
 400                 # POST RENAME SIGNAL
 401                 filebrowser_post_delete.send(sender=request, path=path, filename=filename)
 402                 # MESSAGE & REDIRECT
 403                 msg = _('Renaming was successful.')
 404                 request.user.message_set.create(message=msg)
 405                 redirect_url = reverse("fb_browse") + query_helper(query, "", "filename")
 406                 return HttpResponseRedirect(redirect_url)
 407             except OSError, (errno, strerror):
 408                 form.errors['name'] = forms.util.ErrorList([_('Error.')])
 409     else:
 410         form = RenameForm(abs_path, file_extension)
 411
 412     return render_to_response('filebrowser/rename.html', {
 413         'form': form,
 414         'query': query,
 415         'file_extension': file_extension,
 416         'title': _(u'Rename "%s"') % filename,
 417         'settings_var': _get_settings_var(),
 418         'breadcrumbs': _get_breadcrumbs(query, path, _(u'Rename')),
 419     }, context_instance=Context(request))
 420 rename = staff_member_required(never_cache(rename))
 421
 422
 423 def versions(request):
 424     """
 425     Show all Versions for an Image according to ADMIN_VERSIONS.
 426     """
 427
 428     # QUERY / PATH CHECK
 429     query = request.GET
 430     path = _get_path(query.get('dir', ''))
 431     filename = _get_file(query.get('dir', ''), query.get('filename', ''))
 432     if path is None or filename is None:
 433         msg = _('Directory/File does not exist.')
 434         request.user.message_set.create(message=msg)
 435         return HttpResponseRedirect(reverse("fb_browse"))
 436     abs_path = os.path.join(MEDIA_ROOT, DIRECTORY, path)
 437
 438     return render_to_response('filebrowser/versions.html', {
 439         'original': _path_to_url(os.path.join(DIRECTORY, path, filename)),
 440         'query': query,
 441         'title': _(u'Versions for "%s"') % filename,
 442         'settings_var': _get_settings_var(),
 443         'breadcrumbs': _get_breadcrumbs(query, path, _(u'Versions for "%s"') % filename),
 444     }, context_instance=Context(request))
 445
 446 versions = staff_member_required(never_cache(versions))