nzbToMedia/nzbtomedia/transcoder/transcoder.py

import errno
import os
import platform
import subprocess
import urllib2
import traceback
import nzbtomedia
from subprocess import call
from nzbtomedia import logger
from nzbtomedia.extractor import extractor

def isVideoGood(videofile):
    fileNameExt = os.path.basename(videofile)
    fileName, fileExt = os.path.splitext(fileNameExt)
    if fileExt not in nzbtomedia.MEDIACONTAINER:
        return True

    if platform.system() == 'Windows':
        bitbucket = open('NUL')
    else:
        bitbucket = open('/dev/null')

    if not nzbtomedia.FFPROBE:
        return True

    command = [nzbtomedia.FFPROBE, videofile]
    try:
        logger.info('Checking [%s] for corruption, please stand by ...' % (fileNameExt), 'TRANSCODER')
        result = call(command, stdout=bitbucket, stderr=bitbucket)
    except:
        logger.error("Checking [%s] for corruption has failed" % (fileNameExt), 'TRANSCODER')
        return False

    if result == 0:
        logger.info("SUCCESS: [%s] has no corruption." % (fileNameExt), 'TRANSCODER')
        return True
    else:
        logger.error("FAILED: [%s] is corrupted!" % (fileNameExt), 'TRANSCODER')
        return False

def Transcode_directory(dirName):
    if platform.system() == 'Windows':
        bitbucket = open('NUL')
    else:
        bitbucket = open('/dev/null')

    if not nzbtomedia.FFMPEG:
        return 1

    logger.info("Checking for files to be transcoded")
    final_result = 0 # initialize as successful
    for file in nzbtomedia.listMediaFiles(dirName):
        name, ext = os.path.splitext(file)
        if ext in [i.replace(".","") for i in nzbtomedia.MEDIACONTAINER]:
            continue
        if ext == nzbtomedia.OUTPUTVIDEOEXTENSION: # we need to change the name to prevent overwriting itself.
            nzbtomedia.OUTPUTVIDEOEXTENSION = '-transcoded' + nzbtomedia.OUTPUTVIDEOEXTENSION # adds '-transcoded.ext'

        newfilePath = os.path.normpath(name + nzbtomedia.OUTPUTVIDEOEXTENSION)

        command = [nzbtomedia.FFMPEG, '-loglevel', 'warning', '-i', file, '-map', '0'] # -map 0 takes all input streams
        if platform.system() != 'Windows':
            command = ['nice', '-%d' % nzbtomedia.NICENESS] + command

        if len(nzbtomedia.OUTPUTVIDEOCODEC) > 0:
            command.append('-c:v')
            command.append(nzbtomedia.OUTPUTVIDEOCODEC)
            if nzbtomedia.OUTPUTVIDEOCODEC == 'libx264' and nzbtomedia.OUTPUTVIDEOPRESET:
                command.append('-pre')
                command.append(nzbtomedia.OUTPUTVIDEOPRESET)
        else:
            command.append('-c:v')
            command.append('copy')
        if len(nzbtomedia.OUTPUTVIDEOFRAMERATE) > 0:
            command.append('-r')
            command.append(str(nzbtomedia.OUTPUTVIDEOFRAMERATE))
        if len(nzbtomedia.OUTPUTVIDEOBITRATE) > 0:
            command.append('-b:v')
            command.append(str(nzbtomedia.OUTPUTVIDEOBITRATE))
        if len(nzbtomedia.OUTPUTAUDIOCODEC) > 0:
            command.append('-c:a')
            command.append(nzbtomedia.OUTPUTAUDIOCODEC)
            if nzbtomedia.OUTPUTAUDIOCODEC == 'aac': # Allow users to use the experimental AAC codec that's built into recent versions of ffmpeg
                command.append('-strict')
                command.append('-2')
        else:
            command.append('-c:a')
            command.append('copy')
        if len(nzbtomedia.OUTPUTAUDIOBITRATE) > 0:
            command.append('-b:a')
            command.append(str(nzbtomedia.OUTPUTAUDIOBITRATE))
        if nzbtomedia.OUTPUTFASTSTART > 0:
            command.append('-movflags')
            command.append('+faststart')
        if nzbtomedia.OUTPUTQUALITYPERCENT > 0:
            command.append('-q:a')
            command.append(str(nzbtomedia.OUTPUTQUALITYPERCENT))
        if len(nzbtomedia.OUTPUTSUBTITLECODEC) > 0: # Not every subtitle codec can be used for every video container format!
            command.append('-c:s')
            command.append(nzbtomedia.OUTPUTSUBTITLECODEC) # http://en.wikibooks.org/wiki/FFMPEG_An_Intermediate_Guide/subtitle_options
        else:
            command.append('-sn')  # Don't copy the subtitles over
        command.append(newfilePath)

        try: # Try to remove the file that we're transcoding to just in case. (ffmpeg will return an error if it already exists for some reason)
            os.remove(newfilePath)
        except OSError, e:
            if e.errno != errno.ENOENT: # Ignore the error if it's just telling us that the file doesn't exist
                logger.debug("Error when removing transcoding target: %s" % (e))
        except Exception, e:
            logger.debug("Error when removing transcoding target: %s" % (e))

        logger.info("Transcoding video: %s" % (file))
        cmd = ""
        for item in command:
            cmd = cmd + " " + item
        logger.debug("calling command:%s" % (cmd))
        result = 1 # set result to failed in case call fails.
        try:
            result = call(command, stdout=bitbucket, stderr=bitbucket)
        except:
            logger.error("Transcoding of video %s has failed" % (file))

        if result == 0:
            logger.info("Transcoding of video %s to %s succeeded" % (file, newfilePath))
            if nzbtomedia.DUPLICATE == 0: # we get rid of the original file
                os.unlink(file)
        else:
            logger.error("Transcoding of video %s to %s failed" % (file, newfilePath))
        # this will be 0 (successful) it all are successful, else will return a positive integer for failure.
        final_result = final_result + result
    return final_result