bitbot-3.11-fork/modules/cve.py
2019-09-18 17:51:08 +01:00

37 lines
1.1 KiB
Python

#--depends-on commands
import datetime, json
from src import ModuleManager, utils
URL_CVE = "https://cve.circl.lu/api/cve/%s"
class Module(ModuleManager.BaseModule):
_name = "CVE"
@utils.hook("received.command.cve", min_args=1)
def cve(self, event):
"""
:help: Get information for a CVE number
:usage: <CVE>
"""
cve_id = event["args_split"][0].upper()
if not cve_id.startswith("CVE-"):
cve_id = "CVE-%s" % cve_id
page = utils.http.request(URL_CVE % cve_id, json=True)
if page and page.data:
cve_id = page.data["id"]
published = "%sZ" % page.data["Published"].rsplit(".", 1)[0]
published = datetime.datetime.strptime(published,
utils.ISO8601_PARSE)
published = datetime.datetime.strftime(published, "%Y-%m-%d")
rank = page.data["cvss"]
summary = page.data["summary"]
event["stdout"].write("%s, %s (%s): %s" %
(cve_id, published, rank, summary))
else:
raise utils.EventsResultsError()