From 29af7886bcb31bebf849297cf8b00600796583d0 Mon Sep 17 00:00:00 2001
From: Ivlev Denis <di-erz@yandex.ru>
Date: Wed, 14 Feb 2018 18:22:40 +0300
Subject: [PATCH] Fix youtube video index

---
 apps/content/models.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/apps/content/models.py b/apps/content/models.py
index 00c15312..345221f4 100644
--- a/apps/content/models.py
+++ b/apps/content/models.py
@@ -1,3 +1,4 @@
+from urllib.parse import urlparse
 from django.db import models
 
 from polymorphic.models import PolymorphicModel
@@ -69,6 +70,12 @@ class Video(Content):
     url = models.URLField('Ссылка')
 
     def video_index(self):
+        if 'youtu.be' in self.url or 'youtube.com' in self.url and 'watch' in self.url:
+            url = urlparse(self.url)
+            query = url.query.split('&')
+            for q in query:
+                if 'v=' in q:
+                    return q.split('=')[-1]
         return self.url.split('/')[-1]