Add conversion from document table to HTML table

2 files changed, 33 insertions, 1 deletions

diff --git a/mammoth/conversion.py b/mammoth/conversion.py
index a21a264..fa1333b 100644
--- a/mammoth/conversion.py
+++ b/mammoth/conversion.py
@@ -26,6 +26,9 @@ class DocumentConverter(object):
             documents.Text: self._convert_text,
             documents.Hyperlink: self._convert_hyperlink,
             documents.Tab: self._convert_tab,
+            documents.Table: self._convert_table,
+            documents.TableRow: self._convert_table_row,
+            documents.TableCell: self._convert_table_cell,
             documents.Image: convert_image or self._convert_image,
         }
 
@@ -72,6 +75,30 @@ class DocumentConverter(object):
         html_generator.text("\t")
     
     
+    def _convert_table(self, table, html_generator):
+        html_generator.end_all()
+        html_generator.start("table")
+        self._convert_elements_to_html(table.children, html_generator)
+        html_generator.end()
+    
+    
+    def _convert_table_row(self, table_row, html_generator):
+        html_generator.start("tr")
+        self._convert_elements_to_html(table_row.children, html_generator)
+        html_generator.end()
+    
+    
+    def _convert_table_cell(self, table_cell, html_generator):
+        html_generator.start("td")
+        for child in table_cell.children:
+            child_generator = HtmlGenerator()
+            self.convert_element_to_html(child, child_generator)
+            child_generator.end_all()
+            html_generator.append(child_generator)
+            
+        html_generator.end()
+    
+    
     def _convert_image(self, image, html_generator):
         with image.open() as image_bytes:
             encoded_src = base64.b64encode(image_bytes.read()).decode("ascii")
diff --git a/mammoth/documents.py b/mammoth/documents.py
index 703e436..7f2750c 100644
--- a/mammoth/documents.py
+++ b/mammoth/documents.py
@@ -6,6 +6,9 @@ Paragraph = dodge.data_class("Paragraph", ["children", "style_id", "style_name",
 Run = dodge.data_class("Run", ["children", "style_id", "style_name", "is_bold", "is_italic"])
 Text = dodge.data_class("Text", ["value"])
 Hyperlink = dodge.data_class("Hyperlink", ["href", "children"])
+Table = dodge.data_class("Table", ["children"])
+TableRow = dodge.data_class("TableRow", ["children"])
+TableCell = dodge.data_class("TableCell", ["children"])
 
 class Tab(object):
     pass
@@ -36,7 +39,9 @@ def tab():
 
 image = Image
 hyperlink = Hyperlink
-
+table = Table
+table_row = TableRow
+table_cell = TableCell
 
 def numbering_level(level_index, is_ordered):
     return _NumberingLevel(str(level_index), bool(is_ordered))