Allow more than one ID to be passed

This commit is contained in:
Klaus-Uwe Mitterer 2017-02-09 00:59:32 +01:00
parent 49ef6f47cb
commit 45c95f45dc

View file

@ -82,10 +82,13 @@ def magazineHandler(driver, id, makepdf = True):
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("id", help="ID of the magazine to be downloaded. May be a URL.")
parser.add_argument("id", help="ID of the magazine to be downloaded. May be a URL.", nargs="+")
id = parser.parse_args().id.split("/")[-1].split("#")[0]
ids = parser.parse_args().id
driver = webdriver.Firefox()
for id in ids:
use = id.split("/")[-1].split("#")[0]
magazineHandler(driver, use)
magazineHandler(driver, id)
driver.quit()