Starting to export tables to docx

Huyston · Mar 19, 2017 · 6d6720b · 6d6720b
1 parent 9fdb7c2
commit 6d6720b
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 0 deletions.
diff --git a/__pycache__/docxExporter.cpython-35.pyc b/__pycache__/docxExporter.cpython-35.pyc
diff --git a/docxExporter.py b/docxExporter.py
@@ -146,6 +146,30 @@ def addFigure(self,img,caption,source='',label='',width='0.5'):
         self.figNumber += 1
         self.document.add_paragraph() #pula linha
 
+    def addTable(self,table,caption,label):
+        cols = 0
+        for row in table.split('\n'):
+            rowCols = 0
+            if '||' in row:
+                rowCols = len(row.split('||'))
+            elif '|' in row:
+                rowCols = len(row.split('|'))
+            if rowCols > cols:
+                cols = rowCols
+
+        docxTable = self.document.add_table(rows=1,cols=cols)
+        for row in table.split('\n'):
+            if '||' in row:
+                headings = row.split('||')
+                hCells = docxTable.add_row().cells
+                for n,heading in enumerate(headings):
+                    hCells[n].text = heading
+            elif '|' in row:
+                colText = row.split('||')
+                hCells = docxTable.add_row().cells
+                for n,text in enumerate(colText):
+                    hCells[n].text = text
+
     def close(self):
         self.document.save(os.getcwd()+'/Archieves/'+self.filename+'.docx')
 

diff --git a/main.py b/main.py
@@ -598,6 +598,19 @@ def saveAs(self,docID,filename,docFormat,article=True):
                     exporter.addFigure(img,caption,source=source,label=label,width=width)
                 else:
                     raise NotImplemented
+            elif '!tab' in content:
+                caption = ''
+                label = ''
+                table = ''
+                for row in content.split('\n'):
+                    if '!tab' in row:
+                        label = row.replace('!tab ','')
+                    elif not '|' in row:
+                        if row:
+                            caption = row
+                    else:
+                        table += row+'\n'
+                exporter.addTable(table,caption,label)
             else:
                 if show:
                     exporter.addText(cell['output'])