From bdadee8447c4936179688291da2b3ebb444ba8ef Mon Sep 17 00:00:00 2001 From: ArmelliniFederico Date: Mon, 26 Jul 2021 23:22:16 +0000 Subject: [PATCH] Update main.py --- main.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index bcf4780..098f7d1 100644 --- a/main.py +++ b/main.py @@ -6,6 +6,7 @@ from bs4 import BeautifulSoup import datetime import codecs +import sys try: from urllib.parse import urlparse, urljoin @@ -308,7 +309,11 @@ def write_index(index_links, base_output): global url_global start = "http://www.risultati-ammissione.polimi.it" - base_output = "D:\\git\\Polimi\\polinetwork.github.io\\graduatorie" + base_output = "" + if sys.arvg is None or len(sys.argv) < 2 or sys.argv[1] is None: + base_output = "D:\\git\\Polimi\\polinetwork.github.io\\graduatorie" + else: + base_output = sys.argv[1] start_len = len(start) generateUrl(start)