X-Git-Url: http://mmka.chem.univ.gda.pl/gitweb/?a=blobdiff_plain;f=filex%2Fviews.py;h=1904d4f91a6b52679ff421f0a5ae4f55c6231ec6;hb=5e547a8cc20ef2bd450b3b839564e56e99099cee;hp=9aeb505fc6145e10d78af39f94289c915902b0cf;hpb=b96be0ed08b8fe0d3c07b7bda0d683e67ff517e4;p=qcg-portal.git diff --git a/filex/views.py b/filex/views.py index 9aeb505..1904d4f 100644 --- a/filex/views.py +++ b/filex/views.py @@ -1,96 +1,220 @@ +from itertools import islice +import logging 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, render 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.uploadhandler import with_ftp_upload_handler +from filex.forms import HostPathNameForm, RenameForm, FavoriteForm, HostPathForm, ExtractForm, HostItemsForm, \ + CompressForm +from filex.ftp import FTPOperation, FTPError +from filex.models import Favorite +from filex.utils import with_ftp_upload_handler, parse_ftp_error -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 + request = None + @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 + self.request = request + try: + return self.handle(FTPOperation(request.session['proxy']), form.cleaned_data) + except FTPError as e: + msg, status = parse_ftp_error(e) - try: - listing = FTPOperation(request.session['proxy']).listing(url) - except FTPException as e: - return JsonResponse({'msg': e.message}, status=400) + logger = logging.getLogger('gridftp') + logger.error(e.verbose, extra={'user': request.user, 'path': request.path, 'params': form.cleaned_data}) + + return JsonResponse({'error': msg}, status=status) + + setattr(cls, cls.method, process) + + return super(FTPView, cls).as_view(**initkwargs) + + def handle(self, ftp, params): + raise NotImplementedError + + +class ListView(FTPView): + def handle(self, ftp, params): + listing = ftp.listing(make_url(params, 'path')) + + 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') + + data.append(item) + + return JsonResponse(data, safe=False) + + +class DownloadView(FTPView): + def handle(self, ftp, params): + url = make_url(params, 'path') + + try: + stats = ftp.info(url) + except FTPError as e: + msg, status = parse_ftp_error(e) + + return render(self.request, 'qcg/download_error.html', {'msg': msg, 'url': url}, status=status) + + data = ftp.get(url) + + 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) + response['Content-Length'] = stats['size'] + + if encoding: + response['Content-Encoding'] = encoding + + return response + + +class InfoView(FTPView): + def handle(self, ftp, params): + return JsonResponse(ftp.info(make_url(params, 'path'))) - data = [] - for item in listing: - item['size'] = filesizeformat(item['size']) - item['date'] = date_format(item['date'], 'CUSTOM_DATETIME_FORMAT') - data.append(item) +class DeleteView(FTPView): + method = 'post' + form_class = HostItemsForm - return JsonResponse(data, safe=False) + 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) -def download(request): - check_auth(request) + for path in params['files']: + try: + ftp.delete(url + path) + except FTPError as e: + fail[path] = e.message + else: + done.append(path) - # 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!") + return JsonResponse({'done': done, 'fail': fail}) - url = 'gsiftp://' + host + path + '/' + name - mime_type, encoding = mimetypes.guess_type(name) +class MkdirView(FTPView): + method = 'post' + form_class = HostPathNameForm - 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 (?) + def handle(self, ftp, params): + ftp.mkdir(make_url(params, 'path', 'name')) - if encoding: - response['Content-Encoding'] = encoding + return JsonResponse({'success': True}) - return response +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}) -def info(request): - check_auth(request) +@require_POST +@login_required +def fav_add(request): + data = request.POST.copy() + data['owner'] = request.user.id - # 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!") + form = FavoriteForm(data) - url = 'gsiftp://' + host + path + if not form.is_valid(): + return JsonResponse({'error': form.errors}, status=400) try: - return JsonResponse(FTPOperation(request.session['proxy']).info(url)) - except FTPException as e: - status = 400 - if 'No such file or directory' in e.message: - status = 404 - elif 'Permission denied' in e.message: - status = 403 - - return JsonResponse({'msg': e.message}, status=status) + FTPOperation(request.session['proxy']).info(make_url(form.cleaned_data, 'path')) + except FTPError as e: + msg, status = parse_ftp_error(e) + + return JsonResponse({'error': msg}, status=status) + + instance = form.save() + + return JsonResponse({'group': 'usr', 'host': instance.host, 'path': instance.path, + 'value': instance.host + '/' + instance.path}) + + +@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})