selectolax 0.3.30__cp310-cp310-macosx_10_9_x86_64.whl → 0.3.32__cp310-cp310-macosx_10_9_x86_64.whl

selectolax/__init__.py

@@ -3,7 +3,7 @@
 
 __author__ = """Artem Golubin"""
 __email__ = 'me@rushter.com'
-__version__ = '0.3.30'
+__version__ = '0.3.32'
 
 from . import parser
 from . import lexbor

selectolax/lexbor/attrs.pxi

@@ -1,5 +1,7 @@
 cimport cython
 
+from typing import Optional
+
 @cython.final
 cdef class LexborAttributes:
     """A dict-like object that represents attributes."""
@@ -23,16 +25,32 @@ cdef class LexborAttributes:
                 yield key.decode(_ENCODING)
             attr = attr.next
 
-    def __setitem__(self, str key, value):
-        value = str(value)
+    def __setitem__(self, str key, object value):
+        value = value
         bytes_key = key.encode(_ENCODING)
-        bytes_value = value.encode(_ENCODING)
-
-        lxb_dom_element_set_attribute(
-            <lxb_dom_element_t *> self.node,
-            <lxb_char_t *> bytes_key, len(bytes_key),
-            <lxb_char_t *> bytes_value, len(bytes_value),
-        )
+        bytes_value = value.encode(_ENCODING) if value else b""
+        cdef lxb_dom_attr_t *attr
+        cdef lxb_dom_document_t *doc
+
+        if value is None:
+            # N.B. This is suboptimal, but there is not API to set empty attributes
+            attr = lxb_dom_element_set_attribute(
+                <lxb_dom_element_t *> self.node,
+                <lxb_char_t *> bytes_key, len(bytes_key),
+                NULL, 0
+            )
+            doc = (<lxb_dom_node_t*>attr).owner_document
+            lexbor_str_destroy(attr.value, doc.text, 0)
+            attr.value = NULL
+
+        elif isinstance(value, str) or isinstance(value, unicode) :
+            lxb_dom_element_set_attribute(
+                <lxb_dom_element_t *> self.node,
+                <lxb_char_t *> bytes_key, len(bytes_key),
+                <lxb_char_t *> bytes_value, len(bytes_value),
+            )
+        else:
+            raise TypeError("Expected str or unicode, got %s" % type(value))
 
     def __delitem__(self, key):
         try:

selectolax/lexbor/node.pxi

@@ -273,6 +273,9 @@ cdef class LexborNode:
         >>>     tag.decompose()
 
         """
+        if self.node == <lxb_dom_node_t *> lxb_dom_document_root(&self.parser.document.dom_document):
+            raise SelectolaxError("Decomposing the root node is not allowed.")
+
         if recursive:
             lxb_dom_node_destroy_deep(<lxb_dom_node_t *> self.node)
         else:

selectolax/lexbor/selection.pxi

@@ -75,6 +75,7 @@ cdef class LexborCSSSelector:
         status = lxb_selectors_find(self.selectors, node.node, selectors_list,
                                     <lxb_selectors_cb_f> css_matcher_callback, <void *> self)
         if status != LXB_STATUS_OK:
+            lxb_css_selector_list_destroy_memory(selectors_list)
             raise SelectolaxError("Can't parse CSS selector.")
         result = bool(self.results)
         self.results = []