a ij+@sdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z ddddZejGd d d ZGd d d Zed ddddZd ddddZdddddZddddddZdS)z Text utils.) annotationsN)AnyIterableIteratorUnion)())[]){})rr r c@s*eZdZUdZded<ded<ded<dS)_Linez Line item.strcontentint indent_lvl indent_sizeN)__name__ __module__ __qualname____doc____annotations__rrR/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/etils/epy/text_utils.pyr "s r c @seZdZdZddddddZddd d d Zd dd ddZddd ddZej ddddZ dddddddZ e d(ddd d!dd"d#dddd$d%d&Z d'S))LinesanUtil to build multi-line text. Useful for pretty-print tools and human readable `__repr__`. Example: ```python d = {'a': 1, 'b': 2} lines = epy.Lines() lines += 'dict(' with lines.indent(): for k, v in d.items(): lines += f'{k}={v},' lines += ')' text = lines.join() ``` Output: ``` dict( a=1, b=2, ) ``` )indentrcCsg|_||_d|_dS)Nr)_lines _indent_size _indent_lvl)selfrrrr__init__IszLines.__init__rNone)linereturncCs6t|tstd||jt||j|jddS)Append a new line `str`.z&Lines should be added with `str`. Got )rrrN) isinstancer TypeErrorrappendr rrr r#rrrr(Ns z Lines.appendz Iterable[str])iterabler$cCs|D]}||qdS)z0Append all the new line `str` from the iterable.Nr()r r*r#rrrextend[sz Lines.extendcCs|||S)r%r+r)rrr__iadd__`s zLines.__iadd__zIterator[None])r$c cs<|jd7_zdVW|jd8_n|jd8_0dS)N)r)r rrrresz Lines.indentFcollapsebool)r0r$cCsXg}|jD]4}|j}|s4d|j|j}t||}||q |rJd}nd}||S)zReturns the lines. Args: collapse: If `True`, all lines are merged together in a single line. Returns: text: All lines merged together   )rrrrtextwraprr(join)r r0linesr#r indentationtokenrrrr6ms    z Lines.joinr3rr=)bracesequallimitz2str | dict[str, Any] | list[Any] | tuple[Any, ...]zUnion[str, tuple[str, str]])headerrr<r=r>r$c s<t|trt|}|\}}t|tr*|g}t|trLfdd|D}n0t|ttfrjdd|D}ntdt|t |dk} t dd|Drd} nt d d|D|krd } |} | ||7} | <| r| d |7} n|D]} | | d 7} qWd n1s0Y| |7} | j | dS)aUtil function to create a code block. Example: ```python epy.Lines.make_block('A', {}) == 'A()' epy.Lines.make_block('A', {'x': '1'}) == 'A(x=1)' epy.Lines.make_block('A', {'x': '1', 'y': '2'}) == '''A( x=1, y=2, )''' ``` Pattern is as: ``` {header}{braces[0]} {k}={v}, ... {braces[1]} ``` Args: header: Prefix before the brace content: Dict of key to values. One line will be displayed per item if `len(content) > 1`. Otherwise the code is collapsed braces: Brace type (`(`, `[`, `{`), can be tuple for custom open/close. equal: The separator (`=`, `: `) limit: Strings smaller than this will be collapsed Returns: The block string cs$g|]\}}|t|qSr pretty_repr.0kvr=rr z$Lines.make_block..cSsg|]}t|qSrr@)rCrErrrrGrHzInvalid fields r.css|]}d|vVqdS)r4NrrCprrr rHz#Lines.make_block..Fcss|]}t|VqdS)N)lenrIrrrrKrHTz, ,Nr/)r&r_BRACE_TO_BRACESdictitemslisttupler'typerLanysumrr6) clsr?rr<r=r>Z brace_startZ brace_endpartsr0r7rJrrFr make_blocks0+     0 zLines.make_blockN)r3r)rrrrr!r(r,r- contextlibcontextmanagerrr6 classmethodrXrrrrr+s  rrr)objr$csttrtStttfvr>tjttr6dnddStturhtjdd DdddSttst rȈj jrt tstjturt }tjjjfd d|Dd StSd S) z9Pretty `repr(obj)` for nested list, dict, dataclasses,...r r)rr<cSsi|]\}}t||qSr)reprrBrrr rHzpretty_repr..r z: )rr<r=cs"i|]}|jr|jt|jqSr)r]namegetattr)rCfieldr\rrr^s)r?rN)r&rr]rSrQrRrrXrOrP dataclasses is_dataclass__dataclass_params__has_default_repr__repr__rAfields __class__r)r\Z all_fieldsrrbrrAs>       rAr1)rVr$cCst|j}|jdkS)z.__repr__)inspectunwraprgr)rVZrepr_fnrrrrfs rf)textr$cCst|S)alWrapper around `textwrap.dedent` which also `strip()` the content. Before: ```python text = textwrap.dedent( \"\"\"\\ A( x=1, )\"\"\" ) ``` After: ```python text = epy.dedent( \"\"\" A( x=1, ) \"\"\" ) ``` Args: text: The text to dedent Returns: The dedented text )r5dedentstrip)rlrrrrms rmz str | object)abr$cCsFt|tst|d}t|ts0t|d}t||}d|S)zPretty diff between 2 objects. Args: a: Object/str to compare b: Object/str to compare Returns: The diff string r4)r&rrAsplitdifflibZndiffr6)rorpdiffrrrdiff_str s   rt)r __future__rrYrcrrrjreprlibr5typingrrrrrN dataclassr rrecursive_reprrArfrmrtrrrrs* ($#