X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=near-rt-ric-simulator%2Fsrc%2FSTD_2.0.0%2Fa1.py;h=9b68a6cc6abfe34cb02350cce6ed761454430512;hb=d9b25321d9bce8a3db88102c127e2aa4436d81c8;hp=28eccc66a5f3d4a27edf81d6e9c33d23e90718f5;hpb=595506e290356d26b8eebfab32ef8d3f625cbb0a;p=sim%2Fa1-interface.git diff --git a/near-rt-ric-simulator/src/STD_2.0.0/a1.py b/near-rt-ric-simulator/src/STD_2.0.0/a1.py old mode 100644 new mode 100755 index 28eccc6..9b68a6c --- a/near-rt-ric-simulator/src/STD_2.0.0/a1.py +++ b/near-rt-ric-simulator/src/STD_2.0.0/a1.py @@ -25,18 +25,19 @@ import time import requests from connexion import NoContent -from flask import Flask, escape, request, Response, make_response +from flask import Flask, request, Response from jsonschema import validate from var_declaration import policy_instances, policy_types, policy_status, callbacks, forced_settings, policy_fingerprint, hosts_set from utils import calcFingerprint from maincommon import check_apipath, apipath, get_supported_interfaces_response, extract_host_name, is_duplicate_check +from models.enforceStatus import EnforceStatus -#Constsants +# Constants APPL_JSON='application/json' APPL_PROB_JSON='application/problem+json' EXT_SRV_URL=os.getenv('EXT_SRV_URL') - +KAFKA_DISPATCHER_URL=os.getenv('KAFKA_DISPATCHER_URL') # API Function: Get all policy type ids def get_all_policy_types(): @@ -47,7 +48,7 @@ def get_all_policy_types(): return r res = list(policy_types.keys()) - return (res, 200) + return Response(json.dumps(res), 200, mimetype=APPL_JSON) # API Function: Get a policy type def get_policy_type(policyTypeId): @@ -79,7 +80,8 @@ def get_all_policy_identities(policyTypeId): pjson=create_problem_json(None, "The policy type does not exist.", 404, None, policy_type_id) return Response(json.dumps(pjson), 404, mimetype=APPL_PROB_JSON) - return (list(policy_instances[policy_type_id].keys()), 200) + res = list(policy_instances[policy_type_id].keys()) + return Response(json.dumps(res), 200, mimetype=APPL_JSON) # API Function: Create or update a policy def put_policy(policyTypeId, policyId): @@ -132,8 +134,15 @@ def put_policy(policyTypeId, policyId): pjson=create_problem_json(None, "Duplicate, the policy json already exists.", 400, None, policy_id) return Response(json.dumps(pjson), 400, mimetype=APPL_PROB_JSON) - #Callout hooks for external server - #When it fails, break and return 419 HTTP status code + #Callout hooks for kafka dispatcher + if (KAFKA_DISPATCHER_URL is not None): + resp = callout_kafka_dispatcher(policy_type_id, policy_id, data, retcode) + if (resp != 200): + pjson=create_error_response(resp) + return Response(json.dumps(pjson), 500, mimetype=APPL_PROB_JSON) + + # Callout hooks for external server + # When it fails, break and return HTTP status code 500 if (EXT_SRV_URL is not None): resp = callout_external_server(policy_id, data, 'PUT') if (resp != retcode): @@ -145,16 +154,14 @@ def put_policy(policyTypeId, policyId): policy_fingerprint[fp]=policy_id - noti=request.args.get('notificationDestination') - callbacks[policy_id]=noti + noti = request.args.get('notificationDestination') + callbacks[policy_id] = noti policy_instances[policy_type_id][policy_id]=data if (policy_types[policy_type_id]['statusSchema'] is not None): - ps = {} - ps["enforceStatus"] = "" - ps["enforceReason"] = "" - policy_status[policy_id] = ps + enforceStatus = EnforceStatus("NOT_ENFORCED", "OTHER_REASON") + policy_status[policy_id] = enforceStatus.to_dict() if (retcode == 200): return Response(json.dumps(data), 200, mimetype=APPL_JSON) @@ -203,8 +210,15 @@ def delete_policy(policyTypeId, policyId): pjson=create_problem_json(None, "The requested policy does not exist.", 404, None, policy_id) return Response(json.dumps(pjson), 404, mimetype=APPL_PROB_JSON) - #Callout hooks for external server - #When it fails, break and return 419 HTTP status code + #Callout hooks for kafka dispatcher + if (KAFKA_DISPATCHER_URL is not None): + resp = callout_kafka_dispatcher(policy_type_id, policy_id, None, 204) + if (resp != 200): + pjson=create_error_response(resp) + return Response(json.dumps(pjson), 500, mimetype=APPL_PROB_JSON) + + # Callout hooks for external server + # When it fails, break and return HTTP status code 500 if (EXT_SRV_URL is not None): resp = callout_external_server(policy_id, None, 'DELETE') if (resp != 204): @@ -241,8 +255,45 @@ def get_policy_status(policyTypeId, policyId): pjson=create_problem_json(None, "The requested policy does not exist.", 404, None, policy_id) return Response(json.dumps(pjson), 404, mimetype=APPL_PROB_JSON) + #Callout hooks for kafka dispatcher + if (KAFKA_DISPATCHER_URL is not None): + resp = callout_kafka_dispatcher(policy_type_id, policy_id, None, 202) + if (resp != 200): + pjson=create_error_response(resp) + return Response(json.dumps(pjson), 500, mimetype=APPL_PROB_JSON) + return Response(json.dumps(policy_status[policy_id]), status=200, mimetype=APPL_JSON) + +# Helper: Callout kafka dispatcher server to notify it for policy operations +def callout_kafka_dispatcher(policy_type_id, policy_id, payload, retcode): + + target_url = KAFKA_DISPATCHER_URL + "/policytypes/" + policy_type_id + "/kafkadispatcher/" + policy_id + try: + # create operation, publish with payload + if (retcode == 201): + resp=requests.put(target_url, json=payload, timeout=30, verify=False) + return resp.status_code + # update operation, publish with payload + elif (retcode == 200): + # add headers an update-flag + headers = {'updateoper' : 'yes'} + resp=requests.put(target_url, json=payload, headers=headers, timeout=30, verify=False) + return resp.status_code + # delete operation, publish without payload + elif (retcode == 204): + resp=requests.delete(target_url, timeout=30, verify=False) + return resp.status_code + # get policy status operation, publish without payload + elif (retcode == 202): + # update endpoint + target_url = target_url + "/status" + resp=requests.get(target_url, timeout=30, verify=False) + return resp.status_code + except Exception: + return 419 + + # Helper: Callout external server to notify it for policy operations # Returns 200, 201 and 204 for the success callout hooks, for the others returns 419 def callout_external_server(policy_id, payload, operation):