diff --git a/swh/search/utils.py b/swh/search/utils.py new file mode 100644 index 0000000..811c9e7 --- /dev/null +++ b/swh/search/utils.py @@ -0,0 +1,10 @@ +def stream_results(f, *args, **kwargs): + if 'cursor' in kwargs: + raise ArgumentError('stream_results has no argument "cursor".') + cursor = None + while True: + results = f(*args, cursor=cursor, **kwargs) + yield from results['results'] + cursor = results['cursor'] + if cursor is None: + break