Change the order for extracting/downloading
Now it gets a video info and directly downloads it, the it pass to the next video founded.main
parent
d39919c03e
commit
146c12a2da
|
@ -419,9 +419,10 @@ class FileDownloader(object):
|
||||||
return u'"' + title + '" title matched reject pattern "' + rejecttitle + '"'
|
return u'"' + title + '" title matched reject pattern "' + rejecttitle + '"'
|
||||||
return None
|
return None
|
||||||
|
|
||||||
def extract_info(self, url):
|
def extract_info(self, url, download = True):
|
||||||
'''
|
'''
|
||||||
Returns a list with a dictionary for each video we find.
|
Returns a list with a dictionary for each video we find.
|
||||||
|
If 'download', also downloads the videos.
|
||||||
'''
|
'''
|
||||||
suitable_found = False
|
suitable_found = False
|
||||||
for ie in self._ies:
|
for ie in self._ies:
|
||||||
|
@ -440,7 +441,12 @@ class FileDownloader(object):
|
||||||
# Extract information from URL and process it
|
# Extract information from URL and process it
|
||||||
try:
|
try:
|
||||||
ie_results = ie.extract(url)
|
ie_results = ie.extract(url)
|
||||||
results = self.process_ie_results(ie_results, ie)
|
results = []
|
||||||
|
for ie_result in ie_results:
|
||||||
|
if not 'extractor' in ie_result:
|
||||||
|
#The extractor has already been set somewhere else
|
||||||
|
ie_result['extractor'] = ie.IE_NAME
|
||||||
|
results.append(self.process_ie_result(ie_result, download))
|
||||||
return results
|
return results
|
||||||
except ExtractorError as de: # An error we somewhat expected
|
except ExtractorError as de: # An error we somewhat expected
|
||||||
self.trouble(u'ERROR: ' + compat_str(de), de.format_traceback())
|
self.trouble(u'ERROR: ' + compat_str(de), de.format_traceback())
|
||||||
|
@ -453,51 +459,51 @@ class FileDownloader(object):
|
||||||
raise
|
raise
|
||||||
if not suitable_found:
|
if not suitable_found:
|
||||||
self.trouble(u'ERROR: no suitable InfoExtractor: %s' % url)
|
self.trouble(u'ERROR: no suitable InfoExtractor: %s' % url)
|
||||||
def extract_info_iterable(self, urls):
|
|
||||||
'''
|
|
||||||
Return the videos founded for the urls
|
|
||||||
'''
|
|
||||||
results = []
|
|
||||||
for url in urls:
|
|
||||||
results.extend(self.extract_info(url))
|
|
||||||
return results
|
|
||||||
|
|
||||||
def process_ie_results(self, ie_results, ie):
|
def process_ie_result(self, ie_result, download = True):
|
||||||
"""
|
"""
|
||||||
Take the results of the ie and return a list of videos.
|
Take the result of the ie and return a list of videos.
|
||||||
For url elements it will seartch the suitable ie and get the videos
|
For url elements it will search the suitable ie and get the videos
|
||||||
For playlist elements it will process each of the elements of the 'entries' key
|
For playlist elements it will process each of the elements of the 'entries' key
|
||||||
|
|
||||||
|
It will also download the videos if 'download'.
|
||||||
"""
|
"""
|
||||||
results = []
|
result_type = ie_result.get('_type', 'video') #If not given we suppose it's a video, support the dafault old system
|
||||||
for result in ie_results or []:
|
|
||||||
result_type = result.get('_type', 'video') #If not given we suppose it's a video, support the dafault old system
|
|
||||||
if result_type == 'video':
|
if result_type == 'video':
|
||||||
if not 'extractor' in result:
|
if 'playlist' not in ie_result:
|
||||||
#The extractor has already been set somewhere else
|
#It isn't part of a playlist
|
||||||
result['extractor'] = ie.IE_NAME
|
ie_result['playlist'] = None
|
||||||
results.append(result)
|
if download:
|
||||||
|
#Do the download:
|
||||||
|
self.process_info(ie_result)
|
||||||
|
return ie_result
|
||||||
elif result_type == 'url':
|
elif result_type == 'url':
|
||||||
#We get the videos pointed by the url
|
#We get the video pointed by the url
|
||||||
results.extend(self.extract_info(result['url']))
|
result = self.extract_info(ie_result['url'], download)[0]
|
||||||
|
return result
|
||||||
elif result_type == 'playlist':
|
elif result_type == 'playlist':
|
||||||
#We process each entry in the playlist
|
#We process each entry in the playlist
|
||||||
entries_result = self.process_ie_results(result['entries'], ie)
|
playlist = ie_result.get('title', None) or ie_result.get('id', None)
|
||||||
result['entries'] = entries_result
|
self.to_screen(u'[download] Downloading playlist: %s' % playlist)
|
||||||
results.extend([result])
|
n_videos = len(ie_result['entries'])
|
||||||
return results
|
playlist_results = []
|
||||||
|
for i,entry in enumerate(ie_result['entries'],1):
|
||||||
|
self.to_screen(u'[download] Downloading video #%s of %s' %(i, n_videos))
|
||||||
|
entry_result = self.process_ie_result(entry, False)
|
||||||
|
entry_result['playlist'] = playlist
|
||||||
|
#We must do the download here to correctly set the 'playlist' key
|
||||||
|
if download:
|
||||||
|
self.process_info(entry_result)
|
||||||
|
playlist_results.append(entry_result)
|
||||||
|
result = ie_result.copy()
|
||||||
|
result['entries'] = playlist_results
|
||||||
|
return result
|
||||||
|
|
||||||
def process_info(self, info_dict):
|
def process_info(self, info_dict):
|
||||||
"""Process a single dictionary returned by an InfoExtractor."""
|
"""Process a single dictionary returned by an InfoExtractor."""
|
||||||
|
|
||||||
if info_dict.get('_type','video') == 'playlist':
|
#We increment the download the download count here to match the previous behaviour.
|
||||||
playlist = info_dict.get('title', None) or info_dict.get('id', None)
|
self.increment_downloads()
|
||||||
self.to_screen(u'[download] Downloading playlist: %s' % playlist)
|
|
||||||
n_videos = len(info_dict['entries'])
|
|
||||||
for i,video in enumerate(info_dict['entries'],1):
|
|
||||||
video['playlist'] = playlist
|
|
||||||
self.to_screen(u'[download] Downloading video #%s of %s' %(i, n_videos))
|
|
||||||
self.process_info(video)
|
|
||||||
return
|
|
||||||
|
|
||||||
# Keep for backwards compatibility
|
# Keep for backwards compatibility
|
||||||
info_dict['stitle'] = info_dict['title']
|
info_dict['stitle'] = info_dict['title']
|
||||||
|
@ -633,12 +639,9 @@ class FileDownloader(object):
|
||||||
raise SameFileError(self.params['outtmpl'])
|
raise SameFileError(self.params['outtmpl'])
|
||||||
|
|
||||||
for url in url_list:
|
for url in url_list:
|
||||||
videos = self.extract_info(url)
|
|
||||||
|
|
||||||
for video in videos or []:
|
|
||||||
try:
|
try:
|
||||||
self.increment_downloads()
|
#It also downloads the videos
|
||||||
self.process_info(video)
|
videos = self.extract_info(url)
|
||||||
except UnavailableVideoError:
|
except UnavailableVideoError:
|
||||||
self.trouble(u'\nERROR: unable to download video')
|
self.trouble(u'\nERROR: unable to download video')
|
||||||
except MaxDownloadsReached:
|
except MaxDownloadsReached:
|
||||||
|
|
Loading…
Reference in New Issue