From f1325f431629f222451bc1ac521f9339853b263d Mon Sep 17 00:00:00 2001 From: isox Date: Tue, 18 Sep 2018 18:36:12 +0300 Subject: [PATCH] Fixed getting of the Vulners AI score. --- misp_modules/modules/expansion/vulners.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/misp_modules/modules/expansion/vulners.py b/misp_modules/modules/expansion/vulners.py index c7d48de..49cb9aa 100644 --- a/misp_modules/modules/expansion/vulners.py +++ b/misp_modules/modules/expansion/vulners.py @@ -26,8 +26,15 @@ def handler(q=False): vulners_api = vulners.Vulners(api_key=key) vulnerability = request.get('vulnerability') vulners_document = vulners_api.document(vulnerability) + + # Get AI scoring from the document if it's already calculated + # There is no need to call AI Scoring method + if 'score' in vulners_document.get('enchantments', {}): + vulners_ai_score = vulners_document['enchantments']['score']['value'] + else: + vulners_ai_score = None + vulners_exploits = vulners_api.searchExploit(vulnerability) - vulners_ai_score = vulners_api.aiScore(vulnerability) if vulners_document: vuln_summary += vulners_document.get('description')