Exemplo n.º 1
0
 def _show_torrent_filter_popup(self):
     self.popup = SelectablePopup(self,"Filter Torrents",self._torrent_filter)
     self.popup.add_line("_All",data=FILTER.ALL)
     self.popup.add_line("Ac_tive",data=FILTER.ACTIVE)
     self.popup.add_line("_Downloading",data=FILTER.DOWNLOADING,foreground="green")
     self.popup.add_line("_Seeding",data=FILTER.SEEDING,foreground="cyan")
     self.popup.add_line("_Paused",data=FILTER.PAUSED)
     self.popup.add_line("_Error",data=FILTER.ERROR,foreground="red")
     self.popup.add_line("_Checking",data=FILTER.CHECKING,foreground="blue")
     self.popup.add_line("Q_ueued",data=FILTER.QUEUED,foreground="yellow")
Exemplo n.º 2
0
 def __update_popup(self):
     self.popup = SelectablePopup(self,"Select Host",self.__host_selected)
     self.popup.add_line("{!white,black,bold!}'Q'=quit, 'r'=refresh, 'a'=add new host, 'D'=delete host",selectable=False)
     for host in self.config["hosts"]:
         if host[0] in self.statuses:
             self.popup.add_line("%s:%d [Online] (%s)"%(host[1],host[2],self.statuses[host[0]]),data=host[0],foreground="green")
         else:
             self.popup.add_line("%s:%d [Offline]"%(host[1],host[2]),data=host[0],foreground="red")
     self.inlist = True
     self.refresh()
Exemplo n.º 3
0
 def show_priority_popup(self):
     if self.marked:
         self.popup = SelectablePopup(self, "Set File Priority",
                                      self.do_priority)
         self.popup.add_line(
             "_Do Not Download",
             data=deluge.common.FILE_PRIORITY["Do Not Download"])
         self.popup.add_line(
             "_Normal Priority",
             data=deluge.common.FILE_PRIORITY["Normal Priority"])
         self.popup.add_line(
             "_High Priority",
             data=deluge.common.FILE_PRIORITY["High Priority"])
         self.popup.add_line(
             "H_ighest Priority",
             data=deluge.common.FILE_PRIORITY["Highest Priority"])
Exemplo n.º 4
0
 def show_priority_popup(self, was_empty):
     func = lambda idx, data, we=was_empty: self.do_priority(idx, data, we)
     if self.marked:
         self.popup = SelectablePopup(self, "Set File Priority", func)
         self.popup.add_line(
             "_Do Not Download",
             data=deluge.common.FILE_PRIORITY["Do Not Download"],
             foreground="red")
         self.popup.add_line(
             "_Normal Priority",
             data=deluge.common.FILE_PRIORITY["Normal Priority"])
         self.popup.add_line(
             "_High Priority",
             data=deluge.common.FILE_PRIORITY["High Priority"],
             foreground="yellow")
         self.popup.add_line(
             "H_ighest Priority",
             data=deluge.common.FILE_PRIORITY["Highest Priority"],
             foreground="green")
         self.popup._selected = 1
