o e@sHdZddlmZddlmZddlmZddlmZGdddeZd S) aTable block object parsed from raw image and text blocks. Data Structure:: { 'type': int 'bbox': (x0, y0, x1, y1), 'rows': [ { "bbox": (x0, y0, x1, y1), "height": float, "cells": [ { 'bbox': (x0, y0, x1, y1), 'border_color': (sRGB,,,), # top, right, bottom, left 'bg_color': sRGB, 'border_width': (,,,), 'merged_cells': (x,y), # this is the bottom-right cell of merged region: x rows, y cols 'blocks': [ {text blocks} ] }, # end of cell {}, None, # merged cell ... ] }, # end of row {...} # more rows ] # end of row } )Row)Rows)Block)docxcseZdZdZd%deffdd ZddZdd Zd d Ze d d Z e ddZ e ddZ e ddZ defddZfddZdefddZdefddZdd Zd!d"Zd#d$ZZS)& TableBlockz Table block.Nrawcs>|duri}t|t|d|dg|_|dS)N)parentrows)super__init__rrestoreget_rowsZset_lattice_table_block)selfr __class__ID:\Projects\ConvertPro\env\Lib\site-packages\pdf2docx/table/TableBlock.pyr *s   zTableBlock.__init__cCs4z|j|}W|Styd|d}t|w)Nz Row index z out of range)r IndexError)ridxrowmsgrrr __getitem__4s  zTableBlock.__getitem__cCdd|jDS)Ncss|]}|VqdSNr.0rrrr >sz&TableBlock.__iter__..rrrrr__iter__=szTableBlock.__iter__cC t|jSrlenrr rrr__len__@s zTableBlock.__len__cCr")zCount of rows.r#r rrrnum_rowsCs zTableBlock.num_rowscCs|jr t|jdSdS)zCount of columns.)r&r$rr rrrnum_colsHszTableBlock.num_colscCr)zGet text contained in each cell. Returns: list: 2D-list with each element representing text in cell. cSsg|] }dd|DqS)cSsg|]}|jqSr)text)rcellrrr Tsz.TableBlock.text...rrrrrr+Tsz#TableBlock.text..rr rrrr)MszTableBlock.textc Csf|j\}}}}|ddj\}}}}|ddj\} } } } ||d||d|| d|| dfS)z"Outer bbox with border considered.r'g@)ZbboxZ border_width) rZx0Zy0x1y1Zw0_topZw0_rightZ w0_bottomZw0_leftZw1_topZw1_rightZ w1_bottomZw1_leftrrr outer_bboxVs,zTableBlock.outer_bboxrcCs|j|dS)zrAppend row to table and update bbox accordingly. Args: row (Row): Target row to add. N)rappend)rrrrrr0_szTableBlock.appendcs"t}|d|ji|S)Nr )r storeupdater)rresrrrr1hs  zTableBlock.storeblockscC*|jD]}|D] }|s q||qqdS)zuAssign ``blocks`` to associated cell. Args: blocks (list): A list of text/table blocks. N)r assign_blocks)rr4rr*rrrr6p  zTableBlock.assign_blocksshapescCr5)ziAssign ``shapes`` to associated cell. Args: shapes (list): A list of Shape. N)r assign_shapes)rr8rr*rrrr9|r7zTableBlock.assign_shapescKs0|jD]}|D] }|s q|jdi|qqdS)znParse layout under cell level. Args: settings (dict): Layout parsing parameters. Nr)rparse)rsettingsrr*rrrr:s zTableBlock.parsecCr5)a`Plot table block, i.e. cell/line/span, for debug purpose. Args: page (fitz.Page): pdf page. content (bool): Plot text blocks contained in cells if True. style (bool): Plot cell style if True, e.g. border width, shading. color (bool): Plot border stroke color if ``style=False``. N)rplot)rpagerr*rrrr<s  zTableBlock.plotcCs8t||jtt|jD] }|j|||qdS)zmCreate docx table. Args: table (Table): ``python-docx`` table instance. N)rZ indent_tableZ left_spaceranger$r r make_docx)rtableZidx_rowrrrr?szTableBlock.make_docxr)__name__ __module__ __qualname____doc__dictr rr!r%propertyr&r(r)r/rr0r1listr6r9r:r<r? __classcell__rrrrr(s*          rN)rDrrZ common.Blockrcommonrrrrrrs