YouTube bells'n'whistles.

[audio.git] / src / youtube / views.py

diff --git a/src/youtube/views.py b/src/youtube/views.py
index 1247c50..4e59f5f 100644 (file)
--- a/src/youtube/views.py
+++ b/src/youtube/views.py
@@ -1,10 +1,11 @@
 from django.contrib.auth.decorators import permission_required
+from django.http import HttpResponse
 from django.shortcuts import redirect, get_object_or_404
 from django.urls import reverse
 from django.views.decorators.http import require_POST
 from archive.constants import status
 from archive.models import Audiobook
-from . import tasks
+from . import models, tasks
 
 
 
@@ -19,3 +20,10 @@ def publish(request, aid, publish=True):
     audiobook.youtube_task = tasks.YouTubeTask.delay(request.user.id, aid, publish).task_id
     audiobook.save(update_fields=['youtube_task'])
     return redirect(reverse('file', args=[aid]))
+
+
+def thumbnail(request, aid):
+    audiobook = get_object_or_404(Audiobook, id=aid)
+    yt = models.YouTube.objects.first()
+    buf = yt.prepare_thumbnail(audiobook)
+    return HttpResponse(buf.getvalue(), content_type='image/png')