diff --git a/src/scribe_data/cli/total.py b/src/scribe_data/cli/total.py index d7538e9b..dc4ba80c 100644 --- a/src/scribe_data/cli/total.py +++ b/src/scribe_data/cli/total.py @@ -20,9 +20,10 @@ --> """ -from SPARQLWrapper import JSON -from urllib.error import HTTPError from http.client import IncompleteRead +from urllib.error import HTTPError + +from SPARQLWrapper import JSON from scribe_data.utils import ( LANGUAGE_DATA_EXTRACTION_DIR, @@ -253,16 +254,21 @@ def get_total_lexemes(language, data_type, doPrint=True): while try_count <= max_retries and results is None: try: results = sparql.query().convert() + except HTTPError as http_err: print(f"HTTPError occurred: {http_err}") + except IncompleteRead as read_err: print(f"Incomplete read error occurred: {read_err}") + try_count += 1 - if results is None and try_count <= max_retries: - print("The query will be retried..") - elif results is None: - print("Query failed after retries.") + if results is None: + if try_count <= max_retries: + print("The query will be retried ...") + + else: + print("Query failed after retries.") # Check if the query returned any results. if (