You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

60 lines
2.3 KiB

4 years ago
  1. # Copyright (c) 2006,2007 Mitch Garnaat http://garnaat.org/
  2. #
  3. # Permission is hereby granted, free of charge, to any person obtaining a
  4. # copy of this software and associated documentation files (the
  5. # "Software"), to deal in the Software without restriction, including
  6. # without limitation the rights to use, copy, modify, merge, publish, dis-
  7. # tribute, sublicense, and/or sell copies of the Software, and to permit
  8. # persons to whom the Software is furnished to do so, subject to the fol-
  9. # lowing conditions:
  10. #
  11. # The above copyright notice and this permission notice shall be included
  12. # in all copies or substantial portions of the Software.
  13. #
  14. # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  15. # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
  16. # ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
  17. # SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
  18. # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19. # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  20. # IN THE SOFTWARE.
  21. import xml.sax
  22. from boto.compat import StringIO
  23. class XmlHandler(xml.sax.ContentHandler):
  24. def __init__(self, root_node, connection):
  25. self.connection = connection
  26. self.nodes = [('root', root_node)]
  27. self.current_text = ''
  28. def startElement(self, name, attrs):
  29. self.current_text = ''
  30. new_node = self.nodes[-1][1].startElement(name, attrs, self.connection)
  31. if new_node is not None:
  32. self.nodes.append((name, new_node))
  33. def endElement(self, name):
  34. self.nodes[-1][1].endElement(name, self.current_text, self.connection)
  35. if self.nodes[-1][0] == name:
  36. if hasattr(self.nodes[-1][1], 'endNode'):
  37. self.nodes[-1][1].endNode(self.connection)
  38. self.nodes.pop()
  39. self.current_text = ''
  40. def characters(self, content):
  41. self.current_text += content
  42. class XmlHandlerWrapper(object):
  43. def __init__(self, root_node, connection):
  44. self.handler = XmlHandler(root_node, connection)
  45. self.parser = xml.sax.make_parser()
  46. self.parser.setContentHandler(self.handler)
  47. self.parser.setFeature(xml.sax.handler.feature_external_ges, 0)
  48. def parseString(self, content):
  49. return self.parser.parse(StringIO(content))