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