From f87d891daa33685870669889ec96f64dc4398456 Mon Sep 17 00:00:00 2001 From: Anatoly Date: Fri, 20 Dec 2019 09:26:45 +0300 Subject: [PATCH] fix transfer --- apps/collection/transfer_data.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/apps/collection/transfer_data.py b/apps/collection/transfer_data.py index efbe52ab..dc9c2bbb 100644 --- a/apps/collection/transfer_data.py +++ b/apps/collection/transfer_data.py @@ -254,15 +254,17 @@ def transfer_guide_elements_bulk(): for guide in tqdm(Guide.objects.all(), desc='update count_of_initial_objects field values'): - descendants = GuideElement.objects.get_root_node(guide).get_descendants() - if descendants: - count_of_initial_objects = descendants.filter( - guide_element_type__name__in=['EstablishmentNode', 'WineNode']).count() - guide.count_related_objects = count_of_initial_objects - objects_to_update.append(guide) + root_node = GuideElement.objects.get_root_node(guide) + if root_node: + descendants = root_node.get_descendants() + if descendants: + count_related_objects = descendants.filter( + guide_element_type__name__in=['EstablishmentNode', 'WineNode']).count() + guide.count_related_objects = count_related_objects + objects_to_update.append(guide) # update count_of_initial_objects field values - Guide.objects.bulk_update(objects_to_update, ['count_objects_during_init', ]) + Guide.objects.bulk_update(objects_to_update, ['count_related_objects', ]) print(f'COUNT OF UPDATED OBJECTS: {len(objects_to_update)}')