From 490064cdf842f132140bbb0631e633cb4daf9bf9 Mon Sep 17 00:00:00 2001 From: Zach Flanders Date: Sat, 29 Apr 2023 09:43:55 -0500 Subject: [PATCH] updating sort title migration --- bookwyrm/migrations/0179_populate_sort_title.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/bookwyrm/migrations/0179_populate_sort_title.py b/bookwyrm/migrations/0179_populate_sort_title.py index 02691f615..c5bdcb25c 100644 --- a/bookwyrm/migrations/0179_populate_sort_title.py +++ b/bookwyrm/migrations/0179_populate_sort_title.py @@ -24,13 +24,15 @@ def populate_sort_title(apps, schema_editor): editions_wo_sort_title = Edition.objects.using(db_alias).filter( Q(sort_title__isnull=True) | Q(sort_title__exact="") ) - batch_size = 50000 + batch_size = 20 start = 0 end = batch_size - while editions_wo_sort_title[start:end]: + while True: + batch = editions_wo_sort_title[start:end] + if not batch.exists(): + break Edition.objects.bulk_update( - (set_sort_title(edition) for edition in editions_wo_sort_title[start:end]), - ["sort_title"], + (set_sort_title(edition) for edition in batch), ["sort_title"] ) start = end end += batch_size