Add very basic test script

cheuph/test.py

@@ -1,33 +0,0 @@
-import curses
-import subprocess
-import tempfile
-from typing import Any
-
-
-def main(stdscr: Any) -> None:
-    while True:
-        key = stdscr.getkey()
-
-        if key in {"\x1b", "q"}:
-            return
-
-        elif key == "e":
-            with tempfile.TemporaryDirectory() as tmpdirname:
-                tmpfilename = tmpdirname + "/" + "tempfile"
-                #stdscr.addstr(f"{curses.COLOR_PAIRS!r}\n")
-                stdscr.addstr(f"{tmpdirname!r} | {tmpfilename!r}\n")
-
-                stdscr.getkey()
-
-                curses.endwin()
-                subprocess.run(["nvim", tmpfilename])
-                stdscr.refresh()
-
-                stdscr.getkey()
-
-                with open(tmpfilename) as f:
-                    for line in f:
-                        stdscr.addstr(line)
-
-
-curses.wrapper(main)

test.py

@@ -0,0 +1,80 @@
+import curses
+import subprocess
+import tempfile
+from typing import Any, List, Optional
+
+from cheuph.element import Element, Id, RenderedElement
+from cheuph.element_supply import MemoryElementSupply
+from cheuph.markup import AttributedText
+from cheuph.tree_display import TreeDisplay
+
+
+class TestElement(Element):
+    DEPTHSTR = "| "
+
+    def __init__(self,
+            id: Id,
+            parent_id: Optional[Id],
+            text: List[str],
+            ) -> None:
+
+        super().__init__(id, parent_id)
+        self.text = text
+
+    def render(self,
+            width: int,
+            depth: int,
+            highlighted: bool = False,
+            folded: bool = False,
+            ) -> RenderedElement:
+
+        depth_text = self.DEPTHSTR * depth
+        lines = [f"{depth_text}{line}" for line in self.text]
+        attributed_lines = [AttributedText(line) for line in lines]
+        return RenderedElement(self, attributed_lines)
+
+def main(stdscr: Any) -> None:
+    messages = MemoryElementSupply()
+    messages.add(TestElement("a", None, ["test element a"]))
+    messages.add(TestElement("b", "a", ["test element b","child of a"]))
+    messages.add(TestElement("c", None, ["test element c"]))
+
+    display = TreeDisplay(messages, 80, 15)
+    display.anchor_id = "a"
+    display.anchor_offset = 5
+
+    display.rerender()
+    display.render_display_lines()
+
+    print("-"*80)
+    for line in display.display_lines:
+        print(str(line))
+    print("-"*80)
+
+#    while True:
+#        key = stdscr.getkey()
+#
+#        if key in {"\x1b", "q"}:
+#            return
+#
+#        elif key == "e":
+#            with tempfile.TemporaryDirectory() as tmpdirname:
+#                tmpfilename = tmpdirname + "/" + "tempfile"
+#                #stdscr.addstr(f"{curses.COLOR_PAIRS!r}\n")
+#                stdscr.addstr(f"{tmpdirname!r} | {tmpfilename!r}\n")
+#
+#                stdscr.getkey()
+#
+#                curses.endwin()
+#                subprocess.run(["nvim", tmpfilename])
+#                stdscr.refresh()
+#
+#                stdscr.getkey()
+#
+#                with open(tmpfilename) as f:
+#                    for line in f:
+#                        stdscr.addstr(line)
+
+
+#curses.wrapper(main)
+main(None)