From eeee92c8302e8e86ff14bfa6c0491ea91c1d9567 Mon Sep 17 00:00:00 2001 From: echel0n Date: Sun, 13 Apr 2014 13:20:39 -0700 Subject: [PATCH] Fixed bugs in autoProcessMusic and autoProcessMovie --- nzbtomedia/autoProcess/autoProcessMovie.py | 4 ++++ nzbtomedia/autoProcess/autoProcessMusic.py | 7 ++++++- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/nzbtomedia/autoProcess/autoProcessMovie.py b/nzbtomedia/autoProcess/autoProcessMovie.py index 6002c41a..6d79f197 100644 --- a/nzbtomedia/autoProcess/autoProcessMovie.py +++ b/nzbtomedia/autoProcess/autoProcessMovie.py @@ -172,6 +172,10 @@ class autoProcessMovie: media_id, download_id, release_id, release_status = self.find_media_id(baseURL, download_id, dirName, nzbName) # get the CPS database movie id for this movie. + if release_status is None: + logger.error("Could not find a current status for %s on CouchPotatoServer", nzbName) + return 1 + # failed to get a download id if release_status != "snatched": logger.postprocess("%s has is marked with a status of [%s] by CouchPotatoServer, skipping ...", nzbName, release_status.upper()) diff --git a/nzbtomedia/autoProcess/autoProcessMusic.py b/nzbtomedia/autoProcess/autoProcessMusic.py index 6ba3b2ef..d74011f9 100644 --- a/nzbtomedia/autoProcess/autoProcessMusic.py +++ b/nzbtomedia/autoProcess/autoProcessMusic.py @@ -88,9 +88,14 @@ class autoProcessMusic: url = baseURL release_status = self.get_status(url, apikey, dirName) + + if release_status is None: + logger.error("Could not find a current status for %s on HeadPhones", nzbName) + return 1 + if release_status != "Unprocessed": logger.error("%s is marked with a status of %s on HeadPhones, skipping ...", nzbName, release_status) - return 1 + return 0 logger.debug("Opening URL: %s", url)