gh-54873: Add support for namespaces prefixes to xml.sax.expatreader

Doc/whatsnew/3.15.rst

@@ -660,6 +660,9 @@ xml.parsers.expat
 
   .. _billion laughs: https://en.wikipedia.org/wiki/Billion_laughs_attack
 
+* Add support for namespace prefixes.
+  (Contributed by Yassir Karroum in :gh:`118317`.)
+
 
 zlib
 ----

Lib/test/test_sax.py

@@ -1,7 +1,7 @@
 # regression test for SAX 2.0
 
 from xml.sax import make_parser, ContentHandler, \
-                    SAXException, SAXReaderNotAvailable, SAXParseException
+                    SAXException, SAXReaderNotAvailable, SAXParseException, handler
 import unittest
 from unittest import mock
 try:
@@ -1307,6 +1307,38 @@ def test_expat_locator_withinfo_nonascii(self):
         self.assertEqual(parser.getSystemId(), fname)
         self.assertEqual(parser.getPublicId(), None)
 
+    def test_namespace_prefix(self):
+        parser = create_parser()
+        parser.setFeature(handler.feature_namespaces, 1)
+        parser.setFeature(handler.feature_namespace_prefixes, 1)
+
+        class Handler(handler.ContentHandler):
+            def startElementNS(self, name, qname, attrs):
+                self.qname = qname
+
+        h = Handler()
+
+        parser.setContentHandler(h)
+        parser.feed("<Q:E xmlns:Q='http://example.org/testuri'/>")
+        parser.close()
+        self.assertEqual(h.qname, "Q:E")
+
+    def test_default_namespace(self):
+        parser = create_parser()
+        parser.setFeature(handler.feature_namespaces, 1)
+
+        class Handler(handler.ContentHandler):
+            def startElementNS(self, name, qname, attrs):
+                self.qname = qname
+
+        h = Handler()
+
+        parser.setContentHandler(h)
+        parser.feed("<E xmlns='http://example.org/testuri'/>")
+        parser.close()
+        self.assertEqual(h.qname, "E")
+
+
 
 # ===========================================================================
 #

Lib/xml/sax/expatreader.py

@@ -91,6 +91,7 @@ def __init__(self, namespaceHandling=0, bufsize=2**16-20):
         self._entity_stack = []
         self._external_ges = 0
         self._interning = None
+        self._namespace_prefixes = 1
 
     # XMLReader methods
 
@@ -126,8 +127,9 @@ def getFeature(self, name):
             return self._namespaces
         elif name == feature_string_interning:
             return self._interning is not None
-        elif name in (feature_validation, feature_external_pes,
-                      feature_namespace_prefixes):
+        elif name == feature_namespace_prefixes:
+            return self._namespace_prefixes
+        elif name in (feature_validation, feature_external_pes):
             return 0
         elif name == feature_external_ges:
             return self._external_ges
@@ -147,6 +149,8 @@ def setFeature(self, name, state):
                     self._interning = {}
             else:
                 self._interning = None
+        elif name == feature_namespace_prefixes:
+            self._namespace_prefixes = state
         elif name == feature_validation:
             if state:
                 raise SAXNotSupportedException(
@@ -155,10 +159,6 @@ def setFeature(self, name, state):
             if state:
                 raise SAXNotSupportedException(
                     "expat does not read external parameter entities")
-        elif name == feature_namespace_prefixes:
-            if state:
-                raise SAXNotSupportedException(
-                    "expat does not report namespace prefixes")
         else:
             raise SAXNotRecognizedException(
                 "Feature '%s' not recognized" % name)
@@ -347,11 +347,14 @@ def start_element_ns(self, name, attrs):
         pair = name.split()
         if len(pair) == 1:
             # no namespace
+            elem_qname = name
             pair = (None, name)
         elif len(pair) == 3:
+            elem_qname = "%s:%s" % (pair[2], pair[1])
             pair = pair[0], pair[1]
         else:
             # default namespace
+            elem_qname = pair[1]
             pair = tuple(pair)
 
         newattrs = {}
@@ -374,7 +377,7 @@ def start_element_ns(self, name, attrs):
             newattrs[apair] = value
             qnames[apair] = qname
 
-        self._cont_handler.startElementNS(pair, None,
+        self._cont_handler.startElementNS(pair, elem_qname,
                                           AttributesNSImpl(newattrs, qnames))
 
     def end_element_ns(self, name):