photofloat/scanner/VideoToolWrapper.py

from CachePath import message
import os
import subprocess


class VideoToolWrapper(object):
    def call(self, *args):
        path = args[-1]
        for tool in self.wrappers:
            try:
                if self.check_output:
                    p = subprocess.check_output((tool,) + args)
                else:
                    p = subprocess.call((tool,) + args)
                    if p > 0:
                        return False
                    else:
                        return "SUCCESS"
            except KeyboardInterrupt:
                if self.cleanup:
                    self.remove(path)
                raise
            except OSError:
                continue
            except:
                if self.cleanup:
                    self.remove(path)
                continue
            return p
        return False

    def remove(self, path):
        try:
            os.unlink(path)
        except:
            pass


class VideoTranscodeWrapper(VideoToolWrapper):
    def __init__(self):
        self.wrappers = ['avconv', 'ffmpeg']
        self.check_output = False
        self.cleanup = True


class VideoProbeWrapper(VideoToolWrapper):
    def __init__(self):
        self.wrappers = ['avprobe', 'ffprobe']
        self.check_output = True
        self.cleanup = False