Решение, которое я предлагаю, делает немного больше, чем вы запрашиваете, потому что оно позволяет находить связанные теги для набора заданных тегов, а не только для одного данного тега.В действительности это, вероятно, то, что вы хотите сделать, хотя.Я не уверен, действительно ли это оптимально с точки зрения производительности, поскольку он использует подзапрос, но он работает, и я нахожу его легким для понимания.
Во-первых, вот контрольный пример:
from django.test import TestCase
from .models import Item, get_related_tags
class RelatedTagsTest(TestCase):
def setUp(self):
article1 = Item.objects.create(title='Python vs. COBOL')
article1.tags.add('programming', 'python', 'cobol')
article2 = Item.objects.create(title='Python vs. Boa Constrictor')
article2.tags.add('zoology', 'python', 'boa')
article3 = Item.objects.create(title='COBOL vs. FORTRAN')
article3.tags.add('cobol', 'fortran', 'programming')
def test_unique_tag(self):
self.assertEquals(get_related_tags('programming'),
['cobol', 'fortran', 'python'])
self.assertEquals(get_related_tags('python'),
['boa', 'cobol', 'programming', 'zoology'])
def test_multiple_tags(self):
self.assertEquals(get_related_tags('boa', 'fortran'),
['cobol', 'programming', 'python', 'zoology'])
Как вы можете видеть, под "связанными тегами" мы понимаем набор тегов, которые связаны с элементами, которые помечены набором заданных тегов.
А вот наша модель с функцией для получениясвязанные теги:
from django.db import models
from taggit.managers import TaggableManager
from taggit.models import Tag
class Item(models.Model):
title = models.CharField(max_length=100)
tags = TaggableManager()
def get_related_tags(*tags):
# Get a QuerySet of related items
related_items = Item.objects.filter(tags__name__in=tags)
# Get tags for those related items (I found the name of the lookup field by
# reading taggit's source code)
qs = Tag.objects.filter(taggit_taggeditem_items__item__in=related_items)
# Exclude the tags we already have
qs = qs.exclude(name__in=tags)
# Order by name and remove duplicates
qs = qs.order_by('name').distinct()
# Return tag names to simplify test code, real code would probably return
# Tag objects
return [t.name for t in qs]
Обратите внимание, что вы можете легко добавить количество элементов в тег, используя qs.annotate(count=Count('name'))
.Он будет доступен в виде атрибута count
для каждого объекта Tag
.