o e $@s`dZddlmZddlmZddlmZdZdZdZ eee fZ dd Z d d Z Gd d d Z dS)z# Tablib - reStructuredText Support ) zip_longest)median) TextWrapperleftcenterrightcCst|tr |dSt|S)Nzutf-8) isinstancebytesdecodestr)valuer F/var/www/html/venv/lib/python3.10/site-packages/tablib/formats/_rst.pyto_strs  rcCs"|rtdd|DddSdS)z} Return the length of the longest word in `text`. >>> _max_word_len('Python Module for Tabular Datasets') 8 css|]}t|VqdSNlen).0wordr r r sz _max_word_len..r)default)maxsplit)textr r r _max_word_lens"rc@seZdZdZdZdZeddZedefddZ eed fd d Z edd dZ edddZ eddZ eddZeddZd S) ReSTFormatrst)rPc Cs|jrdd|jD}dd|jD}nddt|jD}ddt|jD}|jD]0}tt|dr7|n|}t|D]\}}t|}|| t |t ||t |||<q>q+||fS)zn Returns a list of string lengths of each column, and a list of maximum word lengths. cSsg|]}t|gqSr rrhr r r +sz9ReSTFormat._get_column_string_lengths..cSsg|]}t|qSr )rrr r rr ,cSsg|]}gqSr r r_r r rr .cSsg|]}dqS)rr r"r r rr /r$values) headersrangewidthdictiterhasattrr% enumeraterappendrrr) clsdatasetcolumn_lengths word_lensrowr%ivalrr r r_get_column_string_lengths$s z%ReSTFormat._get_column_string_lengths|c stvrtddtfdd|r|dnd|r"d|ndd|dg}t|D]\}}||_t|} || } || q1t |ddi} fdd | D} fd d | D} | S) zJ Returns a table row of wrapped values as a list of lines z&Value of "justify" must be one of "{}"z", "cs.tkr ||Stkr||S||Sr) JUSTIFY_LEFTljustJUSTIFY_CENTERrrjust)text_width_justifyr rjustBs    z&ReSTFormat._row_to_lines..just fillvaluec3s(|]}fddt|DVqdS)c3s"|] \}}||VqdSrr )rr3 cell_liner?widthsr rrTs z5ReSTFormat._row_to_lines...N)r,rlinerDr rrSs  z+ReSTFormat._row_to_lines..cs"g|] }d|fqS)rA)joinrF)lpadpadrpadr rr Ws"z,ReSTFormat._row_to_lines..) JUSTIFY_VALUES ValueErrorformatrHzipr(rwrapr-r) r.r%rEwrappersepr>cellsr r(rcelllinesr )r?r>rIrJrKrEr _row_to_lines8s(     zReSTFormat._row_to_linescs||\}}dd|D}t|t|kr'fdd|D}ndd|D}fdd|D}ddt||D}|S)zw Returns a list of column widths proportional to the median length of the text in their cells. cSsg|]}tt|qSr )intrrlensr r rr asz1ReSTFormat._get_column_widths..c3s|] }|VqdSrr rY)max_table_widthtotalr rrdsz0ReSTFormat._get_column_widths..css|]}|VqdSrr rYr r rrfsc3s$|] }|kr |n|VqdSrr rwpad_lenr rrhs"cSsg|] \}}t||qSr )r)rr^lr r rr js)r5sumrrO)r.r/r[r`str_lensr1 median_lens column_widthsr )r[r`r\r_get_column_widthsZszReSTFormat._get_column_widthsNc Csg}t}|dur|j|dd}ddd|D}|||jr5||j|j||dtd|||jD]}t t |d rD| n|}|||||dq8||d |S) zP Returns reStructuredText grid table representation of dataset. Nr_z cSg|]}d|qS=r r]r r rr vr!z9ReSTFormat.export_set_as_simple_table..rA)rRr>r% rrfrHr-r&extendrVr9r)r*r+r%)r.r/rerUrQborderr2r%r r rexport_set_as_simple_tablems(     z%ReSTFormat.export_set_as_simple_tablec Csg}t}|dur||}dddd|Dd}ddd d|Dd }|||jrD||j|j||td |||jD]}t t |d rS| n|}|||||||qGd |S)a Returns reStructuredText grid table representation of dataset. >>> from tablib import Dataset >>> from tablib.formats import registry >>> bits = ((0, 0), (1, 0), (0, 1), (1, 1)) >>> data = Dataset() >>> data.headers = ['A', 'B', 'A and B'] >>> for a, b in bits: ... data.append([bool(a), bool(b), bool(a * b)]) >>> rst = registry.get_format('rst') >>> print(rst.export_set(data, force_grid=True)) +-------+-------+-------+ | A | B | A and | | | | B | +=======+=======+=======+ | False | False | False | +-------+-------+-------+ | True | False | False | +-------+-------+-------+ | False | True | False | +-------+-------+-------+ | True | True | True | +-------+-------+-------+ Nz+=z=+=cSrhrir r]r r rr r!z7ReSTFormat.export_set_as_grid_table..z=+z+-z-+-cSrh)-r r]r r rr r!z-+r=r%rkrl) r.r/rerUrQ header_seprow_sepr2r%r r rexport_set_as_grid_tables(      z#ReSTFormat.export_set_as_grid_tablecCsH|durt|}t||krdS|D]}t|}t||kr!dSqdS)a[ Use a simple table if the text in the first column is never wrapped >>> from tablib.formats import registry >>> rst = registry.get_format('rst') >>> rst._use_simple_table('menu', ['egg', 'bacon'], 10) True >>> rst._use_simple_table(None, ['lobster thermidor', 'spam'], 10) False NFT)rr)r.head0col0width0rTr r r_use_simple_tables  zReSTFormat._use_simple_tablecKsx|jsdS|dd}|d|j}|||}||jr"|jdnd|d|d}|r6|s6|||S|||S)a Returns reStructuredText table representation of dataset. Returns a simple table if the text in the first column is never wrapped, otherwise returns a grid table. >>> from tablib import Dataset >>> bits = ((0, 0), (1, 0), (0, 1), (1, 1)) >>> data = Dataset() >>> data.headers = ['A', 'B', 'A and B'] >>> for a, b in bits: ... data.append([bool(a), bool(b), bool(a * b)]) >>> table = data.rst >>> table.split('\n') == [ ... '===== ===== =====', ... ' A B A and', ... ' B ', ... '===== ===== =====', ... 'False False False', ... 'True False False', ... 'False True False', ... 'True True True ', ... '===== ===== =====', ... ] True rA force_gridFr[rN) r)getMAX_TABLE_WIDTHrfrwr&get_colrors)r.r/kwargsrxr[reuse_simple_tabler r r export_sets    zReSTFormat.export_setcsdfdd|jDS)z reStructuredText representation of a Databook. Tables are separated by a blank line. All tables use the grid format. z c3s|] }j|ddVqdS)T)rxN)r~)rr/r.r rr sz)ReSTFormat.export_book..)rH _datasets)r.databookr rr export_bookszReSTFormat.export_bookr)__name__ __module__ __qualname__title extensionsrz classmethodr5r7rVrfrorsrwr~rr r r rrs( !  3  -rN)__doc__ itertoolsr statisticsrtextwraprr7r9 JUSTIFY_RIGHTrLrrrr r r rs