[downloader/fragment] Fix download finalization when writing file to stdout (closes...
authorSergey M․ <dstftw@gmail.com>
Sat, 24 Mar 2018 08:59:48 +0000 (15:59 +0700)
committerSergey M․ <dstftw@gmail.com>
Sat, 24 Mar 2018 09:04:23 +0000 (16:04 +0700)
youtube_dl/downloader/fragment.py

index ea5e3a4b5df9f957328557b6f822ef5494cfc9a6..927c7e491655f950bb1a1c316fcd7911b4b3f2fe 100644 (file)
@@ -241,12 +241,16 @@ class FragmentFD(FileDownloader):
             if os.path.isfile(ytdl_filename):
                 os.remove(ytdl_filename)
         elapsed = time.time() - ctx['started']
-        self.try_rename(ctx['tmpfilename'], ctx['filename'])
-        fsize = os.path.getsize(encodeFilename(ctx['filename']))
+
+        if ctx['tmpfilename'] == '-':
+            downloaded_bytes = ctx['complete_frags_downloaded_bytes']
+        else:
+            self.try_rename(ctx['tmpfilename'], ctx['filename'])
+            downloaded_bytes = os.path.getsize(encodeFilename(ctx['filename']))
 
         self._hook_progress({
-            'downloaded_bytes': fsize,
-            'total_bytes': fsize,
+            'downloaded_bytes': downloaded_bytes,
+            'total_bytes': downloaded_bytes,
             'filename': ctx['filename'],
             'status': 'finished',
             'elapsed': elapsed,