Update python script

baerchen201 · baerchen201 · commit 1eabf1038920 · 2024-10-15T20:12:12.000+02:00
diff --git a/build.py b/build.py
@@ -2,12 +2,24 @@
 import shutil
 
 from html.parser import HTMLParser
+from html import escape as html_escape
+from urllib.parse import quote_plus
 
 
-class _HTMLParser(HTMLParser):
+class NavHTMLParser(HTMLParser):
+    def __init__(self, *_, **__) -> None:
+        super().__init__(*_, *__)
+        self.title = None
+
     def handle_starttag(self, tag, attrs):
-        print("HTML Tag:", tag, attrs, sep="\n  ")
+        print("  HTML Tag:", tag, attrs, sep="\n    ")
+        if tag == "meta":
+            for attr in attrs:
+                if attr[0] == "nav-title":
+                    self.title = attr[1] if attr[1] else None
+
 
+html_files = {}
 
 for dir, subdirs, files in os.walk("./www"):
     if (not subdirs and not files) or (
@@ -24,9 +36,23 @@ def handle_starttag(self, tag, attrs):
 
         if file.endswith(".html"):
             try:
-                p = _HTMLParser()
+                print(os.path.join(dir, file), "START")
+                p = NavHTMLParser()
                 p.feed(open(os.path.join(dir, file), "rb").read().decode())
                 p.close()
+                html_files[os.path.join(dir[5:], file)] = p.title
                 print(os.path.join(dir, file), "SUCCESS")
             except Exception as e:
                 print(os.path.join(dir, file), e)
+
+if html_files:
+    with open("./www/nav.html", "wb") as f:
+        f.write("<><head><title>Navigation</title></head><>".encode())
+        for file, title in html_files.items():
+            f.write(
+                f'<div><a href="{quote_plus(file)}">{html_escape(file)}</a>'.encode()
+            )
+            if title:
+                f.write(f"&#58; {html_escape(title)}".encode())
+            f.write(f"</div>".encode())
+        f.write("</body></html>".encode())
