nzbToMedia/nzbtomedia/Transcoder.py

import errno
import logging
import os
import sys
from subprocess import call
from nzbtomedia.nzbToMediaConfig import config


class Transcoder:
    def Transcode_directory(self, dirName):
        Logger = logging.getLogger()

        if os.name == 'nt':
            ffmpeg = os.path.join(os.path.dirname(sys.argv[0]), 'ffmpeg\\bin\\ffmpeg.exe') # note, will need to package in this dir.
            useNiceness = False
            if not os.path.isfile(ffmpeg): # problem
                Logger.error("ffmpeg not found. ffmpeg needs to be located at: %s", ffmpeg)
                Logger.info("Cannot transcode files in folder %s", dirName)
                return 1 # failure
        else:
            if call(['which', 'ffmpeg']) != 0:
                res = call([os.path.join(os.path.dirname(sys.argv[0]),'getffmpeg.sh')])
                if res or call(['which', 'ffmpeg']) != 0: # did not install or ffmpeg still not found.
                    Logger.error("Failed to install ffmpeg. Please install manually")
                    Logger.info("Cannot transcode files in folder %s", dirName)
                    return 1 # failure
                else:
                    ffmpeg = 'ffmpeg'
            else:
                ffmpeg = 'ffmpeg'
            useNiceness = True

        Logger.info("Loading config from %s", config.CONFIG_FILE)

        if not config():
            Logger.error("You need an autoProcessMedia.cfg file - did you rename and edit the .sample?")
            return 1 # failure

        mediaContainer = (config().get("Extensions", "mediaExtensions")).split(',')
        duplicate = int(config().get("Transcoder", "duplicate"))
        ignoreExtensions = (config().get("Transcoder", "ignoreExtensions")).split(',')
        outputVideoExtension = config().get("Transcoder", "outputVideoExtension").strip()
        outputVideoCodec = config().get("Transcoder", "outputVideoCodec").strip()
        outputVideoPreset = config().get("Transcoder", "outputVideoPreset").strip()
        outputVideoFramerate = config().get("Transcoder", "outputVideoFramerate").strip()
        outputVideoBitrate = config().get("Transcoder", "outputVideoBitrate").strip()
        outputAudioCodec = config().get("Transcoder", "outputAudioCodec").strip()
        outputAudioBitrate = config().get("Transcoder", "outputAudioBitrate").strip()
        outputSubtitleCodec = config().get("Transcoder", "outputSubtitleCodec").strip()
        outputFastStart = int(config().get("Transcoder", "outputFastStart"))
        outputQualityPercent = int(config().get("Transcoder", "outputQualityPercent"))

        niceness = None
        if useNiceness:niceness = int(config().get("Transcoder", "niceness"))

        map(lambda ext: ext.strip(), mediaContainer)
        map(lambda ext: ext.strip(), ignoreExtensions)

        Logger.info("Checking for files to be transcoded")
        final_result = 0 # initialize as successful
        for dirpath, dirnames, filenames in os.walk(dirName):
            for file in filenames:
                filePath = os.path.join(dirpath, file)
                name, ext = os.path.splitext(filePath)
                if ext in mediaContainer:  # If the file is a video file
                    if ext in ignoreExtensions:
                        Logger.info("No need to transcode video type %s", ext)
                        continue
                    if ext == outputVideoExtension: # we need to change the name to prevent overwriting itself.
                        outputVideoExtension = '-transcoded' + outputVideoExtension # adds '-transcoded.ext'
                    newfilePath = os.path.normpath(name + outputVideoExtension)

                    command = [ffmpeg, '-loglevel', 'warning', '-i', filePath, '-map', '0'] # -map 0 takes all input streams

                    if useNiceness:
                        command = ['nice', '-%d' % niceness] + command

                    if len(outputVideoCodec) > 0:
                        command.append('-c:v')
                        command.append(outputVideoCodec)
                        if outputVideoCodec == 'libx264' and outputVideoPreset:
                            command.append('-preset')
                            command.append(outputVideoPreset)
                    else:
                        command.append('-c:v')
                        command.append('copy')
                    if len(outputVideoFramerate) > 0:
                        command.append('-r')
                        command.append(str(outputVideoFramerate))
                    if len(outputVideoBitrate) > 0:
                        command.append('-b:v')
                        command.append(str(outputVideoBitrate))
                    if len(outputAudioCodec) > 0:
                        command.append('-c:a')
                        command.append(outputAudioCodec)
                        if 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(outputAudioBitrate) > 0:
                        command.append('-b:a')
                        command.append(str(outputAudioBitrate))
                    if outputFastStart > 0:
                        command.append('-movflags')
                        command.append('+faststart')
                    if outputQualityPercent > 0:
                        command.append('-q:a')
                        command.append(str(outputQualityPercent))
                    if len(outputSubtitleCodec) > 0: # Not every subtitle codec can be used for every video container format!
                        command.append('-c:s')
                        command.append(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)
                    except:
                        Logger.exception("Transcoding of video %s has failed", filePath)
                    if result == 0:
                        Logger.info("Transcoding of video %s to %s succeeded", filePath, newfilePath)
                        if duplicate == 0: # we get rid of the original file
                            os.unlink(filePath)
                    else:
                        Logger.error("Transcoding of video %s to %s failed", filePath, 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