Fixed getting of the Vulners AI score.

pull/231/head
isox 2018-09-18 18:36:12 +03:00
parent 84124f819d
commit f1325f4316
1 changed files with 8 additions and 1 deletions

View File

@ -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')