aboutsummaryrefslogtreecommitdiff
path: root/scripts/rpki/sax_handler.py
blob: b4259d2aae3e92c083e310f8cdf801616cbd21f4 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
# $Id$

import xml.sax

def snarf(obj, attrs, key, func=None):
  """
  Utility function to consolidate the steps needed to extract a field
  from the SAX XML parse and insert it as an object attribute of the
  same name.
  """

  try:
    val = attrs.getValue(key).encode("ascii")
    if func:
      val = func(val)
  except KeyError:
    val = None
  setattr(obj, key, val)

class handler(xml.sax.handler.ContentHandler):
  """
  SAX handler for RPKI protocols.  Handles a few tasks
  common to all of these protocols, needs to be subtyped
  to handle protocol-specific details.
  """

  def __init__(self):
    self.text = ""
    self.obj = None

  def startElementNS(self, name, qname, attrs):
    return self.startElement(name[1], attrs)

  def endElementNS(self, name, qname):
    return self.endElement(name[1])

  def characters(self, content):
    self.text += content