1 from functools import wraps
4 from django.core.files.uploadedfile import UploadedFile
5 from django.core.files.uploadhandler import FileUploadHandler, StopUpload, StopFutureHandlers
6 from django.http import JsonResponse
7 from django.views.decorators.csrf import csrf_exempt, csrf_protect
9 from filex.forms import HostPathForm
10 from filex.ftp import FTPOperation, FTPError
13 class FtpUploadHandler(FileUploadHandler):
18 def new_file(self, file_name, *args, **kwargs):
19 super(FtpUploadHandler, self).new_file(file_name, *args, **kwargs)
21 form = HostPathForm(self.request.GET)
23 if self.request.user.is_anonymous() or not form.is_valid():
24 raise StopUpload(connection_reset=True)
27 self.ftp = FTPOperation(self.request.session['proxy'], self.chunk_size)
29 self.url = 'gsiftp://{}/{}'.format(form.cleaned_data['host'],
30 os.path.join(form.cleaned_data['path'], self.file_name))
31 self.tmp_url = self.url + '.part'
33 self.ftp.put(self.tmp_url)
37 def receive_data_chunk(self, raw_data, start):
38 self.ftp.stream.put(raw_data)
40 def file_complete(self, file_size):
41 # signal end of data and wait for finish
42 self.ftp.stream.put(None)
45 self.ftp.move(self.tmp_url, self.url)
47 return UploadedFile(name=self.file_name, size=file_size, charset=self.charset,
48 content_type=self.content_type, content_type_extra=self.content_type_extra)
51 def with_ftp_upload_handler(view_func):
53 def wrapped_view(request, *args, **kwargs):
54 request.upload_handlers = [FtpUploadHandler(request)]
57 return csrf_protect(view_func)(request, *args, **kwargs)
60 if 'No such file or directory' in e.message:
62 elif 'Permission denied' in e.message:
65 return JsonResponse({'error': e.message}, status=status)
67 return csrf_exempt(wrapped_view)