X-Git-Url: http://mmka.chem.univ.gda.pl/gitweb/?a=blobdiff_plain;f=filex%2Fviews.py;h=f5aabb80d3d4b00ecde10db47132ce51d9e9e4f2;hb=87c84cec7b5b78b145a61953408b6cadffcd9735;hp=6dd2d9f5f4f79aacc7d142447d999c1b97ff6cdf;hpb=be284bd6256a7a095d62786d9b58f815eefb769a;p=qcg-portal.git diff --git a/filex/views.py b/filex/views.py index 6dd2d9f..f5aabb8 100644 --- a/filex/views.py +++ b/filex/views.py @@ -1,70 +1,203 @@ +from itertools import islice import mimetypes +import os -from django.core.exceptions import PermissionDenied, SuspiciousOperation +from django.contrib.auth.decorators import login_required +from django.core.exceptions import PermissionDenied from django.http import JsonResponse, StreamingHttpResponse +from django.shortcuts import get_object_or_404 from django.template.defaultfilters import filesizeformat from django.utils.formats import date_format +from django.views.decorators.http import require_POST +from django.views.generic import View -from filex.ftp import FTPOperation, FTPException +from filex.forms import HostPathNameForm, RenameForm, FavoriteForm, HostPathForm, ExtractForm, HostItemsForm, \ + CompressForm +from filex.ftp import FTPOperation, FTPError +from filex.models import Favorite from filex.uploadhandler import with_ftp_upload_handler -def check_auth(request): - if not request.user.is_authenticated(): - raise PermissionDenied("Login required!") - if not request.session['proxy']: - raise PermissionDenied("No proxy found!") +class FTPView(View): + method = 'get' + form_class = HostPathForm + @classmethod + def as_view(cls, **initkwargs): + def process(self, request): + if not request.user.is_authenticated(): + raise PermissionDenied("Login required!") + if not request.session['proxy']: + raise PermissionDenied("No proxy found!") -def list_content(request): - check_auth(request) + form = self.form_class(request.POST if self.method == 'post' else request.GET) - # TODO data validation - host = request.GET.get('host') - path = request.GET.get('path') - if not host or not path: - raise SuspiciousOperation("No path or host given!") + if not form.is_valid(): + return JsonResponse({'error': form.errors}, status=400) - url = 'gsiftp://' + host + path + try: + return self.handle(FTPOperation(request.session['proxy']), form.cleaned_data) + except FTPError as e: + status = 400 + if 'No such file or directory' in e.message: + status = 404 + elif 'Permission denied' in e.message: + status = 403 - try: - listing = FTPOperation(request.session['proxy']).listing(url) - except FTPException as e: - return JsonResponse({'msg': e.message}, status=400) + return JsonResponse({'error': e.message}, status=status) - for item in listing: - item['size'] = filesizeformat(item['size']) - item['date'] = date_format(item['date'], 'DATETIME_FORMAT') + setattr(cls, cls.method, process) - return JsonResponse(listing, safe=False) + return super(FTPView, cls).as_view(**initkwargs) + def handle(self, ftp, params): + raise NotImplementedError -def download(request): - check_auth(request) - # TODO data validation - host = request.GET.get('host') - path = request.GET.get('path') - name = request.GET.get('name') - if not host or not path or not name: - raise SuspiciousOperation("No path or host or name given!") +class ListView(FTPView): + def handle(self, ftp, params): + listing = ftp.listing(make_url(params, 'path')) - url = 'gsiftp://' + host + path + '/' + name + data = [] + # ignore . and .. from beginning of the listing + for item in islice(listing, 2, None): + item['size'] = filesizeformat(item['size']) + item['date'] = date_format(item['date'], 'CUSTOM_DATETIME_FORMAT') - mime_type, encoding = mimetypes.guess_type(name) + data.append(item) - response = StreamingHttpResponse(FTPOperation(request.session['proxy']).get(url), - content_type=mime_type or 'application/octet-stream') - response['Content-Disposition'] = 'attachment; filename={}'.format(name) - # TODO Content-Length (?) + return JsonResponse(data, safe=False) - if encoding: - response['Content-Encoding'] = encoding - return response +class DownloadView(FTPView): + def handle(self, ftp, params): + data = ftp.get(make_url(params, 'path')) + name = os.path.basename(params['path']) + mime_type, encoding = mimetypes.guess_type(name) + response = StreamingHttpResponse(data, content_type=mime_type or 'application/octet-stream') + response['Content-Disposition'] = u'attachment; filename={}'.format(name) + # TODO Content-Length (?) + + if encoding: + response['Content-Encoding'] = encoding + + return response + + +class InfoView(FTPView): + def handle(self, ftp, params): + return JsonResponse(ftp.info(make_url(params, 'path'))) + + +class DeleteView(FTPView): + method = 'post' + form_class = HostItemsForm + + def handle(self, ftp, params): + url = make_url(params) + done, fail = [], {} + + for path in params['dirs']: + try: + ftp.rmdir(url + path) + except FTPError as e: + fail[path] = e.message + else: + done.append(path) + + for path in params['files']: + try: + ftp.delete(url + path) + except FTPError as e: + fail[path] = e.message + else: + done.append(path) + + return JsonResponse({'done': done, 'fail': fail}) + + +class MkdirView(FTPView): + method = 'post' + form_class = HostPathNameForm + + def handle(self, ftp, params): + ftp.mkdir(make_url(params, 'path', 'name')) + + return JsonResponse({'success': True}) + + +class MoveView(FTPView): + method = 'post' + form_class = RenameForm + + def handle(self, ftp, params): + ftp.move(make_url(params, 'src'), make_url(params, 'dst')) + + return JsonResponse({'success': True}) + + +class CompressView(FTPView): + method = 'post' + form_class = CompressForm + + def handle(self, ftp, params): + try: + # consume generator with command output + list(ftp.compress(make_url(params), params['path'], params['files'], params['archive'])) + except ValueError as e: + return JsonResponse({'error': e.message}, status=400) + + return JsonResponse({'success': True}) + + +class ExtractView(FTPView): + method = 'post' + form_class = ExtractForm + + def handle(self, ftp, params): + try: + # consume generator with command output + list(ftp.extract(make_url(params), params['path'], params['dst'])) + except ValueError as e: + return JsonResponse({'error': e.message}, status=400) + + return JsonResponse({'success': True}) + + +def make_url(params, *parts): + return 'gsiftp://{}/{}'.format(params['host'], os.path.join(*[params[part] for part in parts]) if parts else '') + + +@require_POST @with_ftp_upload_handler def upload(request): - # TODO error handling + return JsonResponse({'success': True}) + + +@require_POST +@login_required +def fav_add(request): + data = request.POST.copy() + data['owner'] = request.user.id + + form = FavoriteForm(data) + + # TODO check if path exists + if form.is_valid(): + instance = form.save() + + return JsonResponse({'group': 'usr', 'host': instance.host, 'path': instance.path, + 'value': instance.host + '/' + instance.path}) + + return JsonResponse({'error': form.errors}, status=400) + + +@require_POST +@login_required +def fav_delete(request): + fav = get_object_or_404(Favorite, owner=request.user, host=request.POST['host'], path=request.POST['path']) + fav.delete() + return JsonResponse({'success': True})