j_dZddlZddlZddlZddlZddlmZejdkrdZndZdd Z d Z dd Z d Z dZ ddZddZddZddZdS)z~ Utilities for working with strings and text. Inheritance diagram: .. inheritance-diagram:: IPython.utils.text :parts: 3 N) Formatterwin32z %B %d, %Yz %B %-d, %YFcF|dSd|zd|zz}|r tjdtj}ntjdtj}tj|||}|t j|zr|dt| S|S)aIndent a string a given number of spaces or tabstops. indent(str,nspaces=4,ntabs=0) -> indent str by ntabs+nspaces. Parameters ---------- instr : basestring The string to be indented. nspaces : int (default: 4) The number of spaces to be indented. ntabs : int (default: 0) The number of tabs to be indented. flatten : bool (default: False) Whether to scrub existing indentation. If True, all lines will be aligned to the same indentation. If False, existing indentation will be strictly increased. Returns ------- str|unicode : string indented by ntabs and nspaces. N  z^\s*^)recompile MULTILINEsubendswithoslineseplen)instrnspacesntabsflattenindpatoutstrs 5lib/python3.11/site-packages/ipython_genutils/text.pyindentrs2 } u*S[ C-j",//jr|,, VCe $ $F rz#~&&jCyj!! c.|drtj|S|dd}t |dkrtj|S|\}}tj|}d||gS)zEquivalent of textwrap.dedent that ignores unindented first line. This means it will still dedent strings like: '''foo is a bar ''' For use in wrap_paragraphs.  ) startswithtextwrapdedentsplitrjoin)textsplitsfirstrests rr!r!?s t%t$$$ZZQ  F 6{{at$$$KE4 ?4 D 99eT] # ##rPctjdtj}t|}||ddd}g}tjdtj}|D]A}||tj||}| |B|S)aWrap multiple paragraphs to fit a specified width. This is equivalent to textwrap.wrap, but with support for multiple paragraphs, as separated by empty lines. Returns ------- list of complete paragraphs, wrapped to fill `ncols` columns. z \n(\s*\n)+Nz\n\s+) r r r r!stripr"searchr fillappend)r$ncols paragraph_re paragraphsout_ps indent_reps rwrap_paragraphsr5Zs:mR\::L $<<    D##D))##A#.J F 8R\22I    A   & a''A a Mrc.tjdd|S)z Remove ansi escape codes from text. Parameters ---------- source : str Source to remove the ansi from z\033\[(\d|;)+?m)r r )sources r strip_ansir9ts 6$b& 1 11rc#jKtdt||D]}||||zVdS)z'Yield successive n-sized chunks from l.rN)ranger)lnis r_chunksr?sH 1c!ffa !A#hrr*c 4tdt|dzD]a}tttt ||}t |}t|}|||dz zz|krnb||dz r ||z |dz z nd||dS)z4Calculate optimal info to columnize a list of stringrr)columns_numbersoptimal_separator_width rows_numbers columns_width)r;rlistmapmaxr?sum)rlistseparator_size displaywidthnrowchk sumlengthr/s r _find_optimalrOsaUA&&3s75$//0011HH C ^U1W- - = = E > %OTVWx&_|i'?%'&J&J^_!!   rc<|t|kr|S||S)z2return list item number, or default if don't exist)r)mylistr>defaults r_get_or_defaultrSs"CKKayrcttttg|Ri|}|d|dcfdt D|fS)aReturns a nested list, and info to columnize items Parameters ---------- items list of strings to columize empty : (default None) default value to fill list if needed separator_size : int (default=2) How much caracters will be used as a separation between each columns. displaywidth : int (default=80) The width of the area onto wich the columns should enter Returns ------- strings_matrix nested list of string, the outer most list contains as many list as rows, the innermost lists have each as many element as colums. If the total number of elements in `items` does not equal the product of rows*columns, the last element of some lists are filled with `None`. dict_info some info to make columnize easier: columns_numbers number of columns rows_numbers number of rows columns_width list of with of each columns optimal_separator_width best separator width between columns Examples -------- :: In [1]: l = ['aaa','b','cc','d','eeeee','f','g','h','i','j','k','l'] ...: compute_item_matrix(l,displaywidth=12) Out[1]: ([['aaa', 'f', 'k'], ['b', 'g', 'l'], ['cc', 'h', None], ['d', 'i', None], ['eeeee', 'j', None]], {'columns_numbers': 3, 'columns_width': [5, 1, 1], 'optimal_separator_width': 2, 'rows_numbers': 5}) rCrAcNg|] fdtD!S)c@g|]}t|zzS))rR)rS).0cemptyr>itemsrLs r z2compute_item_matrix...s0 U U U1uafQh>>> U U Ur)r;)rWr>rYrZncolrLs @rr[z'compute_item_matrix..s? l l lZ[ U U U U U U Ud U U U l l lr)rOrErFrr;)rZrYargskwargsinfor\rLs`` @@rcompute_item_matrixr`sl c#uoo.. @ @ @ @ @ @Dn%t,='>JD$ l l l l l l l_dei_j_j l l lnr ssr c|sdSt|t|\}d|D}fd}dt||dzS)a Transform a list of strings into a single string with columns. Parameters ---------- items : sequence of strings The strings to process. separator : str, optional [default is two spaces] The string that separates columns. displaywidth : int, optional [default is 80] Width of the display in number of characters. Returns ------- The formatted string. r)rJrKc.g|]}td|SN)filter)rWxs rr[zcolumnize..s ///1vdA///rcjdt|dDS)Nc@g|]\}}||dS)r)ljust)rWyws rr[z/columnize....s(&b&b&bDAq3&b&b&brrD)r#zip)rfr_ separators rzcolumnize..s2y~~&b&bC4P_K`DaDa&b&b&bccr)r`rr#rF)rZrmrKmatrixfmatrixsjoinr_s ` @r columnizerrsw$ t&uS^^ZfgggLFD/////G c c c c cE 99S(( ) )$ ..r)rrF)r()r*r(rd)rar()__doc__rr sysr stringrplatform date_formatrr!r5r9r?rOrSr`rrrrrys <7KKK$$$$N$$$64 2 2 2      8t8t8t8tv//////r