restore_find_data: print progression

1 files changed, 9 insertions, 3 deletions

diff --git a/scripts/restore_find_data.py b/scripts/restore_find_data.py
index 090647162..df5c38069 100644
--- a/scripts/restore_find_data.py
+++ b/scripts/restore_find_data.py
@@ -2,12 +2,18 @@
 Restore find data on destructive import
 """
 
-from ishtar_common.utils import update_data
+import sys
 
+from ishtar_common.utils import update_data
 from archaeological_finds.models import Find
 
+ln = Find.objects.count()
 
-for item in Find.objects.all():
+sys.stdout.write("\n")
+for idx, item in enumerate(Find.objects.all()):
+    sys.stdout.write("\r* processing {}/{} ({}%)".format(
+        idx + 1, ln, (idx + 1)*100.0 / ln))
+    sys.stdout.flush()
     data = item.data
     for h in item.history.order_by('-history_modifier_id', '-history_date',
                                    '-history_id').all():
@@ -15,5 +21,5 @@ for item in Find.objects.all():
     item.data = data
     item.skip_history_when_saving = True
     item.save()
-
+sys.stdout.write("\n")