U d#@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-_3ph8cfy/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 s|tkrtddt|tkr.ddn|tkr@ddndd|rT|dnd|rdd|ndd|dg}t|D]*\}}||_t|} | | } | | qt |d di} fd d | 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", "cSs ||Sr)ljustrr&r r rBz*ReSTFormat._row_to_lines..cSs ||Sr)rr5r r rr6Dr7cSs ||Sr)rjustr5r r rr6Fr7 fillvaluec3s&|]}fddt|DVqdS)c3s |]\}}||VqdSrr )rr0Z cell_linejustwidthsr rrRsz5ReSTFormat._row_to_lines...N)r*rliner<r rrQsz+ReSTFormat._row_to_lines..cs"g|]}d|fqS)r:)joinr?)lpadpadrpadr rr Usz,ReSTFormat._row_to_lines..) JUSTIFY_VALUES ValueErrorformatrA JUSTIFY_LEFTJUSTIFY_CENTERzipr&rwrapr+r) r,r#r>wrappersepjustifycellsr r&rcelllinesr )r=rBrCrDr>r _row_to_lines8s0      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 )intr)rZlensr r rr _sz1ReSTFormat._get_column_widths..c3s|]}|VqdSrr rl)max_table_widthtotalr rrbsz0ReSTFormat._get_column_widths..css|] }|VqdSrr rUr r rrdsc3s"|]}|kr|n|VqdSrr rwpad_lenr rrfscSsg|]\}}t||qSr )r)rrZrVr r rr hs)r2sumrrJ)r,r-rWr\Zstr_lensr.Z median_lens column_widthsr )rWr\rXr_get_column_widthsXszReSTFormat._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. Nr[z cSsg|] }d|qS=r rYr r rr tsz9ReSTFormat.export_set_as_simple_table..r:)rMrNr# rr_rAr+r$extendrRrIr'r(r)r#)r,r-r^rQrLZborderr/r#r r rexport_set_as_simple_tableks(    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|qSrar rYr r rr sz7ReSTFormat.export_set_as_grid_table..z=+z+-z-+-cSsg|] }d|qS)-r rYr r rr sz-+)rNr#rcrd) r,r-r^rQrLZ 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,Zhead0Zcol0Zwidth0rPr 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_gridFrWrN) r'getMAX_TABLE_WIDTHr_rir$Zget_colrfrh)r,r-kwargsrjrWr^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)rjN)rn)rr-r,r rrsz)ReSTFormat.export_book..)rAZ _datasets)r,Zdatabookr ror export_bookszReSTFormat.export_book)N)N)__name__ __module__ __qualname__title extensionsrl classmethodr2rHrRr_rfrhrirnrpr r r rrs&   3  -rN)__doc__ itertoolsrZ statisticsrtextwraprrHrIZ JUSTIFY_RIGHTrErrrr r r rs