storagepools.py

#
# Project Kimchi
#
# Copyright IBM Corp, 2015-2017
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
import cherrypy
from wok.control.base import Collection
from wok.control.base import Resource
from wok.control.utils import get_class_name
from wok.control.utils import model_fn
from wok.control.utils import UrlSubNode
from wok.control.utils import validate_params
from wok.plugins.kimchi.control.storagevolumes import IsoVolumes
from wok.plugins.kimchi.control.storagevolumes import StorageVolumes
from wok.plugins.kimchi.model.storagepools import ISO_POOL_NAME


STORAGEPOOLS_REQUESTS = {'POST': {'default': 'KCHPOOL0001L'}}

STORAGEPOOL_REQUESTS = {
    'DELETE': {'default': 'KCHPOOL0002L'},
    'PUT': {'default': 'KCHPOOL0003L'},
    'POST': {'activate': 'KCHPOOL0004L', 'deactivate': 'KCHPOOL0005L'},
}


@UrlSubNode('storagepools', True)
class StoragePools(Collection):
    def __init__(self, model):
        super(StoragePools, self).__init__(model)
        self.admin_methods = ['POST']
        self.resource = StoragePool
        isos = IsoPool(model)
        setattr(self, ISO_POOL_NAME, isos)

        # set user log messages and make sure all parameters are present
        self.log_map = STORAGEPOOLS_REQUESTS
        self.log_args.update({'name': '', 'type': ''})

    def create(self, params, *args):
        try:
            create = getattr(self.model, model_fn(self, 'create'))
        except AttributeError:
            error = 'Create is not allowed for %s' % get_class_name(self)
            raise cherrypy.HTTPError(405, error)

        validate_params(params, self, 'create')
        args = self.model_args + [params]
        name = create(*args)
        args = self.resource_args + [name]
        res = self.resource(self.model, *args)
        res.lookup()
        resp = res.get()

        if 'task_id' in res.data:
            cherrypy.response.status = 202
        else:
            cherrypy.response.status = 201

        return resp

    def _get_resources(self, filter_params):
        try:
            res_list = super(StoragePools, self)._get_resources(filter_params)
            # Append reserved pools
            isos = getattr(self, ISO_POOL_NAME)
            isos.lookup()
            res_list.append(isos)
        except AttributeError:
            pass

        return res_list


class StoragePool(Resource):
    def __init__(self, model, ident):
        super(StoragePool, self).__init__(model, ident)
        self.admin_methods = ['PUT', 'POST', 'DELETE']
        self.uri_fmt = '/storagepools/%s'
        self.activate = self.generate_action_handler('activate')
        self.deactivate = self.generate_action_handler(
            'deactivate', destructive=True)
        self.storagevolumes = StorageVolumes(self.model, ident)
        self.log_map = STORAGEPOOL_REQUESTS

    @property
    def data(self):
        res = {
            'name': self.ident,
            'state': self.info['state'],
            'capacity': self.info['capacity'],
            'allocated': self.info['allocated'],
            'available': self.info['available'],
            'path': self.info['path'],
            'source': self.info['source'],
            'type': self.info['type'],
            'nr_volumes': self.info['nr_volumes'],
            'autostart': self.info['autostart'],
            'persistent': self.info['persistent'],
            'in_use': self.info['in_use'],
        }

        val = self.info.get('task_id')
        if val:
            res['task_id'] = val

        return res


class IsoPool(Resource):
    def __init__(self, model):
        super(IsoPool, self).__init__(model, ISO_POOL_NAME)
        self.storagevolumes = IsoVolumes(self.model, ISO_POOL_NAME)

    @property
    def data(self):
        return {
            'name': self.ident,
            'state': self.info['state'],
            'type': self.info['type'],
        }