log warning, prevent crash, and exit gracefully if no manual directories defined.

This commit is contained in:
clinton-hall 2014-04-04 09:39:26 +10:30
commit 5165ee31ba
3 changed files with 14 additions and 11 deletions

View file

@ -467,14 +467,17 @@ def get_dirnames(section, category):
outputDirectory = ""
# set dirName
dirNames = None
dirNames = []
if watch_dir != "":
if os.path.exists(watch_dir):
dirNames = [os.path.join(watch_dir, o) for o in os.listdir(watch_dir) if
os.path.isdir(os.path.join(watch_dir, o))]
elif outputDirectory != "":
dirNames.extend([os.path.join(watch_dir, o) for o in os.listdir(watch_dir) if
os.path.isdir(os.path.join(watch_dir, o))])
if outputDirectory != "":
if os.path.exists(outputDirectory):
dirNames = [os.path.join(outputDirectory, o) for o in os.listdir(outputDirectory) if
os.path.isdir(os.path.join(outputDirectory, o))]
dirNames.extend([os.path.join(outputDirectory, o) for o in os.listdir(outputDirectory) if
os.path.isdir(os.path.join(outputDirectory, o))])
if not dirNames:
Logger.warn("No Directories identified to Scan.")
return dirNames

View file

@ -414,7 +414,7 @@ if inputCategory in cpsCategory:
Logger.info("MAIN: Calling CouchPotatoServer to post-process: %s", inputName)
result = autoProcessMovie.process(nzbDir, inputName, status, clientAgent, download_id, inputCategory)
elif inputCategory in sbCategory:
result = 0
result = 1
if isinstance(nzbDir, list):
for dirName in nzbDir:
Logger.info("MAIN: Calling Sick-Beard to post-process: %s", inputName)

View file

@ -229,13 +229,13 @@ elif len(sys.argv) >= SABNZB_0717_NO_OF_ARGUMENTS:
clientAgent = "sabnzbd"
result = autoProcessTV.processEpisode(sys.argv[1], sys.argv[2], sys.argv[7], clientAgent, sys.argv[5])
else:
result = 0
sbCategory = (config().get("SickBeard", "sbCategory")).split(',') # tv
dirNames = get_dirnames("SickBeard", sbCategory[0])
Logger.debug("MAIN: Invalid number of arguments received from client.")
Logger.info("MAIN: Running autoProcessTV as a manual run...")
result = 1
sbCategory = (config().get("SickBeard", "sbCategory")).split(',') # tv
dirNames = get_dirnames("SickBeard", sbCategory[0])
for dirName in dirNames:
Logger.info("MAIN: Calling Sick-Beard to post-process: %s", dirName)
result = autoProcessTV.processEpisode(dirName, dirName, 0)