diff --git a/apps/establishment/management/commands/add_establishment_description.py b/apps/establishment/management/commands/add_establishment_description.py index 380e2f1f..533a8bd7 100644 --- a/apps/establishment/management/commands/add_establishment_description.py +++ b/apps/establishment/management/commands/add_establishment_description.py @@ -24,17 +24,17 @@ class Command(BaseCommand): for r_id, establishment_id, new_date in queryset: try: - es_id, date = valid_reviews[r_id] + review_id, date = valid_reviews[establishment_id] except KeyError: - valid_reviews[r_id] = (establishment_id, new_date) + valid_reviews[establishment_id] = (r_id, new_date) else: if new_date > date: - valid_reviews[r_id] = (establishment_id, new_date) + valid_reviews[establishment_id] = (r_id, new_date) text_qs = ReviewTexts.objects.exclude( locale__isnull=True ).filter( - review_id__in=(r_id for r_id in valid_reviews.keys()), + review_id__in=(value[0] for value in valid_reviews.values()), ).values_list( 'review__establishment_id', 'locale', @@ -42,7 +42,7 @@ class Command(BaseCommand): ) for es_id, locale, text in text_qs: - establishment = Establishment.objects.filter(old_id=es_id, description__isnull=True).first() + establishment = Establishment.objects.filter(old_id=es_id).first() if establishment: description = establishment.description description.update({