From aacaccaba82a350771ac4acda732188ab2d2e805 Mon Sep 17 00:00:00 2001
From: Azalea <22280294+hykilpikonna@users.noreply.github.com>
Date: Thu, 13 Mar 2025 22:07:20 -0400
Subject: [PATCH] [+] Full tree generator script

---
 gentree.py | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 gentree.py

diff --git a/gentree.py b/gentree.py
new file mode 100644
index 0000000..387d33b
--- /dev/null
+++ b/gentree.py
@@ -0,0 +1,41 @@
+import time
+from pathlib import Path
+
+from hypy_utils import write
+
+import db
+
+
+src = Path(__file__).parent
+
+
+def indent(string: str, level: int):
+    # Indent each line by level spaces
+    return "\n".join(" " * level + line for line in string.split("\n"))
+
+
+def dfs(channel: str):
+    info = db.channel_info(channel)
+    out = f"""<span class="tree l{info.height}"><a href="https://t.me/{channel}">@{channel}</a> - {info.name}</span>\n"""
+    if not info.children:
+        return out
+
+    out += f"""<div class="container l{info.height}">\n"""
+    for child in info.children:
+        out += indent(dfs(child.username), 2)
+    out += f"""</div>\n"""
+    return out
+
+
+def gen_tree():
+    of = dfs("hykilp")
+    write(src / "public/index.html", (src / "public/layout-full-tree.html").read_text()
+          .replace("{{CONTENT}}", of))
+
+
+if __name__ == '__main__':
+    while True:
+        print("Generating tree...")
+        gen_tree()
+        print("Done! Sleeping for 60 seconds...")
+        time.sleep(60)