From e52861b6e2c674239acdf9ab166571e8b63aa1bb Mon Sep 17 00:00:00 2001 From: daniel-Jones Date: Tue, 5 Jun 2018 16:50:48 +0930 Subject: added argument -o to specify output directory --- imagescraper/imagescraper.py | 9 ++++++--- 1 file 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); -- cgit v1.2.3