Add try except to fetchSubtitle and fetchSubtitleUrls.

fetchYoutubeSubtitle.py

@@ -24,28 +24,21 @@ def getUrlFromSubtitles(item, lang='en', subType="vtt"):
         return None
 
     l = lang if lang in langs else ('en' if 'en' in langs else list(langs)[0] )
-    print("getUrlFromSubtitles l: %s, item: %s" % (l, item))
+    if l is  None:
+        return
 
     for subtitle in item[l]:
-        print("getUrlFromSubtitles subtitle: %s" % subtitle)
+        # print("getUrlFromSubtitles subtitle: %s" % subtitle)
         if l != "live_chat" and subType =="xml":
             return subtitle.get("url").replace("fmt="+subtitle.get("ext"),"")
         if subtitle.get("ext") == subType:
             return subtitle.get("url")
     return None
 
-async def fetchSubtitle(url: str, lang: Optional[str] = 'en', subType: Optional[str] = "vtt") -> Optional[str]:
-    if subType == "srt":
-        subtitle = await fetchSubtitlebyType(url, lang, subType, True)
-        if subtitle:
-            return subtitle
-        subtitle = await fetchSubtitlebyType(url, lang, "xml", True)
-        print(subtitle)
-        return xml_caption_to_srt(subtitle)
-    else:
-        return await fetchSubtitlebyType(url, lang, subType, True)
+async def fetchSubtitle(url: str, lang: Optional[str] = 'en', subType: Optional[str] = "vtt") -> dict:
+    return await fetchSubtitlebyType(url, lang, subType)
 
-async def fetchSubtitlebyType(url: str, lang: Optional[str] = 'en', subType="vtt", decode: bool = False) -> Optional[str]:
+async def fetchSubtitlebyType(url: str, lang: Optional[str] = 'en', subType="vtt") -> dict:
     ydl_opts = {
         "writesubtitles": True,
         "allsubtitles": True,
@@ -54,16 +47,25 @@ async def fetchSubtitlebyType(url: str, lang: Optional[str] = 'en', subType="vtt
         "socket_timeout": 20
     }
 
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info_dict = ydl.extract_info(url, download=False)
-        for subtitle_item in ["subtitles", "automatic_captions"]: # "requested_subtitles" item is dict
-            if info_dict.get(subtitle_item) :
-                subtitle_url = getUrlFromSubtitles(info_dict.get(subtitle_item), lang, subType)
-                if subtitle_url:
-                    with ydl.urlopen(subtitle_url) as subtitle:
-                        return subtitle.read().decode() if decode else subtitle.read()
+    title = "unknow"
+    try:
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info_dict = ydl.extract_info(url, download=False)
+            title = info_dict.get("title", "unknow")
+            if info_dict.get("extractor") == "youtube" and subType == "srt":
+                subType = "xml"
 
-    return None
+            for subtitle_item in ["subtitles", "automatic_captions"]: # "requested_subtitles" item is dict
+                if info_dict.get(subtitle_item):
+                    subtitle_url = getUrlFromSubtitles(info_dict.get(subtitle_item), lang, subType)
+                    if subtitle_url:
+                        with ydl.urlopen(subtitle_url) as response:
+                            subtitle = xml_caption_to_srt(response.read().decode()) if subType == "xml" else response.read().decode()
+                            print("url{}, title:{} len(subtitle): {}".format(url, title, len(subtitle)))
+                            return {"title": title, "subtitle": subtitle}
+    except Exception as e:
+        return {"error": str(e)}
+    return {"title": title,"error": "No subtitles"}
 
 def float_to_srt_time_format(d: float) -> str:
     """Convert decimal durations into proper srt format.
@@ -109,13 +111,18 @@ async def fetchSubtitleUrls(url: str) -> json:
         "allsubtitles": True,
         "skip_download": True,
     }
+    title = "unknow"
+    try:
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info_dict = ydl.extract_info(url, download=False)
+            title = info_dict.get("title", "unknow")
 
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        info_dict = ydl.extract_info(url, download=False)
-        if info_dict.get("subtitles"):
-            langs = info_dict.get("subtitles").keys()
-            if not (len(langs) == 1 and "live_chat" in langs):
-                return info_dict.get("subtitles")
-            if info_dict.get("automatic_captions"):
-                return info_dict.get("automatic_captions")
-    return None
+            if info_dict.get("subtitles"):
+                langs = info_dict.get("subtitles").keys()
+                if not (len(langs) == 1 and "live_chat" in langs):
+                    return {"title": info_dict.get("title", "unknow"), "subtitles": info_dict.get("subtitles")}
+                if info_dict.get("automatic_captions"):
+                    return {"title": info_dict.get("title", "unknow"), "subtitles": info_dict.get("automatic_captions")}
+    except Exception as e:
+        return {"error": str(e)}
+    return {"title": title,"error": "No subtitles"}