Exemplo n.º 5
0
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")
        elif data==ACTION.TORRENT_OPTIONS:
            mode.popup = Popup(mode, "Torrent options")
            mode.popup.add_line("Querying core, please wait...")

            torrents = ids

            options = {}

            def _do_set_torrent_options(ids, result):
                options = {}
                for opt in result:
                    if result[opt] not in ["multiple", None]:
                        options[opt] = result[opt]
                client.core.set_torrent_options( ids, options )
                for tid in ids:
                    if "move_on_completed_path" in options:
                        client.core.set_torrent_move_completed_path(tid, options["move_on_completed_path"])
                    if "move_on_completed" in options:
                        client.core.set_torrent_move_completed(tid, options["move_on_completed"])
                    if "is_auto_managed" in options:
                        client.core.set_torrent_auto_managed(tid, options["is_auto_managed"])
                    if "remove_at_ratio" in options:
                        client.core.set_torrent_remove_at_ratio(tid, options["remove_at_ratio"])
                    if "prioritize_first_last" in options:
                        client.core.set_torrent_prioritize_first_last(tid, options["prioritize_first_last"])

            def on_torrent_status(status):
                for key in status:
                    if   key not in options:
                        options[key] = status[key]
                    elif options[key] != status[key]:
                        options[key] = "multiple"

            def create_popup(status):
                cb = lambda result, ids=ids: _do_set_torrent_options(ids, result)
                option_popup = InputPopup(mode,"Set torrent options (Esc to cancel)",close_cb=cb, height_req=22)

                for (field, field_type) in torrent_options:
                    caption = "{!info!}" + torrent_options_to_names[field]
                    value = options[field]
                    if   field_type == str:
                        if not isinstance(value, basestring):
                            value = str(value)
                        option_popup.add_text_input(caption, field, value)
                    elif field_type == bool:
                        if options[field] == "multiple":
                            choices = (
                                ["Yes", "No", "Mixed"],
                                [True, False, None],
                                2
                            )
                        else:
                            choices = (
                                ["Yes", "No"],
                                [True, False],
                                [True, False].index(options[field])
                            )
                        option_popup.add_select_input(caption, field, choices[0], choices[1], choices[2])
                    elif field_type == float:
                        option_popup.add_float_spin_input(caption, field, value, min_val = -1)
                    elif field_type == int:
                        option_popup.add_int_spin_input(caption, field, value, min_val = -1)

                mode.set_popup(option_popup)
                mode.refresh()

            callbacks = []

            field_list = map(lambda t: t[0], torrent_options)

            for tid in torrents:
                deferred = component.get("SessionProxy").get_torrent_status(tid, field_list)
                callbacks.append( deferred.addCallback(on_torrent_status) )

            callbacks = defer.DeferredList(callbacks)
            callbacks.addCallback(create_popup)

    if len(ids) == 1:
        mode.clear_marks()
    return True
Exemplo n.º 6
0
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)
    if details:
        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)
    popup.add_divider()
    if details:
        popup.add_line("Torrent _Details",data=ACTION.DETAILS)
    popup.add_line("Torrent _Options",data=ACTION.TORRENT_OPTIONS)
    mode.set_popup(popup)
Exemplo n.º 7
0
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")
        elif data == ACTION.TORRENT_OPTIONS:
            mode.popup = Popup(mode, "Torrent options")
            mode.popup.add_line("Querying core, please wait...")

            torrents = ids

            options = {}

            def _do_set_torrent_options(ids, result):
                options = {}
                for opt in result:
                    if result[opt] not in ["multiple", None]:
                        options[opt] = result[opt]
                client.core.set_torrent_options(ids, options)
                for tid in ids:
                    if "move_on_completed_path" in options:
                        client.core.set_torrent_move_completed_path(
                            tid, options["move_on_completed_path"])
                    if "move_on_completed" in options:
                        client.core.set_torrent_move_completed(
                            tid, options["move_on_completed"])
                    if "is_auto_managed" in options:
                        client.core.set_torrent_auto_managed(
                            tid, options["is_auto_managed"])
                    if "remove_at_ratio" in options:
                        client.core.set_torrent_remove_at_ratio(
                            tid, options["remove_at_ratio"])
                    if "prioritize_first_last" in options:
                        client.core.set_torrent_prioritize_first_last(
                            tid, options["prioritize_first_last"])

            def on_torrent_status(status):
                for key in status:
                    if key not in options:
                        options[key] = status[key]
                    elif options[key] != status[key]:
                        options[key] = "multiple"

            def create_popup(status):
                cb = lambda result, ids=ids: _do_set_torrent_options(
                    ids, result)
                option_popup = InputPopup(
                    mode,
                    "Set torrent options (Esc to cancel)",
                    close_cb=cb,
                    height_req=22)

                for (field, field_type) in torrent_options:
                    caption = "{!info!}" + torrent_options_to_names[field]
                    value = options[field]
                    if field_type == str:
                        if not isinstance(value, basestring):
                            value = str(value)
                        option_popup.add_text_input(caption, field, value)
                    elif field_type == bool:
                        if options[field] == "multiple":
                            choices = (["Yes", "No",
                                        "Mixed"], [True, False, None], 2)
                        else:
                            choices = (["Yes", "No"], [True, False],
                                       [True, False].index(options[field]))
                        option_popup.add_select_input(caption, field,
                                                      choices[0], choices[1],
                                                      choices[2])
                    elif field_type == float:
                        option_popup.add_float_spin_input(caption,
                                                          field,
                                                          value,
                                                          min_val=-1)
                    elif field_type == int:
                        option_popup.add_int_spin_input(caption,
                                                        field,
                                                        value,
                                                        min_val=-1)

                mode.set_popup(option_popup)
                mode.refresh()

            callbacks = []

            field_list = map(lambda t: t[0], torrent_options)

            for tid in torrents:
                deferred = component.get("SessionProxy").get_torrent_status(
                    tid, field_list)
                callbacks.append(deferred.addCallback(on_torrent_status))

            callbacks = defer.DeferredList(callbacks)
            callbacks.addCallback(create_popup)

    if len(ids) == 1:
        mode.clear_marks()
    return True
