lanbogao commited on
Commit
49a4a29
1 Parent(s): 4fb8d8a

Fix getVttUrlFromSubtitles not correct.

Browse files
Files changed (1) hide show
  1. fetchYoutubeSubtitle.py +3 -3
fetchYoutubeSubtitle.py CHANGED
@@ -5,9 +5,9 @@ import yt_dlp
5
  def getVttUrlFromSubtitles(item, lang='en', vttType="vtt"):
6
  langs = item.keys()
7
  key = lang if lang in langs else ('en' if 'en' in langs else list(langs)[0] )
8
- for item in langs[key]:
9
- if(item.get("ext") == vttType):
10
- return item.get("url")
11
  return None
12
 
13
  async def fetchSubtitle(url: str, lang: Optional[str] = 'en', vttType="vtt") -> Optional[str]:
 
5
  def getVttUrlFromSubtitles(item, lang='en', vttType="vtt"):
6
  langs = item.keys()
7
  key = lang if lang in langs else ('en' if 'en' in langs else list(langs)[0] )
8
+ for subtitle in item[key]:
9
+ if(subtitle.get("ext") == vttType):
10
+ return subtitle.get("url")
11
  return None
12
 
13
  async def fetchSubtitle(url: str, lang: Optional[str] = 'en', vttType="vtt") -> Optional[str]: