Skip to content

Commit 770e310

Browse files
committed
Fix accumulation of results from generators
1 parent 0405604 commit 770e310

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

list_secret_scanning_alerts.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
"""List secret scanning alerts for a GitHub repository, organization or Enterprise."""
44

5+
import itertools
56
import sys
67
import argparse
78
import logging
@@ -260,14 +261,14 @@ def list_secret_scanning_alerts(
260261
)
261262
alerts.append(specific_alerts)
262263

263-
alerts = decorate_alerts(g, alerts, include_locations=include_locations, include_commit=include_commit)
264+
decorated_alerts = decorate_alerts(g, itertools.chain.from_iterable(alerts), include_locations=include_locations, include_commit=include_commit)
264265

265266
if raw:
266-
for alert in alerts:
267+
for alert in decorated_alerts:
267268
yield alert
268269
return None
269270
else:
270-
for alert in alerts:
271+
for alert in decorated_alerts:
271272
result = make_result(g, alert, scope, name, include_secret=include_secret, include_locations=include_locations, include_commit=include_commit)
272273
if result is not None:
273274
yield result

0 commit comments

Comments
 (0)