diff --git a/generate.py b/generate.py index d7b99a9..ba7e5d8 100644 --- a/generate.py +++ b/generate.py @@ -18,17 +18,17 @@ except FileExistsError: rmtree(os.path.join(root, d)) #generate navigation bar -navbarfound = False +def generateNavigationBar(lines): + global navbar + navbar = E + for line in lines: + split = line.strip().split(";", 1) + navbar.li(HTML("" + split[0])) + if os.path.exists("./navbar"): with open("./navbar", 'r') as navbarfile: - navbarfound = True - navbar = E - for line in navbarfile.readlines(): - split = line.strip().split(";", 1) - navbar = navbar.li(HTML("" + split[0])) - -#checking false instead of else to be more sure -if navbarfound == False: + generateNavigationBar(navbarfile.readlines()) +else: print("No 'navbar' file found, there will be no navigation bar.") #generate pages @@ -41,13 +41,13 @@ for file in os.listdir("./"): doc.p(line.strip()) #creates ./website-output/pagetitle/index.html file if it is not homepage - foldername = "" - if pagetitle =! homepage: + foldername = "" + if pagetitle != homepage: foldername = "/" + pagetitle.replace(" ", "-").lower() os.mkdir("./website-output" + foldername) with open("./website-output" + foldername + "/index.html", 'w') as newpage: - if navbarfound: + if 'navbar' in globals(): newpage.write(str(E.ul(navbar))) else: newpage.write(str(doc))