gent1x.py 1.64 KB
Newer Older
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 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66
from xml.sax import handler, saxutils, make_parser
import sys

alt_other = "other"

alternative = sys.argv[1]
dixfile     = sys.argv[2]

if alternative == "other":
  alt_other = alt_other[::-1] # reverse string

def printAttrs(attrsmap):
  if len(attrsmap) == 0:
    return ""
  output = []
  for (i,j) in attrsmap.items():
    output.append(i)
    output.append("=\"")
    output.append(saxutils.escape(j))
    output.append("\" ")
  
  return " " + "".join(output).strip()    

class DixHandler(handler.ContentHandler):
  def __init__(self, out = sys.stdout):
    handler.ContentHandler.__init__(self)
    self._out = out
    self._buf = []
    
  def flush(self):
    for i in self._buf:
      self._out.write(i.encode("utf-8"))
    self._buf = []
    
        
  def startElement(self, name, attrs):
    if "alt" in attrs:
      attrs2 = {i:j for (i,j) in attrs.items() if i != "alt"}
      vals = attrs["alt"].split()
      if alternative in vals:
        attrs2["alt"] = alternative
      else:
        attrs2["alt"] = alt_other
      attrs = attrs2
    self._buf.append("<" + name + printAttrs(attrs) + ">")
    
  def endElement(self, name):
    if len(self._buf) > 0 and self._buf[-1][0:1] == "<" and self._buf[-1][-1:] == ">" and self._buf[-1][1:2] != "/":
      self._buf[-1] = self._buf[-1][0:-1]+"/>"
      self.flush()
    else:
      self.flush()
      self._out.write("</")
      self._out.write(name)
      self._out.write(">")
  
  def characters(self, content):
    self.flush()
    self._out.write(saxutils.escape(content).encode("utf-8"))
        
parser = make_parser()
parser.setContentHandler(DixHandler())

with open(dixfile, "r") as f:
  parser.parse(f)