From fa0d62772076210f840998726dfb0822fceeb5fb Mon Sep 17 00:00:00 2001 From: Vladimir Berezhnoy Date: Sat, 26 Nov 2011 23:15:34 +0400 Subject: [PATCH] Add --max-files option --- youtube-dl | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/youtube-dl b/youtube-dl index 1ce120007..b98d3b42a 100755 --- a/youtube-dl +++ b/youtube-dl @@ -701,7 +701,11 @@ class FileDownloader(object): def process_info(self, info_dict): """Process a single dictionary returned by an InfoExtractor.""" filename = self.prepare_filename(info_dict) - + + # Stop downloading at max_files + if int(self._num_downloads) > int(self.params.get('max_files', 0)): + return + # Forced printings if self.params.get('forcetitle', False): print info_dict['title'].encode(preferredencoding(), 'xmlcharrefreplace') @@ -4075,6 +4079,8 @@ def parseOpts(): filesystem.add_option('--write-info-json', action='store_true', dest='writeinfojson', help='write video metadata to a .info.json file', default=False) + filesystem.add_option('--max-files', + dest='max_files', help='Maximum number of files to download', default=0) postproc.add_option('--extract-audio', action='store_true', dest='extractaudio', default=False, @@ -4265,6 +4271,7 @@ def _real_main(): 'writeinfojson': opts.writeinfojson, 'matchtitle': opts.matchtitle, 'rejecttitle': opts.rejecttitle, + 'max_files': opts.max_files, }) for extractor in extractors: fd.add_info_extractor(extractor)