diff options
Diffstat (limited to 'ms/py-executor/blueprints_grpc/executor_utils.py')
| -rw-r--r-- | ms/py-executor/blueprints_grpc/executor_utils.py | 138 |
1 files changed, 138 insertions, 0 deletions
diff --git a/ms/py-executor/blueprints_grpc/executor_utils.py b/ms/py-executor/blueprints_grpc/executor_utils.py new file mode 100644 index 000000000..224e35ef9 --- /dev/null +++ b/ms/py-executor/blueprints_grpc/executor_utils.py @@ -0,0 +1,138 @@ +#!/usr/bin/python +# +# Copyright © 2018-2019 AT&T Intellectual Property. +# +# 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 google.protobuf.timestamp_pb2 import Timestamp +from google.protobuf import struct_pb2 +from google.protobuf import json_format +import sys, importlib, importlib.util, json, time, datetime +import logging +from .proto.BluePrintProcessing_pb2 import ExecutionServiceInput, ExecutionServiceOutput +from .proto.BluePrintCommon_pb2 import Status, EVENT_COMPONENT_TRACE, EVENT_COMPONENT_PROCESSING, \ + EVENT_COMPONENT_EXECUTED, EVENT_COMPONENT_NOTIFICATION +from .script_executor_configuration import ScriptExecutorConfiguration + +logger = logging.getLogger("Utils") + + +def current_time(): + ts = time.time() + return datetime.datetime.fromtimestamp(ts).strftime("yyyy-MM-dd'T'HH:mm:ss.SSS'Z'") + + +def blueprint_id(input: ExecutionServiceInput): + blueprint_name = input.actionIdentifiers.blueprintName + blueprint_version = input.actionIdentifiers.blueprintVersion + return blueprint_name + '/' + blueprint_version + + +def blueprint_location(config: ScriptExecutorConfiguration, input: ExecutionServiceInput): + blueprint_name = input.actionIdentifiers.blueprintName + blueprint_version = input.actionIdentifiers.blueprintVersion + return config.blueprints_processor('blueprintDeployPath') + '/' + blueprint_name + '/' + blueprint_version + + +def instance_for_input(config: ScriptExecutorConfiguration, input: ExecutionServiceInput): + blueprint_name = input.actionIdentifiers.blueprintName + blueprint_version = input.actionIdentifiers.blueprintVersion + action_name = input.actionIdentifiers.actionName + # Get Blueprint python script location + script_location = blueprint_location(config, input) + '/' + 'Scripts/python/__init__.py' + logger.info(script_location) + + # Create Dynamic Module Name + module_name = blueprint_name + '-' + blueprint_version + spec = importlib.util.spec_from_file_location(module_name, script_location) + logger.info(spec) + dynamic_module = importlib.util.module_from_spec(spec) + # Add blueprint modules + sys.modules[spec.name] = dynamic_module + spec.loader.exec_module(dynamic_module) + script_clazz = getattr(dynamic_module, action_name) + return script_clazz() + + +def log_response(input: ExecutionServiceInput, message: str): + payload = struct_pb2.Struct() + payload['message'] = message + status = Status() + status.timestamp = current_time() + status.eventType = EVENT_COMPONENT_TRACE + return ExecutionServiceOutput(commonHeader=input.commonHeader, + actionIdentifiers=input.actionIdentifiers, + payload=payload, status=status) + + +def send_notification(input: ExecutionServiceInput, message: str): + payload = struct_pb2.Struct() + payload['message'] = message + status = Status() + status.timestamp = current_time() + status.eventType = EVENT_COMPONENT_NOTIFICATION + return ExecutionServiceOutput(commonHeader=input.commonHeader, + actionIdentifiers=input.actionIdentifiers, + payload=payload, status=status) + + +def ack_response(input: ExecutionServiceInput): + timestamp = Timestamp() + timestamp.GetCurrentTime() + response_common_header = input.commonHeader + status = Status() + status.timestamp = current_time() + status.eventType = EVENT_COMPONENT_PROCESSING + return ExecutionServiceOutput(commonHeader=response_common_header, + actionIdentifiers=input.actionIdentifiers, + status=status) + + +def success_response(input: ExecutionServiceInput, property_json: json, code: int): + timestamp = Timestamp() + timestamp.GetCurrentTime() + status = Status() + status.timestamp = current_time() + status.eventType = EVENT_COMPONENT_EXECUTED + status.code = code + status.message = 'success' + payload_struct = create_response_payload_from_json(input.actionIdentifiers.actionName, property_json) + return ExecutionServiceOutput(commonHeader=input.commonHeader, + actionIdentifiers=input.actionIdentifiers, status=status, payload=payload_struct) + + +def failure_response(input: ExecutionServiceInput, property_json: json, error_code: int, + error_message: str): + timestamp = Timestamp() + timestamp.GetCurrentTime() + status = Status() + status.timestamp = current_time() + status.eventType = EVENT_COMPONENT_EXECUTED + status.code = error_code + status.message = 'failure' + payload_struct = create_response_payload_from_json(input.actionIdentifiers.actionName, property_json) + status.errorMessage = error_message + return ExecutionServiceOutput(commonHeader=input.commonHeader, + actionIdentifiers=input.actionIdentifiers, status=status, payload=payload_struct) + + +def create_response_payload_from_json(action_name, property_json: json): + # Create response Pay load json from property Json + payload_key = action_name + '-response' + response_payload = {} + response_payload[payload_key] = property_json + response_payload_json = json.dumps(response_payload) + # Create Struct from Json + payload_struct = struct_pb2.Struct() + json_format.Parse(str(response_payload_json), payload_struct, ignore_unknown_fields=True) + return payload_struct |