X-Git-Url: https://git.mdrn.pl/audio.git/blobdiff_plain/576253ba3c8b54dd509de0dbf7373b310fa9b71c..1c72d045af4d7c5607a872a678eadec17498bd6d:/src/youtube/models.py

diff --git a/src/youtube/models.py b/src/youtube/models.py
index 356e335..df071f1 100644
--- a/src/youtube/models.py
+++ b/src/youtube/models.py
@@ -24,6 +24,7 @@ YOUTUBE_TITLE_LIMIT = 100
 
 
 class YouTube(models.Model):
+    name = models.CharField(max_length=255)
     title_template = models.CharField(max_length=1024, blank=True)
     description_template = models.TextField(blank=True)
     category = models.IntegerField(null=True, blank=True, choices=[
@@ -44,6 +45,9 @@ class YouTube(models.Model):
         verbose_name = _("YouTube configuration")
         verbose_name_plural = _("YouTube configurations")
 
+    def __str__(self):
+        return self.name
+
     def get_context(self, audiobook):
         return Context(dict(
             audiobook=audiobook,
@@ -212,6 +216,7 @@ class Font(models.Model):
 
 
 class ThumbnailTemplate(models.Model):
+    youtube = models.ForeignKey(YouTube, models.CASCADE)
     order = models.SmallIntegerField()
     is_active = models.BooleanField()
     background = models.FileField(upload_to='youtube/thumbnail')