Read colspan for table cells

1 files changed, 7 insertions, 2 deletions

diff --git a/mammoth/documents.py b/mammoth/documents.py
index d837556..7f17f3e 100644
--- a/mammoth/documents.py
+++ b/mammoth/documents.py
@@ -48,7 +48,7 @@ class TableRow(HasChildren):
 
 @cobble.data
 class TableCell(HasChildren):
-    pass
+    colspan = cobble.field()
 
 @cobble.data
 class LineBreak(Element):
@@ -111,7 +111,12 @@ bookmark = Bookmark
 
 table = Table
 table_row = TableRow
-table_cell = TableCell
+def table_cell(children, colspan=None):
+    if colspan is None:
+        colspan = 1
+    return TableCell(children=children, colspan=colspan)
+
+
 line_break = LineBreak
 
 def numbering_level(level_index, is_ordered):