github/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-30 16:10:24 -05:00
Code Issues Projects Releases Packages Wiki Activity

Abort when --max-downloads is reached.

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

12
youtube_dl/__init__.py
View file

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