42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
|
'''
|
||
|
|
Wrapper for the youtube-dl library. Used so if there are any library interface
|
||
|
|
updates we only need to udpate them in one place.
|
||
|
|
'''
|
||
|
|
|
||
|
|
|
||
|
|
from django.conf import settings
|
||
|
|
import youtube_dl
|
||
|
|
|
||
|
|
|
||
|
|
_defaults = getattr(settings, 'YOUTUBE_DEFAULTS', {})
|
||
|
|
|
||
|
|
|
||
|
|
class YouTubeError(youtube_dl.utils.DownloadError):
|
||
|
|
'''
|
||
|
|
Generic wrapped error for all errors that could be raised by youtube-dl.
|
||
|
|
'''
|
||
|
|
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def extract_info(url):
|
||
|
|
'''
|
||
|
|
Extracts information from a YouTube URL and returns it as a dict. For a channel
|
||
|
|
or playlist this returns a dict of all the videos on the channel or playlist
|
||
|
|
as well as associated metadata.
|
||
|
|
'''
|
||
|
|
opts = _defaults.update({
|
||
|
|
'skip_download': True,
|
||
|
|
'forcejson': True,
|
||
|
|
'simulate': True,
|
||
|
|
'extract_flat': 'in_playlist',
|
||
|
|
'playlist_items': 1,
|
||
|
|
})
|
||
|
|
response = {}
|
||
|
|
with youtube_dl.YoutubeDL(opts) as y:
|
||
|
|
try:
|
||
|
|
response = y.extract_info(url, download=False)
|
||
|
|
except youtube_dl.utils.DownloadError as e:
|
||
|
|
raise YouTubeError(f'Failed to extract_info for "{url}": {e}') from e
|
||
|
|
return response
|