a7455a8bf7
The BaseException class defined in exceptions.py is shadowing the built-in BaseException class of the Python exception hierarchy, which could potentially cause confusion. This removes the BaseException definition and replaces it with the existing WatcherException object. Instantiations of the IllegalArgumentException are also changed to use the message kwarg. Change-Id: I20abf135805c7a354924de8a5194b59fc040460a Closes-Bug: #1535504
75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
# -*- encoding: utf-8 -*-
|
|
# Copyright (c) 2015 b<>com
|
|
#
|
|
# Authors: Jean-Emile DARTOIS <jean-emile.dartois@b-com.com>
|
|
#
|
|
# 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 watcher._i18n import _
|
|
from watcher.applier.primitives import base
|
|
from watcher.applier import promise
|
|
from watcher.common import exception
|
|
from watcher.common import keystone as kclient
|
|
from watcher.common import nova as nclient
|
|
from watcher.decision_engine.model import hypervisor_state as hstate
|
|
|
|
|
|
class ChangeNovaServiceState(base.BasePrimitive):
|
|
def __init__(self):
|
|
"""This class allows us to change the state of nova-compute service."""
|
|
super(ChangeNovaServiceState, self).__init__()
|
|
self._host = self.applies_to
|
|
self._state = self.input_parameters.get('state')
|
|
|
|
@property
|
|
def host(self):
|
|
return self._host
|
|
|
|
@property
|
|
def state(self):
|
|
return self._state
|
|
|
|
@promise.Promise
|
|
def execute(self):
|
|
target_state = None
|
|
if self.state == hstate.HypervisorState.OFFLINE.value:
|
|
target_state = False
|
|
elif self.status == hstate.HypervisorState.ONLINE.value:
|
|
target_state = True
|
|
return self.nova_manage_service(target_state)
|
|
|
|
@promise.Promise
|
|
def undo(self):
|
|
target_state = None
|
|
if self.state == hstate.HypervisorState.OFFLINE.value:
|
|
target_state = True
|
|
elif self.state == hstate.HypervisorState.ONLINE.value:
|
|
target_state = False
|
|
return self.nova_manage_service(target_state)
|
|
|
|
def nova_manage_service(self, state):
|
|
if state is None:
|
|
raise exception.IllegalArgumentException(
|
|
message=_("The target state is not defined"))
|
|
|
|
keystone = kclient.KeystoneClient()
|
|
wrapper = nclient.NovaClient(keystone.get_credentials(),
|
|
session=keystone.get_session())
|
|
if state is True:
|
|
return wrapper.enable_service_nova_compute(self.host)
|
|
else:
|
|
return wrapper.disable_service_nova_compute(self.host)
|