54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
import os
|
|
from html5tagger import Document, E, HTML
|
|
from shutil import rmtree
|
|
|
|
#if homepage is at Home.page, set homepage to "Home"
|
|
homepage = "Home"
|
|
|
|
try:
|
|
os.mkdir("./website-output")
|
|
except FileExistsError:
|
|
print("Output directory already exists, let's clean it!")
|
|
|
|
#deleting contents of folder without deleting the folder, to increase compatibility with various systems
|
|
for root, dirs, files in os.walk('./website-output'):
|
|
for f in files:
|
|
os.unlink(os.path.join(root, f))
|
|
for d in dirs:
|
|
rmtree(os.path.join(root, d))
|
|
|
|
#generate navigation bar
|
|
def generateNavigationBar(lines):
|
|
global navbar
|
|
navbar = E
|
|
for line in lines:
|
|
split = line.strip().split(";", 1)
|
|
navbar = navbar.li(HTML("<a href='" + split[1] + "'>" + split[0] + "</a>"))
|
|
|
|
if os.path.exists("./navbar"):
|
|
with open("./navbar", 'r') as navbarfile:
|
|
generateNavigationBar(navbarfile.readlines())
|
|
else:
|
|
print("No 'navbar' file found, there will be no navigation bar.")
|
|
|
|
#generate pages
|
|
for file in os.listdir("./"):
|
|
if file.endswith(".page"):
|
|
with open(file, 'r') as page:
|
|
pagetitle = os.path.basename(file).replace(".page", "")
|
|
doc = Document(pagetitle, lang="en")
|
|
for line in page.readlines():
|
|
doc.p(line.strip())
|
|
|
|
#creates ./website-output/pagetitle/index.html file if it is not homepage
|
|
foldername = ""
|
|
if pagetitle != homepage:
|
|
foldername = "/" + pagetitle.replace(" ", "-").lower()
|
|
os.mkdir("./website-output" + foldername)
|
|
print(foldername)
|
|
with open("./website-output" + foldername + "/index.html", 'w') as newpage:
|
|
if 'navbar' in globals():
|
|
newpage.write(str(E.ul(navbar)) + str(doc))
|
|
else:
|
|
newpage.write(str(doc))
|