Showing related pages with similar tags in Wagtail

In my blog, I list a number of related blog posts for each post. This is based on the tags I apply to the posts.

I'm using the taggit module to manage the tags. I'm also using wagtailmarkdown for having Markdown support in the post body.

My model looks like this:

blog/models.py

 1from django.db import models
 2from django.db.models.aggregates import Count
 3from django.db.models.fields import DateTimeField
 4from django.utils import timezone
 5
 6from modelcluster.fields import ParentalKey
 7from modelcluster.contrib.taggit import ClusterTaggableManager
 8from taggit.models import TaggedItemBase
 9
10from wagtail.admin.edit_handlers import FieldPanel, MultiFieldPanel
11from wagtail.core.models import Page, PageManager
12
13from wagtailmarkdown.edit_handlers import MarkdownPanel
14from wagtailmarkdown.fields import MarkdownField
15
16class BlogPostTag(TaggedItemBase):
17    content_object = ParentalKey('BlogPost', related_name='tagged_items', on_delete=models.CASCADE)
18
19class BlogPostManager(PageManager):
20
21    def related_posts(self, post, max_items=5):
22        tags = post.tags.all()
23
24        matches = BlogPost.objects.filter(tags__in=tags).live().annotate(Count('title'))
25        matches = matches.exclude(pk=post.pk)
26
27        related = matches.order_by('-title__count')
28        return related[:max_items]
29
30class BlogPost(Page):
31
32    objects = BlogPostManager()
33
34    body = MarkdownField(blank=True)
35    tags = ClusterTaggableManager(through=BlogPostTag, blank=True)
36
37    date = DateTimeField(blank=True, default=None, null=True, db_index=True)
38    date.verbose_name = 'Publish Date'
39
40    content_panels = [
41        MultiFieldPanel(
42            [
43                FieldPanel('title', classname="full title"),
44                FieldPanel('date'),
45                FieldPanel('tags'),
46            ],
47            heading="Post Details",
48            classname="collapsible"
49        ),
50        MarkdownPanel('body'),
51    ]
52    
53    def get_context(self, request, *args, **kwargs):
54        context = super(BlogPost, self).get_context(request)
55        context['related_posts'] = BlogPost.objects.related_posts(self)
56        return context

A whole lot of code, but lets look at it step by step.

At the top, you'll find the list of imports of all the items we need.

We first define BlogPostTag which is a single tag which can be added to a blog post. The blog posts themselves are defined in the class BlogPost which inherits from the base Page class. I do this to get base functionality of a Wagtail page in there. So far, that's all pretty standard.

We also define a couple of field such as body and date. To add tags to the blog post, we define a ClusterTaggableManager through the BlogPostTag class. This essentially creates a many-to-many relation between the blog posts and the tags. To make them editable, we also need to add them to the content_panels.

I also created a class BlogPostManager which inherits from Wagtail's PageManager class. I like this approach as it keeps the functions to access the blog posts nicely separated from the actual blog post definition. By using my custom manager as the objects variable in the BlogPost class, it makes it nice and clean.

In this example, BlogPostManager only defines one single extra method called related_posts. This is the method responsible for finding the related posts. It will first get the list of tags assigned to the given post. Based on that list, it will filter out all live blog posts which have tags in common. It also annotates the them by the number of tags they have in common. We also exclude the blog posts itself as we don't want it to show in the listing.

The list is ordered starting with the posts that have most tags in common.

I also added a limit so that we don't get all posts, but just a subset.

This is then used in the get_context function of the blog post so that they are available in the template.

In the template, we can show the related posts with a loop statement:

blog/templates/blog/blog_post.html

 1<h1>{{ page.title }}</h1>
 2
 3{{ page.body | markdown | safe }} 
 4
 5{% if related_posts %}
 6    <h1 class="related">Related Posts</h1
 7    <ul>
 8        {% for post in related_posts %}
 9            <li><a href="{% pageurl post %}">{{ post.title }}</a></li>
10        {% endfor %}
11    </ul>
12{% endif %}