mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-07-31 03:50:06 -07:00
5c593b2d15
Turns out "failed" is passed as a string and then handled by the failed fork branch. Status here is used to determine what the actual state is, since "failed" will always be True. This script now treats the failed branch the same as the autoProcessTV from that branch.
139 lines
4.4 KiB
Python
139 lines
4.4 KiB
Python
# Author: Nic Wolfe <nic@wolfeden.ca>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# This file is part of Sick Beard.
|
|
#
|
|
# Sick Beard is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Sick Beard is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Sick Beard. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
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, failed=False):
|
|
|
|
status = int(failed)
|
|
config = ConfigParser.ConfigParser()
|
|
configFilename = os.path.join(os.path.dirname(sys.argv[0]), "autoProcessTV.cfg")
|
|
print "Loading config from", configFilename
|
|
|
|
if not os.path.isfile(configFilename):
|
|
print "ERROR: You need an autoProcessTV.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)
|
|
|
|
watch_dir = ""
|
|
host = config.get("SickBeard", "host")
|
|
port = config.get("SickBeard", "port")
|
|
username = config.get("SickBeard", "username")
|
|
password = config.get("SickBeard", "password")
|
|
try:
|
|
ssl = int(config.get("SickBeard", "ssl"))
|
|
except (ConfigParser.NoOptionError, ValueError):
|
|
ssl = 0
|
|
|
|
try:
|
|
web_root = config.get("SickBeard", "web_root")
|
|
except ConfigParser.NoOptionError:
|
|
web_root = ""
|
|
|
|
try:
|
|
watch_dir = config.get("SickBeard", "watch_dir")
|
|
except ConfigParser.NoOptionError:
|
|
watch_dir = ""
|
|
|
|
try:
|
|
failed_fork = int(config.get("SickBeard", "failed_fork"))
|
|
except (ConfigParser.NoOptionError, ValueError):
|
|
failed_fork = 0
|
|
|
|
#allows us to specify the default watch directory and call the postproecssing on another PC with different directory structure.
|
|
if watch_dir != "":
|
|
dirName = watch_dir
|
|
|
|
params = {}
|
|
|
|
params['quiet'] = 1
|
|
|
|
# if you have specified you are using development branch from fork https://github.com/Tolstyak/Sick-Beard.git
|
|
if failed_fork:
|
|
params['dirName'] = dirName
|
|
if nzbName != None:
|
|
params['nzbName'] = nzbName
|
|
params['failed'] = failed
|
|
if status:
|
|
print "The download failed. Sending 'failed' process request to SickBeard's failed branch"
|
|
else:
|
|
print "The download succeeded. Sending process request to SickBeard's failed branch"
|
|
|
|
# this is our default behaviour to work with the standard Master branch of SickBeard
|
|
else:
|
|
params['dir'] = dirName
|
|
if nzbName != None:
|
|
params['nzbName'] = nzbName
|
|
# the standard Master bamch of SickBeard cannot process failed downloads. So Exit here.
|
|
if status:
|
|
print "The download failed. Nothing to process"
|
|
sys.exit()
|
|
else:
|
|
print "The download succeeded. Sending process request to SickBeard"
|
|
|
|
myOpener = AuthURLOpener(username, password)
|
|
|
|
if ssl:
|
|
protocol = "https://"
|
|
else:
|
|
protocol = "http://"
|
|
|
|
url = protocol + host + ":" + port + web_root + "/home/postprocess/processEpisode?" + 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
|
|
|