6 from flask import Flask, escape, request, make_response
7 from var_declaration import policy_instances, policy_types, policy_status
9 app = connexion.App(__name__, specification_dir='.')
11 @app.route('/policytypes/<string:policyTypeId>', methods=['PUT','DELETE'])
12 def policy_type(policyTypeId):
13 if request.method == 'PUT':
14 data = request.data.decode("utf-8")
15 data = data.replace("'", "\"")
16 data = json.loads(data)
17 policy_types[policyTypeId] = data
18 return ('The policy type was either created or updated for policy type id: ' + policyTypeId)
19 elif request.method == 'DELETE':
20 if policyTypeId in policy_types.keys():
21 policy_types.pop(policyTypeId)
22 return make_response("policy type successfully deleted for policy type id: " + policyTypeId, 200)
24 return make_response("No policy type defined for the specified id", 404)
26 @app.route('/', methods=['GET'])
28 if "STD_QoSNud_0.1.0" in policy_types.keys():
29 return("Something fishy!", 200)
31 return(str(list(policy_types.keys())), 200)
33 @app.route('/deleteinstances', methods=['DELETE'])
34 def delete_instances():
35 global policy_instances
37 policy_instances.clear()
39 return("All policy instances deleted", 200)
41 @app.route('/deletetypes', methods=['DELETE'])
45 return("All policy types deleted", 200)
47 @app.route('/<string:policyId>/<string:enforceStatus>', methods=['PUT'])
48 def set_status(policyId, enforceStatus):
49 if policyId in policy_instances.keys():
50 if enforceStatus in ["UNDEFINED", "ENFORCED", "NOT_ENFORCED"]:
51 policy_status.pop(policyId)
53 ps["policyId"] = policyId
54 ps["enforceStatus"] = enforceStatus
55 policy_status[policyId] = ps
56 return("Status updated for policy: " + policyId, 200)
58 return("enforceStatus should be one of \"UNDEFINED\", \"ENFORCED\" or \"NOT_ENFORCED\"", 400)
60 return("The policy id does not correspond to any existing policy instance", 400)
62 @app.route('/<string:policyId>/<string:enforceStatus>/<string:enforceReason>', methods=['PUT'])
63 def set_status_with_reason(policyId, enforceStatus, enforceReason):
64 if policyId in policy_instances.keys():
65 if enforceStatus == "NOT_ENFORCED":
66 if enforceReason in ["100", "200", "300", "800"]:
67 policy_status.pop(policyId)
69 ps["policyId"] = policyId
70 ps["enforceStatus"] = enforceStatus
71 ps["enforceReason"] = enforceReason
72 policy_status[policyId] = ps
73 return("Status updated for policy: " + policyId, 200)
75 return("enforceReason should be one of \"100\", \"200\", \"300\" or \"800\"", 400)
77 return("A status provided together with an enforcement reason should be \"NOT_ENFORCED\"", 400)
79 return("The policy id does not correspond to any existing policy instance", 404)
83 if len(sys.argv) >= 2:
84 if isinstance(sys.argv[1], int):
85 port_number = sys.argv[1]
87 app.add_api('a1-openapi.yaml')
88 app.run(port=port_number)