[downloader/fragment] Output ad fragment count
authorSergey M․ <dstftw@gmail.com>
Sat, 14 Oct 2017 23:13:07 +0000 (06:13 +0700)
committerSergey M․ <dstftw@gmail.com>
Sat, 14 Oct 2017 23:13:07 +0000 (06:13 +0700)
youtube_dl/downloader/fragment.py

index 6f6fb4a77a9dc2b3e4e4278bc51b12755bc709ee..7e891b92a3b6a05257484c5d8ba81590013bb74b 100644 (file)
@@ -117,9 +117,15 @@ class FragmentFD(FileDownloader):
     def _prepare_frag_download(self, ctx):
         if 'live' not in ctx:
             ctx['live'] = False
+        if not ctx['live']:
+            total_frags_str = '%d' % ctx['total_frags']
+            ad_frags = ctx.get('ad_frags', 0)
+            if ad_frags:
+                total_frags_str += ' (not including %d ad)' % ad_frags
+        else:
+            total_frags_str = 'unknown (live)'
         self.to_screen(
-            '[%s] Total fragments: %s'
-            % (self.FD_NAME, ctx['total_frags'] if not ctx['live'] else 'unknown (live)'))
+            '[%s] Total fragments: %s' % (self.FD_NAME, total_frags_str))
         self.report_destination(ctx['filename'])
         dl = HttpQuietDownloader(
             self.ydl,