Exemplo n.º 8
0
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)
    if details:
        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)
    popup.add_divider()
    if details:
        popup.add_line("Torrent _Details", data=ACTION.DETAILS)
    popup.add_line("Torrent _Options", data=ACTION.TORRENT_OPTIONS)
    mode.set_popup(popup)
Exemplo n.º 9
0
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):
                if qact == ACTION.QUEUE_TOP:
                    log.debug("Queuing torrents top")
                    client.core.queue_top(ids)
                elif qact == ACTION.QUEUE_UP:
                    log.debug("Queuing torrents up")
                    client.core.queue_up(ids)
                elif qact == ACTION.QUEUE_DOWN:
                    log.debug("Queuing torrents down")
                    client.core.queue_down(ids)
                elif qact == ACTION.QUEUE_BOTTOM:
                    log.debug("Queuing torrents bottom")
                    client.core.queue_bottom(ids)
                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(idx, data, mode, ids):
                if data:
                    wd = data == ACTION.REMOVE_DATA
                    for tid in ids:
                        log.debug("Removing torrent: %s,%d", tid, wd)
                        client.core.remove_torrent(tid, wd).addErrback(action_error, mode)
                if len(ids) == 1:
                    mode.clear_marks()
                return True

            popup = SelectablePopup(mode, "Confirm Remove", do_remove, mode, ids)
            popup.add_line("Are you sure you want to remove the marked torrents?", selectable=False)
            popup.add_line("Remove with _data", data=ACTION.REMOVE_DATA)
            popup.add_line("Remove _torrent", data=ACTION.REMOVE_NODATA)
            popup.add_line("_Cancel", data=0)
            mode.set_popup(popup)
            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
Exemplo n.º 10
0
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):
                if qact == ACTION.QUEUE_TOP:
                    log.debug("Queuing torrents top")
                    client.core.queue_top(ids)
                elif qact == ACTION.QUEUE_UP:
                    log.debug("Queuing torrents up")
                    client.core.queue_up(ids)
                elif qact == ACTION.QUEUE_DOWN:
                    log.debug("Queuing torrents down")
                    client.core.queue_down(ids)
                elif qact == ACTION.QUEUE_BOTTOM:
                    log.debug("Queuing torrents bottom")
                    client.core.queue_bottom(ids)
                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(idx, data, mode, ids):
                if data:
                    wd = data == ACTION.REMOVE_DATA
                    for tid in ids:
                        log.debug("Removing torrent: %s,%d", tid, wd)
                        client.core.remove_torrent(tid, wd).addErrback(
                            action_error, mode)
                if len(ids) == 1:
                    mode.clear_marks()
                return True

            popup = SelectablePopup(mode, "Confirm Remove", do_remove, mode,
                                    ids)
            popup.add_line(
                "Are you sure you want to remove the marked torrents?",
                selectable=False)
            popup.add_line("Remove with _data", data=ACTION.REMOVE_DATA)
            popup.add_line("Remove _torrent", data=ACTION.REMOVE_NODATA)
            popup.add_line("_Cancel", data=0)
            mode.set_popup(popup)
            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