def external_resource_download(self, file_mgr_name, username, src_file_id, dest_file_id):
"""
:param self:
:param username:
:param src_file_id:
:param dest_file_id:
:return:
"""
logger.debug('Downloading %s://%s for user %s to %s',
file_mgr_name, src_file_id, username, dest_file_id)
from designsafe.apps.api.external_resources.box.filemanager.manager \
import FileManager as BoxFileManager
from designsafe.apps.api.external_resources.dropbox.filemanager.manager \
import FileManager as DropboxFileManager
from designsafe.apps.api.external_resources.googledrive.filemanager.manager \
import FileManager as GoogleDriveFileManager
user = get_user_model().objects.get(username=username)
if file_mgr_name == 'box':
fmgr = BoxFileManager(user)
elif file_mgr_name == 'dropbox':
fmgr = DropboxFileManager(user)
elif file_mgr_name == 'googledrive':
fmgr = GoogleDriveFileManager(user)
fmgr.copy(username, src_file_id, dest_file_id)
def external_resource_upload(self, username, dest_resource, src_file_id, dest_file_id):
"""
:param self:
:param username:
:param dest_resource:
:param src_file_id:
:param dest_file_id:
:return:
"""
logger.debug('Initializing external_resource_upload. username: %s, src_file_id: %s, dest_resource: %s, dest_file_id: %s ', username, src_file_id, dest_resource, dest_file_id)
from designsafe.apps.api.external_resources.box.filemanager.manager \
import FileManager as BoxFileManager
from designsafe.apps.api.external_resources.dropbox.filemanager.manager \
import FileManager as DropboxFileManager
from designsafe.apps.api.external_resources.googledrive.filemanager.manager \
import FileManager as GoogleDriveFileManager
user = get_user_model().objects.get(username=username)
if dest_resource == 'box':
fmgr = BoxFileManager(user)
elif dest_resource == 'dropbox':
fmgr = DropboxFileManager(user)
elif dest_resource == 'googledrive':
fmgr = GoogleDriveFileManager(user)
logger.debug('fmgr.upload( %s, %s, %s)', username, src_file_id, dest_file_id)
fmgr.upload(username, src_file_id, dest_file_id)
def put(self, request, file_mgr_name, file_id):
if request.is_ajax():
body = json.loads(request.body)
else:
body = request.POST.copy()
action = body.get('action')
if file_mgr_name not in [
BoxFileManager.NAME, DropboxFileManager.NAME,
GoogleDriveFileManager.NAME
] or action is None:
return HttpResponseBadRequest("Bad Request.")
if file_mgr_name == 'box':
fmgr = BoxFileManager(request.user)
elif file_mgr_name == 'dropbox':
fmgr = DropboxFileManager(request.user)
elif file_mgr_name == 'googledrive':
fmgr = GoogleDriveFileManager(request.user)
if action == 'preview':
try:
return fmgr.preview(file_id, file_mgr_name=file_mgr_name)
except HTTPError as e:
logger.exception('Unable to preview file')
return HttpResponseBadRequest(e.response.text)
elif action == 'copy' or action == 'move':
try:
tasks.external_resource_download.apply_async(kwargs={
'file_mgr_name':
file_mgr_name,
'username':
request.user.username,
'src_file_id':
file_id,
'dest_file_id':
os.path.join(body['system'], body['path'].strip('/'))
},
queue='files')
return JsonResponse({'status': 200, 'message': 'OK'})
except HTTPError as e:
logger.exception('Unable to copy file')
return HttpResponseBadRequest(e.response.text)
elif action == 'download':
try:
download_dict = fmgr.get_download_url(
file_id, username=request.user.username)
if not download_dict:
HttpResponseBadRequest('Operation not permitted')
return JsonResponse(download_dict)
except HTTPError as err:
logger.exception(
'Unable to download {} file'.format(file_mgr_name))
return HttpResponseBadRequest(err.response.text)
return HttpResponseBadRequest("Operation not implemented.")
def get(self, request, file_mgr_name, file_id=None):
if file_mgr_name not in [
BoxFileManager.NAME, DropboxFileManager.NAME,
GoogleDriveFileManager.NAME
]:
return HttpResponseBadRequest('Incorrect file manager.')
if file_mgr_name == 'box':
fmgr = BoxFileManager(request.user)
elif file_mgr_name == 'dropbox':
fmgr = DropboxFileManager(request.user)
elif file_mgr_name == 'googledrive':
fmgr = GoogleDriveFileManager(request.user)
listing = fmgr.listing(file_id)
return JsonResponse(listing, safe=False)
def get(self, request, file_mgr_name, file_id):
if file_mgr_name not in [
BoxFileManager.NAME, DropboxFileManager.NAME,
GoogleDriveFileManager.NAME
]:
return HttpResponseBadRequest("Incorrect file manager.")
if file_mgr_name == 'box':
fmgr = BoxFileManager(request.user)
elif file_mgr_name == 'dropbox':
fmgr = DropboxFileManager(request.user)
elif file_mgr_name == 'googledrive':
fmgr = GoogleDriveFileManager(request.user)
f = fmgr.listing(file_id)
if request.GET.get('preview', False):
context = {'file': f}
preview_url = fmgr.get_preview_url(file_id)
if preview_url is not None:
context['preview_url'] = preview_url['href']
return render(request, 'designsafe/apps/api/box/preview.html',
context)
else:
return HttpResponseRedirect(fmgr.get_download_url(file_id))