From e0b022e52d720758023bb0248ea0a17a4cbbb822 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sat, 7 Feb 2015 18:42:15 +0000 Subject: Pass function to create writer into HtmlGenerator --- mammoth/html_generation.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) (limited to 'mammoth/html_generation.py') diff --git a/mammoth/html_generation.py b/mammoth/html_generation.py index 6dd0b33..a0c35b7 100644 --- a/mammoth/html_generation.py +++ b/mammoth/html_generation.py @@ -1,12 +1,11 @@ from __future__ import unicode_literals -from .writers.html import HtmlWriter - class HtmlGenerator(object): - def __init__(self): + def __init__(self, create_writer): self._stack = [] - self._writer = HtmlWriter() + self._create_writer = create_writer + self._writer = create_writer() def text(self, text): if text: @@ -50,7 +49,7 @@ class HtmlGenerator(object): return self._writer.as_string() def child(self): - return HtmlGenerator() + return HtmlGenerator(self._create_writer) class _Element(object): -- cgit v1.2.3