ft:comment model -- add comment fields

BrianMwevi · Jun 16, 2022 · 503cdb0 · 503cdb0
1 parent eec4910
commit 503cdb0
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/posts/models.py b/posts/models.py
@@ -2,6 +2,8 @@
 from django.contrib.auth.models import User
 from neighborhoods.models import Neighborhood
 
+# Create your models here.
+
 
 class Post(models.Model):
     user = models.ForeignKey(
@@ -16,3 +18,19 @@ class Post(models.Model):
 
     def __str__(self):
         return self.post
+
+
+class Comment(models.Model):
+    user = models.ForeignKey(
+        User, on_delete=models.CASCADE)
+    post = models.ForeignKey(
+        Post, on_delete=models.SET_NULL, null=True)
+    comment = models.TextField()
+    likes = models.ManyToManyField(
+        User, blank=True, related_name='comment_likes')
+    posted_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+
+    def __str__(self):
+        return self.comment
+