diff --git a/apps/tag/management/commands/add_tags.py b/apps/tag/management/commands/add_tags.py index 89ee89a1..496cb85b 100644 --- a/apps/tag/management/commands/add_tags.py +++ b/apps/tag/management/commands/add_tags.py @@ -10,7 +10,9 @@ class Command(BaseCommand): def handle(self, *args, **kwargs): - existing_establishment = Establishment.objects.filter(old_id__isnull=False) + existing_establishment = Establishment.objects.filter( + old_id__isnull=False + ) ESTABLISHMENT = 1 SHOP = 2 RESTAURANT = 3 @@ -52,7 +54,9 @@ class Command(BaseCommand): # create Tag for tag in key_value.metadata_set.filter( - establishment__id__in=list(existing_establishment.values_list('old_id', flat=True))): + establishment__id__in=list( + existing_establishment.values_list('old_id', flat=True) + )): new_tag, _ = Tag.objects.get_or_create( label={ @@ -63,7 +67,8 @@ class Command(BaseCommand): value=tag.value, category=tag_category, ) - est = existing_establishment.get(old_id=tag.establishment_id) - est.tags.add(new_tag) - est.save() - + est = existing_establishment.filter( + old_id=tag.establishment_id, tags__isnull=True).first() + if est: + est.tags.add(new_tag) + est.save()