[Docubay] Improve IE

Ashish0804 · May 20, 2022 · 713117c · 713117c
1 parent fc18502
commit 713117c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/yt_dlp/extractor/docubay.py b/yt_dlp/extractor/docubay.py
@@ -33,10 +33,13 @@ def _real_extract(self, url):
             })
 
         webpage = self._download_webpage(url, video_id)
-        title = self._og_search_title(webpage)
-        if title:
-            title = title.replace(' | Online at DocuBay', '').replace('Watch ', '')
+        title = self._search_regex(r'class\s*=\s*"shows-title"[^>]*>([^<]+)', webpage, 'title', None, False)
+        if not title:
+            title = self._og_search_title(webpage).replace(' | Online at DocuBay', '').replace('Watch ', '')
 
+        upload_date = self._search_regex(r'class\s*=\s*"show-duration"[^>]*>\s*(\d+)', webpage, 'upload date', None, False)
+        if upload_date:
+            upload_date += '0101'
         subtitles = self._merge_subtitles(subs, subtitles)
         self._sort_formats(formats)
 
@@ -45,6 +48,7 @@ def _real_extract(self, url):
             'title': title,
             'description': self._og_search_description(webpage),
             'thumbnail': self._og_search_thumbnail(webpage),
+            'upload_date': upload_date,
             'formats': formats,
             'subtitles': subtitles
 

diff --git a/yt_dlp/postprocessor/renamer.py b/yt_dlp/postprocessor/renamer.py
@@ -25,6 +25,7 @@ def __init__(self, downloader=None, group='ytdlP'):
         'SonyLiv': 'SONY',
         'Zee5': 'ZEE',
         'DiscoveryPlusIndia': 'DSCP',
+        'Docubay': 'DOCBAY'
     }
 
     def run(self, info):