diff options
Diffstat (limited to 'youtube_dl/postprocessor')
| -rw-r--r-- | youtube_dl/postprocessor/__init__.py | 2 | ||||
| -rw-r--r-- | youtube_dl/postprocessor/execafterdownload.py | 31 | ||||
| -rw-r--r-- | youtube_dl/postprocessor/ffmpeg.py | 8 | 
3 files changed, 39 insertions, 2 deletions
diff --git a/youtube_dl/postprocessor/__init__.py b/youtube_dl/postprocessor/__init__.py index 08e6ddd00..15aa0daa9 100644 --- a/youtube_dl/postprocessor/__init__.py +++ b/youtube_dl/postprocessor/__init__.py @@ -9,6 +9,7 @@ from .ffmpeg import (      FFmpegEmbedSubtitlePP,  )  from .xattrpp import XAttrMetadataPP +from .execafterdownload import ExecAfterDownloadPP  __all__ = [      'AtomicParsleyPP', @@ -19,4 +20,5 @@ __all__ = [      'FFmpegExtractAudioPP',      'FFmpegEmbedSubtitlePP',      'XAttrMetadataPP', +    'ExecAfterDownloadPP',  ] diff --git a/youtube_dl/postprocessor/execafterdownload.py b/youtube_dl/postprocessor/execafterdownload.py new file mode 100644 index 000000000..08419a3d4 --- /dev/null +++ b/youtube_dl/postprocessor/execafterdownload.py @@ -0,0 +1,31 @@ +from __future__ import unicode_literals + +import subprocess + +from .common import PostProcessor +from ..utils import ( +    shlex_quote, +    PostProcessingError, +) + + +class ExecAfterDownloadPP(PostProcessor): +    def __init__(self, downloader=None, verboseOutput=None, exec_cmd=None): +        self.verboseOutput = verboseOutput +        self.exec_cmd = exec_cmd + +    def run(self, information): +        cmd = self.exec_cmd +        if not '{}' in cmd: +            cmd += ' {}' + +        cmd = cmd.replace('{}', shlex_quote(information['filepath'])) + +        self._downloader.to_screen("[exec] Executing command: %s" % cmd) +        retCode = subprocess.call(cmd, shell=True) +        if retCode != 0: +            raise PostProcessingError( +                'Command returned error code %d' % retCode) + +        return None, information  # by default, keep file and do nothing + diff --git a/youtube_dl/postprocessor/ffmpeg.py b/youtube_dl/postprocessor/ffmpeg.py index 45328ed43..8c5f7c43b 100644 --- a/youtube_dl/postprocessor/ffmpeg.py +++ b/youtube_dl/postprocessor/ffmpeg.py @@ -18,14 +18,15 @@ from ..utils import (  ) -  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 ipath in input_paths: +                os.remove(ipath)      def run_ffmpeg(self, path, out_path, opts):          self.run_ffmpeg_multiple_files([path], out_path, opts)  | 
