music-kraken/music-kraken-core
3
2
Fork 0
Code Issues 10 Pull Requests 1 Packages Projects 2 Releases Wiki Activity

fixed crashing in youtube search

Browse Source
This commit is contained in:
lars 2022-11-08 23:49:21 +01:00
parent 5486f7f8d1
commit dbf47c1937
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/scraping/youtube_music.py
View File

@ -1,5 +1,6 @@
from typing import List
import youtube_dl
import pandas as pd
import logging
import time
@ -15,10 +16,13 @@ WAIT_BETWEEN_BLOCK = 10
MAX_TRIES = 3
def get_youtube_from_isrc(isrc: str):
def get_youtube_from_isrc(isrc: str) -> List[dict]:
# https://stackoverflow.com/questions/63388364/searching-youtube-videos-using-youtube-dl
with youtube_dl.YoutubeDL(YDL_OPTIONS) as ydl:
try:
videos = ydl.extract_info(f"ytsearch:{isrc}", download=False)['entries']
except youtube_dl.utils.DownloadError:
return []
return [{
'url': video[YOUTUBE_URL_KEY],
@ -27,8 +31,6 @@ def get_youtube_from_isrc(isrc: str):
def get_youtube_url(row):
if pd.isna(row['isrc']):
return None
real_title = row['title'].lower()
final_result = None