feat: add confidence filtering

Author: Saharsh1123

cli.py

@@ -24,6 +24,13 @@
     help="Only show findings matching the selected severity",
 )
 
+# Limit displayed findings to a selected confidence level.
+parser.add_argument(
+    "--confidence",
+    choices=["LOW", "MEDIUM", "HIGH"],
+    help="Only show findings matching the selected confidence level",
+)
+
 # Mask detected secret values in text or JSON output.
 parser.add_argument(
     "--redact",
@@ -38,3 +45,4 @@
 use_json = args.json
 redact_secrets = args.redact
 chosen_severity = args.severity
+chosen_confidence = args.confidence

main.py

@@ -4,7 +4,7 @@
 Coordinates CLI input, file discovery, scanning, filtering, and output.
 """
 
-from cli import input_path, chosen_severity, use_json, redact_secrets
+from cli import input_path, chosen_severity, use_json, redact_secrets, chosen_confidence
 from scanner import check_path, scan, list_python_files
 from output import filter_results, output
 
@@ -21,7 +21,7 @@
         results = scan(files)
 
         # Apply optional severity filtering.
-        filtered_findings = filter_results(results, chosen_severity)
+        filtered_findings = filter_results(results, chosen_severity, chosen_confidence)
 
         # Render findings as text or JSON, with optional redaction.
         output(filtered_findings, use_json, redact_secrets, files)

output.py

@@ -17,7 +17,7 @@ def redact_value(value):
     return f"{value[:2]}{'*' * (len(value) - 4)}{value[-2:]}"
 
 
-def filter_results(results, chosen_severity):
+def filter_results(results, chosen_severity, chosen_confidence):
     """
     Filter findings by severity.
 
@@ -31,7 +31,7 @@ def filter_results(results, chosen_severity):
     filtered_findings = []
 
     for finding in results:
-        if finding.severity == chosen_severity or chosen_severity is None:
+        if (finding.severity == chosen_severity or chosen_severity is None) and (finding.confidence == chosen_confidence or chosen_confidence is None):
             filtered_findings.append(finding)
 
     return filtered_findings