deb-mistral/mistral/services/workflows.py
Nikolay Mahotkin 1c702e2555 Adding 'is_system' to definition model
* Fixed work of mistral-db-manage populate
 * Fixed modifying of standard workflows

Closes-Bug: #1478448
Closes-Bug: #1396121

Change-Id: Ia2825408ee43a6dfc80e33fa9614e11f3ff6d97b
2015-09-17 13:31:27 +03:00

102 lines
2.8 KiB
Python

# Copyright 2013 - Mirantis, 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 mistral.db.v2 import api as db_api
from mistral import exceptions as exc
from mistral import utils
from mistral.workbook import parser as spec_parser
STD_WF_PATH = 'resources/workflows'
def register_standard_workflows():
workflow_paths = utils.get_file_list(STD_WF_PATH)
for wf_path in workflow_paths:
workflow_definition = open(wf_path).read()
create_workflows(workflow_definition, scope='public', is_system=True)
def _clear_system_workflow_db():
db_api.delete_workflow_definitions(is_system=True)
def sync_db():
_clear_system_workflow_db()
register_standard_workflows()
def create_workflows(definition, scope='private', is_system=False):
wf_list_spec = spec_parser.get_workflow_list_spec_from_yaml(definition)
db_wfs = []
with db_api.transaction():
for wf_spec in wf_list_spec.get_workflows():
db_wfs.append(
_create_workflow(wf_spec, definition, scope, is_system)
)
return db_wfs
def update_workflows(definition, scope='private'):
wf_list_spec = spec_parser.get_workflow_list_spec_from_yaml(definition)
db_wfs = []
with db_api.transaction():
for wf_spec in wf_list_spec.get_workflows():
db_wfs.append(_update_workflow(
wf_spec,
definition,
scope
))
return db_wfs
def _get_workflow_values(wf_spec, definition, scope, is_system=False):
values = {
'name': wf_spec.get_name(),
'tags': wf_spec.get_tags(),
'definition': definition,
'spec': wf_spec.to_dict(),
'scope': scope,
'is_system': is_system
}
return values
def _create_workflow(wf_spec, definition, scope, is_system):
return db_api.create_workflow_definition(
_get_workflow_values(wf_spec, definition, scope, is_system)
)
def _update_workflow(wf_spec, definition, scope):
workflow = db_api.load_workflow_definition(wf_spec.get_name())
if workflow and workflow.is_system:
raise exc.InvalidActionException(
"Attempt to modify a system workflow: %s" %
workflow.name
)
values = _get_workflow_values(wf_spec, definition, scope)
return db_api.update_workflow_definition(values['name'], values)