How to add comments to model instances as generic relations

As a newbie of Django’s Generic Relation and Generic Foreign Key,I found that the documentation can be kind of confusing and hard to grab. Nevertheless, Generic Relations helped us a lot, and so I decided to write about it in this blog post :)

When we have a foreign key, we are linking an instance of another model in the current model. So, we can access those very easily. Below code would work like this:

class Author(models.Model):
    name = models.CharField(max_length=50)

class Book(models.Model):
    author = models.ForeignKey(Author)
    title = models.CharField(max_length=250)
    pages = models.IntegerField()

Here you will have an instance of Author associated with all the information we have on the Book instance. That's cool because you can store several books, all linked to the same author.

Imagine now that you want the same thing: associate an instance of a model with more information. But instead of having just an instance of Author (for example), you want the same information through several model classes (such as books and cds).

GenericRelations is the saviour here. In a simple way we can say that GenericRelation is a foreign key that can store any instance of any of your models at any of your apps.

Question can be arise here: Why is this useful? You can simply add more fields in your original model to accomplish your goal. Yes, this is true in most cases. However, for some specific cases, Generic Relations can be really handy. In our case, we needed the ability to add comments in two apps of our system. We needed the same functionality in different part of the system in a way it would be easily maintainable and ignore duplication: perfect time for Generic Relations.

We started by creating an Comment model such as (Django 2.2):

from django.contrib.contenttypes.fields import GenericForeignKey
from django.contrib.contenttypes.models import ContentType
from django.db import models

class Comment(models.Model):
    content_type = models.ForeignKey(ContentType)
    object_id = models.Charfield(max_length=50)
    content_object = GenericForeignKey('content_type', 'object_id')
    text = models.TextField(blank=True)

On this model the text field is a normal TextField from Django, used to store the comment itself. The other fields, content_type, object_id and content_objects are part of the Generic Relation we are adding here.

Instances of ContentType represent and store information about the models installed on your project. Every time a new model is created, new instances of ContentTypes are automatically created. Here, the content_type will be a Foreign Key to the model you want to associate.

The object_id, by the other end, is a simple Charfield that will store an id of an object that is stored in your model. On the official Django documentation, you will find that the suggestion is to use PositiveIntegerField on this field. However, we use uuid as our id fields so we had to change this to Charfield.

You have the model, you have the id of the object you want to access so the content_objects will actually represent the instance of that particular object on that particular model. The GenericForeignKey does the magic for you!

Let’s apply this model for something useful. Imagine that you have models for Books and CDs, and you want to to be able to add comments from your users in each book or cd available on your database.

To create a new comment in a specific book all you need to do is:

from django.contrib.contenttypes.models import ContentType
from .models import Book, Comment

book = Book.objects.first()
text = 'The message goes here'

new_comment = Comment(text=text,
                       content_object=book)
new_comment.save()

So you can first recover the instance you want to associate your comment with and send it as the content_object.

You can do this with the Book model, the Cd model or any other model in any other app you have on your system. You won’t need to rewrite this comment to every app or every model you want to add a series of comments.

Now you can ask: How can I recover the comments information in my Book or Cd instance? Here comes the easy part!

from django.contrib.contenttypes.fields import GenericRelation
from .models import Comment

class Book(models.Model):
    author = models.ForeignKey(Author)
    title = models.CharField(max_length=250)
    pages = models.IntegerField()
    comments = GenericRelation(Comment)

class Cd(models.Model):
    artist = models.ForeignKey(Artists)
    title = models.CharField(max_length=250)
    comments = GenericRelation(Comment)

Done! You can add comments to any Book or Cd instance, are retrieve it by simply doing:

book.comments.all()

I hope this would help you out. :)