Browse Source

Merge pull request #601 from paullik/no-post-overwrites

No post-processing overwrites
Philipp Hagemeister 12 years ago
parent
commit
dc23886a77
2 changed files with 11 additions and 5 deletions
  1. 8 4
      youtube_dl/PostProcessor.py
  2. 3 1
      youtube_dl/__init__.py

+ 8 - 4
youtube_dl/PostProcessor.py

@@ -62,13 +62,14 @@ class AudioConversionError(BaseException):
         self.message = message
         self.message = message
 
 
 class FFmpegExtractAudioPP(PostProcessor):
 class FFmpegExtractAudioPP(PostProcessor):
-    def __init__(self, downloader=None, preferredcodec=None, preferredquality=None, keepvideo=False):
+    def __init__(self, downloader=None, preferredcodec=None, preferredquality=None, keepvideo=False, nopostoverwrites=False):
         PostProcessor.__init__(self, downloader)
         PostProcessor.__init__(self, downloader)
         if preferredcodec is None:
         if preferredcodec is None:
             preferredcodec = 'best'
             preferredcodec = 'best'
         self._preferredcodec = preferredcodec
         self._preferredcodec = preferredcodec
         self._preferredquality = preferredquality
         self._preferredquality = preferredquality
         self._keepvideo = keepvideo
         self._keepvideo = keepvideo
+        self._nopostoverwrites = nopostoverwrites
         self._exes = self.detect_executables()
         self._exes = self.detect_executables()
 
 
     @staticmethod
     @staticmethod
@@ -102,7 +103,7 @@ class FFmpegExtractAudioPP(PostProcessor):
 
 
     def run_ffmpeg(self, path, out_path, codec, more_opts):
     def run_ffmpeg(self, path, out_path, codec, more_opts):
         if not self._exes['ffmpeg'] and not self._exes['avconv']:
         if not self._exes['ffmpeg'] and not self._exes['avconv']:
-            raise AudioConversionError('ffmpeg or avconv not found. Please install one.')   
+            raise AudioConversionError('ffmpeg or avconv not found. Please install one.')
         if codec is None:
         if codec is None:
             acodec_opts = []
             acodec_opts = []
         else:
         else:
@@ -171,9 +172,12 @@ class FFmpegExtractAudioPP(PostProcessor):
 
 
         prefix, sep, ext = path.rpartition(u'.') # not os.path.splitext, since the latter does not work on unicode in all setups
         prefix, sep, ext = path.rpartition(u'.') # not os.path.splitext, since the latter does not work on unicode in all setups
         new_path = prefix + sep + extension
         new_path = prefix + sep + extension
-        self._downloader.to_screen(u'[' + (self._exes['avconv'] and 'avconv' or 'ffmpeg') + '] Destination: ' + new_path)
         try:
         try:
-            self.run_ffmpeg(path, new_path, acodec, more_opts)
+            if self._nopostoverwrites and os.path.exists(encodeFilename(new_path)):
+                self._downloader.to_screen(u'[youtube] Post-process file %s exists, skipping' % new_path)
+            else:
+                self._downloader.to_screen(u'[' + (self._exes['avconv'] and 'avconv' or 'ffmpeg') + '] Destination: ' + new_path)
+                self.run_ffmpeg(path, new_path, acodec, more_opts)
         except:
         except:
             etype,e,tb = sys.exc_info()
             etype,e,tb = sys.exc_info()
             if isinstance(e, AudioConversionError):
             if isinstance(e, AudioConversionError):

+ 3 - 1
youtube_dl/__init__.py

@@ -343,6 +343,8 @@ def parseOpts():
             help='ffmpeg/avconv audio quality specification, insert a value between 0 (better) and 9 (worse) for VBR or a specific bitrate like 128K (default 5)')
             help='ffmpeg/avconv audio quality specification, insert a value between 0 (better) and 9 (worse) for VBR or a specific bitrate like 128K (default 5)')
     postproc.add_option('-k', '--keep-video', action='store_true', dest='keepvideo', default=False,
     postproc.add_option('-k', '--keep-video', action='store_true', dest='keepvideo', default=False,
             help='keeps the video file on disk after the post-processing; the video is erased by default')
             help='keeps the video file on disk after the post-processing; the video is erased by default')
+    postproc.add_option('--no-post-overwrites', action='store_true', dest='nopostoverwrites', default=False,
+            help='do not overwrite post-processed files; the post-processed files are overwritten by default')
 
 
 
 
     parser.add_option_group(general)
     parser.add_option_group(general)
@@ -571,7 +573,7 @@ def _real_main():
 
 
     # PostProcessors
     # PostProcessors
     if opts.extractaudio:
     if opts.extractaudio:
-        fd.add_post_processor(FFmpegExtractAudioPP(preferredcodec=opts.audioformat, preferredquality=opts.audioquality, keepvideo=opts.keepvideo))
+        fd.add_post_processor(FFmpegExtractAudioPP(preferredcodec=opts.audioformat, preferredquality=opts.audioquality, keepvideo=opts.keepvideo, nopostoverwrites=opts.nopostoverwrites))
 
 
     # Update version
     # Update version
     if opts.update_self:
     if opts.update_self: