def _read_html(self, doctree, rawdata): htmlfile = StringIO() htmlfile.write(publish_from_doctree( doctree, writer_name='html', settings_overrides={'output_encoding': 'UTF-8'})) htmlfile.seek(0) return HtmlReader().read(htmlfile, rawdata)
class _MockLogger(object): def __init__(self): self._output = StringIO() def message(self, msg): self._output.write(msg.message) def value(self): return self._output.getvalue()
class ClosableOutput(object): def __init__(self, path): self._output = StringIO() self._path = path def __enter__(self): return self def __exit__(self, *args): self.close() def write(self, data): self._output.write(data) def close(self): self.value = self._output.getvalue() self._output.close() def __str__(self): return self._path