mirror of
https://github.com/gryf/mistral-evacuate.git
synced 2026-02-07 08:45:48 +01:00
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
"""
|
|
FilterVmAction - custom action.
|
|
|
|
Simple action for filtering VM on the presence of metadata/extra spec
|
|
"evacuate" flag
|
|
"""
|
|
from mistral.actions.openstack.actions import NovaAction
|
|
from mistral.workflow.utils import Result
|
|
|
|
|
|
class FilterVmException(Exception):
|
|
pass
|
|
|
|
|
|
class FilterVmAction(NovaAction):
|
|
"""
|
|
Filter and return VMs whith the flag 'evacuate' either on vm metadtata
|
|
or flavor extra spec.
|
|
"""
|
|
|
|
def __init__(self, metadata, flavor, uuid):
|
|
"""init."""
|
|
self._metadata = metadata
|
|
self._flavor = flavor
|
|
self._uuid = uuid
|
|
|
|
def run(self):
|
|
"""Entry point for the action execution."""
|
|
client = self._get_client()
|
|
metadata = self._metadata
|
|
|
|
if str(metadata.get('evacuate')).upper() == 'TRUE':
|
|
return Result(data={'status': 0, 'uuid': self._uuid})
|
|
elif str(metadata.get('evacuate')).upper() == 'FALSE':
|
|
return Result(error='evacuate for vm %s is disabled' % self._uuid)
|
|
|
|
# ether is no metadata for vm - check flavor
|
|
try:
|
|
flavor = [x for x in client.flavors.list() if x.id == '1'][0]
|
|
except IndexError:
|
|
raise FilterVmException('Flavor not found')
|
|
|
|
evacuate = flavor.get_keys().get('evacuation:evacuate')
|
|
|
|
if str(evacuate).upper() == 'TRUE':
|
|
return Result(data={'status': 0, 'uuid': self._uuid})
|
|
|
|
return Result(error='evacuate for vm %s is disabled' % self._uuid)
|