239 lines
8.7 KiB
Python
239 lines
8.7 KiB
Python
# 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.utils.translation import ugettext_lazy as _
|
|
|
|
from horizon import exceptions
|
|
from horizon import forms
|
|
from horizon import workflows
|
|
|
|
from openstack_dashboard import api
|
|
from openstack_dashboard.dashboards.admin.aggregates import constants
|
|
|
|
|
|
class SetAggregateInfoAction(workflows.Action):
|
|
name = forms.CharField(label=_("Name"),
|
|
max_length=255)
|
|
|
|
availability_zone = forms.CharField(label=_("Availability Zone"),
|
|
max_length=255,
|
|
required=False)
|
|
|
|
class Meta:
|
|
name = _("Host Aggregate Info")
|
|
help_text = _("From here you can create a new "
|
|
"host aggregate to organize instances.")
|
|
slug = "set_aggregate_info"
|
|
|
|
def clean(self):
|
|
cleaned_data = super(SetAggregateInfoAction, self).clean()
|
|
name = cleaned_data.get('name')
|
|
|
|
try:
|
|
aggregates = api.nova.aggregate_details_list(self.request)
|
|
except Exception:
|
|
msg = _('Unable to get host aggregate list')
|
|
exceptions.check_message(["Connection", "refused"], msg)
|
|
raise
|
|
if aggregates is not None:
|
|
for aggregate in aggregates:
|
|
if aggregate.name.lower() == name.lower():
|
|
raise forms.ValidationError(
|
|
_('The name "%s" is already used by '
|
|
'another host aggregate.')
|
|
% name
|
|
)
|
|
return cleaned_data
|
|
|
|
|
|
class SetAggregateInfoStep(workflows.Step):
|
|
action_class = SetAggregateInfoAction
|
|
contributes = ("availability_zone",
|
|
"name")
|
|
|
|
|
|
class AddHostsToAggregateAction(workflows.MembershipAction):
|
|
def __init__(self, request, *args, **kwargs):
|
|
super(AddHostsToAggregateAction, self).__init__(request,
|
|
*args,
|
|
**kwargs)
|
|
err_msg = _('Unable to get the available hosts')
|
|
|
|
default_role_field_name = self.get_default_role_field_name()
|
|
self.fields[default_role_field_name] = forms.CharField(required=False)
|
|
self.fields[default_role_field_name].initial = 'member'
|
|
|
|
field_name = self.get_member_field_name('member')
|
|
self.fields[field_name] = forms.MultipleChoiceField(required=False)
|
|
|
|
hosts = []
|
|
try:
|
|
hosts = api.nova.host_list(request)
|
|
except Exception:
|
|
exceptions.handle(request, err_msg)
|
|
|
|
host_names = []
|
|
for host in hosts:
|
|
if host.host_name not in host_names and host.service == u'compute':
|
|
host_names.append(host.host_name)
|
|
host_names.sort()
|
|
|
|
self.fields[field_name].choices = \
|
|
[(host_name, host_name) for host_name in host_names]
|
|
|
|
class Meta:
|
|
name = _("Hosts within aggregate")
|
|
slug = "add_host_to_aggregate"
|
|
|
|
|
|
class ManageAggregateHostsAction(workflows.MembershipAction):
|
|
def __init__(self, request, *args, **kwargs):
|
|
super(ManageAggregateHostsAction, self).__init__(request,
|
|
*args,
|
|
**kwargs)
|
|
err_msg = _('Unable to get the available hosts')
|
|
|
|
default_role_field_name = self.get_default_role_field_name()
|
|
self.fields[default_role_field_name] = forms.CharField(required=False)
|
|
self.fields[default_role_field_name].initial = 'member'
|
|
|
|
field_name = self.get_member_field_name('member')
|
|
self.fields[field_name] = forms.MultipleChoiceField(required=False)
|
|
|
|
aggregate_id = self.initial['id']
|
|
aggregate = api.nova.aggregate_get(request, aggregate_id)
|
|
aggregate_hosts = aggregate.hosts
|
|
|
|
hosts = []
|
|
try:
|
|
hosts = api.nova.host_list(request)
|
|
except Exception:
|
|
exceptions.handle(request, err_msg)
|
|
|
|
host_names = []
|
|
for host in hosts:
|
|
if host.host_name not in host_names and host.service == u'compute':
|
|
host_names.append(host.host_name)
|
|
host_names.sort()
|
|
|
|
self.fields[field_name].choices = \
|
|
[(host_name, host_name) for host_name in host_names]
|
|
|
|
self.fields[field_name].initial = aggregate_hosts
|
|
|
|
class Meta:
|
|
name = _("Hosts within aggregate")
|
|
|
|
|
|
class AddHostsToAggregateStep(workflows.UpdateMembersStep):
|
|
action_class = AddHostsToAggregateAction
|
|
help_text = _("You can add hosts to this aggregate. One host can be added "
|
|
"to one or more aggregate. You can also add the hosts later "
|
|
"by editing the aggregate.")
|
|
available_list_title = _("All available hosts")
|
|
members_list_title = _("Selected hosts")
|
|
no_available_text = _("No hosts found.")
|
|
no_members_text = _("No host selected.")
|
|
show_roles = False
|
|
contributes = ("hosts_aggregate",)
|
|
|
|
def contribute(self, data, context):
|
|
if data:
|
|
member_field_name = self.get_member_field_name('member')
|
|
context['hosts_aggregate'] = data.get(member_field_name, [])
|
|
return context
|
|
|
|
|
|
class ManageAggregateHostsStep(workflows.UpdateMembersStep):
|
|
action_class = ManageAggregateHostsAction
|
|
help_text = _("You can add hosts to this aggregate, as well as remove "
|
|
"hosts from it.")
|
|
available_list_title = _("All Available Hosts")
|
|
members_list_title = _("Selected Hosts")
|
|
no_available_text = _("No Hosts found.")
|
|
no_members_text = _("No Host selected.")
|
|
show_roles = False
|
|
depends_on = ("id",)
|
|
contributes = ("hosts_aggregate",)
|
|
|
|
def contribute(self, data, context):
|
|
if data:
|
|
member_field_name = self.get_member_field_name('member')
|
|
context['hosts_aggregate'] = data.get(member_field_name, [])
|
|
return context
|
|
|
|
|
|
class CreateAggregateWorkflow(workflows.Workflow):
|
|
slug = "create_aggregate"
|
|
name = _("Create Host Aggregate")
|
|
finalize_button_name = _("Create Host Aggregate")
|
|
success_message = _('Created new host aggregate "%s".')
|
|
failure_message = _('Unable to create host aggregate "%s".')
|
|
success_url = constants.AGGREGATES_INDEX_URL
|
|
default_steps = (SetAggregateInfoStep, AddHostsToAggregateStep)
|
|
|
|
def format_status_message(self, message):
|
|
return message % self.context['name']
|
|
|
|
def handle(self, request, context):
|
|
try:
|
|
self.object = \
|
|
api.nova.aggregate_create(
|
|
request,
|
|
name=context['name'],
|
|
availability_zone=context['availability_zone'])
|
|
except Exception:
|
|
exceptions.handle(request, _('Unable to create host aggregate.'))
|
|
return False
|
|
|
|
hosts = context['hosts_aggregate']
|
|
for host in hosts:
|
|
try:
|
|
api.nova.add_host_to_aggregate(request, self.object.id, host)
|
|
except Exception:
|
|
exceptions.handle(
|
|
request, _('Error adding Hosts to the aggregate.'))
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
class ManageAggregateHostsWorkflow(workflows.Workflow):
|
|
slug = "manage_hosts_aggregate"
|
|
name = _("Add/Remove Hosts to Aggregate")
|
|
finalize_button_name = _("Save")
|
|
success_message = _('The Aggregate was updated.')
|
|
failure_message = _('Unable to update the aggregate.')
|
|
success_url = constants.AGGREGATES_INDEX_URL
|
|
default_steps = (ManageAggregateHostsStep, )
|
|
|
|
def format_status_message(self, message):
|
|
return message
|
|
|
|
def handle(self, request, context):
|
|
hosts_aggregate = context['hosts_aggregate']
|
|
aggregate_id = context['id']
|
|
aggregate = api.nova.aggregate_get(request, aggregate_id)
|
|
aggregate_hosts = aggregate.hosts
|
|
for host in aggregate_hosts:
|
|
api.nova.remove_host_from_aggregate(request, aggregate_id, host)
|
|
|
|
for host in hosts_aggregate:
|
|
try:
|
|
api.nova.add_host_to_aggregate(request, aggregate_id, host)
|
|
except Exception:
|
|
exceptions.handle(
|
|
request, _('Error updating the aggregate.'))
|
|
return False
|
|
|
|
return True
|