e85b19e619
E127/E128 checks helps us to keep code more readable, so these rules should be allowed, when pep8 runs. Enabled check E128 - continuation line under-indented for visual indent. Change-Id: Ic17a7ba47d555183ff26a5dc5f6ec27c0461725a
355 lines
13 KiB
Python
355 lines
13 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# 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.
|
|
|
|
from django.core import urlresolvers
|
|
from django.utils.translation import ugettext_lazy as _ # noqa
|
|
|
|
from horizon import exceptions
|
|
from horizon import forms
|
|
from horizon import workflows
|
|
from openstack_dashboard.api import glance
|
|
|
|
from tuskar_ui import api as tuskar
|
|
from tuskar_ui.infrastructure.resource_management.flavors\
|
|
import forms as flavors_forms
|
|
from tuskar_ui.infrastructure.resource_management.resource_classes\
|
|
import tables
|
|
import tuskar_ui.workflows
|
|
|
|
|
|
TEMPLATE_PREFIX = 'infrastructure/resource_management/resource_classes/'
|
|
URL_PREFIX = 'horizon:infrastructure:resource_management:'
|
|
|
|
|
|
class ResourceClassInfoAction(workflows.Action):
|
|
name = forms.CharField(max_length=255,
|
|
label=_("Class Name"),
|
|
help_text="",
|
|
required=True)
|
|
service_type = forms.ChoiceField(label=_('Class Type'),
|
|
required=True,
|
|
choices=[('', ''),
|
|
('controller',
|
|
('Controller')),
|
|
('compute',
|
|
('Compute')),
|
|
('object_storage',
|
|
('Object Storage')),
|
|
('block_storage',
|
|
('Block Storage')),
|
|
],
|
|
widget=forms.Select(
|
|
attrs={'class': 'switchable'})
|
|
)
|
|
image_id = forms.ChoiceField(
|
|
label=_('Provisioning Image'),
|
|
required=True,
|
|
choices=[],
|
|
widget=forms.Select(attrs={'class': 'switchable'}),
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(ResourceClassInfoAction, self).__init__(*args, **kwargs)
|
|
try:
|
|
images, more = glance.image_list_detailed(self.request)
|
|
except Exception:
|
|
exceptions.handle(self.request,
|
|
_('Unable to retrieve image list.'))
|
|
else:
|
|
self.fields['image_id'].choices = [
|
|
(image.id, image.name) for image in images]
|
|
|
|
def clean(self):
|
|
cleaned_data = super(ResourceClassInfoAction, self).clean()
|
|
|
|
name = cleaned_data.get('name')
|
|
resource_class_id = self.initial.get('resource_class_id', None)
|
|
try:
|
|
resource_classes = tuskar.ResourceClass.list(self.request)
|
|
except Exception:
|
|
resource_classes = []
|
|
msg = _('Unable to get resource class list')
|
|
exceptions.check_message(["Connection", "refused"], msg)
|
|
raise
|
|
for resource_class in resource_classes:
|
|
if resource_class.name == name and \
|
|
resource_class_id != resource_class.id:
|
|
raise forms.ValidationError(
|
|
_('The name "%s" is already used by'
|
|
' another resource class.')
|
|
% name
|
|
)
|
|
return cleaned_data
|
|
|
|
class Meta:
|
|
name = _("Class Settings")
|
|
help_text = _("From here you can fill in the class settings.")
|
|
|
|
|
|
class CreateResourceClassInfo(workflows.Step):
|
|
action_class = ResourceClassInfoAction
|
|
contributes = ("name", "service_type", "image_id")
|
|
|
|
|
|
class ResourceClassFlavorsAction(workflows.Action):
|
|
def clean(self):
|
|
cleaned_data = super(ResourceClassFlavorsAction, self).clean()
|
|
table = self.initial.get('_tables', {}).get('flavors')
|
|
if table:
|
|
formset = table.get_formset()
|
|
if formset.is_valid():
|
|
cleaned_data['flavors'] = [form.cleaned_data
|
|
for form in formset
|
|
if form.cleaned_data
|
|
and not form.cleaned_data['DELETE']]
|
|
else:
|
|
raise forms.ValidationError(_('Errors in the flavors list.'))
|
|
return cleaned_data
|
|
|
|
class Meta:
|
|
name = _("Flavors")
|
|
help_text = _("From here you can add flavors to the class.")
|
|
|
|
|
|
class CreateResourceClassFlavors(tuskar_ui.workflows.TableStep):
|
|
table_classes = (tables.FlavorsFormsetTable,)
|
|
|
|
action_class = ResourceClassFlavorsAction
|
|
template_name = TEMPLATE_PREFIX + '_flavors_step.html'
|
|
contributes = ("flavors",)
|
|
|
|
def get_flavors_data(self):
|
|
try:
|
|
resource_class_id = self.workflow.context.get("resource_class_id")
|
|
if resource_class_id:
|
|
resource_class = tuskar.ResourceClass.get(
|
|
self.workflow.request, resource_class_id)
|
|
flavors = resource_class.list_flavors
|
|
else:
|
|
flavors = []
|
|
except Exception:
|
|
flavors = []
|
|
exceptions.handle(self.workflow.request,
|
|
_('Unable to retrieve resource flavors list.'))
|
|
return flavors
|
|
|
|
|
|
class RacksAction(workflows.Action):
|
|
class Meta:
|
|
name = _("Racks")
|
|
|
|
def clean(self):
|
|
cleaned_data = super(RacksAction, self).clean()
|
|
table = self.initial.get('_tables', {}).get('racks')
|
|
if table:
|
|
formset = table.get_formset()
|
|
if formset.is_valid():
|
|
cleaned_data['racks_object_ids'] = [
|
|
form.cleaned_data['id'] for form in formset
|
|
if form.cleaned_data and
|
|
form.cleaned_data.get('selected') and
|
|
not form.cleaned_data.get('DELETE')]
|
|
else:
|
|
raise forms.ValidationError(_('Errors in the racks table.'))
|
|
return cleaned_data
|
|
|
|
|
|
class CreateRacks(tuskar_ui.workflows.TableStep):
|
|
table_classes = (tables.RacksFormsetTable,)
|
|
|
|
action_class = RacksAction
|
|
contributes = ("racks_object_ids")
|
|
template_name = TEMPLATE_PREFIX + '_table_step.html'
|
|
|
|
def contribute(self, data, context):
|
|
context.update(data)
|
|
return context
|
|
|
|
def get_racks_data(self):
|
|
try:
|
|
resource_class_id = self.workflow.context.get("resource_class_id")
|
|
if resource_class_id:
|
|
resource_class = tuskar.ResourceClass.get(
|
|
self.workflow.request,
|
|
resource_class_id)
|
|
selected_racks = resource_class.racks_ids
|
|
racks = resource_class.all_racks
|
|
for rack in racks:
|
|
rack.selected = (rack.id in selected_racks)
|
|
else:
|
|
racks = tuskar.Rack.list(self.workflow.request, True)
|
|
except Exception:
|
|
racks = []
|
|
exceptions.handle(self.workflow.request,
|
|
_('Unable to retrieve racks list.'))
|
|
|
|
return racks
|
|
|
|
|
|
class ResourceClassWorkflowMixin:
|
|
# FIXME active tabs coflict
|
|
# When on page with tabs, the workflow with more steps is used,
|
|
# there is a conflict of active tabs and it always shows the
|
|
# first tab after an action. So I explicitly specify to what
|
|
# tab it should redirect after action, until the coflict will
|
|
# be fixed in Horizon.
|
|
def get_index_url(self):
|
|
# This url is used both as success and failure url
|
|
return "%s?tab=resource_management_tabs__resource_classes_tab" % (
|
|
urlresolvers.reverse(URL_PREFIX + 'index'))
|
|
|
|
def get_success_url(self):
|
|
return self.get_index_url()
|
|
|
|
def get_failure_url(self):
|
|
return self.get_index_url()
|
|
|
|
def format_status_message(self, message):
|
|
name = self.context.get('name')
|
|
return message % name
|
|
|
|
def _get_flavors(self, request, data):
|
|
flavors = []
|
|
resource_class_name = data['name']
|
|
for flavor in data.get('flavors') or []:
|
|
capacities = []
|
|
for name, info in flavors_forms.CAPACITIES.items():
|
|
label, unit, required = info
|
|
value = flavor.get(name, '')
|
|
if value is None:
|
|
value = ''
|
|
capacities.append({'name': name, 'value': str(value),
|
|
'unit': unit})
|
|
# FIXME: tuskar uses resource-class-name prefix for flavors,
|
|
# e.g. m1.large, we add rc name to the template name:
|
|
flavor_name = "%s.%s" % (resource_class_name, flavor['name'])
|
|
# FIXME: for now just use blank max_vms
|
|
flavors.append({'name': flavor_name, 'capacities': capacities,
|
|
'max_vms': None, 'id': data.get('id')})
|
|
return flavors
|
|
|
|
def _add_racks(self, request, data, resource_class):
|
|
ids_to_add = data.get('racks_object_ids') or []
|
|
resource_class.set_racks(request, ids_to_add)
|
|
|
|
|
|
class CreateResourceClass(ResourceClassWorkflowMixin, workflows.Workflow):
|
|
default_steps = (CreateResourceClassInfo, CreateResourceClassFlavors,
|
|
CreateRacks)
|
|
|
|
slug = "create_resource_class"
|
|
name = _("Create Class")
|
|
finalize_button_name = _("Create Class")
|
|
success_message = _('Created class "%s".')
|
|
failure_message = _('Unable to create class "%s".')
|
|
|
|
def _create_resource_class_info(self, request, data):
|
|
try:
|
|
if data['service_type'] == 'compute':
|
|
flavors = self._get_flavors(request, data)
|
|
else:
|
|
flavors = []
|
|
return tuskar.ResourceClass.create(
|
|
request,
|
|
name=data['name'],
|
|
service_type=data['service_type'],
|
|
image_id=data['image_id'],
|
|
flavors=flavors)
|
|
except Exception:
|
|
redirect = self.get_failure_url()
|
|
exceptions.handle(request,
|
|
_('Unable to create resource class.'),
|
|
redirect=redirect)
|
|
return None
|
|
|
|
def handle(self, request, data):
|
|
resource_class = self._create_resource_class_info(request, data)
|
|
self._add_racks(request, data, resource_class)
|
|
return True
|
|
|
|
|
|
class UpdateResourceClassInfo(CreateResourceClassInfo):
|
|
depends_on = ("resource_class_id",)
|
|
|
|
|
|
class UpdateResourceClassFlavors(CreateResourceClassFlavors):
|
|
depends_on = ("resource_class_id",)
|
|
|
|
|
|
class UpdateRacks(CreateRacks):
|
|
depends_on = ("resource_class_id",)
|
|
|
|
|
|
class UpdateResourceClass(ResourceClassWorkflowMixin, workflows.Workflow):
|
|
default_steps = (UpdateResourceClassInfo, UpdateResourceClassFlavors,
|
|
UpdateRacks)
|
|
|
|
slug = "update_resource_class"
|
|
name = _("Update Class")
|
|
finalize_button_name = _("Update Class")
|
|
success_message = _('Updated class "%s".')
|
|
failure_message = _('Unable to update class "%s".')
|
|
|
|
def _update_resource_class_info(self, request, data):
|
|
try:
|
|
if data['service_type'] == 'compute':
|
|
flavors = self._get_flavors(request, data)
|
|
else:
|
|
flavors = []
|
|
return tuskar.ResourceClass.update(
|
|
request,
|
|
data['resource_class_id'],
|
|
name=data['name'],
|
|
service_type=data['service_type'],
|
|
image_id=data['image_id'],
|
|
flavors=flavors)
|
|
except Exception:
|
|
redirect = self.get_failure_url()
|
|
exceptions.handle(request,
|
|
_('Unable to create resource class.'),
|
|
redirect=redirect)
|
|
return None
|
|
|
|
def handle(self, request, data):
|
|
resource_class = self._update_resource_class_info(request, data)
|
|
self._add_racks(request, data, resource_class)
|
|
return True
|
|
|
|
|
|
class DetailUpdateWorkflow(UpdateResourceClass):
|
|
def get_index_url(self):
|
|
# This url is used both as success and failure url
|
|
url = URL_PREFIX + "resource_classes:detail"
|
|
resource_class_id = self.context["resource_class_id"]
|
|
return "%s?tab=resource_class_details__overview" % (
|
|
urlresolvers.reverse(url, args=(resource_class_id,)))
|
|
|
|
|
|
class UpdateFlavorsWorkflow(UpdateResourceClass):
|
|
def get_index_url(self):
|
|
# This url is used both as success and failure url
|
|
url = URL_PREFIX + "resource_classes:detail"
|
|
resource_class_id = self.context["resource_class_id"]
|
|
return "%s?tab=resource_class_details__flavors" % (
|
|
urlresolvers.reverse(url, args=(resource_class_id,)))
|
|
|
|
|
|
class UpdateRacksWorkflow(UpdateResourceClass):
|
|
def get_index_url(self):
|
|
# This url is used both as success and failure url
|
|
url = URL_PREFIX + "resource_classes:detail"
|
|
resource_class_id = self.context["resource_class_id"]
|
|
return "%s?tab=resource_class_details__racks" % (
|
|
urlresolvers.reverse(url, args=(resource_class_id,)))
|