You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.7 KiB
53 lines
1.7 KiB
14 years ago
|
from couchpotato.core.downloaders.base import Downloader
|
||
|
|
from couchpotato.core.helpers.encoding import isInt
|
||
|
|
from couchpotato.core.logger import CPLog
|
||
|
|
from libs import transmissionrpc
|
||
|
|
|
||
|
|
log = CPLog(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class Transmission(Downloader):
|
||
|
|
|
||
|
|
type = ['torrent']
|
||
|
|
|
||
|
|
def download(self, data = {}, movie = {}):
|
||
|
|
|
||
|
|
if self.isDisabled() or not self.isCorrectType(data.get('type')):
|
||
|
|
return
|
||
|
|
|
||
|
|
log.info('Sending "%s" to Transmission.' % data.get('name'))
|
||
|
|
|
||
|
|
|
||
|
|
# Load host from config and split out port.
|
||
|
|
host = self.conf('host').split(':')
|
||
|
|
if not isInt(host[1]):
|
||
|
|
log.error("Config properties are not filled in correctly, port is missing.")
|
||
|
|
return False
|
||
|
|
|
||
|
|
# Set parameters for Transmission
|
||
|
|
params = {
|
||
|
|
'paused': self.conf('paused', 0),
|
||
|
|
'download_dir': self.conf('directory', None)
|
||
|
|
}
|
||
|
|
change_params = {
|
||
|
|
'seedRatioLimit': self.conf('ratio'),
|
||
|
|
'seedRatioMode': 1 if self.conf('ratio') else 0
|
||
|
|
}
|
||
|
|
|
||
|
|
try:
|
||
|
|
tc = transmissionrpc.Client(host[0], port = host[1], user = self.conf('username'), password = self.conf('password'))
|
||
|
|
tr_id = tc.add_uri(data.get('url'), **params)
|
||
|
|
|
||
|
|
# Change settings of added torrents
|
||
|
|
for item in tr_id:
|
||
|
|
try:
|
||
|
|
tc.change(item, timeout = None, **change_params)
|
||
|
|
except transmissionrpc.TransmissionError, e:
|
||
|
|
log.error('Failed to change settings for transfer in transmission: %s' % e)
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
except transmissionrpc.TransmissionError, e:
|
||
|
|
log.error('Failed to send link to transmission: %s' % e)
|
||
|
|
return False
|