Add reading of rowspan

1 files changed, 5 insertions, 2 deletions

diff --git a/mammoth/documents.py b/mammoth/documents.py
index 7f17f3e..36e3458 100644
--- a/mammoth/documents.py
+++ b/mammoth/documents.py
@@ -49,6 +49,7 @@ class TableRow(HasChildren):
 @cobble.data
 class TableCell(HasChildren):
     colspan = cobble.field()
+    rowspan = cobble.field()
 
 @cobble.data
 class LineBreak(Element):
@@ -111,10 +112,12 @@ bookmark = Bookmark
 
 table = Table
 table_row = TableRow
-def table_cell(children, colspan=None):
+def table_cell(children, colspan=None, rowspan=None):
     if colspan is None:
         colspan = 1
-    return TableCell(children=children, colspan=colspan)
+    if rowspan is None:
+        rowspan = 1
+    return TableCell(children=children, colspan=colspan, rowspan=rowspan)
 
 
 line_break = LineBreak