X-Git-Url: http://mmka.chem.univ.gda.pl/gitweb/?a=blobdiff_plain;f=filex%2Fviews.py;h=ada8b7b1d43e91b6c5d436c81565819d3c36611d;hb=53b5d55ae94941e7621982e56438025527fa2950;hp=1a02ddb483b6c38a4a270382f4a93751acdf5e95;hpb=2e226ed0e2a8bbf0cedbcaf5fff9e2906e858588;p=qcg-portal.git diff --git a/filex/views.py b/filex/views.py index 1a02ddb..ada8b7b 100644 --- a/filex/views.py +++ b/filex/views.py @@ -1,211 +1,222 @@ +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.utils.http import urlquote from django.views.decorators.http import require_POST -from filex.forms import NewDirForm, RenameForm +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}) - data = [] - for item in listing: - item['size'] = filesizeformat(item['size']) - item['date'] = date_format(item['date'], 'CUSTOM_DATETIME_FORMAT') + return JsonResponse({'error': msg}, status=status) - data.append(item) + setattr(cls, cls.method, process) - return JsonResponse(data, 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): + url = make_url(params, 'path') + try: + stats = ftp.info(url) + except FTPError as e: + msg, status = parse_ftp_error(e) -@with_ftp_upload_handler -def upload(request): - # TODO error handling - return JsonResponse({'success': True}) + return render(self.request, 'qcg/download_error.html', {'msg': msg, 'url': url}, status=status) + data = ftp.get(url) -def info(request): - check_auth(request) + name = os.path.basename(params['path']) + mime_type, encoding = mimetypes.guess_type(name) - # 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!") + response = StreamingHttpResponse(data, content_type=mime_type or 'application/octet-stream') + response['Content-Disposition'] = u'attachment; filename={}'.format(name) + response['Content-Length'] = stats['size'] - url = 'gsiftp://' + host + path + if encoding: + response['Content-Encoding'] = encoding - 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 response - return JsonResponse({'msg': e.message}, status=status) +class InfoView(FTPView): + def handle(self, ftp, params): + return JsonResponse(ftp.info(make_url(params, 'path'))) -@require_POST -def delete(request): - check_auth(request) - # TODO data validation - host = request.POST.get('host') - path = request.POST.get('path') - dirs = request.POST.getlist('dirs') - files = request.POST.getlist('files') - if not host or not path or not (files or dirs): - raise SuspiciousOperation("No path or host or files given!") +class DeleteView(FTPView): + method = 'post' + form_class = HostItemsForm - url = 'gsiftp://' + host + path + '/' - ftp = FTPOperation(request.session['proxy']) + def handle(self, ftp, params): + url = make_url(params) + done, fail = [], {} - done, fail = [], {} + for path in params['dirs']: + try: + ftp.rmdir(url + urlquote(path)) + except FTPError as e: + fail[path] = e.message + else: + done.append(path) - for name in dirs: - try: - ftp.rmdir(url + name) - except FTPException as e: - fail[name] = e.message - else: - done.append(name) + for path in params['files']: + try: + ftp.delete(url + urlquote(path)) + except FTPError as e: + fail[path] = e.message + else: + done.append(path) - for name in files: - try: - ftp.delete(url + name) - except FTPException as e: - fail[name] = e.message - else: - done.append(name) + return JsonResponse({'done': done, 'fail': fail}) - return JsonResponse({'done': done, 'fail': fail}) +class MkdirView(FTPView): + method = 'post' + form_class = HostPathNameForm -@require_POST -def mkdir(request): - check_auth(request) + def handle(self, ftp, params): + ftp.mkdir(make_url(params, 'path', 'name')) - # TODO actual data validation - form = NewDirForm(request.POST) + return JsonResponse({'success': True}) - if form.is_valid(): - host = form.cleaned_data['host'] - path = form.cleaned_data['path'] - name = form.cleaned_data['name'] - url = 'gsiftp://' + host + path + '/' + name +class MoveView(FTPView): + method = 'post' + form_class = RenameForm - try: - FTPOperation(request.session['proxy']).mkdir(url) - except FTPException as e: - msg = e.message - else: - return JsonResponse({'success': True}) - else: - msg = form.errors + def handle(self, ftp, params): + ftp.move(make_url(params, 'src'), make_url(params, 'dst')) - return JsonResponse({'msg': msg}, status=400) + return JsonResponse({'success': True}) -@require_POST -def move(request): - check_auth(request) +class CompressView(FTPView): + method = 'post' + form_class = CompressForm - # TODO actual data validation - form = RenameForm(request.POST) + 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}) - if form.is_valid(): - host = form.cleaned_data['host'] - path = form.cleaned_data['path'] - src = form.cleaned_data['src'] - dst = form.cleaned_data['dst'] - src_url = 'gsiftp://' + host + path + '/' + src - dst_url = 'gsiftp://' + host + path + '/' + dst +class ExtractView(FTPView): + method = 'post' + form_class = ExtractForm + def handle(self, ftp, params): try: - FTPOperation(request.session['proxy']).move(src_url, dst_url) - except FTPException as e: - msg = e.message - else: - return JsonResponse({'success': True}) - else: - msg = form.errors + # 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({'msg': msg}, status=400) + return JsonResponse({'success': True}) + + +def make_url(params, *parts): + return 'gsiftp://{}/{}'.format(params['host'], + urlquote(os.path.join(*[params[part] for part in parts]), safe='/~') if parts else '') @require_POST -def compress(request): - check_auth(request) +@with_ftp_upload_handler +def upload(request): + return JsonResponse({'success': True}) - # TODO data validation - host = request.POST.get('host') - path = request.POST.get('path') - files = request.POST.getlist('files') - archive = request.POST.get('archive') - if not host or not path or not files: - raise SuspiciousOperation("No path or host or files or archive given!") - server = 'gsiftp://' + host +@require_POST +@login_required +def fav_add(request): + data = request.POST.copy() + data['owner'] = request.user.id + + form = FavoriteForm(data) + + if not form.is_valid(): + return JsonResponse({'error': form.errors}, status=400) try: - # consume generator with command output - list(FTPOperation(request.session['proxy']).compress(server, path, files, archive)) - except FTPException as e: - return JsonResponse({'msg': e.message}, status=400) - else: - return JsonResponse({'success': True}) + 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})