Source code for openstack_dashboard.dashboards.project.volumes.views

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

# 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 volumes.
"""

import logging

from django.core.urlresolvers import reverse_lazy
from django.utils.translation import ugettext_lazy as _
from django.utils.datastructures import SortedDict

from horizon import exceptions
from horizon import forms
from horizon import tables
from horizon import tabs

from openstack_dashboard import api
from openstack_dashboard.api import cinder
from openstack_dashboard.usage import quotas
from .forms import CreateForm, AttachForm, CreateSnapshotForm
from .tables import AttachmentsTable, VolumesTable
from .tabs import VolumeDetailTabs


LOG = logging.getLogger(__name__)


[docs]class VolumeTableMixIn(object): def _get_volumes(self, search_opts=None): try: return cinder.volume_list(self.request, search_opts=search_opts) except: exceptions.handle(self.request, _('Unable to retrieve volume list.')) return [] def _get_instances(self): try: return api.nova.server_list(self.request) except: exceptions.handle(self.request, _("Unable to retrieve volume/instance " "attachment information")) return [] def _set_id_if_nameless(self, volumes, instances): for volume in volumes: # It is possible to create a volume with no name through the # EC2 API, use the ID in those cases. if not volume.display_name: volume.display_name = volume.id def _set_attachments_string(self, volumes, instances): instances = SortedDict([(inst.id, inst) for inst in instances]) for volume in volumes: for att in volume.attachments: server_id = att.get('server_id', None) att['instance'] = instances.get(server_id, None)
[docs]class IndexView(tables.DataTableView, VolumeTableMixIn): table_class = VolumesTable template_name = 'project/volumes/index.html'
[docs] def get_data(self): volumes = self._get_volumes() instances = self._get_instances() self._set_id_if_nameless(volumes, instances) self._set_attachments_string(volumes, instances) return volumes
[docs]class DetailView(tabs.TabView): tab_group_class = VolumeDetailTabs template_name = 'project/volumes/detail.html'
[docs]class CreateView(forms.ModalFormView): form_class = CreateForm template_name = 'project/volumes/create.html' success_url = reverse_lazy("horizon:project:volumes:index")
[docs] def get_context_data(self, **kwargs): context = super(CreateView, self).get_context_data(**kwargs) try: context['usages'] = quotas.tenant_quota_usages(self.request) except: exceptions.handle(self.request) return context
[docs]class CreateSnapshotView(forms.ModalFormView): form_class = CreateSnapshotForm template_name = 'project/volumes/create_snapshot.html' success_url = reverse_lazy("horizon:project:images_and_snapshots:index")
[docs] def get_context_data(self, **kwargs): context = super(CreateSnapshotView, self).get_context_data(**kwargs) context['volume_id'] = self.kwargs['volume_id'] return context
[docs] def get_initial(self): return {'volume_id': self.kwargs["volume_id"]}
[docs]class EditAttachmentsView(tables.DataTableView, forms.ModalFormView): table_class = AttachmentsTable form_class = AttachForm template_name = 'project/volumes/attach.html' success_url = reverse_lazy("horizon:project:volumes:index")
[docs] def get_object(self): if not hasattr(self, "_object"): volume_id = self.kwargs['volume_id'] try: self._object = cinder.volume_get(self.request, volume_id) except: self._object = None exceptions.handle(self.request, _('Unable to retrieve volume information.')) return self._object
[docs] def get_data(self): try: volumes = self.get_object() attachments = [att for att in volumes.attachments if att] except: attachments = [] exceptions.handle(self.request, _('Unable to retrieve volume information.')) return attachments
[docs] def get_initial(self): try: instances = api.nova.server_list(self.request) except: instances = [] exceptions.handle(self.request, _("Unable to retrieve attachment information.")) return {'volume': self.get_object(), 'instances': instances}
[docs] def get_form(self): if not hasattr(self, "_form"): form_class = self.get_form_class() self._form = super(EditAttachmentsView, self).get_form(form_class) return self._form
[docs] def get_context_data(self, **kwargs): context = super(EditAttachmentsView, self).get_context_data(**kwargs) context['form'] = self.get_form() volume = self.get_object() if volume and volume.status == 'available': context['show_attach'] = True else: context['show_attach'] = False context['volume'] = volume if self.request.is_ajax(): context['hide'] = True return context
[docs] def get(self, request, *args, **kwargs): # Table action handling handled = self.construct_tables() if handled: return handled return self.render_to_response(self.get_context_data(**kwargs))
[docs] def post(self, request, *args, **kwargs): form = self.get_form() if form.is_valid(): return self.form_valid(form) else: return self.get(request, *args, **kwargs)