U g $@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 7/tmp/pip-unpacked-wheel-2jt74g4_/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_lensrc@seZdZdZdZdZeddZedefddZ eed fd d Z edd dZ edddZ eddZ eddZeddZd S) ReSTFormatrst)rPc Cs|jr(dd|jD}dd|jD}n(ddt|jD}ddt|jD}|jD]`}tt|drn|n|}t|D]8\}}t|}|| t |t ||t |||<q|qV||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 ,scSsg|]}gqSr r r_r r rr .scSsg|]}dqS)rr r!r r rr /svalues) headersrangewidthdictiterhasattrr# enumeraterappendrrr) clsdatasetZcolumn_lengths word_lensrowr#ivalrr r r_get_column_string_lengths$s z%ReSTFormat._get_column_string_lengths|c stkrtddtfdd|r4|dnd|rDd|ndd|dg}t|D]*\}}||_t|} || } || qbt |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", "cs2tkr||Stkr$||S||SdSr) JUSTIFY_LEFTljustJUSTIFY_CENTERrrjust)Ztext_Zwidth_justifyr rjustBs   z&ReSTFormat._row_to_lines..just fillvaluec3s&|]}fddt|DVqdS)c3s |]\}}||VqdSrr )rr0Z cell_liner:widthsr rrTsz5ReSTFormat._row_to_lines...N)r*rliner>r rrSsz+ReSTFormat._row_to_lines..cs"g|]}d|fqS)r<)joinr@)lpadpadrpadr rr Wsz,ReSTFormat._row_to_lines..) JUSTIFY_VALUES ValueErrorformatrBzipr&rwrapr+r) r,r#r?wrappersepr9cellsr r&rcelllinesr )r:r9rCrDrEr?r _row_to_lines8s(     zReSTFormat._row_to_linescs||\}}dd|D}t|t|krNfdd|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 )intrrZlensr r rr asz1ReSTFormat._get_column_widths..c3s|]}|VqdSrr rS)max_table_widthtotalr rrdsz0ReSTFormat._get_column_widths..css|] }|VqdSrr rSr r rrfsc3s"|]}|kr|n|VqdSrr rwpad_lenr rrhscSsg|]\}}t||qSr )r)rrWlr r rr js)r2sumrrI)r,r-rTrYZstr_lensr.Z median_lens column_widthsr )rTrYrUr_get_column_widthsZszReSTFormat._get_column_widthsNc Csg}t}|dkr |j|dd}ddd|D}|||jrj||j|j||dtd|||jD]4}t t |d r| n|}|||||dqp||d |S) zP Returns reStructuredText grid table representation of dataset. NrXz cSsg|] }d|qS=r rVr r rr vsz9ReSTFormat.export_set_as_simple_table..r<)rLr9r# rr]rBr+r$extendrPr6r'r(r)r#)r,r-r\rOrKZborderr/r#r r rexport_set_as_simple_tablems(    z%ReSTFormat.export_set_as_simple_tablec Csg}t}|dkr||}dddd|Dd}ddd d|Dd }|||jr||j|j||td |||jD]<}t t |d r| n|}|||||||qd |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=+=cSsg|] }d|qSr_r rVr r rr sz7ReSTFormat.export_set_as_grid_table..z=+z+-z-+-cSsg|] }d|qS)-r rVr r rr sz-+r8r#rarb) r,r-r\rOrKZ header_sepZrow_sepr/r#r r rexport_set_as_grid_tables(     z#ReSTFormat.export_set_as_grid_tablecCsH|dk r t|}t||kr dS|D]}t|}t||kr$dSq$dS)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,Zhead0Zcol0Zwidth0rNr r r_use_simple_tables  zReSTFormat._use_simple_tablecKs||js dS|dd}|d|j}|||}||jrD|jdnd|d|d}|rl|sl|||S|||SdS)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 r< force_gridFrTrN) r'getMAX_TABLE_WIDTHr]rgr$Zget_colrdrf)r,r-kwargsrhrTr\Zuse_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)rhN)rl)rr-r,r rr sz)ReSTFormat.export_book..)rBZ _datasets)r,Zdatabookr rmr export_bookszReSTFormat.export_book)N)N)__name__ __module__ __qualname__title extensionsrj classmethodr2r4rPr]rdrfrgrlrnr r r rrs& !  3  -rN)__doc__ itertoolsrZ statisticsrtextwraprr4r6Z JUSTIFY_RIGHTrFrrrr r r rs