chg: [lookup] add --unique option to return the best matching CPE

cve-search · Nov 23, 2024 · 22fd8e1 · 22fd8e1
1 parent 30d9321
commit 22fd8e1
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/bin/lookup.py b/bin/lookup.py
@@ -21,7 +21,21 @@
         nargs="+",
         help="One or more keyword(s) to lookup",
     )
+    parser.add_argument(
+        "--unique",
+        action="store_true",
+        help="Return the best CPE matching the keywords given",
+        default=False,
+    )
     args = parser.parse_args()
 
     cpeGuesser = CPEGuesser()
-    print(json.dumps(cpeGuesser.guessCpe(args.word)))
+    r = cpeGuesser.guessCpe(args.word)
+    if not args.unique:
+        print(json.dumps(r))
+    else:
+        try:
+            r = r[:1][0][1]
+        except:
+            r = []
+        print(json.dumps(r))