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)