aboutsummaryrefslogtreecommitdiff
path: root/mammoth/html_generation.py
diff options
context:
space:
mode:
Diffstat (limited to 'mammoth/html_generation.py')
-rw-r--r--mammoth/html_generation.py9
1 files changed, 4 insertions, 5 deletions
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):