Coverage for deluge/ui/console/commands/add : 0%

# -*- coding: utf-8 -*- 

# 

# Copyright (C) 2008-2009 Ido Abramovich <ido.deluge@gmail.com> 

# Copyright (C) 2009 Andrew Resch <andrewresch@gmail.com> 

# 

# This file is part of Deluge and is licensed under GNU General Public License 3.0, or later, with 

# the additional special exception to link portions of this program with the OpenSSL library. 

# See LICENSE for more details. 

# 

import base64 

import os 

from optparse import make_option 

from urllib import url2pathname 

from urlparse import urlparse 

from twisted.internet import defer 

import deluge.common 

import deluge.component as component 

from deluge.ui.client import client 

from deluge.ui.console.main import BaseCommand 

class Command(BaseCommand): 

    """Add a torrent""" 

    option_list = BaseCommand.option_list + ( 

        make_option('-p', '--path', dest='path', help='download folder for torrent'), 

    ) 

    usage = "Usage: add [-p <download-folder>] <torrent-file> [<torrent-file> ...]\n"\ 

            "             <torrent-file> arguments can be file paths, URLs or magnet uris" 

    def handle(self, *args, **options): 

        self.console = component.get("ConsoleUI") 

        t_options = {} 

        if options["path"]: 

            t_options["download_location"] = os.path.expanduser(options["path"]) 

        def on_success(result): 

            if not result: 

                self.console.write("{!error!}Torrent was not added: Already in session") 

            else: 

                self.console.write("{!success!}Torrent added!") 

        def on_fail(result): 

            self.console.write("{!error!}Torrent was not added: %s" % result) 

        # Keep a list of deferreds to make a DeferredList 

        deferreds = [] 

        for arg in args: 

            if not arg.strip(): 

                continue 

            if deluge.common.is_url(arg): 

                self.console.write("{!info!}Attempting to add torrent from url: %s" % arg) 

                deferreds.append(client.core.add_torrent_url(arg, t_options).addCallback(on_success).addErrback( 

                    on_fail)) 

            elif deluge.common.is_magnet(arg): 

                self.console.write("{!info!}Attempting to add torrent from magnet uri: %s" % arg) 

                deferreds.append(client.core.add_torrent_magnet(arg, t_options).addCallback(on_success).addErrback( 

                    on_fail)) 

            else: 

                # Just a file 

                if urlparse(arg).scheme == "file": 

                    arg = url2pathname(urlparse(arg).path) 

                path = os.path.abspath(os.path.expanduser(arg)) 

                if not os.path.exists(path): 

                    self.console.write("{!error!}%s doesn't exist!" % path) 

                    continue 

                if not os.path.isfile(path): 

                    self.console.write("{!error!}This is a directory!") 

                    continue 

                self.console.write("{!info!}Attempting to add torrent: %s" % path) 

                filename = os.path.split(path)[-1] 

                filedump = base64.encodestring(open(path, "rb").read()) 

                deferreds.append(client.core.add_torrent_file(filename, filedump, t_options).addCallback( 

                    on_success).addErrback(on_fail)) 

        return defer.DeferredList(deferreds) 

    def complete(self, line): 

        return component.get("ConsoleUI").tab_complete_path(line, ext=".torrent", sort="date")