140 lines
4.5 KiB
Python
140 lines
4.5 KiB
Python
# Copyright 2013 - Mirantis, Inc.
|
|
# Copyright 2015 - StackStorm, Inc.
|
|
#
|
|
# 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 oslo_log import log as logging
|
|
import pecan
|
|
from pecan import hooks
|
|
from pecan import rest
|
|
from wsme import types as wtypes
|
|
import wsmeext.pecan as wsme_pecan
|
|
|
|
from mistral.api import access_control as acl
|
|
from mistral.api.controllers import resource
|
|
from mistral.api.controllers.v2 import validation
|
|
from mistral.api.hooks import content_type as ct_hook
|
|
from mistral import context
|
|
from mistral.db.v2 import api as db_api
|
|
from mistral.services import workbooks
|
|
from mistral.utils import rest_utils
|
|
from mistral.workbook import parser as spec_parser
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
SCOPE_TYPES = wtypes.Enum(str, 'private', 'public')
|
|
|
|
|
|
class Workbook(resource.Resource):
|
|
"""Workbook resource."""
|
|
|
|
id = wtypes.text
|
|
name = wtypes.text
|
|
|
|
definition = wtypes.text
|
|
"workbook definition in Mistral v2 DSL"
|
|
tags = [wtypes.text]
|
|
scope = SCOPE_TYPES
|
|
"'private' or 'public'"
|
|
|
|
created_at = wtypes.text
|
|
updated_at = wtypes.text
|
|
|
|
@classmethod
|
|
def sample(cls):
|
|
return cls(id='123e4567-e89b-12d3-a456-426655440000',
|
|
name='book',
|
|
definition='HERE GOES'
|
|
'WORKBOOK DEFINITION IN MISTRAL DSL v2',
|
|
tags=['large', 'expensive'],
|
|
scope='private',
|
|
created_at='1970-01-01T00:00:00.000000',
|
|
updated_at='1970-01-01T00:00:00.000000')
|
|
|
|
|
|
class Workbooks(resource.Resource):
|
|
"""A collection of Workbooks."""
|
|
|
|
workbooks = [Workbook]
|
|
|
|
@classmethod
|
|
def sample(cls):
|
|
return cls(workbooks=[Workbook.sample()])
|
|
|
|
|
|
class WorkbooksController(rest.RestController, hooks.HookController):
|
|
__hooks__ = [ct_hook.ContentTypeHook("application/json", ['POST', 'PUT'])]
|
|
|
|
validate = validation.SpecValidationController(
|
|
spec_parser.get_workbook_spec_from_yaml)
|
|
|
|
@rest_utils.wrap_wsme_controller_exception
|
|
@wsme_pecan.wsexpose(Workbook, wtypes.text)
|
|
def get(self, name):
|
|
"""Return the named workbook."""
|
|
acl.enforce('workbooks:get', context.ctx())
|
|
LOG.info("Fetch workbook [name=%s]" % name)
|
|
|
|
db_model = db_api.get_workbook(name)
|
|
|
|
return Workbook.from_dict(db_model.to_dict())
|
|
|
|
@rest_utils.wrap_pecan_controller_exception
|
|
@pecan.expose(content_type="text/plain")
|
|
def put(self):
|
|
"""Update a workbook."""
|
|
acl.enforce('workbooks:update', context.ctx())
|
|
definition = pecan.request.text
|
|
LOG.info("Update workbook [definition=%s]" % definition)
|
|
|
|
wb_db = workbooks.update_workbook_v2(definition)
|
|
|
|
return Workbook.from_dict(wb_db.to_dict()).to_string()
|
|
|
|
@rest_utils.wrap_pecan_controller_exception
|
|
@pecan.expose(content_type="text/plain")
|
|
def post(self):
|
|
"""Create a new workbook."""
|
|
acl.enforce('workbooks:create', context.ctx())
|
|
definition = pecan.request.text
|
|
LOG.info("Create workbook [definition=%s]" % definition)
|
|
|
|
wb_db = workbooks.create_workbook_v2(definition)
|
|
pecan.response.status = 201
|
|
|
|
return Workbook.from_dict(wb_db.to_dict()).to_string()
|
|
|
|
@rest_utils.wrap_wsme_controller_exception
|
|
@wsme_pecan.wsexpose(None, wtypes.text, status_code=204)
|
|
def delete(self, name):
|
|
"""Delete the named workbook."""
|
|
acl.enforce('workbooks:delete', context.ctx())
|
|
LOG.info("Delete workbook [name=%s]" % name)
|
|
|
|
db_api.delete_workbook(name)
|
|
|
|
@wsme_pecan.wsexpose(Workbooks)
|
|
def get_all(self):
|
|
"""Return all workbooks.
|
|
|
|
Where project_id is the same as the requestor or
|
|
project_id is different but the scope is public.
|
|
"""
|
|
acl.enforce('workbooks:list', context.ctx())
|
|
LOG.info("Fetch workbooks.")
|
|
|
|
workbooks_list = [Workbook.from_dict(db_model.to_dict())
|
|
for db_model in db_api.get_workbooks()]
|
|
|
|
return Workbooks(workbooks=workbooks_list)
|