42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from django.db import models
|
|
|
|
from django.conf import settings
|
|
import magic
|
|
|
|
mime = magic.Magic(mime=True)
|
|
|
|
|
|
class Post(models.Model):
|
|
content = models.TextField(blank=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
author = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL,
|
|
on_delete=models.CASCADE,
|
|
related_name='posts'
|
|
)
|
|
|
|
hub = models.ForeignKey(
|
|
'pages.Hub',
|
|
on_delete=models.CASCADE,
|
|
null=True,
|
|
blank=True
|
|
)
|
|
|
|
def __str__(self):
|
|
return f"Post {self.id}"
|
|
|
|
class PostContent(models.Model):
|
|
post = models.ForeignKey(Post, related_name='contents', on_delete=models.CASCADE)
|
|
mime_type = models.CharField(max_length=100)
|
|
file = models.FileField(upload_to='post_contents/', null=True, blank=True) # For images/videos
|
|
alt_text = models.TextField(blank=True, null=True) # For text content
|
|
|
|
def __str__(self):
|
|
return f"Content for Post {self.post.id} - Type: {self.mime_type}"
|
|
|
|
def save(self, *args, **kwargs):
|
|
if self.file and not self.file._committed:
|
|
self.mime_type = mime.from_buffer(self.file.read(1024))
|
|
return super().save(*args, **kwargs) |