Add _repr_html_() as an alias of __str__()

tvst · Nov 15, 2024 · 7e38bd0 · 7e38bd0
1 parent 39ca583
commit 7e38bd0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/htbuilder/__init__.py b/htbuilder/__init__.py
@@ -163,6 +163,9 @@ def __str__(self):
             else:
                 return "<%(tag)s>%(children)s</%(tag)s>" % args
 
+    def _repr_html_(self):
+        return str(self)
+
 
 def _clean_name(k):
     # This allows you to use reserved words by prepending/appending underscores.

diff --git a/tests/htbuild_test.py b/tests/htbuild_test.py
@@ -269,6 +269,13 @@ def test_repeat(self):
             <div foo="bar">hello</div>
         '''))
 
+    def test_repr_html(self):
+        dom = div("Exists!")
+        self.assertEqual(
+            dom._repr_html_(),
+            normalize_whitespace('<div>Exists!</div>'),
+        )
+
 
 if __name__ == '__main__':
     unittest.main()