flangr/flangr/posts/models.py

from django.db import models

from django.conf import settings

# Create your models here.


class Post(models.Model):
    img = models.ImageField(upload_to="posts/%Y/%m")
    posted = models.DateTimeField(auto_now_add=True)
    public = models.BooleanField(default=False)

    title = models.CharField(max_length=255, null=True)
    body = models.TextField(null=True)

    user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)

    def __str__(self):
        return f"Post: {self.title} at {self.posted.strftime('%Y-%m-%d %H:%S')}"


class Collection(models.Model):
    title = models.CharField(max_length=255)
    description = models.TextField()

    posts = models.ManyToManyField("Post", related_name="collections")
    user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)

    def __str__(self):
        return f"Collection: {self.title}"


class Comment(models.Model):
    post = models.ForeignKey("Post", on_delete=models.CASCADE, related_name="comments")
    comment = models.CharField(max_length=255)

    user = models.ForeignKey(
        settings.AUTH_USER_MODEL, on_delete=models.SET_NULL, null=True
    )

    def __str__(self):
        return f"Comment by {self.user} on {self.post}"