mirror of
https://github.com/correl/Transmission-XBMC.git
synced 2024-11-25 11:09:57 +00:00
310 lines
12 KiB
Python
310 lines
12 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (c) 2010 Correl J. Roush
|
|
|
|
import sys
|
|
import base64
|
|
import threading
|
|
import xbmc
|
|
import xbmcgui
|
|
from basictypes.bytes import Bytes
|
|
import transmissionrpc
|
|
import search
|
|
|
|
_ = sys.modules[ "__main__" ].__language__
|
|
__settings__ = sys.modules[ "__main__" ].__settings__
|
|
|
|
KEY_BUTTON_BACK = 275
|
|
KEY_KEYBOARD_ESC = 61467
|
|
KEY_MENU_ID = 92
|
|
|
|
EXIT_SCRIPT = ( 6, 10, 247, 275, 61467, 216, 257, 61448, )
|
|
CANCEL_DIALOG = EXIT_SCRIPT + ( 216, 257, 61448, )
|
|
|
|
UPDATE_INTERVAL = 1.0
|
|
|
|
STATUS_ICONS = {'stopped': 'pause.png',
|
|
'seeding': 'ok.png',
|
|
'downloading': 'down.png'}
|
|
|
|
class TransmissionGUI(xbmcgui.WindowXMLDialog):
|
|
def __init__(self, strXMLname, strFallbackPath, strDefaultName, bforeFallback=0):
|
|
self.torrents = {}
|
|
self.timer = None
|
|
self.filter = lambda x : True
|
|
def get_settings(self):
|
|
params = {
|
|
'address': __settings__.getSetting('rpc_host'),
|
|
'port': __settings__.getSetting('rpc_port'),
|
|
'user': __settings__.getSetting('rpc_user'),
|
|
'password': __settings__.getSetting('rpc_password')
|
|
}
|
|
return params
|
|
def set_settings(self, params):
|
|
__settings__.setSetting('rpc_host', params['address'])
|
|
__settings__.setSetting('rpc_port', params['port'])
|
|
__settings__.setSetting('rpc_user', params['user'])
|
|
__settings__.setSetting('rpc_password', params['password'])
|
|
def get_rpc_client(self):
|
|
params = self.get_settings()
|
|
return transmissionrpc.Client(**params)
|
|
def onInit(self):
|
|
self.getControl(125).setSelected(True)
|
|
self.getControl(126).setSelected(False)
|
|
self.getControl(127).setSelected(False)
|
|
p = xbmcgui.DialogProgress()
|
|
p.create(_(0), _(1)) # 'Transmission', 'Connecting to Transmission'
|
|
try:
|
|
self.transmission = self.get_rpc_client()
|
|
except:
|
|
p.close()
|
|
self.close()
|
|
(type, e, traceback) = sys.exc_info()
|
|
message = _(9000) # Unexpected error
|
|
if type is transmissionrpc.TransmissionError:
|
|
if e.original:
|
|
if e.original.code is 401:
|
|
message = _(9002) # Invalid auth
|
|
else:
|
|
message = _(9001) # Unable to connect
|
|
if xbmcgui.Dialog().yesno(_(2), message, _(3)):
|
|
__settings__.openSettings()
|
|
elif type is ValueError:
|
|
# In python 2.4, urllib2.HTTPDigestAuthHandler will barf up a lung
|
|
# if auth fails and the server wants non-digest authentication
|
|
message = _(9002) # Invalid auth
|
|
if xbmcgui.Dialog().yesno(_(2), message, _(3)):
|
|
__settings__.openSettings()
|
|
else:
|
|
message = _(9000) # Unexpected error
|
|
xbmcgui.Dialog().ok(_(2), message)
|
|
return False
|
|
self.updateTorrents()
|
|
p.close()
|
|
self.timer = threading.Timer(UPDATE_INTERVAL, self.updateTorrents)
|
|
self.timer.start()
|
|
def updateTorrents(self):
|
|
list = self.getControl(120)
|
|
self.torrents = self.transmission.info()
|
|
self.torrents = dict((k, v) for (k, v) in self.torrents.items() if self.filter(v))
|
|
i = 0
|
|
for torrentid, torrent in self.torrents.iteritems():
|
|
statusline = "[%(status)s] %(down)s down (%(pct).2f%%), %(up)s up (Ratio: %(ratio).2f)" % \
|
|
{'down': Bytes.format(torrent.downloadedEver), 'pct': torrent.progress, \
|
|
'up': Bytes.format(torrent.uploadedEver), 'ratio': torrent.ratio, \
|
|
'status': torrent.status}
|
|
if i >= list.size():
|
|
# Create a new list item
|
|
l = xbmcgui.ListItem(label=torrent.name, label2=statusline)
|
|
list.addItem(l)
|
|
else:
|
|
# Update existing list item
|
|
l = list.getListItem(i)
|
|
l.setLabel(torrent.name)
|
|
l.setLabel2(statusline)
|
|
l.setProperty('TorrentStatusIcon', STATUS_ICONS.get(torrent.status, 'pending.png'))
|
|
l.setProperty('TorrentID', str(i))
|
|
l.setProperty('TorrentProgress', "%3d%%" % torrent.progress)
|
|
l.setInfo('torrent', torrent.fields)
|
|
l.setInfo('video', {'episode': int(torrent.progress)})
|
|
i += 1
|
|
|
|
for i in range(list.size() - len(self.torrents)):
|
|
list.removeItem(list.size() - 1)
|
|
|
|
# Update again, after an interval
|
|
self.timer = threading.Timer(UPDATE_INTERVAL, self.updateTorrents)
|
|
self.timer.start()
|
|
def onClick(self, controlID):
|
|
list = self.getControl(120)
|
|
if (controlID == 111):
|
|
# Add torrent
|
|
engines = [
|
|
(_(200), None),
|
|
(_(202), search.TPB),
|
|
(_(203), search.Mininova),
|
|
(_(204), search.TorrentReactor),
|
|
]
|
|
selected = xbmcgui.Dialog().select(_(0), [i[0] for i in engines])
|
|
if selected < 0:
|
|
return
|
|
engine = engines[selected][1]
|
|
if not engine:
|
|
filename = xbmcgui.Dialog().browse(1, _(0), 'files', '.torrent')
|
|
try:
|
|
f = open(filename, 'r')
|
|
data = base64.b64encode(f.read())
|
|
self.transmission.add(data)
|
|
except:
|
|
pass
|
|
else:
|
|
kb = xbmc.Keyboard('', engines[selected][0])
|
|
kb.doModal()
|
|
if not kb.isConfirmed():
|
|
return
|
|
terms = kb.getText()
|
|
p = xbmcgui.DialogProgress()
|
|
p.create(_(0), _(290))
|
|
try:
|
|
results = engine().search(terms)
|
|
except:
|
|
p.close()
|
|
xbmcgui.Dialog().ok(_(0), _(292))
|
|
return
|
|
p.close()
|
|
if not results:
|
|
xbmcgui.Dialog().ok(_(0), _(291))
|
|
return
|
|
selected = xbmcgui.Dialog().select(_(0), ['[S:%d L:%d] %s' % (t['seeds'], t['leechers'], t['name']) for t in results])
|
|
if selected < 0:
|
|
return
|
|
try:
|
|
self.transmission.add_uri(results[selected]['url'])
|
|
except:
|
|
xbmcgui.Dialog().ok(_(0), _(293))
|
|
return
|
|
if (controlID == 112):
|
|
# Remove selected torrent
|
|
item = list.getSelectedItem()
|
|
if item and xbmcgui.Dialog().yesno(_(0), 'Remove \'%s\'?' % self.torrents[int(item.getProperty('TorrentID'))].name):
|
|
remove_data = xbmcgui.Dialog().yesno(_(0), 'Remove data as well?')
|
|
self.transmission.remove(int(item.getProperty('TorrentID')), remove_data)
|
|
if (controlID == 113):
|
|
# Stop selected torrent
|
|
item = list.getSelectedItem()
|
|
if item:
|
|
self.transmission.stop(int(item.getProperty('TorrentID')))
|
|
if (controlID == 114):
|
|
# Start selected torrent
|
|
item = list.getSelectedItem()
|
|
if item:
|
|
self.transmission.start(int(item.getProperty('TorrentID')))
|
|
if (controlID == 115):
|
|
# Stop all torrents
|
|
self.transmission.stop(self.torrents.keys())
|
|
if (controlID == 116):
|
|
# Start all torrents
|
|
self.transmission.start(self.torrents.keys())
|
|
if (controlID == 117):
|
|
# Exit button
|
|
self.close()
|
|
if (controlID == 118):
|
|
# Settings button
|
|
prev_settings = self.get_settings()
|
|
__settings__.openSettings()
|
|
p = xbmcgui.DialogProgress()
|
|
p.create(_(0), _(1)) # 'Transmission', 'Connecting to Transmission'
|
|
try:
|
|
self.transmission = self.get_rpc_client()
|
|
self.updateTorrents()
|
|
p.close()
|
|
except:
|
|
p.close()
|
|
xbmcgui.Dialog().ok(_(2), _(9001))
|
|
# restore settings
|
|
self.set_settings(prev_settings)
|
|
try:
|
|
self.transmission = self.get_rpc_client()
|
|
except err:
|
|
xbmcgui.Dialog().ok(_(2), _(9001))
|
|
self.close()
|
|
if (controlID == 120):
|
|
# Do nothing, just select the item
|
|
pass
|
|
if (controlID == 125):
|
|
# Show all torrents
|
|
self.getControl(126).setSelected(False)
|
|
self.getControl(127).setSelected(False)
|
|
self.filter = lambda torrent : True
|
|
if (controlID == 126):
|
|
# Show only active downloads
|
|
self.getControl(125).setSelected(False)
|
|
self.getControl(127).setSelected(False)
|
|
self.filter = lambda torrent : True if torrent.status != "seeding" else False
|
|
if (controlID == 127):
|
|
# Show finished downloads
|
|
self.getControl(125).setSelected(False)
|
|
self.getControl(126).setSelected(False)
|
|
self.filter = lambda torrent : True if torrent.status == "seeding" else False
|
|
def onDoubleClick(self, controlID):
|
|
if (controlID == 120):
|
|
# A torrent was chosen, show details
|
|
list = self.getControl(120)
|
|
item = list.getSelectedItem()
|
|
w = TorrentInfoGUI("script-Transmission-details.xml", __settings__.getAddonInfo('path') ,"Default")
|
|
w.setTorrent(self.transmission, int(item.getProperty('TorrentID')))
|
|
w.doModal()
|
|
del w
|
|
def onFocus(self, controlID):
|
|
pass
|
|
|
|
def onAction( self, action ):
|
|
if ( action.getButtonCode() in CANCEL_DIALOG ) or (action.getId() == KEY_MENU_ID):
|
|
self.close()
|
|
def close(self):
|
|
if self.timer:
|
|
self.timer.cancel()
|
|
super(TransmissionGUI, self).close()
|
|
|
|
|
|
class TorrentInfoGUI(xbmcgui.WindowXMLDialog):
|
|
def __init__(self, strXMLname, strFallbackPath, strDefaultName, bforeFallback=0):
|
|
self.transmission = None
|
|
self.torrent_id = None
|
|
self.list = {}
|
|
self.timer = None
|
|
def setTorrent(self, transmission, t_id):
|
|
self.transmission = transmission
|
|
self.torrent_id = t_id
|
|
self.timer = threading.Timer(UPDATE_INTERVAL, self.updateTorrent)
|
|
self.timer.start()
|
|
def updateTorrent(self):
|
|
pbar = self.getControl(219)
|
|
list = self.getControl(220)
|
|
labelName = self.getControl(1)
|
|
labelStatus = self.getControl(2)
|
|
labelProgress = self.getControl(11)
|
|
torrent = self.transmission.info()[self.torrent_id]
|
|
files = self.transmission.get_files(self.torrent_id)[self.torrent_id]
|
|
|
|
statusline = "[%(status)s] %(down)s down, %(up)s up (Ratio: %(ratio).2f)" % \
|
|
{'down': Bytes.format(torrent.downloadedEver), 'pct': torrent.progress, \
|
|
'up': Bytes.format(torrent.uploadedEver), 'ratio': torrent.ratio, \
|
|
'status': torrent.status}
|
|
if torrent.status is 'downloading':
|
|
statusline += " ETA: %(eta)s" % \
|
|
{'eta': torrent.eta}
|
|
|
|
labelName.setLabel(torrent.name)
|
|
labelStatus.setLabel(statusline)
|
|
labelProgress.setLabel('%3d%%' % (torrent.progress))
|
|
pbar.setPercent(torrent.progress)
|
|
|
|
for i, file in files.iteritems():
|
|
if i not in self.list:
|
|
# Create a new list item
|
|
l = xbmcgui.ListItem(label=file['name'])
|
|
list.addItem(l)
|
|
self.list[i] = l
|
|
else:
|
|
# Update existing list item
|
|
l = self.list[i]
|
|
l.setProperty('Progress', '[%3d%%]' % (file['completed'] * 100 / file['size']))
|
|
|
|
# Update again, after an interval
|
|
self.timer = threading.Timer(UPDATE_INTERVAL, self.updateTorrent)
|
|
self.timer.start()
|
|
def onInit(self):
|
|
self.updateTorrent()
|
|
def close(self):
|
|
if self.timer:
|
|
self.timer.cancel()
|
|
super(TorrentInfoGUI, self).close()
|
|
def onAction(self, action):
|
|
if (action.getButtonCode() in CANCEL_DIALOG) or (action.getId() == KEY_MENU_ID):
|
|
self.close()
|
|
pass
|
|
def onClick(self, controlID):
|
|
if controlID == 111:
|
|
self.close()
|
|
def onFocus(self, controlID):
|
|
pass
|