Add "__getattr__()" method to ObjectifiedElement to support arbitrary…

… attribute lookups. Closes #61
lxml · Mar 4, 2022 · d29a5ee · d29a5ee
1 parent 0b1659f
commit d29a5ee
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/lxml-stubs/objectify.pyi b/lxml-stubs/objectify.pyi
@@ -6,7 +6,7 @@ from lxml.etree import ElementBase, XMLParser
 def __getattr__(name) -> Any: ...
 
 class ObjectifiedElement(ElementBase):
-    pass
+    def __getattr__(self, name) -> ObjectifiedElement: ...
 
 def fromstring(
     text: Union[bytes, str],

diff --git a/test-data/test_objectify.yml b/test-data/test_objectify.yml
@@ -0,0 +1,13 @@
+-   case: etree_from_string_returns_element
+    disable_cache: true
+    main: |
+        from lxml import objectify
+        document = objectify.fromstring("<doc></doc>")
+        reveal_type(document)  # N: Revealed type is "lxml.objectify.ObjectifiedElement"
+
+-   case: etree_from_string_returns_element
+    disable_cache: true
+    main: |
+        from lxml import objectify
+        xyz = objectify.fromstring("<doc><xyz/></doc>").xyz
+        reveal_type(xyz)  # N: Revealed type is "lxml.objectify.ObjectifiedElement"