mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-01 08:53:37 +00:00
151 lines
No EOL
4.5 KiB
Python
Executable file
151 lines
No EOL
4.5 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
# Author: Nic Wolfe <nic@wolfeden.ca>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# This file is part of SickGear.
|
|
#
|
|
# SickGear 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.
|
|
#
|
|
# SickGear 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 SickGear. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from __future__ import print_function
|
|
from __future__ import with_statement
|
|
|
|
import os.path
|
|
import sys
|
|
|
|
sickbeardPath = os.path.split(os.path.split(sys.argv[0])[0])[0]
|
|
sys.path.insert(1, os.path.join(sickbeardPath, 'lib'))
|
|
sys.path.insert(1, sickbeardPath)
|
|
|
|
try:
|
|
import requests
|
|
except ImportError:
|
|
print ('You need to install python requests library')
|
|
sys.exit(1)
|
|
|
|
|
|
# Try importing Python 2 modules using new names
|
|
try:
|
|
import ConfigParser as configparser
|
|
import urllib2
|
|
from urllib import urlencode
|
|
|
|
# On error import Python 3 modules
|
|
except ImportError:
|
|
import configparser
|
|
import urllib.request as urllib2
|
|
from urllib.parse import urlencode
|
|
|
|
def processEpisode(dir_to_process, org_NZB_name=None, status=None):
|
|
# Default values
|
|
host = 'localhost'
|
|
port = '8081'
|
|
username = ''
|
|
password = ''
|
|
ssl = 0
|
|
web_root = '/'
|
|
|
|
default_url = host + ':' + port + web_root
|
|
if ssl:
|
|
default_url = 'https://' + default_url
|
|
else:
|
|
default_url = 'http://' + default_url
|
|
|
|
# Get values from config_file
|
|
config = configparser.RawConfigParser()
|
|
config_filename = os.path.join(os.path.dirname(sys.argv[0]), 'autoProcessTV.cfg')
|
|
|
|
if not os.path.isfile(config_filename):
|
|
print ('ERROR: ' + config_filename + " doesn't exist")
|
|
print ('copy /rename ' + config_filename + '.sample and edit\n')
|
|
print ('Trying default url: ' + default_url + '\n')
|
|
|
|
else:
|
|
try:
|
|
print ('Loading config from ' + config_filename + '\n')
|
|
|
|
with open(config_filename, 'r') as fp:
|
|
config.readfp(fp)
|
|
|
|
# Replace default values with config_file values
|
|
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):
|
|
pass
|
|
|
|
try:
|
|
web_root = config.get('SickBeard', 'web_root')
|
|
if not web_root.startswith('/'):
|
|
web_root = '/' + web_root
|
|
|
|
if not web_root.endswith('/'):
|
|
web_root = web_root + '/'
|
|
|
|
except configparser.NoOptionError:
|
|
pass
|
|
|
|
except EnvironmentError:
|
|
e = sys.exc_info()[1]
|
|
print ('Could not read configuration file: ' + str(e))
|
|
# There was a config_file, don't use default values but exit
|
|
sys.exit(1)
|
|
|
|
params = {}
|
|
|
|
params['quiet'] = 1
|
|
|
|
params['dir'] = dir_to_process
|
|
if org_NZB_name != None:
|
|
params['nzbName'] = org_NZB_name
|
|
|
|
if status != None:
|
|
params['failed'] = status
|
|
|
|
if ssl:
|
|
protocol = 'https://'
|
|
else:
|
|
protocol = 'http://'
|
|
|
|
url = protocol + host + ':' + port + web_root + 'home/postprocess/processEpisode'
|
|
login_url = protocol + host + ':' + port + web_root + 'login'
|
|
|
|
print ('Opening URL: ' + url)
|
|
|
|
try:
|
|
sess = requests.Session()
|
|
sess.post(login_url, data={'username': username, 'password': password}, stream=True, verify=False)
|
|
result = sess.get(url, params=params, stream=True, verify=False)
|
|
if result.status_code == 401:
|
|
print('Verify and use correct username and password in autoProcessTV.cfg')
|
|
else:
|
|
for line in result.iter_lines():
|
|
if line:
|
|
print (line.strip())
|
|
|
|
except IOError:
|
|
e = sys.exc_info()[1]
|
|
print ('Unable to open URL: ' + str(e))
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print ('This module is supposed to be used as import in other scripts and not run standalone.')
|
|
print ('Use sabToSickBeard instead.')
|
|
sys.exit(1) |