Abort when --max-downloads is reached.

This commit is contained in:
Philipp Hagemeister 2011-12-08 20:59:02 +01:00
parent 0b3f3e1ad9
commit 94fd3201b2
1 changed files with 11 additions and 3 deletions

View File

@ -317,6 +317,10 @@ class PostProcessingError(Exception):
"""
pass
class MaxDownloadsReached(Exception):
""" --max-downloads limit has been reached. """
pass
class UnavailableVideoError(Exception):
"""Unavailable Format exception.
@ -730,8 +734,7 @@ class FileDownloader(object):
max_downloads = self.params.get('max_downloads')
if max_downloads is not None:
if self._num_downloads > int(max_downloads):
self.to_screen(u'[download] Maximum number of downloads reached. Skipping ' + info_dict['title'])
return
raise MaxDownloadsReached()
filename = self.prepare_filename(info_dict)
@ -4447,7 +4450,12 @@ def _real_main():
parser.error(u'you must provide at least one URL')
else:
sys.exit()
retcode = fd.download(all_urls)
try:
retcode = fd.download(all_urls)
except MaxDownloadsReached:
fd.to_screen(u'--max-download limit reached, aborting.')
retcode = 101
# Dump cookie jar if requested
if opts.cookiefile is not None: