49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
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}"
|