added argument -o to specify output directory

1 files changed, 6 insertions, 3 deletions

diff --git a/imagescraper/imagescraper.py b/imagescraper/imagescraper.py
index 7d83227..da97ae6 100755
--- a/imagescraper/imagescraper.py
+++ b/imagescraper/imagescraper.py
@@ -75,12 +75,12 @@ def scrapelinks(animelinks):
             break;
     return links;
 
-def getcoverimage(link, animeid):
+def getcoverimage(link, directory, animeid):
     """
     download the cover image file and save it
     """
     print("downloading {}".format(link));
-    urllib.request.urlretrieve(link, "covers/{}.jpg".format(animeid))
+    urllib.request.urlretrieve(link, directory + "/{}.jpg".format(animeid))
 
 if __name__ == "__main__":
     """
@@ -93,6 +93,9 @@ if __name__ == "__main__":
     parser.add_argument("-d", "--database", type=str, action="store", dest="dbfile",
             default="../userdb.db", required=True,
             help="sqlite3 database file containing anime information");
+    parser.add_argument("-o", "--output", type=str, action="store", dest="outdir",
+            default="../output/covers", required=False,
+            help="directory to save images");
     args = parser.parse_args();
 
     # if our database (file) doesn't exist, exit
@@ -113,7 +116,7 @@ if __name__ == "__main__":
     # download cover images
     x = 0;
     for link in coverlinks:
-        getcoverimage(link, animeids[x]);
+        getcoverimage(link, args.outdir, animeids[x]);
         x += 1;
 
     dbclose(db);