from flask import Flask, escape, request, Response, jsonify
from jsonschema import validate
from var_declaration import policy_instances, policy_types, policy_status, policy_fingerprint, forced_settings, hosts_set
from flask import Flask, escape, request, Response, jsonify
from jsonschema import validate
from var_declaration import policy_instances, policy_types, policy_status, policy_fingerprint, forced_settings, hosts_set
forced_settings['code']=None
forced_settings['delay']=None
policy_fingerprint.clear()
forced_settings['code']=None
forced_settings['delay']=None
policy_fingerprint.clear()
forced_settings['code']=None
forced_settings['delay']=None
policy_fingerprint.clear()
forced_settings['code']=None
forced_settings['delay']=None
policy_fingerprint.clear()
- policyTypeId=request.args.get('id')
- if (policyTypeId is None):
- return Response('Parameter <id> missing in request', status=400, mimetype='text/plain')
+ policy_type_id=request.args.get('id')
+ if (policy_type_id is None):
+ return Response('Parameter <id> missing in request', status=400, mimetype=TEXT_PLAIN)
- if (len(policy_instances[policyTypeId]) > 0):
- return Response("The policy type already exists and instances exists", 400, mimetype='text/plain')
+ if (len(policy_instances[policy_type_id]) > 0):
+ return Response("The policy type already exists and instances exists", 400, mimetype=TEXT_PLAIN)
- policy_types[policyTypeId]=data
- policy_instances[policyTypeId]={}
- return Response("Policy type " + policyTypeId + " is OK.", retcode, mimetype='text/plain')
+ policy_types[policy_type_id]=data
+ policy_instances[policy_type_id]={}
+ return Response("Policy type " + policy_type_id + " is OK.", retcode, mimetype=TEXT_PLAIN)
- policyTypeId=request.args.get('id')
- if (policyTypeId is None):
- return Response('Parameter <id> missing in request', status=400, mimetype='text/plain')
+ policy_type_id=request.args.get('id')
+ if (policy_type_id is None):
+ return Response('Parameter <id> missing in request', status=400, mimetype=TEXT_PLAIN)
- if (policyTypeId in policy_types.keys()):
- if (len(policy_instances[policyTypeId]) > 0):
- return Response("The policy type already exists and instances exists", 400, mimetype='text/plain')
+ if (policy_type_id in policy_types.keys()):
+ if (len(policy_instances[policy_type_id]) > 0):
+ return Response("The policy type already exists and instances exists", 400, mimetype=TEXT_PLAIN)
- del policy_types[policyTypeId]
- del policy_instances[policyTypeId]
- return Response("Policy type " + policyTypeId + " is OK.", 204, mimetype='text/plain')
+ del policy_types[policy_type_id]
+ del policy_instances[policy_type_id]
+ return Response("Policy type " + policy_type_id + " is OK.", 204, mimetype=TEXT_PLAIN)
- return Response("Force response code: " + str(forced_settings['code']) + " set for one single A1 response", 200, mimetype='text/plain')
+ return Response("Force response code: " + str(forced_settings['code']) + " set for one single A1 response", 200, mimetype=TEXT_PLAIN)
#Set force delay response, in seconds, for all A1 responses
#/froceesponse?delay=<seconds>
#Set force delay response, in seconds, for all A1 responses
#/froceesponse?delay=<seconds>
- return Response("Force delay: " + str(forced_settings['delay']) + " sec set for all A1 responses", 200, mimetype='text/plain')
+ return Response("Force delay: " + str(forced_settings['delay']) + " sec set for all A1 responses", 200, mimetype=TEXT_PLAIN)
- policyId=request.args.get('policyid')
- if (policyId is None):
- return Response('Parameter <policyid> missing in request', status=400, mimetype='text/plain')
+ policy_id=request.args.get('policyid')
+ if (policy_id is None):
+ return Response('Parameter <policyid> missing in request', status=400, mimetype=TEXT_PLAIN)
- if policyId not in policy_status.keys():
- return Response('Policyid: '+policyId+' not found.', status=404, mimetype='text/plain')
+ if policy_id not in policy_status.keys():
+ return Response('Policyid: '+policy_id+' not found.', status=404, mimetype=TEXT_PLAIN)
- return Response('Parameter <status> missing in request', status=400, mimetype='text/plain')
- policy_status[policyId]["instance_status"]=status
+ return Response('Parameter <status> missing in request', status=400, mimetype=TEXT_PLAIN)
+ policy_status[policy_id]["instance_status"]=status
msg = "Status set to "+status
deleted_policy=request.args.get('deleted')
if (deleted_policy is not None):
msg = "Status set to "+status
deleted_policy=request.args.get('deleted')
if (deleted_policy is not None):
msg = msg + " and has_been_deleted set to "+deleted_policy
created_at = request.args.get('created_at')
if (created_at is not None):
msg = msg + " and has_been_deleted set to "+deleted_policy
created_at = request.args.get('created_at')
if (created_at is not None):
- msg=msg + " for policy: " + policyId
- return Response(msg, 200, mimetype='text/plain')
+ msg=msg + " for policy: " + policy_id
+ return Response(msg, 200, mimetype=TEXT_PLAIN)
elif (countername == "remote_hosts"):
hosts=",".join(hosts_set)
return str(hosts),200
else:
elif (countername == "remote_hosts"):
hosts=",".join(hosts_set)
return str(hosts),200
else:
if isinstance(sys.argv[1], int):
port_number = sys.argv[1]
app.add_api('openapi.yaml')
if isinstance(sys.argv[1], int):
port_number = sys.argv[1]
app.add_api('openapi.yaml')
-app.run(port=port_number, host="127.0.0.1", threaded=False)
\ No newline at end of file
+if __name__ == '__main__':
+ app.run(port=port_number, host="127.0.0.1", threaded=False)
\ No newline at end of file