102 lines
3.4 KiB
Python
102 lines
3.4 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2012 OpenStack Foundation
|
|
# Copyright 2011 Canonical Ltd.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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.
|
|
|
|
"""The Flavor extra data extension
|
|
|
|
OpenStack API version 1.1 lists "name", "ram", "disk", "vcpus" as flavor
|
|
attributes. This extension adds to that list:
|
|
|
|
- OS-FLV-EXT-DATA:ephemeral
|
|
"""
|
|
|
|
from nova.api.openstack import extensions
|
|
from nova.api.openstack import wsgi
|
|
from nova.api.openstack import xmlutil
|
|
|
|
|
|
authorize = extensions.soft_extension_authorizer('compute', 'flavorextradata')
|
|
|
|
|
|
class FlavorextradataController(wsgi.Controller):
|
|
def _extend_flavors(self, req, flavors):
|
|
for flavor in flavors:
|
|
db_flavor = req.get_db_flavor(flavor['id'])
|
|
key = "%s:ephemeral" % Flavorextradata.alias
|
|
flavor[key] = db_flavor['ephemeral_gb']
|
|
|
|
def _show(self, req, resp_obj):
|
|
if not authorize(req.environ['nova.context']):
|
|
return
|
|
if 'flavor' in resp_obj.obj:
|
|
resp_obj.attach(xml=FlavorextradatumTemplate())
|
|
self._extend_flavors(req, [resp_obj.obj['flavor']])
|
|
|
|
@wsgi.extends
|
|
def show(self, req, resp_obj, id):
|
|
return self._show(req, resp_obj)
|
|
|
|
@wsgi.extends(action='create')
|
|
def create(self, req, resp_obj, body):
|
|
return self._show(req, resp_obj)
|
|
|
|
@wsgi.extends
|
|
def detail(self, req, resp_obj):
|
|
if not authorize(req.environ['nova.context']):
|
|
return
|
|
resp_obj.attach(xml=FlavorextradataTemplate())
|
|
self._extend_flavors(req, list(resp_obj.obj['flavors']))
|
|
|
|
|
|
class Flavorextradata(extensions.ExtensionDescriptor):
|
|
"""Provide additional data for flavors."""
|
|
|
|
name = "FlavorExtraData"
|
|
alias = "OS-FLV-EXT-DATA"
|
|
namespace = ("http://docs.openstack.org/compute/ext/"
|
|
"flavor_extra_data/api/v1.1")
|
|
updated = "2011-09-14T00:00:00+00:00"
|
|
|
|
def get_controller_extensions(self):
|
|
controller = FlavorextradataController()
|
|
extension = extensions.ControllerExtension(self, 'flavors', controller)
|
|
return [extension]
|
|
|
|
|
|
def make_flavor(elem):
|
|
elem.set('{%s}ephemeral' % Flavorextradata.namespace,
|
|
'%s:ephemeral' % Flavorextradata.alias)
|
|
|
|
|
|
class FlavorextradatumTemplate(xmlutil.TemplateBuilder):
|
|
def construct(self):
|
|
root = xmlutil.TemplateElement('flavor', selector='flavor')
|
|
make_flavor(root)
|
|
alias = Flavorextradata.alias
|
|
namespace = Flavorextradata.namespace
|
|
return xmlutil.SlaveTemplate(root, 1, nsmap={alias: namespace})
|
|
|
|
|
|
class FlavorextradataTemplate(xmlutil.TemplateBuilder):
|
|
def construct(self):
|
|
root = xmlutil.TemplateElement('flavors')
|
|
elem = xmlutil.SubTemplateElement(root, 'flavor', selector='flavors')
|
|
make_flavor(elem)
|
|
alias = Flavorextradata.alias
|
|
namespace = Flavorextradata.namespace
|
|
return xmlutil.SlaveTemplate(root, 1, nsmap={alias: namespace})
|