mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-21 20:46:36 -05:00
Move the code to process an InfoExtractor result to its own method
This commit is contained in:
parent
257453b92b
commit
c8619e0163
1 changed files with 56 additions and 42 deletions
98
youtube-dl
98
youtube-dl
|
@ -239,6 +239,48 @@ class FileDownloader(object):
|
||||||
"""Report download finished."""
|
"""Report download finished."""
|
||||||
self.to_stdout(u'')
|
self.to_stdout(u'')
|
||||||
|
|
||||||
|
def process_info(self, info_dict):
|
||||||
|
"""Process a single dictionary returned by an InfoExtractor."""
|
||||||
|
# Forced printings
|
||||||
|
if self.params.get('forcetitle', False):
|
||||||
|
print info_dict['title']
|
||||||
|
if self.params.get('forceurl', False):
|
||||||
|
print info_dict['url']
|
||||||
|
|
||||||
|
# Do nothing else if in simulate mode
|
||||||
|
if self.params.get('simulate', False):
|
||||||
|
return 0
|
||||||
|
|
||||||
|
try:
|
||||||
|
filename = self.params['outtmpl'] % info_dict
|
||||||
|
self.report_destination(filename)
|
||||||
|
except (ValueError, KeyError), err:
|
||||||
|
return self.trouble('ERROR: invalid output template or system charset: %s' % str(err))
|
||||||
|
if self.params['nooverwrites'] and os.path.exists(filename):
|
||||||
|
self.to_stderr('WARNING: file exists: %s; skipping' % filename)
|
||||||
|
return 0
|
||||||
|
try:
|
||||||
|
self.pmkdir(filename)
|
||||||
|
except (OSError, IOError), err:
|
||||||
|
return self.trouble('ERROR: unable to create directories: %s' % str(err))
|
||||||
|
try:
|
||||||
|
outstream = open(filename, 'wb')
|
||||||
|
except (OSError, IOError), err:
|
||||||
|
return self.trouble('ERROR: unable to open for writing: %s' % str(err))
|
||||||
|
try:
|
||||||
|
self._do_download(outstream, info_dict['url'])
|
||||||
|
outstream.close()
|
||||||
|
except (OSError, IOError), err:
|
||||||
|
return self.trouble('ERROR: unable to write video data: %s' % str(err))
|
||||||
|
except (urllib2.URLError, httplib.HTTPException, socket.error), err:
|
||||||
|
return self.trouble('ERROR: unable to download video data: %s' % str(err))
|
||||||
|
try:
|
||||||
|
self.post_process(filename, info_dict)
|
||||||
|
except (PostProcessingError), err:
|
||||||
|
return self.trouble('ERROR: postprocessing: %s' % str(err))
|
||||||
|
|
||||||
|
return 0
|
||||||
|
|
||||||
def download(self, url_list):
|
def download(self, url_list):
|
||||||
"""Download a given list of URLs."""
|
"""Download a given list of URLs."""
|
||||||
retcode = 0
|
retcode = 0
|
||||||
|
@ -248,64 +290,36 @@ class FileDownloader(object):
|
||||||
for url in url_list:
|
for url in url_list:
|
||||||
suitable_found = False
|
suitable_found = False
|
||||||
for ie in self._ies:
|
for ie in self._ies:
|
||||||
|
# Go to next InfoExtractor if not suitable
|
||||||
if not ie.suitable(url):
|
if not ie.suitable(url):
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# Suitable InfoExtractor found
|
# Suitable InfoExtractor found
|
||||||
suitable_found = True
|
suitable_found = True
|
||||||
|
|
||||||
|
# Extract information from URL
|
||||||
all_results = ie.extract(url)
|
all_results = ie.extract(url)
|
||||||
results = [x for x in all_results if x is not None]
|
results = [x for x in all_results if x is not None]
|
||||||
|
|
||||||
|
# See if there were problems extracting any information
|
||||||
if len(results) != len(all_results):
|
if len(results) != len(all_results):
|
||||||
retcode = self.trouble()
|
retcode = self.trouble()
|
||||||
|
|
||||||
|
# Two results could go to the same file
|
||||||
if len(results) > 1 and self.fixed_template():
|
if len(results) > 1 and self.fixed_template():
|
||||||
raise SameFileError(self.params['outtmpl'])
|
raise SameFileError(self.params['outtmpl'])
|
||||||
|
|
||||||
|
# Process each result
|
||||||
for result in results:
|
for result in results:
|
||||||
# Forced printings
|
result = self.process_info(result)
|
||||||
if self.params.get('forcetitle', False):
|
|
||||||
print result['title']
|
|
||||||
if self.params.get('forceurl', False):
|
|
||||||
print result['url']
|
|
||||||
|
|
||||||
# Do nothing else if in simulate mode
|
# Do not overwrite an error code with a success code
|
||||||
if self.params.get('simulate', False):
|
if result != 0:
|
||||||
continue
|
retcode = result
|
||||||
|
|
||||||
try:
|
|
||||||
filename = self.params['outtmpl'] % result
|
|
||||||
self.report_destination(filename)
|
|
||||||
except (ValueError, KeyError), err:
|
|
||||||
retcode = self.trouble('ERROR: invalid output template or system charset: %s' % str(err))
|
|
||||||
continue
|
|
||||||
if self.params['nooverwrites'] and os.path.exists(filename):
|
|
||||||
self.to_stderr('WARNING: file exists: %s; skipping' % filename)
|
|
||||||
continue
|
|
||||||
try:
|
|
||||||
self.pmkdir(filename)
|
|
||||||
except (OSError, IOError), err:
|
|
||||||
retcode = self.trouble('ERROR: unable to create directories: %s' % str(err))
|
|
||||||
continue
|
|
||||||
try:
|
|
||||||
outstream = open(filename, 'wb')
|
|
||||||
except (OSError, IOError), err:
|
|
||||||
retcode = self.trouble('ERROR: unable to open for writing: %s' % str(err))
|
|
||||||
continue
|
|
||||||
try:
|
|
||||||
self._do_download(outstream, result['url'])
|
|
||||||
outstream.close()
|
|
||||||
except (OSError, IOError), err:
|
|
||||||
retcode = self.trouble('ERROR: unable to write video data: %s' % str(err))
|
|
||||||
continue
|
|
||||||
except (urllib2.URLError, httplib.HTTPException, socket.error), err:
|
|
||||||
retcode = self.trouble('ERROR: unable to download video data: %s' % str(err))
|
|
||||||
continue
|
|
||||||
try:
|
|
||||||
self.post_process(filename, result)
|
|
||||||
except (PostProcessingError), err:
|
|
||||||
retcode = self.trouble('ERROR: postprocessing: %s' % str(err))
|
|
||||||
continue
|
|
||||||
|
|
||||||
|
# Suitable InfoExtractor had been found; go to next URL
|
||||||
break
|
break
|
||||||
|
|
||||||
if not suitable_found:
|
if not suitable_found:
|
||||||
retcode = self.trouble('ERROR: no suitable InfoExtractor: %s' % url)
|
retcode = self.trouble('ERROR: no suitable InfoExtractor: %s' % url)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue