make cElementTree an alias for ElementTree

1 files changed, 14 insertions, 20 deletions

diff --git a/defusedxml/cElementTree.py b/defusedxml/cElementTree.py
index 2b6364e..565ab66 100644
--- a/defusedxml/cElementTree.py
+++ b/defusedxml/cElementTree.py
@@ -5,20 +5,20 @@
 # See https://www.python.org/psf/license for licensing details.
 """Defused xml.etree.cElementTree
 """
-from __future__ import absolute_import
-
 import warnings
 
-from .common import _generate_etree_functions
-
-from xml.etree.cElementTree import TreeBuilder as _TreeBuilder
-from xml.etree.cElementTree import parse as _parse
-from xml.etree.cElementTree import tostring
-
-# iterparse from ElementTree!
-from xml.etree.ElementTree import iterparse as _iterparse
-
-from .ElementTree import DefusedXMLParser
+# This module is an alias for ElementTree just like xml.etree.cElementTree
+from .ElementTree import (
+    XML,
+    XMLParse,
+    XMLParser,
+    XMLTreeBuilder,
+    fromstring,
+    iterparse,
+    parse,
+    tostring,
+    DefusedXMLParser,
+)
 
 __origin__ = "xml.etree.cElementTree"
 
@@ -29,14 +29,6 @@ warnings.warn(
     stacklevel=2,
 )
 
-# XMLParse is a typo, keep it for backwards compatibility
-XMLTreeBuilder = XMLParse = XMLParser = DefusedXMLParser
-
-parse, iterparse, fromstring = _generate_etree_functions(
-    DefusedXMLParser, _TreeBuilder, _parse, _iterparse
-)
-XML = fromstring
-
 __all__ = [
     "XML",
     "XMLParse",
@@ -46,4 +38,6 @@ __all__ = [
     "iterparse",
     "parse",
     "tostring",
+    # backwards compatibility
+    "DefusedXMLParser",
 ]