genshi/mirror: genshi/filters/transform.py comparison

comparison genshi/filters/transform.py @ 720:acf7c5ee36e7 experimental-newctxt

newctxt branch: Merged revisions [678:835] via svnmerge from [source:trunk].

author	cmlenz
date	Fri, 11 Apr 2008 08:42:11 +0000
parents	4d486f15c986
children

comparison

equal deleted inserted replaced

-:837786a584d5
+:acf7c5ee36e7
 </body>
 </html>
 The ``Transformer`` support a large number of useful transformations out of the
 box, but custom transformations can be added easily.
+:since: version 0.5
 """
 import re
 import sys
 from genshi.builder import Element
-from genshi.core import Stream, Attrs, QName, TEXT, START, END, _ensure
+from genshi.core import Stream, Attrs, QName, TEXT, START, END, _ensure, Markup
 from genshi.path import Path
 __all__ = ['Transformer', 'StreamBuffer', 'InjectorTransformation', 'ENTER',
 'EXIT', 'INSIDE', 'OUTSIDE']
 class="emphasis">body</em> text.</body></html>
 """
 __slots__ = ['transforms']
-def __init__(self, path=None):
+def __init__(self, path='.'):
 """Construct a new transformation filter.
 :param path: an XPath expression (as string) or a `Path` instance
 """
-self.transforms = []
+self.transforms = [SelectTransformation(path)]
-if path:
-self.transforms.append(SelectTransformation(path))
 def __call__(self, stream):
 """Apply the transform filter to the marked stream.
 :param stream: the marked event stream to filter
 :rtype: `Stream`
 """
 transforms = self._mark(stream)
 for link in self.transforms:
 transforms = link(transforms)
-return Stream(self._unmark(transforms))
+return Stream(self._unmark(transforms),
+serializer=getattr(stream, 'serializer', None))
 def apply(self, function):
 """Apply a transformation to the stream.
 Transformations can be chained, similar to stream filters. Any callable
 >>> html = HTML('<html><body>Some text, some more text and '
 ...             '<b>some bold text</b></body></html>')
 >>> print html | Transformer('body').substitute('(?i)some', 'SOME')
 <html><body>SOME text, some more text and <b>SOME bold text</b></body></html>
+>>> tags = tag.html(tag.body('Some text, some more text and ',
+...      Markup('<b>some bold text</b>')))
+>>> print tags.generate() | Transformer('body').substitute('(?i)some', 'SOME')
+<html><body>SOME text, some more text and <b>SOME bold text</b></body></html>
 :param pattern: A regular expression object or string.
 :param replace: Replacement pattern.
 :param count: Number of replacements to make in each text fragment.
 :rtype: `Transformer`
 """
 return self.apply(SubstituteTransformation(pattern, replace, count))
+def rename(self, name):
+"""Rename matching elements.
+>>> html = HTML('<html><body>Some text, some more text and '
+...             '<b>some bold text</b></body></html>')
+>>> print html | Transformer('body/b').rename('strong')
+<html><body>Some text, some more text and <strong>some bold text</strong></body></html>
+"""
+return self.apply(RenameTransformation(name))
 def trace(self, prefix='', fileobj=None):
 """Print events as they pass through the transform.
 >>> html = HTML('<body>Some <em>test</em> text</body>')
 element = list(self.element.generate())
 for prefix in element[:-1]:
 yield None, prefix
 yield mark, event
 while True:
-mark, event = stream.next()
+try:
+mark, event = stream.next()
+except StopIteration:
+yield None, element[-1]
 if not mark:
 break
 yield mark, event
 yield None, element[-1]
 yield mark, event
 queue = []
 for mark, event in stream:
 if mark:
 queue.append(event)
 else:
-for event in flush(queue):
+for queue_event in flush(queue):
-yield event
+yield queue_event
 yield None, event
 for event in flush(queue):
 yield event
 :param stream: The marked event stream to filter
 """
 for mark, (kind, data, pos) in stream:
 if kind is TEXT:
-data = self.pattern.sub(self.replace, data, self.count)
+new_data = self.pattern.sub(self.replace, data, self.count)
+if isinstance(data, Markup):
+data = Markup(new_data)
+else:
+data = new_data
+yield mark, (kind, data, pos)
+class RenameTransformation(object):
+"""Rename matching elements."""
+def __init__(self, name):
+"""Create the transform.
+:param name: New element name.
+"""
+self.name = QName(name)
+def __call__(self, stream):
+"""Apply the transform filter to the marked stream.
+:param stream: The marked event stream to filter
+"""
+for mark, (kind, data, pos) in stream:
+if mark is ENTER:
+data = self.name, data[1]
+elif mark is EXIT:
+data = self.name
 yield mark, (kind, data, pos)
 class InjectorTransformation(object):
 """Abstract base class for transformations that inject content into a
 """Apply the transform filter to the marked stream.
 :param stream: The marked event stream to filter
 """
 for mark, event in stream:
-if mark in (ENTER, OUTSIDE):
+if mark is not None:
 for subevent in self._inject():
 yield subevent
+yield mark, event
+while True:
+mark, event = stream.next()
+if not mark:
+break
+yield mark, event
 yield mark, event
 class AfterTransformation(InjectorTransformation):
 """Insert content after selection."""
 """
 for mark, event in stream:
 yield mark, event
 if mark:
 while True:
-mark, event = stream.next()
+try:
+mark, event = stream.next()
+except StopIteration:
+break
 if not mark:
 break
 yield mark, event
 for subevent in self._inject():
 yield subevent

Mercurial > genshi > mirror

comparison genshi/filters/transform.py @ 720:acf7c5ee36e7 experimental-newctxt