Code modifications regarding MISP integration

This commit is contained in:
Félix Aime
2021-06-09 18:24:37 +02:00
parent a481e88251
commit f189f2e100
5 changed files with 138 additions and 325 deletions

View File

@ -20,12 +20,9 @@ def add_instance():
"""
data = json.loads(request.data)
instance = data["data"]["instance"]
res = MISP.add_instance(instance["name"], instance["url"],
instance["key"], instance["ssl"])
res = misp.add_instance(instance)
return jsonify(res)
@misp_bp.route('/delete/<misp_id>', methods=['GET'])
@require_header_token
def delete_instance(misp_id):
@ -33,16 +30,16 @@ def delete_instance(misp_id):
Delete a MISP instance by its id to the database.
:return: status of the operation in JSON
"""
res = MISP.delete_instance(misp_id)
res = misp.delete_instance(misp_id)
return jsonify(res)
@misp_bp.route('/get_all', methods=['GET'])
# @require_header_token
@require_header_token
def get_all():
"""
Retreive a list of all MISP instances.
:return: list of MISP instances in JSON.
"""
res = MISP().get_instances()
res = misp.get_instances()
return jsonify({"results": [i for i in res]})

View File

@ -17,12 +17,17 @@ class MISP(object):
def __init__(self):
return None
def add_instance(self, name, url, apikey, verify):
def add_instance(self, instance):
"""
Parse and add a MISP instance to the database.
:return: status of the operation in JSON
"""
url = instance["url"]
name = instance["name"]
apikey = instance["key"]
verify = instance["ssl"]
sameinstances = db.session.query(MISPInst).filter(
MISPInst.url == url, MISPInst.apikey == apikey)
if sameinstances.count():