From c8917f51aad82e6ef29e44fa44000e7bf235cab2 Mon Sep 17 00:00:00 2001 From: Henrik Levkowetz Date: Thu, 25 Aug 2016 05:48:04 +0000 Subject: [PATCH] Changed the migration progress indication to a tqdm progress bar. - Legacy-Id: 11862 --- ietf/doc/migrations/0013_auto_20151027_1127.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/ietf/doc/migrations/0013_auto_20151027_1127.py b/ietf/doc/migrations/0013_auto_20151027_1127.py index 01dce5678..279793f85 100644 --- a/ietf/doc/migrations/0013_auto_20151027_1127.py +++ b/ietf/doc/migrations/0013_auto_20151027_1127.py @@ -2,6 +2,7 @@ from __future__ import unicode_literals import sys +from tqdm import tqdm from django.db import migrations @@ -60,17 +61,14 @@ def save_all_documents_in_history(apps, schema_editor): if objs.exists(): try: dochist = objs.get(**fields) - sys.stderr.write('.') except DocHistory.MultipleObjectsReturned: dochist_list = list(objs) for dochist in dochist_list[1:]: dochist.delete() dochist = dochist_list[0] - sys.stderr.write('-') else: dochist = DocHistory(**fields) dochist.save() - sys.stderr.write('+') # copy many to many for field in doc._meta.many_to_many: @@ -97,9 +95,13 @@ def save_all_documents_in_history(apps, schema_editor): from django.conf import settings settings.DEBUG = False # prevent out-of-memory problems - for d in Document.objects.iterator(): + docs = Document.objects.all() + for d in tqdm(docs): save_document_in_history(d) +def noop_backward(apps, schema_editor): + pass + class Migration(migrations.Migration): dependencies = [ @@ -108,6 +110,6 @@ class Migration(migrations.Migration): ] operations = [ - migrations.RunPython(fix_buggy_author_foreignkey), - migrations.RunPython(save_all_documents_in_history) + migrations.RunPython(fix_buggy_author_foreignkey, noop_backward), + migrations.RunPython(save_all_documents_in_history, noop_backward) ]