def _get_auth(self):
self.auth = None
if self.auth is not None:
return self.auth
if not self.host:
return
tp_kwargs = {}
if sickrage.app.config.torrent_auth_type.lower() != 'none':
tp_kwargs['authtype'] = sickrage.app.config.torrent_auth_type
if not sickrage.app.config.torrent_verify_cert:
tp_kwargs['check_ssl_cert'] = False
if self.username and self.password:
self.auth = RTorrent(self.host,
self.username,
self.password,
True,
tp_kwargs=tp_kwargs)
else:
self.auth = RTorrent(self.host, None, None, True)
return self.auth
def _get_auth(self):
self.auth = None
if self.auth is not None:
return self.auth
if not self.host:
return
tp_kwargs = {}
if sickrage.app.config.torrent_auth_type.lower() != 'none':
tp_kwargs['authtype'] = sickrage.app.config.torrent_auth_type
if not sickrage.app.config.torrent_verify_cert:
tp_kwargs['check_ssl_cert'] = False
if self.username and self.password:
self.auth = RTorrent(self.host, self.username, self.password, True, tp_kwargs=tp_kwargs)
else:
self.auth = RTorrent(self.host, None, None, True)
return self.auth
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
tp_kwargs = {}
if sickrage.app.config.torrent_auth_type.lower() != 'none':
tp_kwargs['authtype'] = sickrage.app.config.torrent_auth_type
if not sickrage.app.config.torrent_verify_cert:
tp_kwargs['check_ssl_cert'] = False
if self.username and self.password:
self.auth = RTorrent(self.host,
self.username,
self.password,
True,
tp_kwargs=tp_kwargs)
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
label = sickrage.app.config.torrent_label
show_object = find_show(result.show_id)
if show_object.is_anime:
label = sickrage.app.config.torrent_label_anime
if label:
torrent.set_custom(1, label)
if sickrage.app.config.torrent_path:
torrent.set_directory(sickrage.app.config.torrent_path)
# Start torrent
torrent.start()
return True
except Exception:
sickrage.app.log.debug(traceback.format_exc())
return False
def _add_torrent_file(self, result):
if not self.auth:
return False
if not result:
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
label = sickrage.app.config.torrent_label
show_object = find_show(result.show_id)
if show_object.is_anime:
label = sickrage.app.config.torrent_label_anime
if label:
torrent.set_custom(1, label)
if sickrage.app.config.torrent_path:
torrent.set_directory(sickrage.app.config.torrent_path)
# Set Ratio Group
# torrent.set_visible(group_name)
# Start torrent
torrent.start()
return True
except Exception:
sickrage.app.log.debug(traceback.format_exc())
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(sickrage.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 test_authentication(self):
try:
if self._get_auth():
return True, 'Success: Connected and Authenticated'
else:
return False, 'Error: Unable to get ' + self.name + ' Authentication, check your config!'
except Exception:
sickrage.app.log.debug(traceback.format_exc())
return False, 'Error: Unable to connect to ' + self.name
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
tp_kwargs = {}
if sickrage.app.config.torrent_auth_type.lower() != 'none':
tp_kwargs['authtype'] = sickrage.app.config.torrent_auth_type
if not sickrage.app.config.torrent_verify_cert:
tp_kwargs['check_ssl_cert'] = False
if self.username and self.password:
self.auth = RTorrent(self.host, self.username, self.password, True, tp_kwargs=tp_kwargs)
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
label = sickrage.app.config.torrent_label
if result.show.is_anime:
label = sickrage.app.config.torrent_label_anime
if label:
torrent.set_custom(1, label)
if sickrage.app.config.torrent_path:
torrent.set_directory(sickrage.app.config.torrent_path)
# Start torrent
torrent.start()
return True
except Exception:
sickrage.app.log.debug(traceback.format_exc())
return False
def _add_torrent_file(self, result):
if not self.auth:
return False
if not result:
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
label = sickrage.app.config.torrent_label
if result.show.is_anime:
label = sickrage.app.config.torrent_label_anime
if label:
torrent.set_custom(1, label)
if sickrage.app.config.torrent_path:
torrent.set_directory(sickrage.app.config.torrent_path)
# Set Ratio Group
# torrent.set_visible(group_name)
# Start torrent
torrent.start()
return True
except Exception:
sickrage.app.log.debug(traceback.format_exc())
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(sickrage.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 test_authentication(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