From 994747c0b41b904081826394bd688ac4b1127a2a Mon Sep 17 00:00:00 2001
From: Sebastian Robinson <sebastian@podtwo.com>
Date: Mon, 16 Mar 2020 00:44:30 +0000
Subject: [PATCH] Add --keep argument to download torrent files to local
 directory

---
 PDMameUpdate.py      | 12 ++++++++++++
 README.md            |  7 ++++---
 config.template.json |  5 +++--
 3 files changed, 19 insertions(+), 5 deletions(-)

diff --git a/PDMameUpdate.py b/PDMameUpdate.py
index 415a992..1cc25cd 100755
--- a/PDMameUpdate.py
+++ b/PDMameUpdate.py
@@ -224,6 +224,10 @@ def update_torrents():
             config['mame-directory'],
             torrent['remote-name']
         )
+        new_torrent = os.path.join(
+            config['torrent-directory'],
+            torrent['remote-name']+'.torrent'
+        )
 
         client.torrent.remove(torrent['transmission-id'])
         os.rename(old_name, new_name)
@@ -234,6 +238,9 @@ def update_torrents():
             paused=False
         )
 
+        if args.keep:
+            t = requests.get(torrent['remote-link'], verify=False)
+            open(new_torrent, 'wb').write(t.content)
 
 if __name__ == '__main__':
     logging.basicConfig(
@@ -253,6 +260,11 @@ if __name__ == '__main__':
         action='store_true',
         help='Display debugging messages'
     )
+    parser.add_argument(
+        '-k', '--keep',
+        action='store_true',
+        help='Keep torrent files localy'
+    )
     parser.add_argument(
         '-c', '--countdown',
         action='store_true',
diff --git a/README.md b/README.md
index 2cf3ae5..436ca6b 100644
--- a/README.md
+++ b/README.md
@@ -1,7 +1,7 @@
 # PDMameUpdate
 
-Checks available MAME Torrents on PleasureDome
-and updates the local versions if more recent
+Checks available MAME Torrents on PleasureDome
+and updates the local versions if more recent
 versions are detected
 
 ![Sample output](https://djib.fr/djib/PDMameUpdate/raw/branch/master/screenshot.png)
@@ -20,7 +20,7 @@ pipenv run python3 PDMameUpdate.py
 
 ## Usage
 ```shell
-usage: PDMameUpdate.py [-h] [-l] [-d] [-c]
+usage: PDMameUpdate.py [-h] [-l] [-d] [-k] [-c]
 
 Update PleasureDome MAME Torrents
 
@@ -28,5 +28,6 @@ optional arguments:
   -h, --help       show this help message and exit
   -l, --log        Display more log messages
   -d, --debug      Display debugging messages
+  -k, --keep       Keep torrent files localy
   -c, --countdown  Start with a 5 second countdown
 ```
diff --git a/config.template.json b/config.template.json
index b02c56d..6198946 100644
--- a/config.template.json
+++ b/config.template.json
@@ -1,8 +1,9 @@
 {
-    "mame-directory":"/path/to/your/locol/mame/torrent/target/folders",
+    "mame-directory":"/path/to/your/local/mame/files/target/folder",
+    "torrent-directory":"/path/to/your/local/mame/torrent/target/folder",
     "transmission-user":"user",
     "transmission-password":"pass",
     "pleasuredome-user":"user",
     "pleasuredome-password":"pass",
     "transmission-port":8080
-}
\ No newline at end of file
+}