diff --git a/scripts/pg_insert.py b/scripts/pg_insert.py index 7e922868c06b3576f4610f82c98f16435d15413c..aed952e2c2be0c07453c2877b89d415cb133dcc9 100755 --- a/scripts/pg_insert.py +++ b/scripts/pg_insert.py @@ -129,7 +129,12 @@ if __name__ == "__main__": uuid_edges = len(getenv("UUID_EDGES", "")) > 0 with psycopg.connect() as connection: + total_batches = len(list(read_zip(file, batch_size))) + info(f"Total batches to process: {total_batches}") + batch_count = 0 for batch in read_zip(file, batch_size): + print(f"Processing batch number {batch_count + 1} of {total_batches}") + batch_count += 1 if interrupted: break batch_iter = (line.replace(HEADER_SCHEMA, b"") for line in batch)