diff --git a/molniya.py b/molniya.py index a222e78..4ad6589 100644 --- a/molniya.py +++ b/molniya.py @@ -21,10 +21,15 @@ CURRENT_URL_PARSED = urllib.parse.urlparse(CURRENT_URL) try: CURRENT_URL_INDEX = URLS.index(CURRENT_URL) except: - CURRENT_URL = MAIN_PAGE - # place the index somewhere random in the list - # this should alleviate the issue of what happens when someone not in the list tries to link to the list - CURRENT_URL_INDEX = random.randrange(len(URLS)) + # I wasn't clear about trailing slashes, maybe add one on? + CURRENT_URL += "/" + try: + CURRENT_URL_INDEX = URLS.index(CURRENT_URL) + except: + CURRENT_URL = MAIN_PAGE + # place the index somewhere random in the list + # this should alleviate the issue of what happens when someone not in the list tries to link to the list + CURRENT_URL_INDEX = random.randrange(len(URLS)) def next_url(): return URLS[(CURRENT_URL_INDEX+1)%len(URLS)]