X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/f3e3374103ee892540927b65acc2231a10e493b7..d329e7e5c88a9445e10083536ae1549e6fc1a1be:/librarian/xmlutils.py diff --git a/librarian/xmlutils.py b/librarian/xmlutils.py index 345e011..bbcc884 100644 --- a/librarian/xmlutils.py +++ b/librarian/xmlutils.py @@ -12,8 +12,9 @@ class Xmill(object): Used instead of XSLT which is difficult and cumbersome. """ - def __init__(self, options=None): + def __init__(self, options=None, state=None): self._options = [] + self.state = state or {} if options: self._options.append(options) self.text_filters = [] @@ -22,6 +23,9 @@ class Xmill(object): def register_text_filter(self, fun): self.text_filters.append(fun) + def register_escaped_text_filter(self, fun): + self.escaped_text_filters.append(fun) + def filter_text(self, text): for flt in self.text_filters: if text is None: @@ -100,6 +104,7 @@ class Xmill(object): if isinstance(element, etree._Comment): return None handler = self._handle_for_element(element) + if self.state.get('mute') and not getattr(handler, 'unmuter', False): return None # How many scopes try: options_scopes = len(self._options) @@ -124,6 +129,7 @@ class Xmill(object): finally: # clean up option scopes if necessary self._options = self._options[0:options_scopes] + return out