diff --git a/enrich_code_scanning_alerts.py b/enrich_code_scanning_alerts.py index 38f7735..eae7cd9 100755 --- a/enrich_code_scanning_alerts.py +++ b/enrich_code_scanning_alerts.py @@ -16,7 +16,6 @@ from typing import Any from pathlib import Path from datetime import datetime, timezone -import time from mistletoe import markdown import humanize from defusedcsv import csv @@ -60,7 +59,7 @@ def fixup_rule_metadata_codeql(rule_metadata: dict) -> None: rule_metadata["security-severity"] = float( rule_metadata["security-severity"] ) - except: + except (FloatingPointError, ValueError): LOG.warning( "Could not parse security-severity in rule %s as float: %s", rule_metadata.get("id", "unknown ID"), diff --git a/list_secret_scanning_alerts.py b/list_secret_scanning_alerts.py index ec1c815..e428cca 100755 --- a/list_secret_scanning_alerts.py +++ b/list_secret_scanning_alerts.py @@ -242,14 +242,16 @@ def list_secret_scanning_alerts( alerts = decorate_alerts(g, alerts, include_locations=include_locations, include_commit=include_commit) if raw: - return alerts + for alert in alerts: + yield alert + return None else: for alert in alerts: result = make_result(g, alert, scope, name, include_secret=include_secret, include_locations=include_locations, include_commit=include_commit) if result is not None: yield result else: - return + return None def add_args(parser: argparse.ArgumentParser) -> None: