Skip to content

Commit fb1fe02

Browse files
committed
fix mysql progress ref
1 parent ab97816 commit fb1fe02

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pynonymizer/database/mysql/__init__.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def __seed(self, qualifier_map):
6060
"""
6161
'Seed' the database with a bunch of pre-generated random records so updates can be performed in batch updates
6262
"""
63-
for i in progress(
63+
for i in self.progress(
6464
range(0, self.seed_rows), desc="Inserting seed data", unit="rows"
6565
):
6666
self.logger.debug(f"Inserting seed row {i}")
@@ -123,7 +123,7 @@ def anonymize_database(self, database_strategy):
123123

124124
anonymization_errors = []
125125

126-
with progress(
126+
with self.progress(
127127
desc="Anonymizing database", total=len(table_strategies)
128128
) as progressbar:
129129
for table_strategy in table_strategies:
@@ -197,7 +197,7 @@ def restore_database(self, input_path):
197197
try:
198198
batch_processor = self.__runner.open_batch_processor()
199199
with input_obj.open() as dumpfile_data:
200-
with progress(
200+
with self.progress(
201201
desc="Restoring",
202202
total=dumpsize,
203203
unit="B",
@@ -222,7 +222,7 @@ def dump_database(self, output_path):
222222

223223
dump_process = self.__dumper.open_dumper()
224224
with output_obj.open() as output_file:
225-
with progress(
225+
with self.progress(
226226
desc="Dumping",
227227
total=dumpsize_estimate,
228228
unit="B",

0 commit comments

Comments
 (0)