Source code for horizon.dashboards.nova.containers.views

# vim: tabstop=4 shiftwidth=4 softtabstop=4

# Copyright 2012 United States Government as represented by the
# Administrator of the National Aeronautics and Space Administration.
# All Rights Reserved.
#
# Copyright 2012 Nebula, Inc.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

"""
Views for managing Swift containers.
"""
import os

from django import http
from django.core.urlresolvers import reverse
from django.utils.translation import ugettext_lazy as _

from horizon import api
from horizon import browsers
from horizon import exceptions
from horizon import forms
from horizon.api import FOLDER_DELIMITER
from .browsers import ContainerBrowser
from .forms import CreateContainer, UploadObject, CopyObject
from .tables import wrap_delimiter


[docs]class ContainerView(browsers.ResourceBrowserView): browser_class = ContainerBrowser template_name = "nova/containers/index.html"
[docs] def get_containers_data(self): containers = [] self._more = None marker = self.request.GET.get('marker', None) try: containers, self._more = api.swift_get_containers(self.request, marker=marker) except: msg = _('Unable to retrieve container list.') exceptions.handle(self.request, msg) return containers
@property
[docs] def objects(self): """ Returns a list of objects given the subfolder's path. The path is from the kwargs of the request. """ if not hasattr(self, "_objects"): objects = [] self._more = None marker = self.request.GET.get('marker', None) container_name = self.kwargs['container_name'] subfolder = self.kwargs['subfolder_path'] prefix = None if container_name: self.navigation_selection = True if subfolder: prefix = subfolder try: objects, self._more = api.swift_get_objects(self.request, container_name, marker=marker, prefix=prefix) except: self._more = None objects = [] msg = _('Unable to retrieve object list.') exceptions.handle(self.request, msg) self._objects = objects return self._objects
[docs] def is_subdir(self, item): return getattr(item, "content_type", None) == "application/directory"
[docs] def get_objects_data(self): """ Returns a list of objects within the current folder. """ filtered_objects = [item for item in self.objects if not self.is_subdir(item)] return filtered_objects
[docs] def get_subfolders_data(self): """ Returns a list of subfolders within the current folder. """ filtered_objects = [item for item in self.objects if self.is_subdir(item)] return filtered_objects
[docs] def get_context_data(self, **kwargs): context = super(ContainerView, self).get_context_data(**kwargs) context['container_name'] = self.kwargs["container_name"] context['subfolders'] = [] if self.kwargs["subfolder_path"]: (parent, slash, folder) = self.kwargs["subfolder_path"].\ strip('/').rpartition('/') while folder: path = "%s%s%s/" % (parent, slash, folder) context['subfolders'].insert(0, (folder, path)) (parent, slash, folder) = parent.rpartition('/') return context
[docs]class CreateView(forms.ModalFormView): form_class = CreateContainer template_name = 'nova/containers/create.html' success_url = "horizon:nova:containers:index"
[docs] def get_success_url(self): parent = self.request.POST.get('parent', None) if parent: container, slash, remainder = parent.partition(FOLDER_DELIMITER) container += FOLDER_DELIMITER if remainder and not remainder.endswith(FOLDER_DELIMITER): remainder = "".join([remainder, FOLDER_DELIMITER]) return reverse(self.success_url, args=(container, remainder)) else: return reverse(self.success_url, args=[self.request.POST['name'] + FOLDER_DELIMITER])
[docs] def get_initial(self): initial = super(CreateView, self).get_initial() initial['parent'] = self.kwargs['container_name'] return initial
[docs]class UploadView(forms.ModalFormView): form_class = UploadObject template_name = 'nova/containers/upload.html' success_url = "horizon:nova:containers:index"
[docs] def get_success_url(self): container_name = self.request.POST['container_name'] return reverse(self.success_url, args=(wrap_delimiter(container_name), self.request.POST.get('path', '')))
[docs] def get_initial(self): return {"container_name": self.kwargs["container_name"], "path": self.kwargs['subfolder_path']}
[docs] def get_context_data(self, **kwargs): context = super(UploadView, self).get_context_data(**kwargs) context['container_name'] = self.kwargs["container_name"] return context
[docs]def object_download(request, container_name, object_path): try: obj = api.swift.swift_get_object(request, container_name, object_path) except: redirect = reverse("horizon:nova:containers:index") exceptions.handle(request, _("Unable to retrieve object."), redirect=redirect) # Add the original file extension back on if it wasn't preserved in the # name given to the object. filename = object_path.rsplit(FOLDER_DELIMITER)[-1] if not os.path.splitext(obj.name)[1] and obj.orig_name: name, ext = os.path.splitext(obj.orig_name) filename = "%s%s" % (filename, ext) response = http.HttpResponse() safe_name = filename.replace(",", "").encode('utf-8') response['Content-Disposition'] = 'attachment; filename=%s' % safe_name response['Content-Type'] = 'application/octet-stream' response.write(obj.data) return response
[docs]class CopyView(forms.ModalFormView): form_class = CopyObject template_name = 'nova/containers/copy.html' success_url = "horizon:nova:containers:index"
[docs] def get_success_url(self): new_container_name = self.request.POST['new_container_name'] return reverse(self.success_url, args=(wrap_delimiter(new_container_name), self.request.POST.get('path', '')))
[docs] def get_form_kwargs(self): kwargs = super(CopyView, self).get_form_kwargs() try: containers = api.swift_get_containers(self.request) except: redirect = reverse("horizon:nova:containers:index") exceptions.handle(self.request, _('Unable to list containers.'), redirect=redirect) kwargs['containers'] = [(c.name, c.name) for c in containers[0]] return kwargs
[docs] def get_initial(self): path = self.kwargs["subfolder_path"] orig = "%s%s" % (path or '', self.kwargs["object_name"]) return {"new_container_name": self.kwargs["container_name"], "orig_container_name": self.kwargs["container_name"], "orig_object_name": orig, "path": path, "new_object_name": "%s copy" % self.kwargs["object_name"]}
[docs] def get_context_data(self, **kwargs): context = super(CopyView, self).get_context_data(**kwargs) context['container_name'] = self.kwargs["container_name"] context['object_name'] = self.kwargs["object_name"] return context