fix: python-openxml#149 KeyError on Document.add_table()

Steve Canny · Steve Canny · commit 280d46ab4f00 · 2015-02-21T14:11:35.000-08:00
diff --git a/HISTORY.rst b/HISTORY.rst
@@ -3,6 +3,12 @@
 Release History
 ---------------
 
+NEXT
+++++++++++++++++++
+
+- Fix #149: KeyError on Document.add_table()
+
+
 0.8.4 (2015-02-20)
 ++++++++++++++++++
 
diff --git a/docx/document.py b/docx/document.py
@@ -89,11 +89,12 @@ def add_section(self, start_type=WD_SECTION.NEW_PAGE):
         new_sectPr.start_type = start_type
         return Section(new_sectPr)
 
-    def add_table(self, rows, cols, style='Light Shading Accent 1'):
+    def add_table(self, rows, cols, style=None):
         """
         Add a table having row and column counts of *rows* and *cols*
-        respectively and table style of *style*. If *style* is |None|, a
-        table with no style is produced.
+        respectively and table style of *style*. *style* may be a paragraph
+        style object or a paragraph style name. If *style* is |None|, the
+        table inherits the default table style of the document.
         """
         table = self._body.add_table(rows, cols)
         table.style = style
diff --git a/features/doc-add-table.feature b/features/doc-add-table.feature
@@ -5,10 +5,10 @@ Feature: Add a table
 
 
   Scenario: Add a table specifying only row and column count
-    Given a document having built-in styles
+    Given a blank document
      When I add a 2 x 2 table specifying only row and column count
      Then the document contains a 2 x 2 table
-      And table.style is styles['Light Shading Accent 1']
+      And table.style is styles['Normal Table']
 
 
   Scenario: Add a table specifying style
