o 1j@sddlmZddlmZmZddlmZedrddlmZedr.dd l m Z dd l m Z ed r8dd l mZd dZedddZeddddZddZedd dddZd ddZddZddZdS)!)absolute_import)function_requires_depsis_dep_availableTablelxml)htmlopenpyxl)Workbook)get_column_letter premailer) PremailercCs|rt|SdS)z( Convert a string to an integer r)isdigitint)sre/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/utils/io/tablepyxl.py string_to_intsrcCs:t|}|d}|D]}|q dd|dDS)z( Find all the tables in the doc z //comment()cSsg|]}t|qSrr).0tablerrr /szget_Tables..z//table)r Z fromstringZxpathZdrop_tag)doctreecommentscommentrrr get_Tables&s   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_row start_columnZend_row end_columnz min-widthz max-width)Zopenpyxl.cell.cellrrowscellscell isinstancerelementgetZ merge_cellsvalueformatZ get_dimensionmaxZcolumn_dimensionsr widthlen) worksheetelemrr rZinitial_columnZ table_rowZ table_cellr)r!r# min_width max_widthr0rrr write_rows2sH              r6cCs&|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)rwbwsrrrtable_to_sheetesr<NcCsH|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)r removeZactiver transformrr<)rr:r=Zinline_styles_docZtablesrrrrdocument_to_workbookns  r@cCst||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 )r=N)r@save)rfilenamer=r:rrrdocument_to_xls rCcCs4|jr t||j||}|jrt||j||}dSdS)zL Inserts a table into the worksheet at the specified column and row N)headr6body)rr2r rrrrr9s r9cCs&|j}|j|j}}t||||dS)zE Inserts a table at the location of an openpyxl Cell object. N)parentr rr9)rr)r;r rrrrinsert_table_at_cellsrG)r)NN)N) __future__rZ utils.depsrrstylerr r r r Zopenpyxl.utilsr rrrrr6r<r@rCr9rGrrrrs*        2