TLDR-773 docx table bug fix (#480)

Co-authored-by: Alexander Golodkov <golodkov@ispras.ru>
ispras · Aug 5, 2024 · ba9b3b4 · ba9b3b4
1 parent 5c597a0
commit ba9b3b4
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/dedoc/readers/docx_reader/data_structures/table.py b/dedoc/readers/docx_reader/data_structures/table.py
@@ -54,11 +54,14 @@ def to_table(self) -> Table:
                 if cell.vMerge:
                     value = cell.vMerge.get("w:val", "continue")
                     if value == "continue":
-                        cell_lines = cell_list[-1][cell_ind].lines
-                        cell_row_list.append(CellWithMeta(lines=cell_lines, colspan=1, rowspan=1, invisible=True))
-                        last_cell_rowspan = cell_list[rowspan_start_info[cell_ind]][cell_ind]
-                        last_cell_rowspan.rowspan += 1
-                        cell_list[rowspan_start_info[cell_ind]][cell_ind] = last_cell_rowspan
+                        if cell_ind in rowspan_start_info:
+                            cell_lines = cell_list[-1][cell_ind].lines
+                            cell_row_list.append(CellWithMeta(lines=cell_lines, colspan=1, rowspan=1, invisible=True))
+                            last_cell_rowspan = cell_list[rowspan_start_info[cell_ind]][cell_ind]
+                            last_cell_rowspan.rowspan += 1
+                            cell_list[rowspan_start_info[cell_ind]][cell_ind] = last_cell_rowspan
+                        else:
+                            cell_row_list.append(CellWithMeta(lines=cell_lines, colspan=grid_span, rowspan=1, invisible=False))
                     elif value == "restart":
                         rowspan_start_info[cell_ind] = row_index
                         cell_row_list.append(CellWithMeta(lines=cell_lines, colspan=grid_span, rowspan=1, invisible=False))