diff options
Diffstat (limited to 'dcae-policy/dcaepolicyplugin/tasks.py')
| -rw-r--r-- | dcae-policy/dcaepolicyplugin/tasks.py | 127 |
1 files changed, 105 insertions, 22 deletions
diff --git a/dcae-policy/dcaepolicyplugin/tasks.py b/dcae-policy/dcaepolicyplugin/tasks.py index 2676864..fb98412 100644 --- a/dcae-policy/dcaepolicyplugin/tasks.py +++ b/dcae-policy/dcaepolicyplugin/tasks.py @@ -18,11 +18,12 @@ # # ECOMP is a trademark and service mark of AT&T Intellectual Property. -# Lifecycle interface calls for DockerContainer +"""tasks are the cloudify operations invoked on interfaces defined in the blueprint""" import json import uuid - +import copy +import traceback import requests from cloudify import ctx @@ -35,12 +36,20 @@ from .discovery import discover_service_url POLICY_ID = 'policy_id' POLICY_REQUIRED = 'policy_required' POLICY_BODY = 'policy_body' +POLICIES_FILTERED = 'policies_filtered' +POLICY_FILTER = 'policy_filter' + +REQUEST_ID = "requestID" + DCAE_POLICY_TYPE = 'dcae.nodes.policy' +DCAE_POLICIES_TYPE = 'dcae.nodes.policies' +DCAE_POLICY_TYPES = [DCAE_POLICY_TYPE, DCAE_POLICIES_TYPE] class PolicyHandler(object): """talk to policy-handler""" SERVICE_NAME_POLICY_HANDLER = "policy_handler" X_ECOMP_REQUESTID = 'X-ECOMP-RequestID' + STATUS_CODE_POLICIES_NOT_FOUND = 404 _url = None @staticmethod @@ -49,53 +58,127 @@ class PolicyHandler(object): if PolicyHandler._url: return - PolicyHandler._url = "{0}/policy_latest".format( - discover_service_url(PolicyHandler.SERVICE_NAME_POLICY_HANDLER) - ) + PolicyHandler._url = discover_service_url(PolicyHandler.SERVICE_NAME_POLICY_HANDLER) @staticmethod def get_latest_policy(policy_id): """retrieve the latest policy for policy_id from policy-handler""" PolicyHandler._lazy_init() - ph_path = "{0}/{1}".format(PolicyHandler._url, policy_id) + ph_path = "{0}/policy_latest/{1}".format(PolicyHandler._url, policy_id) headers = {PolicyHandler.X_ECOMP_REQUESTID: str(uuid.uuid4())} ctx.logger.info("getting latest policy from {0} headers={1}".format( \ ph_path, json.dumps(headers))) res = requests.get(ph_path, headers=headers) + + if res.status_code == PolicyHandler.STATUS_CODE_POLICIES_NOT_FOUND: + return + res.raise_for_status() + return res.json() - if res.status_code == requests.codes.ok: - return res.json() - return {} + @staticmethod + def find_latest_policies(policy_filter): + """retrieve the latest policies by policy filter (selection criteria) from policy-handler""" + PolicyHandler._lazy_init() -######################################################### -@operation -def policy_get(**kwargs): - """retrieve the latest policy_body for policy_id property and save it in runtime_properties""" - if ctx.type != NODE_INSTANCE or DCAE_POLICY_TYPE not in ctx.node.type_hierarchy: - error = "can only invoke policy_get on node of type {0}".format(DCAE_POLICY_TYPE) - ctx.logger.error(error) - raise NonRecoverableError(error) + ph_path = "{0}/policies_latest".format(PolicyHandler._url) + headers = { + PolicyHandler.X_ECOMP_REQUESTID: policy_filter.get(REQUEST_ID, str(uuid.uuid4())) + } + + ctx.logger.info("finding the latest polices from {0} by {1} headers={2}".format( \ + ph_path, json.dumps(policy_filter), json.dumps(headers))) + + res = requests.post(ph_path, json=policy_filter, headers=headers) - if POLICY_ID not in ctx.node.properties: + if res.status_code == PolicyHandler.STATUS_CODE_POLICIES_NOT_FOUND: + return + + res.raise_for_status() + return res.json() + +def _policy_get(): + """ + dcae.nodes.policy - + retrieve the latest policy_body for policy_id property + and save policy_body in runtime_properties + """ + if DCAE_POLICY_TYPE not in ctx.node.type_hierarchy: + return + + policy_id = ctx.node.properties.get(POLICY_ID) + if not policy_id: error = "no {0} found in ctx.node.properties".format(POLICY_ID) ctx.logger.error(error) raise NonRecoverableError(error) try: - policy_id = ctx.node.properties[POLICY_ID] policy = PolicyHandler.get_latest_policy(policy_id) if not policy: raise NonRecoverableError("policy not found for policy_id {0}".format(policy_id)) - ctx.logger.info("found policy {0}".format(json.dumps(policy))) + ctx.logger.info("found policy {0}: {1}".format(policy_id, json.dumps(policy))) if POLICY_BODY in policy: ctx.instance.runtime_properties[POLICY_BODY] = policy[POLICY_BODY] except Exception as ex: - error = "failed to get policy: {0}".format(str(ex)) - ctx.logger.error(error) + error = "failed to get policy({0}): {1}".format(policy_id, str(ex)) + ctx.logger.error("{0}: {1}".format(error, traceback.format_exc())) if ctx.node.properties.get(POLICY_REQUIRED, True): raise NonRecoverableError(error) + + return True + +def _policies_find(): + """ + dcae.nodes.policies - + retrieve the latest policies for selection criteria + and save found policies in runtime_properties + """ + if DCAE_POLICIES_TYPE not in ctx.node.type_hierarchy: + return + + try: + policy_filter = copy.deepcopy(dict( + (k, v) for (k, v) in dict(ctx.node.properties.get(POLICY_FILTER, {})).iteritems() + if v or isinstance(v, (int, float)) + )) + if REQUEST_ID not in policy_filter: + policy_filter[REQUEST_ID] = str(uuid.uuid4()) + + policies_filtered = PolicyHandler.find_latest_policies(policy_filter) + + if not policies_filtered: + ctx.logger.info("policies not found by {0}".format(json.dumps(policy_filter))) + return True + + ctx.logger.info("found policies by {0}: {1}".format( + json.dumps(policy_filter), json.dumps(policies_filtered) + )) + ctx.instance.runtime_properties[POLICIES_FILTERED] = policies_filtered + + except Exception as ex: + error = "failed to find policies: {0}".format(str(ex)) + ctx.logger.error("{0}: {1}".format(error, traceback.format_exc())) + raise NonRecoverableError(error) + + return True + +######################################################### +@operation +def policy_get(**kwargs): + """retrieve the policy or policies and save it in runtime_properties""" + if ctx.type != NODE_INSTANCE: + error = "can only invoke policy_get on node of types: {0}".format(DCAE_POLICY_TYPES) + ctx.logger.error(error) + raise NonRecoverableError(error) + + if not _policy_get() and not _policies_find(): + error = "unexpected node type {0} for policy_get - expected types: {1}" \ + .format(ctx.node.type_hierarchy, DCAE_POLICY_TYPES) + ctx.logger.error(error) + raise NonRecoverableError(error) + + ctx.logger.info("exit policy_get") |