heckyel/yt-local
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Invidious fallback: Use original format info and just substitute invidious urls

Browse Source 
Because the invidious formats don't have all the information
This commit is contained in:
James Taylor 2020-02-04 19:08:56 -08:00
parent 9f090dbbf8
commit 14b9c30daf
1 changed files with 12 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
youtube/watch.py
View File

@ -263,30 +263,20 @@ def extract_info(video_id):
return info
video_info = json.loads(video_info.decode('utf-8'))
info['formats'] = []
for fmt in (video_info['adaptiveFormats']
# collect invidious urls for each itag
itag_to_url = {}
for invidious_fmt in (video_info['adaptiveFormats']
+ video_info['formatStreams']):
# adjust keys to match our conventions
fmt['file_size'] = fmt.get('clen')
fmt['ext'] = fmt.get('container')
if 'resolution' in fmt:
fmt['height'] = int(fmt['resolution'].rstrip('p'))
itag_to_url[invidious_fmt['itag']] = invidious_fmt['url']
# update with information from _formats table such as ext
itag = fmt.get('itag')
fmt.update(yt_data_extract._formats.get(itag, {}))
# extract acodec, vcodec, and ext
# (need for 'ext' because 'container' not always present)
yt_data_extract.update_format_with_type_info(fmt, fmt)
# ensure keys are present
for key in ('ext', 'audio_bitrate', 'acodec', 'vcodec',
'width', 'height', 'audio_sample_rate', 'fps'):
if key not in fmt:
fmt[key] = None
info['formats'].append(fmt)
# replace urls with urls from invidious
for fmt in info['formats']:
itag = str(fmt['itag'])
if itag not in itag_to_url:
print(('Warning: itag '
+ itag + ' not found in invidious urls'))
continue
fmt['url'] = itag_to_url[itag]
return info
def video_quality_string(format):