fix bugs with saving magnets and torrents

vikstrous · Aug 31, 2015 · 82b9c77 · 82b9c77
1 parent cfb4a6c
commit 82b9c77
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pirate/print.py b/pirate/print.py
@@ -31,7 +31,7 @@ def print(*args, **kwargs):
         return builtins.print(*args, **kwargs)
 
 
-def search_results(mags, sizes, uploaded, local):
+def search_results(mags, sizes, uploaded, local=None):
     columns = int(os.popen('stty size', 'r').read().split()[1])
     cur_color = 'zebra_0'
 

diff --git a/pirate/torrent.py b/pirate/torrent.py
@@ -4,8 +4,10 @@
 import urllib.request as request
 import urllib.parse as parse
 import urllib.error
+import os.path
 
 import pirate.data
+from pirate.print import print
 
 from io import BytesIO
 
@@ -152,4 +154,4 @@ def save_magnets(chosen_links, mags, folder):
 
         print('Saved {:X} in {}'.format(info_hash, file))
         with open(file, 'w') as f:
-            f.write(magnet + '\n')
+            f.write(magnet + '\n')