[drtuber] Extract display_id

[youtube-dl] / youtube_dl / extractor / beeg.py

diff --git a/youtube_dl/extractor/beeg.py b/youtube_dl/extractor/beeg.py
index 775f6be6b82e9aaf351a5b9b5cc1ce9e1540543d..d7301fe18ccf7f2324e3cb233ac93102ed076ba6 100644 (file)
--- a/youtube_dl/extractor/beeg.py
+++ b/youtube_dl/extractor/beeg.py
@@ -15,6 +15,9 @@ class BeegIE(InfoExtractor):
             'ext': 'mp4',
             'title': 'Sultry Striptease',
             'description': 'md5:6db3c6177972822aaba18652ff59c773',
+            'categories': list,  # NSFW
+            'thumbnail': 're:https?://.*\.jpg$',
+            'age_limit': 18,
         }
     }
 
@@ -24,19 +27,24 @@ class BeegIE(InfoExtractor):
 
         webpage = self._download_webpage(url, video_id)
 
-        video_url = self._html_search_regex(r"'480p'\s*:\s*'([^']+)'", webpage, 'video URL')
+        video_url = self._html_search_regex(
+            r"'480p'\s*:\s*'([^']+)'", webpage, 'video URL')
 
-        title = self._html_search_regex(r'<title>([^<]+)\s*-\s*beeg\.?</title>', webpage, 'title')
+        title = self._html_search_regex(
+            r'<title>([^<]+)\s*-\s*beeg\.?</title>', webpage, 'title')
         
         description = self._html_search_regex(
-            r'<meta name="description" content="([^"]*)"', webpage, 'description', fatal=False)
-        
+            r'<meta name="description" content="([^"]*)"',
+            webpage, 'description', fatal=False)
         thumbnail = self._html_search_regex(
-            r'\'previewer.url\'\s*:\s*"([^"]*)"', webpage, 'thumbnail', fatal=False)
+            r'\'previewer.url\'\s*:\s*"([^"]*)"',
+            webpage, 'thumbnail', fatal=False)
 
         categories_str = self._html_search_regex(
             r'<meta name="keywords" content="([^"]+)"', webpage, 'categories', fatal=False)
-        categories = categories_str.split(',')
+        categories = (
+            None if categories_str is None
+            else categories_str.split(','))
 
         return {
             'id': video_id,
@@ -45,4 +53,5 @@ class BeegIE(InfoExtractor):
             'description': description,
             'thumbnail': thumbnail,
             'categories': categories,
+            'age_limit': 18,
         }