1 files changed, 117 insertions, 0 deletions
diff --git a/deluge/ui/console/cmdline/commands/add.py b/deluge/ui/console/cmdline/commands/add.py
new file mode 100644
index 0000000..706ae16
--- /dev/null
+++ b/deluge/ui/console/cmdline/commands/add.py
@@ -0,0 +1,117 @@
+#
+# 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 os
+from base64 import b64encode
+from urllib.parse import urlparse
+from urllib.request import url2pathname
+
+from twisted.internet import defer
+
+import deluge.common
+import deluge.component as component
+from deluge.ui.client import client
+
+from . import BaseCommand
+
+
+class Command(BaseCommand):
+    """Add torrents"""
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            '-p', '--path', dest='path', help=_('Download folder for torrent')
+        )
+        parser.add_argument(
+            '-m',
+            '--move-path',
+            dest='move_completed_path',
+            help=_('Move the completed torrent to this folder'),
+        )
+        parser.add_argument(
+            'torrents',
+            metavar='<torrent>',
+            nargs='+',
+            help=_('One or more torrent files, URLs or magnet URIs'),
+        )
+
+    def handle(self, options):
+        self.console = component.get('ConsoleUI')
+
+        t_options = {}
+        if options.path:
+            t_options['download_location'] = os.path.abspath(
+                os.path.expanduser(options.path)
+            )
+
+        if options.move_completed_path:
+            t_options['move_completed'] = True
+            t_options['move_completed_path'] = os.path.abspath(
+                os.path.expanduser(options.move_completed_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 torrent in options.torrents:
+            if not torrent.strip():
+                continue
+            if deluge.common.is_url(torrent):
+                self.console.write(
+                    '{!info!}Attempting to add torrent from URL: %s' % torrent
+                )
+                deferreds.append(
+                    client.core.add_torrent_url(torrent, t_options)
+                    .addCallback(on_success)
+                    .addErrback(on_fail)
+                )
+            elif deluge.common.is_magnet(torrent):
+                self.console.write(
+                    '{!info!}Attempting to add torrent from magnet URI: %s' % torrent
+                )
+                deferreds.append(
+                    client.core.add_torrent_magnet(torrent, t_options)
+                    .addCallback(on_success)
+                    .addErrback(on_fail)
+                )
+            else:
+                # Just a file
+                if urlparse(torrent).scheme == 'file':
+                    torrent = url2pathname(urlparse(torrent).path)
+                path = os.path.abspath(os.path.expanduser(torrent))
+                if not os.path.exists(path):
+                    self.console.write('{!error!}%s does not 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]
+                with open(path, 'rb') as _file:
+                    filedump = b64encode(_file.read())
+                deferreds.append(
+                    client.core.add_torrent_file_async(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'
+        )