66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
from django.db import models
|
|
from tp.models import UUIDAsIDModel
|
|
from django.contrib.auth.models import User
|
|
from django.utils.text import slugify
|
|
from tp.storage_backends import PublicImageStorage, PublicMP3Storage
|
|
import string, random
|
|
|
|
|
|
def rand_slug():
|
|
return ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(8))
|
|
|
|
|
|
def slugify_file_name(instance, filename):
|
|
fname, dot, extension = filename.rpartition('.')
|
|
slug = slugify(fname)
|
|
return f'{slug}.{extension}'
|
|
|
|
|
|
class Feed(UUIDAsIDModel):
|
|
user = models.ForeignKey(User, on_delete=models.CASCADE)
|
|
created_on = models.DateTimeField(auto_now_add=True)
|
|
title = models.CharField(max_length=255)
|
|
author = models.CharField(max_length=255)
|
|
slug = models.SlugField(max_length=255, unique=True)
|
|
description = models.TextField(null=False)
|
|
image = models.ImageField(
|
|
storage=PublicImageStorage(),
|
|
upload_to=slugify_file_name,
|
|
null=True, blank=True)
|
|
|
|
def save(self, *args, **kwargs):
|
|
if not self.slug:
|
|
self.slug = slugify(rand_slug() + "-" + self.title)
|
|
super(Feed, self).save(*args, **kwargs)
|
|
|
|
def __str__(self):
|
|
return str(self.title)
|
|
|
|
|
|
class Episode(UUIDAsIDModel):
|
|
user = models.ForeignKey(User, on_delete=models.CASCADE)
|
|
feed = models.ForeignKey(Feed, on_delete=models.CASCADE)
|
|
title = models.CharField(max_length=255)
|
|
author = models.CharField(max_length=255)
|
|
slug = models.SlugField(max_length=255, unique=True)
|
|
description = models.TextField(null=False)
|
|
created_on = models.DateTimeField(auto_now_add=True)
|
|
pub_date = models.DateField()
|
|
episode_number = models.IntegerField(null=True)
|
|
image = models.ImageField(
|
|
storage=PublicImageStorage(),
|
|
upload_to=slugify_file_name,
|
|
null=True, blank=True)
|
|
mp3 = models.FileField(
|
|
storage=PublicMP3Storage(),
|
|
upload_to=slugify_file_name,
|
|
null=True, blank=True)
|
|
|
|
def save(self, *args, **kwargs):
|
|
if not self.slug:
|
|
self.slug = slugify(rand_slug() + "-" + self.title)
|
|
super(Episode, self).save(*args, **kwargs)
|
|
|
|
def __str__(self):
|
|
return str(self.title)
|