Update to pitree.git commit 5cba525

[c_to_python.git] / xml_to_y.py

diff --git a/xml_to_y.py b/xml_to_y.py
index cff1c55..d056189 100755 (executable)
--- a/xml_to_y.py
+++ b/xml_to_y.py
@@ -1,8 +1,24 @@
 #!/usr/bin/env python3
 
-import element
 import sys
+import xml.etree.ElementTree
+
+def get_text(root, i):
+  if i < 0:
+    i += len(root) + 1
+  text = root.text if i == 0 else root[i - 1].tail
+  return '' if text is None else text
+
+def to_text(root):
+  return ''.join(
+    [
+      j
+      for i in range(len(root))
+      for j in [get_text(root, i), to_text(root[i])]
+    ] +
+    [get_text(root, len(root))]
+  )
 
 sys.stdout.write(
-  element.to_text(element.deserialize(sys.stdin))
+  to_text(xml.etree.ElementTree.parse(sys.stdin).getroot()[0])
 )