diff --git a/TorrentToMedia.py b/TorrentToMedia.py index b6f0b25f..322ee7bc 100755 --- a/TorrentToMedia.py +++ b/TorrentToMedia.py @@ -39,15 +39,27 @@ def main(inputDirectory, inputName, inputCategory, inputHash, inputID): utorrentClass = "" TransmissionClass = "" - # init autoFork - fork, fork_params = autoFork(inputCategory) - Logger.debug("MAIN: Received Directory: %s | Name: %s | Category: %s", inputDirectory, inputName, inputCategory) inputDirectory, inputName, inputCategory, root = category_search(inputDirectory, inputName, inputCategory, root, categories) # Confirm the category by parsing directory structure Logger.debug("MAIN: Determined Directory: %s | Name: %s | Category: %s", inputDirectory, inputName, inputCategory) + # init autoFork + if inputCategory in sbCategory: + fork, fork_params = autoFork("SickBeard") + elif inputCategory in cpsCategory: + fork, fork_params = autoFork("CouchPotato") + elif inputCategory in hpCategory: + fork, fork_params = autoFork("HeadPhones") + elif inputCategory in gzCategory: + fork, fork_params = autoFork("Gamez") + elif inputCategory in mlCategory: + fork, fork_params = autoFork("Mylar") + else: + fork = config.FORKS.items()[config.FORKS.keys().index(config.FORK_DEFAULT)][0] + fork_params = config.FORKS.items()[config.FORKS.keys().index(config.FORK_DEFAULT)][0] + if inputCategory in sbCategory: if fork in config.SICKBEARD_TORRENT and Torrent_ForceLink != 1: Logger.info("MAIN: Calling SickBeard's %s branch to post-process: %s",fork ,inputName) @@ -511,10 +523,10 @@ if __name__ == "__main__": Logger.debug("arg %s is: %s", n, arg) n = n+1 - if not len(sys.argv) > 1: + try: + inputDirectory, inputName, inputCategory, inputHash, inputID = parse_args(clientAgent) + except: Logger.exception("MAIN: There was a problem loading variables") sys.exit(-1) - # process torrent - inputDirectory, inputName, inputCategory, inputHash, inputID = parse_args(clientAgent) main(inputDirectory, inputName, inputCategory, inputHash, inputID) diff --git a/nzbtomedia/nzbToMediaAutoFork.py b/nzbtomedia/nzbToMediaAutoFork.py index 5d5e621d..6c29e381 100644 --- a/nzbtomedia/nzbToMediaAutoFork.py +++ b/nzbtomedia/nzbToMediaAutoFork.py @@ -10,8 +10,13 @@ def autoFork(section): # config settings host = config().get(section, "host") port = config().get(section, "port") - username = config().get(section, "username") - password = config().get(section, "password") + + try: + username = config().get(section, "username") + password = config().get(section, "password") + except: + username = None + password = None try: ssl = int(config().get(section, "ssl")) @@ -41,7 +46,10 @@ def autoFork(section): # attempting to auto-detect fork try: - r = requests.get(url, auth=(username, password)) + if username and password: + r = requests.get(url, auth=(username, password)) + else: + r = requests.get(url) except requests.ConnectionError: Logger.info("Could not connect to SickBeard to perform auto-fork detection!") break