youtube-dl

commit b7f8116406d7d7165b01c02b4f3a5dc63476a294
parent 6db274e057873feb256568f0e27c9a03a2b8d16c
Author: rupertbaxter2 <rupertbaxter2@yahoo.com>
Date:   Wed, 23 Jul 2014 02:53:44 +0200

Deletes temp files after postprocess merge unless -k option is specified

Diffstat:
M
youtube_dl/YoutubeDL.py
 | 
2
+-
M
youtube_dl/postprocessor/ffmpeg.py
 | 
6
+++++-

2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/youtube_dl/YoutubeDL.py b/youtube_dl/YoutubeDL.py
@@ -999,7 +999,7 @@ class YoutubeDL(object):
                     if info_dict.get('requested_formats') is not None:
                         downloaded = []
                         success = True
-                        merger = FFmpegMergerPP(self)
+                        merger = FFmpegMergerPP(self, not self.params.get('keepvideo'))
                         if not merger._get_executable():
                             postprocessors = []
                             self.report_warning('You have requested multiple '
diff --git a/youtube_dl/postprocessor/ffmpeg.py b/youtube_dl/postprocessor/ffmpeg.py
@@ -23,9 +23,10 @@ class FFmpegPostProcessorError(PostProcessingError):
     pass
 
 class FFmpegPostProcessor(PostProcessor):
-    def __init__(self,downloader=None):
+    def __init__(self,downloader=None,deletetempfiles=False):
         PostProcessor.__init__(self, downloader)
         self._exes = self.detect_executables()
+        self._deletetempfiles = deletetempfiles
 
     @staticmethod
     def detect_executables():
@@ -60,6 +61,9 @@ class FFmpegPostProcessor(PostProcessor):
             stderr = stderr.decode('utf-8', 'replace')
             msg = stderr.strip().split('\n')[-1]
             raise FFmpegPostProcessorError(msg)
+        if self._deletetempfiles:
+            for rempath in input_paths:
+                os.remove(rempath)
 
     def run_ffmpeg(self, path, out_path, opts):
         self.run_ffmpeg_multiple_files([path], out_path, opts)