diff options
Diffstat (limited to 'azure/aria/aria-extension-cloudify/src/aria/aria/parser/consumption/modeling.py')
| -rw-r--r-- | azure/aria/aria-extension-cloudify/src/aria/aria/parser/consumption/modeling.py | 198 |
1 files changed, 198 insertions, 0 deletions
diff --git a/azure/aria/aria-extension-cloudify/src/aria/aria/parser/consumption/modeling.py b/azure/aria/aria-extension-cloudify/src/aria/aria/parser/consumption/modeling.py new file mode 100644 index 0000000..221b308 --- /dev/null +++ b/azure/aria/aria-extension-cloudify/src/aria/aria/parser/consumption/modeling.py @@ -0,0 +1,198 @@ +# Licensed to the Apache Software Foundation (ASF) under one or more +# contributor license agreements. See the NOTICE file distributed with +# this work for additional information regarding copyright ownership. +# The ASF licenses this file to You 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 .consumer import Consumer, ConsumerChain +from ...utils.formatting import json_dumps, yaml_dumps + + +class DeriveServiceTemplate(Consumer): + """ + Derives the service template from the presenter. + """ + + def consume(self): + if self.context.presentation.presenter is None: + self.context.validation.report('DeriveServiceTemplate consumer: missing presenter') + return + + if not hasattr(self.context.presentation.presenter, '_get_model'): + self.context.validation.report('DeriveServiceTemplate consumer: presenter does not' + ' support "_get_model"') + return + + self.context.modeling.template = \ + self.context.presentation.presenter._get_model(self.context) + + +class CoerceServiceTemplateValues(Consumer): + """ + Coerces values in the service template. + """ + + def consume(self): + self.topology.coerce(self.context.modeling.template, report_issues=True) + + +class ValidateServiceTemplate(Consumer): + """ + Validates the service template. + """ + + def consume(self): + self.topology.validate(self.context.modeling.template) + + +class ServiceTemplate(ConsumerChain): + """ + Generates the service template from the presenter. + """ + + def __init__(self, context): + super(ServiceTemplate, self).__init__(context, (DeriveServiceTemplate, + CoerceServiceTemplateValues, + ValidateServiceTemplate)) + + def dump(self): + if self.context.has_arg_switch('yaml'): + indent = self.context.get_arg_value_int('indent', 2) + raw = self.context.modeling.template_as_raw + self.context.write(yaml_dumps(raw, indent=indent)) + elif self.context.has_arg_switch('json'): + indent = self.context.get_arg_value_int('indent', 2) + raw = self.context.modeling.template_as_raw + self.context.write(json_dumps(raw, indent=indent)) + else: + self.context.write(self.topology.dump(self.context.modeling.template)) + + +class Types(Consumer): + """ + Used to just dump the types. + """ + + def dump(self): + if self.context.has_arg_switch('yaml'): + indent = self.context.get_arg_value_int('indent', 2) + raw = self.context.modeling.types_as_raw + self.context.write(yaml_dumps(raw, indent=indent)) + elif self.context.has_arg_switch('json'): + indent = self.context.get_arg_value_int('indent', 2) + raw = self.context.modeling.types_as_raw + self.context.write(json_dumps(raw, indent=indent)) + else: + self.topology.dump_types(self.context, self.context.modeling.template) + + +class InstantiateServiceInstance(Consumer): + """ + Instantiates the service template into a service instance. + """ + + def consume(self): + if self.context.modeling.template is None: + self.context.validation.report('InstantiateServiceInstance consumer: missing service ' + 'template') + return + self.context.modeling.instance = self.topology.instantiate( + self.context.modeling.template, + inputs=dict(self.context.modeling.inputs) + ) + + +class CoerceServiceInstanceValues(Consumer): + """ + Coerces values in the service instance. + """ + + def consume(self): + self.topology.coerce(self.context.modeling.instance, report_issues=True) + + +class ValidateServiceInstance(Consumer): + """ + Validates the service instance. + """ + + def consume(self): + self.topology.validate(self.context.modeling.instance) + + +class SatisfyRequirements(Consumer): + """ + Satisfies node requirements in the service instance. + """ + + def consume(self): + self.topology.satisfy_requirements(self.context.modeling.instance) + + +class ValidateCapabilities(Consumer): + """ + Validates capabilities in the service instance. + """ + + def consume(self): + self.topology.validate_capabilities(self.context.modeling.instance) + + +class FindHosts(Consumer): + """ + Find hosts for all nodes in the service instance. + """ + + def consume(self): + self.topology.assign_hosts(self.context.modeling.instance) + + +class ConfigureOperations(Consumer): + """ + Configures all operations in the service instance. + """ + + def consume(self): + self.topology.configure_operations(self.context.modeling.instance) + + +class ServiceInstance(ConsumerChain): + """ + Generates the service instance by instantiating the service template. + """ + + def __init__(self, context): + super(ServiceInstance, self).__init__(context, (InstantiateServiceInstance, + CoerceServiceInstanceValues, + ValidateServiceInstance, + CoerceServiceInstanceValues, + SatisfyRequirements, + CoerceServiceInstanceValues, + ValidateCapabilities, + FindHosts, + ConfigureOperations, + CoerceServiceInstanceValues)) + + def dump(self): + if self.context.has_arg_switch('graph'): + self.context.modeling.instance.dump_graph() + elif self.context.has_arg_switch('yaml'): + indent = self.context.get_arg_value_int('indent', 2) + raw = self.context.modeling.instance_as_raw + self.context.write(yaml_dumps(raw, indent=indent)) + elif self.context.has_arg_switch('json'): + indent = self.context.get_arg_value_int('indent', 2) + raw = self.context.modeling.instance_as_raw + self.context.write(json_dumps(raw, indent=indent)) + else: + str_rep = self.topology.dump(self.context.modeling.instance) + self.context.write(str_rep) |