ooi/ooi/api/query.py

127 lines
4.7 KiB
Python

# Copyright 2015 Spanish National Research Council
# Copyright 2015 LIP - INDIGO-DataCloud
#
# 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 ooi.api import base
import ooi.api.helpers
from ooi.occi.core import collection
from ooi.occi.core import entity
from ooi.occi.core import link
from ooi.occi.core import resource
from ooi.occi.infrastructure import compute
from ooi.occi.infrastructure import contextualization
from ooi.occi.infrastructure import ip_reservation
from ooi.occi.infrastructure import network
from ooi.occi.infrastructure import network_link
from ooi.occi.infrastructure import storage
from ooi.occi.infrastructure import storage_link
from ooi.occi.infrastructure import templates as infra_templates
from ooi.openstack import contextualization as os_contextualization
from ooi.openstack import network as os_network
from ooi.openstack import templates
class Controller(base.Controller):
def __init__(self, app, openstack_version,
neutron_ooi_endpoint=None):
self.neutron_ooi_endpoint = neutron_ooi_endpoint
super(Controller, self).__init__(
app=app,
openstack_version=openstack_version)
self.os_helper = ooi.api.helpers.OpenStackHelper(
self.app,
self.openstack_version
)
def _resource_tpls(self, req):
flavors = self.os_helper.get_flavors(req)
occi_resource_templates = []
if flavors:
for f in flavors:
tpl = templates.OpenStackResourceTemplate(f["id"],
f["name"],
f["vcpus"],
f["ram"],
f["disk"])
occi_resource_templates.append(tpl)
return occi_resource_templates
def _os_tpls(self, req):
images = self.os_helper.get_images(req)
occi_os_templates = []
if images:
for i in images:
tpl = templates.OpenStackOSTemplate(i["id"], i["name"])
occi_os_templates.append(tpl)
return occi_os_templates
def _ip_pools(self, req):
pools = self.os_helper.get_floating_ip_pools(req)
occi_ip_pools = []
if pools:
for p in pools:
occi_ip_pools.append(os_network.OSFloatingIPPool(p["name"]))
occi_ip_pools.append(os_network.os_floatingip_pool)
return occi_ip_pools
def index(self, req):
# OCCI Core Kinds:
kinds = []
actions = []
mixins = []
kinds.append(entity.Entity.kind)
kinds.append(resource.Resource.kind)
kinds.append(link.Link.kind)
# OCCI infra Compute:
kinds.append(compute.ComputeResource.kind)
actions.extend(compute.ComputeResource.actions)
# OCCI infra Storage
kinds.append(storage.StorageResource.kind)
kinds.append(storage_link.StorageLink.kind)
actions.extend(storage.StorageResource.actions)
# OCCI infra network
kinds.append(network.NetworkResource.kind)
actions.extend(network.NetworkResource.actions)
if self.neutron_ooi_endpoint:
mixins.append(os_network.neutron_network)
mixins.append(network.ip_network)
kinds.append(network_link.NetworkInterface.kind)
mixins.append(network_link.ip_network_interface)
kinds.append(ip_reservation.IPReservation.kind)
# OCCI infra compute mixins
mixins.append(infra_templates.os_tpl)
mixins.append(infra_templates.resource_tpl)
# OpenStack flavors & images
mixins.extend(self._resource_tpls(req))
mixins.extend(self._os_tpls(req))
# OpenStack Contextualization
mixins.append(os_contextualization.user_data)
mixins.append(os_contextualization.public_key)
# OCCI Contextualization
mixins.append(contextualization.user_data)
mixins.append(contextualization.ssh_key)
# OpenStack Floating IP Pools
mixins.extend(self._ip_pools(req))
return collection.Collection(kinds=kinds,
mixins=mixins,
actions=actions)