229 lines
9.5 KiB
Python
229 lines
9.5 KiB
Python
# torrent_actions.py
|
|
#
|
|
# Copyright (C) 2011 Nick Lanham <nick@afternight.org>
|
|
#
|
|
# Deluge is free software.
|
|
#
|
|
# You may 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.
|
|
#
|
|
# deluge 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 deluge. If not, write to:
|
|
# The Free Software Foundation, Inc.,
|
|
# 51 Franklin Street, Fifth Floor
|
|
# Boston, MA 02110-1301, USA.
|
|
#
|
|
# In addition, as a special exception, the copyright holders give
|
|
# permission to link the code of portions of this program with the OpenSSL
|
|
# library.
|
|
# You must obey the GNU General Public License in all respects for all of
|
|
# the code used other than OpenSSL. If you modify file(s) with this
|
|
# exception, you may extend this exception to your version of the file(s),
|
|
# but you are not obligated to do so. If you do not wish to do so, delete
|
|
# this exception statement from your version. If you delete this exception
|
|
# statement from all source files in the program, then also delete it here.
|
|
#
|
|
#
|
|
|
|
from deluge.ui.client import client
|
|
from popup import SelectablePopup
|
|
from input_popup import InputPopup
|
|
|
|
|
|
from deluge.ui.console import colors, modes
|
|
from twisted.internet import defer
|
|
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
|
|
class ACTION:
|
|
PAUSE=0
|
|
RESUME=1
|
|
REANNOUNCE=2
|
|
EDIT_TRACKERS=3
|
|
RECHECK=4
|
|
REMOVE=5
|
|
REMOVE_DATA=6
|
|
REMOVE_NODATA=7
|
|
DETAILS=8
|
|
MOVE_STORAGE=9
|
|
QUEUE=10
|
|
QUEUE_TOP=11
|
|
QUEUE_UP=12
|
|
QUEUE_DOWN=13
|
|
QUEUE_BOTTOM=14
|
|
|
|
def action_error(error,mode):
|
|
rerr = error.value
|
|
mode.report_message("An Error Occurred","%s got error %s: %s"%(rerr.method,rerr.exception_type,rerr.exception_msg))
|
|
mode.refresh()
|
|
|
|
def torrent_action(idx, data, mode, ids):
|
|
if ids:
|
|
if data==ACTION.PAUSE:
|
|
log.debug("Pausing torrents: %s",ids)
|
|
client.core.pause_torrent(ids).addErrback(action_error,mode)
|
|
elif data==ACTION.RESUME:
|
|
log.debug("Resuming torrents: %s", ids)
|
|
client.core.resume_torrent(ids).addErrback(action_error,mode)
|
|
elif data==ACTION.QUEUE:
|
|
def do_queue(idx,qact,mode,ids):
|
|
def move_selection(r):
|
|
if mode.config["move_selection"]:
|
|
queue_length = 0
|
|
selected_num = 0
|
|
for tid in mode.curstate:
|
|
tq = mode.curstate.get(tid)["queue"]
|
|
if tq != -1:
|
|
queue_length += 1
|
|
if tq in mode.marked:
|
|
selected_num += 1
|
|
if qact == ACTION.QUEUE_TOP:
|
|
if mode.marked:
|
|
mode.cursel = 1 + sorted(mode.marked).index(mode.cursel)
|
|
else:
|
|
mode.cursel = 1
|
|
mode.marked = range(1, selected_num + 1)
|
|
elif qact == ACTION.QUEUE_UP:
|
|
mode.cursel = max(1, mode.cursel - 1)
|
|
mode.marked = map(lambda v: v-1, mode.marked)
|
|
mode.marked = filter(lambda v: v>0, mode.marked)
|
|
elif qact == ACTION.QUEUE_DOWN:
|
|
mode.cursel = min(queue_length, mode.cursel + 1)
|
|
mode.marked = map(lambda v: v+1, mode.marked)
|
|
mode.marked = filter(lambda v: v<=queue_length, mode.marked)
|
|
elif qact == ACTION.QUEUE_BOTTOM:
|
|
if mode.marked:
|
|
mode.cursel = queue_length - selected_num + 1 + sorted(mode.marked).index(mode.cursel)
|
|
else:
|
|
mode.cursel = queue_length
|
|
mode.marked = range(queue_length - selected_num + 1, queue_length+1)
|
|
|
|
if qact == ACTION.QUEUE_TOP:
|
|
log.debug("Queuing torrents top")
|
|
client.core.queue_top(ids).addCallback(move_selection)
|
|
elif qact == ACTION.QUEUE_UP:
|
|
log.debug("Queuing torrents up")
|
|
client.core.queue_up(ids).addCallback(move_selection)
|
|
elif qact == ACTION.QUEUE_DOWN:
|
|
log.debug("Queuing torrents down")
|
|
client.core.queue_down(ids).addCallback(move_selection)
|
|
elif qact == ACTION.QUEUE_BOTTOM:
|
|
log.debug("Queuing torrents bottom")
|
|
client.core.queue_bottom(ids).addCallback(move_selection)
|
|
|
|
if len(ids) == 1:
|
|
mode.clear_marks()
|
|
return True
|
|
popup = SelectablePopup(mode,"Queue Action",do_queue,mode,ids)
|
|
popup.add_line("_Top",data=ACTION.QUEUE_TOP)
|
|
popup.add_line("_Up",data=ACTION.QUEUE_UP)
|
|
popup.add_line("_Down",data=ACTION.QUEUE_DOWN)
|
|
popup.add_line("_Bottom",data=ACTION.QUEUE_BOTTOM)
|
|
mode.set_popup(popup)
|
|
return False
|
|
elif data==ACTION.REMOVE:
|
|
def do_remove(data):
|
|
if not data: return
|
|
mode.clear_marks()
|
|
|
|
wd = data["remove_files"]
|
|
for tid in ids:
|
|
log.debug("Removing torrent: %s, %d", tid, wd)
|
|
client.core.remove_torrent(tid,wd).addErrback(action_error,mode)
|
|
|
|
rem_msg = ""
|
|
|
|
|
|
def got_status(status):
|
|
return (status["name"], status["state"])
|
|
|
|
callbacks = []
|
|
for tid in ids:
|
|
d = client.core.get_torrent_status(tid, ["name", "state"])
|
|
callbacks.append( d.addCallback(got_status) )
|
|
|
|
|
|
def finish_up(status):
|
|
status = map(lambda x: x[1], status)
|
|
|
|
if len(ids) == 1:
|
|
rem_msg = "{!info!}Removing the following torrent:{!input!}"
|
|
else:
|
|
rem_msg = "{!info!}Removing the following torrents:{!input!}"
|
|
|
|
for i, (name, state) in enumerate(status):
|
|
color = colors.state_color[state]
|
|
rem_msg += "\n %s* {!input!}%s" % (color, name)
|
|
if i == 5:
|
|
if i < len(status):
|
|
rem_msg += "\n {!red!}And %i more" % (len(status) - 5)
|
|
break
|
|
|
|
popup = InputPopup(mode, "(Esc to cancel, Enter to remove)", close_cb=do_remove)
|
|
popup.add_text(rem_msg)
|
|
popup.add_spaces(1)
|
|
popup.add_select_input("{!info!}Torrent files:", 'remove_files', ["Keep", "Remove"], [False, True], False)
|
|
mode.set_popup(popup)
|
|
defer.DeferredList(callbacks).addCallback(finish_up)
|
|
return False
|
|
elif data==ACTION.MOVE_STORAGE:
|
|
def do_move(res):
|
|
import os.path
|
|
if os.path.exists(res["path"]) and not os.path.isdir(res["path"]):
|
|
mode.report_message("Cannot Move Storage","{!error!}%s exists and is not a directory"%res["path"])
|
|
else:
|
|
log.debug("Moving %s to: %s",ids,res["path"])
|
|
client.core.move_storage(ids,res["path"]).addErrback(action_error,mode)
|
|
if len(ids) == 1:
|
|
mode.clear_marks()
|
|
return True
|
|
popup = InputPopup(mode,"Move Storage (Esc to cancel)",close_cb=do_move)
|
|
popup.add_text_input("Enter path to move to:","path")
|
|
mode.set_popup(popup)
|
|
return False
|
|
elif data==ACTION.RECHECK:
|
|
log.debug("Rechecking torrents: %s", ids)
|
|
client.core.force_recheck(ids).addErrback(action_error,mode)
|
|
elif data==ACTION.REANNOUNCE:
|
|
log.debug("Reannouncing torrents: %s",ids)
|
|
client.core.force_reannounce(ids).addErrback(action_error,mode)
|
|
elif data==ACTION.DETAILS:
|
|
log.debug("Torrent details")
|
|
tid = mode.current_torrent_id()
|
|
if tid:
|
|
mode.show_torrent_details(tid)
|
|
else:
|
|
log.error("No current torrent in _torrent_action, this is a bug")
|
|
if len(ids) == 1:
|
|
mode.clear_marks()
|
|
return True
|
|
|
|
# Creates the popup. mode is the calling mode, tids is a list of torrents to take action upon
|
|
def torrent_actions_popup(mode,tids,details=False, action = None):
|
|
if action != None:
|
|
torrent_action(-1, action, mode, tids)
|
|
return
|
|
popup = SelectablePopup(mode,"Torrent Actions",torrent_action,mode,tids)
|
|
popup.add_line("_Pause",data=ACTION.PAUSE)
|
|
popup.add_line("_Resume",data=ACTION.RESUME)
|
|
popup.add_divider()
|
|
popup.add_line("Queue",data=ACTION.QUEUE)
|
|
popup.add_divider()
|
|
popup.add_line("_Update Tracker",data=ACTION.REANNOUNCE)
|
|
popup.add_divider()
|
|
popup.add_line("Remo_ve Torrent",data=ACTION.REMOVE)
|
|
popup.add_line("_Force Recheck",data=ACTION.RECHECK)
|
|
popup.add_line("_Move Storage",data=ACTION.MOVE_STORAGE)
|
|
if details:
|
|
popup.add_divider()
|
|
popup.add_line("Torrent _Details",data=ACTION.DETAILS)
|
|
mode.set_popup(popup)
|