use attr instead of data
[redakcja.git] / 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))