nzbToMedia/nzbtomedia/Transcoder.py

import errno
import os
import platform
import nzbtomedia
from subprocess import call, Popen
from lib import guessit
from nzbtomedia import logger
from nzbToMediaUtil import listMediaFiles

class Transcoder:
    def isVideoGood(self, videofile):
        if platform.system() == 'Windows':
            bitbucket = open('NUL')
        else:
            bitbucket = open('/dev/null')

        if not nzbtomedia.FFPROBE:
            logger.error("Cannot detect corrupt video files!, set your ffmpeg_path in your autoProcessMedia.cfg ...")
            return False

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

        if result == 0:
            logger.info("Video %s has no corruption." % (videofile))
            return True
        else:
            logger.error("Video %s is corrupted!" % (videofile))
            return False

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

        if not nzbtomedia.FFMPEG:
            logger.error("Cannot transcode files!, set your ffmpeg_path in your autoProcessMedia.cfg ...")
            return 1

        logger.info("Checking for files to be transcoded")
        final_result = 0 # initialize as successful
        for file in 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('-preset')
                    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