mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-11 05:33:37 +00:00
43778d8edd
Add BitMeTV torrent provider. Add TVChaosUK torrent provider. Add HD-Space torrent provider. Add Shazbat torrent provider. Remove Animenzb provider. Change use tvdbid for searching usenet providers. Change consolidate global and per show ignore and require words functions. Change "Require word" title and notes on Config Search page to properly describe its functional logic. Add "Reject Blu-ray M2TS releases" to BTN provider. Add regular expression capability to ignore and require words by starting wordlist with "regex:". Add list shows with custom ignore and require words under the global counterparts on the Search Settings page. Fix failure to search for more than one selected wanted episode.
160 lines
4.4 KiB
Python
160 lines
4.4 KiB
Python
# Author: jkaberg <joel.kaberg@gmail.com>, based on fuzemans work (https://github.com/RuudBurger/CouchPotatoServer/blob/develop/couchpotato/core/downloaders/rtorrent/main.py)
|
|
# 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/>.
|
|
|
|
import sickbeard
|
|
from sickbeard.clients.generic import GenericClient
|
|
from lib.rtorrent import RTorrent
|
|
|
|
|
|
class rTorrentAPI(GenericClient):
|
|
def __init__(self, host=None, username=None, password=None):
|
|
super(rTorrentAPI, self).__init__('rTorrent', host, username, password)
|
|
|
|
def _get_auth(self):
|
|
self.auth = None
|
|
|
|
if self.auth is not None:
|
|
return self.auth
|
|
|
|
if not self.host:
|
|
return
|
|
|
|
if self.username and self.password:
|
|
self.auth = RTorrent(self.host, self.username, self.password)
|
|
else:
|
|
self.auth = RTorrent(self.host, None, None, True)
|
|
|
|
return self.auth
|
|
|
|
def _add_torrent_uri(self, result):
|
|
|
|
if not self.auth:
|
|
return False
|
|
|
|
if not result:
|
|
return False
|
|
|
|
try:
|
|
# Send magnet to rTorrent
|
|
torrent = self.auth.load_magnet(result.url, result.hash)
|
|
|
|
if not torrent:
|
|
return False
|
|
|
|
# Set label
|
|
if sickbeard.TORRENT_LABEL:
|
|
torrent.set_custom(1, sickbeard.TORRENT_LABEL)
|
|
|
|
if sickbeard.TORRENT_PATH:
|
|
torrent.set_directory(sickbeard.TORRENT_PATH)
|
|
|
|
# Start torrent
|
|
torrent.start()
|
|
|
|
return True
|
|
|
|
except:
|
|
return False
|
|
|
|
def _add_torrent_file(self, result):
|
|
|
|
if not self.auth:
|
|
return False
|
|
|
|
if not result:
|
|
return False
|
|
|
|
# group_name = 'sb_test' ##### Use provider instead of _test
|
|
# if not self._set_torrent_ratio(group_name):
|
|
# return False
|
|
|
|
# Send request to rTorrent
|
|
try:
|
|
# Send torrent to rTorrent
|
|
torrent = self.auth.load_torrent(result.content)
|
|
|
|
if not torrent:
|
|
return False
|
|
|
|
# Set label
|
|
if sickbeard.TORRENT_LABEL:
|
|
torrent.set_custom(1, sickbeard.TORRENT_LABEL)
|
|
|
|
if sickbeard.TORRENT_PATH:
|
|
torrent.set_directory(sickbeard.TORRENT_PATH)
|
|
|
|
# Set Ratio Group
|
|
# torrent.set_visible(group_name)
|
|
|
|
# Start torrent
|
|
torrent.start()
|
|
|
|
return True
|
|
|
|
except:
|
|
return False
|
|
|
|
def _set_torrent_ratio(self, name):
|
|
|
|
# if not name:
|
|
# return False
|
|
#
|
|
# if not self.auth:
|
|
# return False
|
|
#
|
|
# views = self.auth.get_views()
|
|
#
|
|
# if name not in views:
|
|
# self.auth.create_group(name)
|
|
|
|
# group = self.auth.get_group(name)
|
|
|
|
# ratio = int(float(sickbeard.TORRENT_RATIO) * 100)
|
|
#
|
|
# try:
|
|
# if ratio > 0:
|
|
#
|
|
# # Explicitly set all group options to ensure it is setup correctly
|
|
# group.set_upload('1M')
|
|
# group.set_min(ratio)
|
|
# group.set_max(ratio)
|
|
# group.set_command('d.stop')
|
|
# group.enable()
|
|
# else:
|
|
# # Reset group action and disable it
|
|
# group.set_command()
|
|
# group.disable()
|
|
#
|
|
# except:
|
|
# return False
|
|
|
|
return True
|
|
|
|
def testAuthentication(self):
|
|
try:
|
|
self._get_auth()
|
|
|
|
if self.auth is not None:
|
|
return True, 'Success: Connected and Authenticated'
|
|
else:
|
|
return False, 'Error: Unable to get ' + self.name + ' Authentication, check your config!'
|
|
except Exception:
|
|
return False, 'Error: Unable to connect to ' + self.name
|
|
|
|
|
|
api = rTorrentAPI()
|