Merge pull request #11 from meddlin:feature/get-epss

Feature/get-epss

Author: meddlin

src/cve_explorer_cli/main.py

@@ -14,6 +14,13 @@
 from yaspin.spinners import Spinners
 from rich import print_json
 
+def __request_epss(cve_id: str) -> Any:
+    res = requests.get(
+        f'https://api.first.org/data/v1/epss?cve={cve_id}',
+        timeout=10
+    )
+    return res.json()
+
 def __request_cve(cve_id: str) -> Any:
     """Request CVE information from First (api.first.org)
 
@@ -24,7 +31,6 @@ def __request_cve(cve_id: str) -> Any:
         _type_: _description_
     """
     res = requests.get(
-        # f'https://api.first.org/data/v1/epss?cve={cve_id}',
         f'https://services.nvd.nist.gov/rest/json/cves/2.0?cveId={cve_id}',
         timeout=10
     )
@@ -54,6 +60,9 @@ def main():
     cve_lookup = subparsers.add_parser("lookup", help="Request a single CVE")
     cve_lookup.add_argument("--id", type=str)
 
+    epss_lookup = subparsers.add_parser("epss", help="Request EPSS info for CVE")
+    epss_lookup.add_argument("--id", type=str)
+
     args = parser.parse_args()
 
     if args.command == "lookup":
@@ -78,6 +87,9 @@ def main():
 
             print("\n")
             custom_cve.display_cve()
+    elif args.command == "epss":
+        data = __request_epss(args.id)
+        print_json(json.dumps(data))
     else:
         parser.print_help()