Merge branch 'vulnersCom-master'

pull/232/head
Alexandre Dulaunoy 2018-09-19 09:04:40 +02:00
commit b67aca09e5
No known key found for this signature in database
GPG Key ID: 09E2CD4944E6CBCD
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(vulners_document.get('description'))
if vulners_document:
vuln_summary += vulners_document.get('description')