Added in some missing code

This commit is contained in:
echel0n 2014-04-19 16:54:43 -07:00
commit 00a5dae4e3
2 changed files with 7 additions and 9 deletions

View file

@ -290,21 +290,19 @@ def process(nzbDir, inputName=None, status=0, clientAgent='manual', download_id=
# auto-detect section # auto-detect section
section = nzbtomedia.CFG.findsection(inputCategory) section = nzbtomedia.CFG.findsection(inputCategory)
if section: if section:
# Check video files for corruption
for video in listMediaFiles(nzbDir):
if not nzbtomedia.FFPROBE and Transcoder().isVideoGood(video):
status = 1
logger.info("Sending %s to %s for post-processing ..." % (inputName, str(section).upper())) logger.info("Sending %s to %s for post-processing ..." % (inputName, str(section).upper()))
else: else:
logger.error("We could not find a section with containing a download category labeled %s in your autoProcessMedia.cfg, Exiting!" % inputCategory) logger.error("We could not find a section with containing a download category labeled %s in your autoProcessMedia.cfg, Exiting!" % inputCategory)
return -1
if nzbtomedia.CFG["CouchPotato"][inputCategory]: if nzbtomedia.CFG["CouchPotato"][inputCategory]:
# Check video files for corruption
for video in listMediaFiles(nzbDir):
if not nzbtomedia.FFPROBE and Transcoder().isVideoGood(video):
status = 1
result = autoProcessMovie().process(nzbDir, inputName, status, clientAgent, download_id, inputCategory) result = autoProcessMovie().process(nzbDir, inputName, status, clientAgent, download_id, inputCategory)
elif nzbtomedia.CFG["SickBeard", "NzbDrone"][inputCategory]: elif nzbtomedia.CFG["SickBeard", "NzbDrone"][inputCategory]:
# Check video files for corruption
for video in listMediaFiles(nzbDir):
if not nzbtomedia.FFPROBE and Transcoder().isVideoGood(video):
status = 1
result = autoProcessTV().processEpisode(nzbDir, inputName, status, clientAgent, inputCategory) result = autoProcessTV().processEpisode(nzbDir, inputName, status, clientAgent, inputCategory)
elif nzbtomedia.CFG["HeadPhones"][inputCategory]: elif nzbtomedia.CFG["HeadPhones"][inputCategory]:
result = autoProcessMusic().process(nzbDir, inputName, status, clientAgent, inputCategory) result = autoProcessMusic().process(nzbDir, inputName, status, clientAgent, inputCategory)