Add __h__() for simple representations

breuleux · Jun 13, 2024 · 00baf4a · 00baf4a
1 parent 8292237
commit 00baf4a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/hrepr/hgen.py b/hrepr/hgen.py
@@ -118,7 +118,10 @@ def process(self, node: Union[str, TextFormatter]):
 
     @ovld
     def process(self, node: object):
-        return str(node)
+        if hasattr(node, "__h__"):
+            return self.process(node.__h__())
+        else:
+            return str(node)
 
     @ovld
     def process(self, node: Tag):

diff --git a/tests/test_h.py b/tests/test_h.py
@@ -93,6 +93,15 @@ def test_style():
     )
 
 
+class Hobbit:
+    def __h__(self):
+        return H.b("preciouss")
+
+
+def test_dunder_h():
+    assert matches(H.div(Hobbit()), "<div><b>preciouss</b></div>")
+
+
 @one_test_per_assert
 def test_voids():
     assert matches(H.area(), "<area />")