Added automated download specification

README.md

@@ -34,8 +34,8 @@ Raincoat is a CLI tool to search torrents using [Jackett](https://github.com/Jac
   - Change the sorting criteria. Valid choices are: 'seeders', 'leechers', 'ratio', 'size' and 'description'. Default/not specified is 'seeders'.
 - -i, --indexer
   - Change the indexer used for your search, in cases where you want to only search one site. Default is "all".
-- -d, --download
-  - Grab the first result and send to the client immediately.
+- -d, --download x
+  - Grab the first x resultd and send to the client immediately. Defaults to 1.
 - -K, --insecure
   - Don't verify certificates  
 - --local

changelog.txt

@@ -1,3 +1,6 @@
+=== 1.1 ===
+- Added the option to specify the number of entries to dowload automatically
+
 === 1.0 ===
 - Fixed a minor typo preventing from using indexer parameter in the command line.
 

raincoat/raincoat.py

@@ -22,7 +22,7 @@
 parser.add_argument("-c", "--config", help="Specify a different config file path.")
 parser.add_argument("-s", "--sort", help="Change sorting criteria.", action="store", dest="sort", choices=['seeders', 'leechers', 'ratio', 'size', 'description'])
 parser.add_argument("-i", "--indexer", help="The Jackett indexer to use for your search.")
-parser.add_argument("-d", "--download", help="Download and send the top result to the client and exit.", action="store_true")
+parser.add_argument("-d", "--download", help="Download and send the top 'x' results (defaults to 1) to the client and exit.", nargs='?', const=1, type=int)
 parser.add_argument("-K", "--insecure", help="Enables to use self-signed certificates.", action="store_true")
 parser.add_argument("--local", help="Override torrent provider with local download.", action="store_true")
 parser.add_argument("--verbose", help="Very verbose output to logs.", action="store_true")
@@ -89,7 +89,7 @@ def set_overrides():
         shared.VERBOSE_MODE = True
 
     if args.download:
-        shared.DOWNLOAD = True
+        shared.DOWNLOAD = args.download
 
     if args.insecure:
         requests.packages.urllib3.disable_warnings(category=InsecureRequestWarning)
@@ -98,7 +98,11 @@ def set_overrides():
 def prompt_torrent():
     if shared.DOWNLOAD:
         if len(shared.TORRENTS) > 0:
-            download(shared.TORRENTS[0].id)
+            # Prevent out of bound results
+            if shared.DOWNLOAD > len(shared.TORRENTS):
+                shared.DOWNLOAD = len(shared.TORRENTS)
+            for i in range(shared.DOWNLOAD):
+                download(shared.TORRENTS[i].id)
             exit()
         else:
             print("Search did not yield any results.")

raincoat/shared.py

@@ -1,7 +1,7 @@
 # Globals
 def init():
     global VERSION
-    VERSION = "1.0"
+    VERSION = "1.1"
     global APP_NAME
     APP_NAME = "Raincoat"
     global TORRENTS
@@ -35,7 +35,7 @@ def init():
     global VERBOSE_MODE
     VERBOSE_MODE = False
     global DOWNLOAD
-    DOWNLOAD = False
+    DOWNLOAD = 0
     global VERIFY
     VERIFY = True
 

setup.py

@@ -34,7 +34,7 @@ def run(self):
 
 setuptools.setup(
     name="raincoat-jackett",
-    version="1.0",
+    version="1.1",
     author="Gabisonfire",
     author_email="gabisonfire@github.com",
     description="Raincoat is a tool to search torrents using Jackett and send them to your client.",