import sys import urllib import os.path import ConfigParser class AuthURLOpener(urllib.FancyURLopener): def __init__(self, user, pw): self.username = user self.password = pw self.numTries = 0 urllib.FancyURLopener.__init__(self) def prompt_user_passwd(self, host, realm): if self.numTries == 0: self.numTries = 1 return (self.username, self.password) else: return ('', '') def openit(self, url): self.numTries = 0 return urllib.FancyURLopener.open(self, url) def processEpisode(dirName, nzbName=None): config = ConfigParser.ConfigParser() configFilename = os.path.join(os.path.dirname(sys.argv[0]), "autoProcessComics.cfg") print "Loading config from", configFilename if not os.path.isfile(configFilename): print "ERROR: You need an autoProcessComics.cfg file - did you rename and edit the .sample?" sys.exit(-1) try: fp = open(configFilename, "r") config.readfp(fp) fp.close() except IOError, e: print "Could not read configuration file: ", str(e) sys.exit(1) host = config.get("Mylar", "host") port = config.get("Mylar", "port") username = config.get("Mylar", "username") password = config.get("Mylar", "password") try: ssl = int(config.get("Mylar", "ssl")) except (ConfigParser.NoOptionError, ValueError): ssl = 0 try: web_root = config.get("Mylar", "web_root") except ConfigParser.NoOptionError: web_root = "" params = {} params['nzb_folder'] = dirName if nzbName != None: params['nzb_name'] = nzbName myOpener = AuthURLOpener(username, password) if ssl: protocol = "https://" else: protocol = "http://" url = protocol + host + ":" + port + web_root + "/post_process?" + urllib.urlencode(params) print "Opening URL:", url try: urlObj = myOpener.openit(url) except IOError, e: print "Unable to open URL: ", str(e) sys.exit(1) result = urlObj.readlines() for line in result: print line if any("Post Processing SUCCESSFULL" in s for s in result): return 0 else: return 1