add config to mdx_video extension to use youtube-nocookie.com for iframe

justincletus · Jul 16, 2020 · 07316b7 · 07316b7
1 parent 8cdc30c
commit 07316b7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/martor/extensions/mdx_video.py b/martor/extensions/mdx_video.py
@@ -20,6 +20,7 @@ def __init__(self, **kwargs):
             'yahoo_height': ['351', 'Height for Yahoo! videos'],
             'youtube_width': ['560', 'Width for Youtube videos'],
             'youtube_height': ['315', 'Height for Youtube videos'],
+            'youtube_nocookie': [False, 'Use youtube-nocookie.com instead of youtube.com']
         }
 
         # Override defaults with user settings
@@ -97,7 +98,10 @@ def handleMatch(self, m):
 class Youtube(markdown.inlinepatterns.Pattern):
 
     def handleMatch(self, m):
-        url = '//www.youtube.com/embed/%s' % m.group('youtubeid')
+        if self.ext.config['youtube_nocookie'][0]:
+            url = '//www.youtube-nocookie.com/embed/%s' % m.group('youtubeid')
+        else:
+            url = '//www.youtube.com/embed/%s' % m.group('youtubeid')
         width = self.ext.config['youtube_width'][0]
         height = self.ext.config['youtube_height'][0]
         return render_iframe(url, width, height)