diff --git a/django_diu/import_command.py b/django_diu/import_command.py index 75932fa65ac1df8ffed963518d34d4ddd54de55b..515b81a1f5f883f47e55872ef7a7b2d4e9a26392 100644 --- a/django_diu/import_command.py +++ b/django_diu/import_command.py @@ -74,7 +74,7 @@ class ImportTask(object): pass def _process_rows(self, rows): - for row in rows: + for idx, row in enumerate(rows): try: new_object = None with transaction.atomic(): @@ -93,6 +93,8 @@ class ImportTask(object): if self.progress_bar == False: self.out_stream.write(self.style.SUCCESS( 'Successfully inserted {}'.format(new_object))) + if self.progress_bar: + self.rows.update(idx) def count_source(self): self.source_count = len(self.rows) @@ -101,7 +103,7 @@ class ImportTask(object): self.open_data_source() self.count_source() if self.progress_bar is True: - with click.progressbar(self.rows, + with click.progressbar(length=self.source_count, label='Importing ' + self.description + ' (' + str(self.source_count) + ' rows to process).') as rows_list: self._process_rows(rows_list) else: