SABnzbd Forums

Code: Select all

#!/usr/bin/env python

# A SABnzbd post-processing script to check if files with video format extensions
# indeed contain video format.
# The tool 'ffprobe' is needed. On Ubuntu it's part of ffmpeg

import sys
import os
from os import walk

if len(sys.argv) < 2:
    sys.exit('Usage: %s directory-name' % sys.argv[0])

dirname = sys.argv[1]

if not os.path.exists(dirname):
    sys.exit('ERROR: Directory not found' % sys.argv[1])

if os.popen('ffprobe -version 2>&1 ').readline().find('ffprobe version') < 0 :
	sys.exit('ffprobe not found. Install it first.')


# Extensions to be scanned (note the dot at the beginning):
extlist = [ '.avi', '.mpg', '.mpeg', '.divx', '.xvid' ]

for path, subdirs, files in os.walk(dirname):
    for file in files:
        extension = os.path.splitext(file)[1].lower()
        if extension in extlist:
                fullfilename = os.path.join(path, file)
		cmd = 'ffprobe ' + fullfilename + ' 2>&1 '
		for thisline in os.popen(cmd).readlines():
			# Invalid data found when processing input ... is not good
			# Metadata ... is good
			if thisline.find('Invalid data') >= 0:
				exitmessage = "No valid video format in " + file
				sys.exit(exitmessage)

# All well
sys.exit(0)