o MEgkj@s>UddlZddlZddlZddlmZddlmZmZmZmZm Z ddl m Z m Z m Z ddlmZddlmZmZmZmZmZmZmZmZmZmZmZddlmZzddlZWn e ybdZYnwded e!fd d Z"ded ed fd dZ#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl$m%Z%ddl0m1Z1m2Z2ddl3m4Z4ddl5m6Z6erddl7m8Z8m9Z9m:Z:m;Z;md#ee>d$e!d dfd%d&Z?Gd'd(d(e1Z@d)eeefd eeAeAeAffd*d+ZBd)d,d eeAeAeAffd-d.ZCejDd/d0eeCeeBed1d0ed2d0eEd3d0ed4d0eFd5d0eGd6d0e d7d0eHd8d0eId9d0ed:d0i ZJeeKeegeeAeAeAfffeLd;<eIeJMZNeEejDeefZOded e!fdd?d?ZQe Gd@dAdAZR dcd)ed"ee>d#ee>d eQfdBdCZSdDdEddddFd)edGe>dHe>d"ee>d#ee>d$e!d eAfdIdJZTddKddddLd)ededd!e!d"ee>d#ee>d$e!d dfdMdNZUeVdOkrGdPdQdQZWee>ZXdReXdS<ddTdUdVdWhdXgeFhdYeeGdZgd[iegd\d]eWd^ZYeYdSZeYdd_l[m\Z\e\e@eYdKd`dadSdS)dN)array)Counter defaultdictdequeUserDictUserList) dataclassfields is_dataclass)islice) DefaultDict TYPE_CHECKINGAnyCallableDictIterableListOptionalSetUnionTuple)MappingProxyTypeobjreturncCstduo tt|S)z1Check if an object was created with attrs module.N) _attr_modulehastyperre/var/www/eduai.edurigo.com/doc_train/edurigo_ai/Puru/venv/lib/python3.10/site-packages/rich/pretty.py_is_attr_objectsr z_attr_module.Attribute[Any]cCstdur tt|SgS)zGet fields for an attrs object.N)rr rrrrr_get_attr_fields"sr!)ReprHighlighter get_console) loop_last) pick_bool)RichRenderablecell_len) JupyterMixinJupyterRenderable) Measurement)Text)ConsoleConsoleOptionsHighlighterType JustifyMethodOverflowMethod RenderResultignoreFconsoler/overflowr3crop indent_guides max_length max_string expand_allc sddlm}ddlmp|dusJdtddffdd }dtddffd d } zt} dd lm} | } | jt | d | | j j d<WdSt yc|t _YdSw)aInstall automatic pretty printing in the Python REPL. Args: console (Console, optional): Console instance or ``None`` to use global console. Defaults to None. overflow (Optional[OverflowMethod], optional): Overflow method. Defaults to "ignore". crop (Optional[bool], optional): Enable cropping of long lines. Defaults to False. indent_guides (bool, optional): Enable indentation guides. Defaults to False. max_length (int, optional): Maximum length of containers before abbreviating, or None for no abbreviation. Defaults to None. max_string (int, optional): Maximum length of string before truncating, or None to disable. Defaults to None. expand_all (bool, optional): Expand all containers. Defaults to False rr$r")ConsoleRenderableNvaluerc sT|dur(dus Jdt_jt|tr|n t|dd|t_dSdS)z?Replacement sys.displayhook which prettifies objects with Rich.N)r7r9r:r;r<r8)builtins_print isinstancer(Prettyr>)r6r8r<r9r:r;r7rr display_hookYs$  zinstall..display_hookc sdusJt|ts|durdSjr!tddt|Dr!dSt|dr(dSt|r1jt|tr:|n t |ddddS)Ncss|]}|dVqdS)_repr_N) startswith).0attrrrr ssz4install..ipy_display_hook.._repr_mimebundle_ )r7r9r:r;r<marginr?) rCr, is_jupyteranydirhasattrlinerBr(rDrEr=r6r8r<r9r:r;r7rripy_display_hookms.    z!install..ipy_display_hook) BaseFormatter)funcz text/plain)richr%r6r=r get_ipythonIPython.core.formattersrVfor_typeobjectdisplay_formatter formatters Exceptionsys displayhook) r6r7r8r9r:r;r<r%rFrUiprVrich_formatterrrTrinstall=s    "$   rdc@seZdZdZ d#ddddddddddd deded d ed ed d eddeededeedeededededdfddZ      d$ddZ      d%d!d"Z dS)&rDaA rich renderable that pretty prints an object. Args: _object (Any): An object to pretty print. highlighter (HighlighterType, optional): Highlighter object to apply to result, or None for ReprHighlighter. Defaults to None. indent_size (int, optional): Number of spaces in indent. Defaults to 4. justify (JustifyMethod, optional): Justify method, or None for default. Defaults to None. overflow (OverflowMethod, optional): Overflow method, or None for default. Defaults to None. no_wrap (Optional[bool], optional): Disable word wrapping. Defaults to False. indent_guides (bool, optional): Enable indentation guides. Defaults to False. max_length (int, optional): Maximum length of containers before abbreviating, or None for no abbreviation. Defaults to None. max_string (int, optional): Maximum length of string before truncating, or None to disable. Defaults to None. expand_all (bool, optional): Expand all containers. Defaults to False. margin (int, optional): Subtrace a margin from width to force containers to expand earlier. Defaults to 0. insert_line (bool, optional): Insert a new line if the output has multiple new lines. Defaults to False. NFr) indent_sizejustifyr7no_wrapr9r:r;r<rN insert_line_object highlighterr1rfrgr2r7r3rhr9r:r;r<rNrirc CsR||_|pt|_||_||_||_||_||_||_| |_ | |_ | |_ | |_ dSN) rjr#rkrfrgr7rhr9r:r;r<rNri) selfrjrkrfrgr7rhr9r:r;r<rNrirrr__init__s  zPretty.__init__r6r/optionsr0r4ccst|j|j|j|j|j|j|jd}t||j p|j |j p |j t |j |j dd}|r1| |n tt|jddd}|jrK|jsK|j|jdd}|jrUd|vrUd V|VdS) N max_widthrfr:r;r<pretty)rgr7rhstylez.__repr__ returned empty stringz dim italic)rsz repr.indent ) pretty_reprrjrqrNrfr:r;r<r.rgr7r'rhrkrr9 ascii_onlywith_indent_guidesri)rmr6ro pretty_str pretty_textrrr__rich_console__s<       zPretty.__rich_console__r-cCsDt|j|j|j|j|jd}|rtdd|Dnd}t||S)N)rqrfr:r;cs|]}t|VqdSrlr)rIrSrrrrKz*Pretty.__rich_measure__..r) rvrjrqrfr:r;max splitlinesr-)rmr6rory text_widthrrr__rich_measure__s zPretty.__rich_measure__rl)r6r/ror0rr4)r6r/ror0rr-) __name__ __module__ __qualname____doc__rrintboolrnr{rrrrrrDsn       "rDrjcCsd|jddd|jdfS)Nz defaultdict(z, {})z, {}))default_factoryrjrrr_get_braces_for_defaultdicts  rz array[Any]cCsd|jdddfS)Nzarray(z, [])zarray({_object.typecode!r}))typecoderrrr_get_braces_for_arraysrcCdS)N)z environ({rz environ({})rrrrrrcCr)N)z Counter({rz Counter()rrrrrr rcCr)N)zdeque([rzdeque()rrrrrr rcCrN){}z{}rrrrrr rcCrrrrrrrr rcCr)N)z frozenset({rz frozenset()rrrrrrrcCrN)[]z[]rrrrrrrcCrrrrrrrrrcCr)N)rrzset()rrrrrrrcCr)N)()()rrrrrrrcCr)N)zmappingproxy({rzmappingproxy({})rrrrrrr_BRACEScCs0t|tpt|ot|t pt|dpt|S)z3Check if an object may be expanded by pretty print. __rich_repr__)rC _CONTAINERSr rrRr rrrr is_expandables rc @seZdZUdZdZeed<dZeed<dZeed<dZ eed<dZ eed<dZ e ed <dZ e ed <d Zeeded <d ZedefddZdeefddZdedede fddZdefddZ d dedede defddZd S)!Nodez4A node in a repr tree. May be atomic or a container.rukey_repr value_repr open_brace close_braceemptyFlastis_tupleNchildrenz: rcCs|jrdSdS)zGet separator between items.ru,)rrmrrr separator1szNode.separatorccs|jr |jV|jV|jr|jVdS|jdurX|jrR|jV|jr9t|jdkr9|jdEdHdVn|jD]}|EdH|jsKdVq<|j VdS|j VdSdS)zGenerate tokens for this node.Nr"rrz, ) r key_separatorrrrrlen iter_tokensrrr)rmchildrrrr6s(     zNode.iter_tokens start_lengthr:cCs0|}|D]}|t|7}||krdSqdS)a*Check the length fits within a limit. Args: start_length (int): Starting length of the line (indent, prefix, suffix). max_length (int): Maximum length. Returns: bool: True if the node can be rendered within max length, otherwise False. FT)rr*)rmrr: total_lengthtokenrrr check_lengthLs  zNode.check_lengthcCsd|}|SNru)joinr)rm repr_textrrr__str__]sz Node.__str__Prerqrfr<cCst|ddg}d}|t|kr5||}|jr+|js+|s ||s+|||||d<|d7}|t|ksddd|D}|S)alRender the node to a pretty repr. Args: max_width (int, optional): Maximum width of the repr. Defaults to 80. indent_size (int, optional): Size of indents. Defaults to 4. expand_all (bool, optional): Expand all levels. Defaults to False. Returns: str: A repr string of the original object. T)nodeis_rootrr"rtcsr|rl)strr}rrrrKwr~zNode.render..)_Liner expandableexpandedrexpandr)rmrqrfr<linesline_norSrepr_strrrrrenderas    z Node.render)rreF)rrrrrr__annotations__rrrrrrrrrrrpropertyrrrrrrrrrrrr#s4        rc@seZdZUdZdZeed<dZee ed<dZ e ed<dZ e ed<dZ e ed <dZeed <ed efd d Zded efddZded edfddZd e fddZdS)rzA line in repr output.FrNrrutextsuffix whitespacerrcCst|jduo |jjS)z"Check if the line may be expanded.N)rrrrrrrrsz_Line.expandabler:cCs:t|jt|jt|j}|jdusJ|j||S)z4Check this line fits within a given number of cells.N)rrr*rrrr)rmr:rrrrrsz_Line.check_lengthrfc cs|j}|dus J|j}|jsJ|jr&t|j|j|j|dVnt|j|dV|jd|}|jo>t|jdk}|jD]}|rHdn|j }t|||d}|VqBt|j ||ra|j sadn|j dVdS)z6Expand this line by adding children on their own line.N)rr r"r)rrr)rrr) rrrrrrrrrrrr) rmrfrrchild_whitespace tuple_of_onerrrSrrrrs4     z _Line.expandcCs |j|j|jp d|jSr)rrrrrrrrrs z _Line.__str__)rrrrrrrrrrrrrrrrrrrrrrrrrrr{s      rcs^dtdtffdd tjjd dtdtdtffdd |d d }|S) aTraverse object and generate a tree. Args: _object (Any): Object to be traversed. max_length (int, optional): Maximum length of containers before abbreviating, or None for no abbreviation. Defaults to None. max_string (int, optional): Maximum length of string before truncating, or None to disable truncating. Defaults to None. Returns: Node: The root of a tree structure which can be used to render a pretty repr. rrc sdur$t|ttfr$t|kr$t|}|dd|}|Szt|}W|StyD}z d|d}WYd}~|Sd}~ww)z0Get repr string for an object, but catch errors.N+z )rCbytesrrreprr_)r truncatedobj_reprerror)r;rrto_reprs    ztraverse..to_reprFrootc sxt}tjjtjjf}dtdttttt tfffdd}t drt | }|rqg}|j }tjjdd||d}t|D]*\} } t| trc| \} } | } | | _| | _d | _|| qD| } | | _|| qDntjjd g|d }ntrg}|j }trtjjdd||d}dttt ttttgt ffffd d }t|D]$\} \}}}|rtt ||d} n|} | | _|| _d | _|| qnUtjjd g|d }nGtrSttsSdjjvs|dkrSt}|vrtddS|g}|j }tjjdd||d}ttD]\} }|j rLt!|j"} |j"| _| | _d | _|| q.|ntt#r,t#D] }t|rg|}nq[t}|vrwtddS|t$|\}}}|jtjkrt|d}nr g}t||||d}|j }t%}|d}tt&rt'(}durt)|}t*|D]\}\} } | } | | _||k| _|| qn%t'}durt)|}t*|D]\}} | } ||k| _|| qdur|kr|td|ddnt|g|d}|nt|d}tt|_+|S)zWalk the object depth first. rich_argsrcss|D]<}t|tr._traverse..iter_rich_argsrrr)rrrr=r)rrrc 3sxD]6}|jr9zt|j}Wnty)}z |j|dfVWYd}~qd}~ww|j|t|jr5|jndfVqdS)z$Iterate over attr fields and values.N)rgetattrnamer_callable)rJr>r attr_fieldsrrr iter_attrssz/traverse.._traverse..iter_attrs)r __create_fn__)rz...)rrr"Nz... +T)rrr),rr` version_infomajorminorrrrrrrRlistrappendr __class__rr&rCrrrrr r!rrr __repr__ridr rrrrrr_MAPPING_CONTAINERSiteritemsr enumerater)rrobj_type py_versionrargsrrrrrrr child_noderrr> repr_callableobj_idfieldcontainer_typerrr num_itemslast_item_index iter_itemsindex iter_values) _traverser: pop_visited push_visitedr visited_idsrrrs&                                ztraverse.._traverseT)rN)F)rrsetaddremoverr)rjr:r;rr)rr:r;rrrrrtraverses& 8rrrerprqrfcCs2t|tr|}nt|||d}|j|||d}|S)aPrettify repr string by expanding on to new lines to fit within a given width. Args: _object (Any): Object to repr. max_width (int, optional): Desired maximum width of repr string. Defaults to 80. indent_size (int, optional): Number of spaces to indent. Defaults to 4. max_length (int, optional): Maximum length of containers before abbreviating, or None for no abbreviation. Defaults to None. max_string (int, optional): Maximum length of string before truncating, or None to disable truncating. Defaults to None. expand_all (bool, optional): Expand all containers regardless of available width. Defaults to False. Returns: str: A possibly multi-line representation of the object. )r:r;)rqrfr<)rCrrr)rjrqrfr:r;r<rrrrrrvs rvT)r6r9r:r;r<c Cs4|durtn|}|jt|||||dddddS)awA convenience function for pretty printing. Args: _object (Any): Object to pretty print. console (Console, optional): Console instance, or None to use default. Defaults to None. max_length (int, optional): Maximum length of containers before abbreviating, or None for no abbreviation. Defaults to None. max_string (int, optional): Maximum length of strings before truncating, or None to disable. Defaults to None. indent_guides (bool, optional): Enable indentation guides. Defaults to True. expand_all (bool, optional): Expand all containers. Defaults to False. Nr5)r:r;r9r<r7T) soft_wrap)r%rBrD)rjr6r9r:r;r<_consolerrrpprints r__main__c@seZdZdefddZdS) BrokenReprrcCs dddS)Nr"rzthis will failrrrrrrszBrokenRepr.__repr__N)rrrrrrrrrr sr fooz Hello World!gZd;Y@gʡE3t@gcA>r"rrrer r>r"rrcrumble)applerhubarbbuttersugarflour)rorangepearkumquatrXduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurian)FTN)r barrcounteratomicBroken)rB)r9r;)Nr5FFNNF)NN)]r@osr`r collectionsrrrrr dataclassesrr r itertoolsr typingr r rrrrrrrrrtypesrrJr ImportErrorrr r!rkr#rur%_loopr&_pickr'abcr(cellsr*jupyterr+r,measurer-rr.r6r/r0r1r2r3r4rrdrDrrr_Environdict frozensetrrrrrrkeysrrrrrrrvrrr ddatarrXrBrrrrs0   4               [$b (  W: c & "