flangr/flangr/posts/models.py

import uuid

from django.conf import settings
from django.db import models

# Create your models here.


def get_img_location(instance, filename):
    uid = uuid.uuid4().hex
    return f"{uid[0:2]}/{uid[2:]}"


class Post(models.Model):
    img = models.ImageField(upload_to=get_img_location)
    posted = models.DateTimeField(auto_now_add=True)

    title = models.CharField(max_length=255, blank=True, default="")
    body = models.TextField(blank=True, default="")

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

    circles = models.ManyToManyField("circles.Circle")

    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}"