def parse(self, message):
"""Given a string of text in the original CIA commit format, return a <colorText>
element representing it as a DOM tree.
"""
# Initialize our model of the current text format in the original message
self.parsedState = ColorState()
self.document = XML.createRootNode()
# Initialize our stack of (element, ColorState) tuples representing
# the state of the XML document being generated. This starts out with
# our root element in it.
self.elementStack = [
(XML.addElement(self.document, "colorText"), ColorState())
]
# Break up the message into lines, each with its whitespace stripped.
# Run our lexical scanner on each line separately, turning it into
# a stream of events. Insert <br/> tags between lines.
lines = []
for line in message.split("\n"):
# Ignore extra whitespace
line = line.strip()
# Ignore blank lines
if line:
lines.append(line)
for i in xrange(len(lines)):
if i != 0:
XML.addElement(self.elementStack[-1][0], 'br')
self.lex(lines[i])
self.closeTags()
return self.document
def preprocess(self):
message = XML.dig(self.xml, "message")
if not message:
raise XML.XMLValidityError("A Message's root node must be named 'message'")
# Stamp it with the current time if it has no timestamp yet
if not XML.dig(message, "timestamp"):
XML.addElement(message, "timestamp", "%d" % time.time())
def getXMLMailHeaders(self, document):
"""Return a <mailHeaders> tag representing a subset of the headers
for this message. This is placed in the <generator> tag of any
message passing through this module, to document and log the
message's true source.
"""
node = document.createElementNS(None, "mailHeaders")
for name, value in self.message.items():
if name in interestingHeaders:
XML.addElement(node, "header", content=str(value)).setAttributeNS(None, 'name', name)
return node
def pushTag(self, name, attributes={}, stateChanges={}):
"""Add a new element to the elementStack, placed at
the end of the children list for the tag currently
at the top of the stack.
name: The name of the new element
attributes: A dict of attributes to set on the new element
stateChanges: A dict of attributes to change in the new tag's state
"""
oldTag, oldState = self.elementStack[-1]
newTag = XML.addElement(self.elementStack[-1][0], name)
for key, value in attributes.iteritems():
newTag.setAttributeNS(None, key, value)
newState = copy.deepcopy(oldState)
newState.__dict__.update(stateChanges)
self.elementStack.append((newTag, newState))
def getLocalGenerator(self, document):
"""Return a <generator> tag for messages produced locally"""
node = document.createElementNS(None, "generator")
XML.addElement(node, "name", content="CIA IncomingMailParser")
return node
