o e@sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddZ d d Z dd d ZddZdddZdddZddZddZdS))absolute_import)html)Workbook)get_column_letter) PremailerTablecCs|rt|SdS)Nr)isdigitint)sr _D:\Projects\ConvertPro\env\Lib\site-packages\paddleocr/ppstructure/table/tablepyxl/tablepyxl.py string_to_int srcCs:t|}|d}|D]}|q dd|dDS)Nz //comment()cSsg|]}t|qSr r).0tabler r r szget_Tables..z//table)rZ fromstringZxpathZdrop_tag)doctreecommentscommentr r r get_Tabless   rcCsRddlm}|}|jD]}|jD]}|j||d}t||r.|d7}|j||d}t||st|jdd} t|jdd} | dksH| dkrY|j |||| d|| dd|j |_ | || d } | d } | dkrt |jt|jpzdt|j d } | r| | kr| } n| r| | kr| } | |jt|_|| 7}q|d7}|}q |S) z} Writes every tr child element of elem to a row in the worksheet returns the next row after all rows are written r) MergedCell)rowcolumnrcolspan1rowspan)Z start_rowZ start_columnZend_rowZ end_columnz min-widthz max-width)Zopenpyxl.cell.cellrrowscellscell isinstancerelementgetZ merge_cellsvalueformatZ get_dimensionmaxZcolumn_dimensionsrwidthlen) worksheetelemrrrZinitial_columnZ table_rowZ table_cellr!rr min_width max_widthr(r r r write_rowss<        $   r.cCs&|j|jdd}t||dddS)z Takes a table and workbook and writes the table to a new sheet. The sheet title will be the same as the table attribute name. name)titlerN)Z create_sheetr#r$ insert_table)rwbwsr r r table_to_sheetDsr4NcCsH|s t}||jt||dd}t|}|D]}t||q|S)z Takes a string representation of an html document and writes one sheet for every table in the document. The workbook is returned F)base_urlZremove_classes)rremoveactiverZ transformrr4)rr2r5Zinline_styles_docZtablesrr r r document_to_workbookMs  r8cCst||d}||dS)z Takes a string representation of an html document and writes one sheet for every table in the document. The workbook is written out to a file called filename )r5N)r8save)rfilenamer5r2r r r document_to_xl`s r;cCs4|jr t||j||}|jrt||j||}dSdSN)headr.body)rr*rrr r r r1is r1cCs&|j}|j|j}}t||||dS)zE Inserts a table at the location of an openpyxl Cell object. N)parentrrr1)rr!r3rrr r r insert_table_at_cellpsr@)r)NNr<) __future__rZlxmlrZopenpyxlrZopenpyxl.utilsrZ premailerrZtablepyxl.stylerrrr.r4r8r;r1r@r r r r s       +