from django.db import models from django.conf import settings from tp.models import UUIDAsIDModel from django.contrib.auth.models import User from django.utils.text import slugify from django.core.files.storage import FileSystemStorage from .choices import LICENSE_CHOICES, get_license_info import string import 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 EpisodeAndFeed(UUIDAsIDModel): user = models.ForeignKey(User, on_delete=models.CASCADE) author = models.CharField(max_length=255) title = models.CharField(max_length=255) created_on = models.DateTimeField(auto_now_add=True) slug = models.SlugField(max_length=255, unique=True) description = models.TextField(null=False) image = models.ImageField( storage=FileSystemStorage(location=settings.IMAGE_ROOT), upload_to=slugify_file_name, null=True, blank=True) image_license = models.SmallIntegerField( choices=LICENSE_CHOICES, default=2, ) image_title = models.CharField(max_length=255, default='') image_attribution = models.CharField(max_length=255, default='') image_attribution_url = models.URLField(max_length=255, blank=True) original_image_url = models.URLField(max_length=255, default='') image_license_jurisdiction = models.TextField(null=False, default='(no jurisdiction specified)') @property def image_license_name(self): return(get_license_info(self.image_license))[0] @property def image_license_url(self): return(get_license_info(self.image_license))[1] def __str__(self): return str(self.title) def save(self, *args, **kwargs): if not self.slug: self.slug = slugify(rand_slug() + "-" + self.title) super().save(*args, **kwargs) class Meta: abstract = True class Feed(EpisodeAndFeed): author_url = models.URLField(max_length=255, default='') ebook_title = models.CharField(max_length=255, default='') ebook_url = models.URLField(max_length=255, default='') translator = models.CharField(max_length=255, blank=True) translator_url = models.URLField(max_length=255, blank=True) intro_author = models.CharField(max_length=255, blank=True) intro_author_url = models.URLField(max_length=255, blank=True) license_jurisdiction = models.TextField(blank=True, default='(no jurisdiction specified)') license = models.SmallIntegerField( choices=LICENSE_CHOICES, default=1, ) @property def get_itpc_rss(self): return(f'itpc://{settings.DOMAIN_NAME}/rss/{self.slug}.xml') @property def license_name(self): return(get_license_info(self.license))[0] @property def license_url(self): return(get_license_info(self.license))[1] class Episode(EpisodeAndFeed): feed = models.ForeignKey(Feed, on_delete=models.CASCADE) pub_date = models.DateField() episode_number = models.IntegerField(null=True) mp3 = models.FileField( storage=FileSystemStorage(location=settings.MP3_ROOT), upload_to=slugify_file_name, null=True, blank=True)