Auto-create model with foreign key when model is created - Django

Question

I am making a comments section on my webpage and want users to be able to upvote or downvote a comment.

My models are as such:

class Comment(models.Model):
    owner = models.ForeignKey(User)
    body = models.TextField(null=True, blank=True, max_length=500)
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)


class Vote(models.Model):
    comment = models.ForeignKey(Comment)
    upvote = models.SmallIntegerField(null=True, blank=True, default=0)
    downvote = models.SmallIntegerField(null=True, blank=True, default=0)
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

When a user posts a comment, I want it to also create a Vote model that is linked to that comment.

I am new to django and programming but from my understanding, I need to create a save hook or something similar?

Answer 1

You can override the save() method of Comment model, ie:

class Comment(models.Model):
    ...
    def save(self, **kwargs):
        super(Comment, self).save(**kwargs)
        vote = Vote(comment=self)
        vote.save()

I suggest you to read the documentation for a better insight.