mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-07-16 02:02:53 -07:00
eb7822b60b
Logging of debug messages is now optional via log_debug option location in autoProcessMedia.cfg Lots of code cleanup has been performed including cleanup log messages and corrections of spelling errors. Improved release lookup code for autoProcessMovie, narrows search results down by making API calls to the download clients to compare results in CouchPotato's database.
121 lines
6.6 KiB
Python
121 lines
6.6 KiB
Python
import errno
|
|
import os
|
|
import sys
|
|
import nzbtomedia
|
|
from subprocess import call
|
|
from nzbtomedia import logger
|
|
|
|
class Transcoder:
|
|
def Transcode_directory(self, dirName):
|
|
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
|
|
|
|
niceness = None
|
|
if useNiceness:niceness = nzbtomedia.NICENESS
|
|
|
|
map(lambda ext: ext.strip(), nzbtomedia.MEDIACONTAINER)
|
|
map(lambda ext: ext.strip(), nzbtomedia.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 nzbtomedia.MEDIACONTAINER: # If the file is a video file
|
|
if ext in nzbtomedia.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(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)
|
|
except:
|
|
logger.error("Transcoding of video %s has failed" % (filePath))
|
|
if result == 0:
|
|
logger.info("Transcoding of video %s to %s succeeded" % (filePath, newfilePath))
|
|
if nzbtomedia.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
|