diff --git a/ldpy.py b/ldpy.py index 4915886..f2bdb12 100755 --- a/ldpy.py +++ b/ldpy.py @@ -115,16 +115,15 @@ def parse_args(args=sys.argv[1:]): def get_last_entries(entries_nb): """Get the last n entries from a stream.""" - if entries_nb in LAST_E_CHOICES: - logger.debug("Wait before getting '%s' entries!", entries_nb) - query = {"size": entries_nb} - last_entries = client.opnsrch_clt.search(body=query) - logger.debug(pf(last_entries)) - - else: + if entries_nb not in LAST_E_CHOICES: logger.critical("'%s' is not in '%s'", entries_nb, LAST_E_CHOICES) raise ValueError + logger.debug("Wait before getting '%s' entries!", entries_nb) + query = {"size": entries_nb} + last_entries = client.opnsrch_clt.search(body=query) + logger.debug(pf(last_entries)) + return last_entries