U 祡c@s$UddlZddlZddlZddlZddlZddlZddlmZddlmZmZm Z m Z m Z ddlm Z m Z mZddlmZddlmZddlmZddlmZmZmZmZmZmZmZmZmZmZmZm Z dd l!m"Z"zddl#Z$d Z%Wne&k rd Z%YnXd d l'm(Z(d dl)m*Z*d dl+m,Z,d dl-m.Z.d dl/m0Z0d dl1m2Z2d dl3m4Z4m5Z5d dl6m7Z7d dl8m9Z9erd dl:m;Z;mZ>m?Z?m@Z@dhZAeeBdddZCeeddddZDeEeBdddZFeGd gZHeEeBdd!d"ZIdeed$d%eBeBeeJeeJeeJeBdd& d'd(ZKeEe eLeeLd)ffeBd*d+d,ZMded$d%eBeBeeJeeJeeJeBdd- d.d/ZNGd0d1d1e4ZOeeefeePePePfd2d3d4ZQd5eePePePfd2d6d7ZRejSd8d9eeRe eQed:d9e d;d9eTdd9eVd?d9e d@d9eWdAd9eXdBd9edCd9i ZYeeLeegeePePePfffeZdD<eXeY[Z\eTejSeefZ]eeBddEdFZ^e GdGdHdHZ_e GdIdJdJZ`eeBddKdLZadeeeJeeJeeJe_dMdNdOZbdPdQdddd dReeJeJeeJeeJeeJeBePdSdTdUZcdd dddd dVeed$eBeeJeeJeeJeBddWdXdYZdeedZkr Gd[d\d\Zfdd]lmgZgGd^d_d_egZhe eJZid`eida<d dbdcddded`dfdgdhdihgeUd djdkhe eVdldmdndodpdqgiedmdrdsdtdtdugdvehdwdxdydzd{d|gefd}Zjejdakejdd~llmmZmGdddZnemeOendS)N)array)CounterUserDictUserList defaultdictdeque) dataclassfields is_dataclass)isclass)islice)MappingProxyType) TYPE_CHECKINGAnyCallable DefaultDictDictIterableListOptionalSequenceSetTupleUnion)RichReprResultTF get_console) loop_last) pick_bool)RichRenderablecell_len)ReprHighlighter) JupyterMixinJupyterRenderable) Measurement)Text)ConsoleConsoleOptionsHighlighterType JustifyMethodOverflowMethod RenderResultzmatplotlib.artist.ArtistobjreturncCstott|S)z1Check if an object was created with attrs module.) _has_attrs _attr_moduleZhastyper/r5//tmp/pip-unpacked-wheel-qkhzhrww/rich/pretty.py_is_attr_objectAsr7z_attr_module.Attribute[Any]cCstrtt|SgS)zGet fields for an attrs object.)r1r2r r3r4r5r5r6_get_attr_fieldsFsr8cCs.z|jjjtjkWStk r(YdSXdS)zCheck if an instance of a dataclass contains the default repr. Args: obj (object): A dataclass instance. Returns: bool: True if the default repr is used, False if there is a custom repr. FN)__repr____code__ co_filename dataclasses__file__ Exceptionr4r5r5r6_is_dataclass_reprKs r?_dummy_namedtuplec CsBd}zt|j}Wnttfk r,YnXttj}||kS)zCheck if an instance of namedtuple contains the default repr Args: obj (object): A namedtuple Returns: bool: True if the default repr is used, False if there's a custom repr. N)inspectgetfiler9OSError TypeErrorr@)r/Zobj_fileZdefault_repr_filer5r5r6_has_default_namedtuple_repr_s  rEignorer(r,) valueconsoleoverflowcrop indent_guides max_length max_string max_depth expand_allr0c  Csddlm} ddlm} t|ts*|dkr.dS|p6t}|jrdddddd d d g} | D]L} t|| d} t | rVz | }Wnt k rYqVYnX|dk rVdSqV| |t rdSt|| r| |jt|tr|nt|||||||d d |dddS)Nr)is_object_one_of_types)ConsoleRenderableZ _repr_html_Z_repr_markdown_Z _repr_json_Z _repr_latex_Z _repr_jpeg_Z _repr_png_Z _repr_svg_Z_repr_mimebundle_ )rIrKrLrMrNrOmarginT)rJZnew_line_start)Z_inspectrPrHrQ_safe_isinstancer%rZ is_jupytergetattrrAismethodr>JUPYTER_CLASSES_TO_NOT_RENDERlineprintr Pretty)rGrHrIrJrKrLrMrNrOrPrQZipython_repr_methodsZ repr_methodmethodZ repr_resultr5r5r6_ipy_display_hookssX           r\.)r/class_or_tupler0cCs(z t||WStk r"YdSXdS)zFisinstance can fail in rare cases, for example types with no __class__FN) isinstancer>)r/r]r5r5r6rTs rT) rHrIrJrKrLrMrNrOr0c sddlmpdk s"ttddfdd }zHt} ddlm} Gfdd d | } | } | | jjd <Wnt k r|t _ YnXdS) 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. max_depth (int, optional): Maximum depth of nested data structures, or None for no maximum. Defaults to None. expand_all (bool, optional): Expand all containers. Defaults to False. max_frames (int): Maximum number of frames to show in a traceback, 0 for no maximum. Defaults to 100. rrNrGr0c sR|dk rNdk stdt_jt|tr,|nt|dd|t_dS)z?Replacement sys.displayhook which prettifies objects with Rich.N)rIrKrLrMrNrO)rJ)AssertionErrorbuiltins_rYrTr rZ)rG)rHrJrOrKrNrLrMrIr5r6 display_hooks$  zinstall..display_hook) BaseFormattercs:eZdZUdZeed<eedfdd ZdS)zinstall..RichFormatterTpprintr_c s,|jr t|dSt|SdS)N)rHrIrKrLrMrNrO)rer\repr)selfrGrOrrKrNrLrMrIr5r6__call__s z'install..RichFormatter.__call__N)__name__ __module__ __qualname__rebool__annotations__rrir5rhr5r6 RichFormatters  roz text/plain) richrr`rZ get_ipythonZIPython.core.formattersrdZdisplay_formatter formattersr>sys displayhook) rHrIrJrKrLrMrNrOrciprdroZrich_formatterr5) rHrJrOrrKrNrLrMrIr6installs   "  ruc@seZdZdZddddddddddddd eedeeded eeeeeeeeeeeedd d d Zd dddddZ d dddddZ dS)rZakA 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. max_depth (int, optional): Maximum depth of nested data structures, or None for no maximum. 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_sizejustifyrIno_wraprKrLrMrNrOrS insert_liner*r+r,)_object highlighterrwrxrIryrKrLrMrNrOrSrzr0c CsX||_|pt|_||_||_||_||_||_||_| |_ | |_ | |_ | |_ | |_ dSN)r{r#r|rwrxrIryrKrLrMrNrOrSrz)rgr{r|rwrxrIryrKrLrMrNrOrSrzr5r5r6__init__+s zPretty.__init__r(r)r-)rHoptionsr0c cst|j|j|j|j|j|j|j|jd}t j ||j p:|j |j pD|j t |j|jdd}|rf||nt t|jddd}|jr|js|j|jdd}|jrd|krd V|VdS) N max_widthrwrLrMrNrOpretty)rxrIrystylez.__repr__ returned empty stringz dim italic)rz repr.indent ) pretty_reprr{rrSrwrLrMrNrOr'Z from_ansirxrIrryr|r3rKZ ascii_onlyZwith_indent_guidesrz)rgrHr pretty_strZ pretty_textr5r5r6__rich_console__Js>        zPretty.__rich_console__r&cCsHt|j|j|j|j|j|jd}|r:tdd|Dnd}t ||S)N)rrwrLrMrOcss|]}t|VqdSr}r!.0rXr5r5r6 ysz*Pretty.__rich_measure__..r) rr{rrwrLrMrOmax splitlinesr&)rgrHrr text_widthr5r5r6__rich_measure__ms zPretty.__rich_measure__)N) rjrkrl__doc__rrintrmr~rrr5r5r5r6rZsL  $rZ)r{r0cCsd|jddd|jdfS)Nz defaultdict(z, {})z, {}))default_factoryr{r5r5r6_get_braces_for_defaultdict~s  rz array[Any]cCsd|jddd|jdfS)Nzarray(z, [])))typecoderr5r5r6_get_braces_for_arraysrcCsdS)N)z environ({rz environ({})r5rr5r5r6rcCsdS)N)z Counter({rz Counter()r5rr5r5r6rrcCsdS)N)zdeque([rzdeque()r5rr5r5r6rrcCsdSN){}z{}r5rr5r5r6rrcCsdSrr5rr5r5r6rrcCsdS)N)z frozenset({rz frozenset()r5rr5r5r6rrcCsdSN)[]z[]r5rr5r5r6rrcCsdSrr5rr5r5r6rrcCsdS)N)rrzset()r5rr5r5r6rrcCsdS)N)(r()r5rr5r5r6rrcCsdS)N)zmappingproxy({rzmappingproxy({})r5rr5r5r6rr_BRACEScCs.t|ts$t|s$t|ds$t|o,t| S)z3Check if an object may be expanded by pretty print. __rich_repr__)rT _CONTAINERSr hasattrr7r r4r5r5r6 is_expandables rc@seZdZUdZdZeed<dZeed<dZeed<dZ eed<dZ eed<dZ e ed <dZ e ed <dZe ed <d Zeeded <dZdZeed<eedddZeee dddZedddZdeee edddZd S)Nodez4A node in a repr tree. May be atomic or a container.rkey_repr value_repr open_brace close_braceemptyFlastis_tuple is_namedtupleNchildrenz: z, separatorr0ccs|jr|jV|jV|jr&|jVn|jdk r|jr|jV|jrt|jstt|jdkrt|jdEdHdVn(|jD] }|EdH|j sz|j Vqz|j Vn|j VdS)zGenerate tokens for this node.Nrr,) r key_separatorrrrrrlen iter_tokensrrrr)rgchildr5r5r6rs"     zNode.iter_tokens) start_lengthrLr0cCs0|}|D]}|t|7}||kr dSq dS)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")rgrrL total_lengthtokenr5r5r6 check_lengths   zNode.check_lengthcCsd|}|SNr)joinr)rgZ repr_textr5r5r6__str__sz Node.__str__Prv)rrwrOr0cCsxt|ddg}d}|t|kr`||}|jrV|jsV|s@||sV|||||d<|d7}qddd|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_rootrrrcss|]}t|VqdSr})strrr5r5r6rszNode.render..)_Liner expandableexpandedrexpandr)rgrrwrOlinesZline_norXrepr_strr5r5r6renders    z Node.render)rrvF)rjrkrlrrrrnrrrrrrmrrrrrrrrrrrrrr5r5r5r6rs0          rc@seZdZUdZdZeded<dZeed<dZ ee ed<dZ e ed<dZ e ed <dZe ed <dZeed <dZeed <eed ddZeedddZeeddddZe d ddZdS)rzA line in repr output.NparentFrrrtextsuffix whitespacerrrcCst|jdk o|jjS)z"Check if the line may be expanded.N)rmrrrgr5r5r6r sz_Line.expandable)rLr0cCs:t|jt|jt|j}|jdk s,t|j||S)z4Check this line fits within a given number of cells.N)rrr"rrrr`r)rgrLrr5r5r6rsz_Line.check_length)rwr0c cs|j}|dk st|j}|js"t|jrJt|j|j|j|dV}nt|j|dV}|jd|}|jozt |jdk}t |jD]4\}}|rdn|j } t|||| |o| d} | Vqt|j ||j |jdVdS)z6Expand this line by adding children on their own line.N)rr rr)rrrrr)rrrr)rr`rrrrrrrrrrrrr) rgrwrrnew_lineZchild_whitespaceZ tuple_of_onerrrrXr5r5r6rs8   z _Line.expandcCsH|jr |j|j|jpdS|j|j|jp4d|jSdSr)rrrrrrstriprr5r5r6r7s"z _Line.__str__)rjrkrlrrrrnrrmrrrrrrrrpropertyrrrrrrr5r5r5r6rs       !rcCs<zt|dd}Wntk r&YdSXt|to:t|tS)aXChecks if an object is most likely a namedtuple. It is possible to craft an object that passes this check and isn't a namedtuple, but there is only a minuscule chance of this happening unintentionally. Args: obj (Any): The object to test Returns: bool: True if the object is a namedtuple. False otherwise. _fieldsNF)rUr>r^tuple)r/r r5r5r6_is_namedtuple@s r)r{rLrMrNr0c s\ttdfdd tjjd ttttdfdd |d d }|S) a`Traverse 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. max_depth (int, optional): Maximum depth of data structures, or None for no maximum. Defaults to None. Returns: Node: The root of a tree structure which can be used to render a pretty repr. r.c sdk rFt|ttfrFt|krFt|}|dd|}n@z t|}Wn2tk r}zdt|d}W5d}~XYnX|S)z0Get repr string for an object, but catch errors.N+z )rTbytesrrrfr>)r/Z truncatedZobj_reprerror)rMr5r6to_reprhs    "ztraverse..to_reprFr)r/rootdepthr0c#st}|krtddSt}tjjtjjf}dk o@|k}ttt tt t tffddd}zt d}Wnt k rd}YnXd} |szt d rts} Wnt k rYnX| dk r|tjd d} t|| } jj} | rg} | j}|r@| r.td | d d}nt| d d}n| rbtd | dd| |dd}nt| dd| |d}t| D]h\}}t|tr|\}}||dd}||_||_d|_||n||dd}||_||qn&t| rd | dn| dg|d}|ntr:|s:|g} | j}tr|rjtjjd d}ntjjdd| |d}tt t ttttgt ffdfdd }t|D]T\}\}}}|rtt ||d}n||dd}||_||_d|_||qntjjdg|d}|njtrtts|st sj|dkr|g} | j}|rtjjd d}nptjjdd| |d}tddt!DD]>\}}t|j"|dd}|j"|_||_d|_||qȈ|nt#rt$r̈|jj} |rPt| d d}npg} | j}t| dd| | dd}t%&D]8\}\}}||dd}||_||_d|_||q|ntt'rt'D]}t|r|}qq܈|t(|\}}}|r2t|d|d}nX|j)tj)krXt|d }n2r|g} t||| |d}| j}t*}|d}tt+rt,&} dk rt-| } t.| D]:\}!\}}||dd}||_|!|k|_||qnRt,}"dk rt-|"}"t.|"D],\}!}||dd}|!|k|_||qdk r|kr|td!|d"d nt|g|d#}|nt|d }tt|_/t#|_0|S)$zWalk the object depth first.z...)rN) rich_argsr0css|D]x}t|trvt|dkr>|\}}}||kr2q||fVq|t|dkr^|\}}||fVq|t|dkr||dVq|VqdS)Nrr)rTrr)rargkeyrdefaultr5r5r6iter_rich_argss        z3traverse.._traverse..iter_rich_argsZ,awehoi234_wdfjwljet234_234wdfoijsdfmmnxpi492Frangularz(...)rr)rrrrrrr)rrrrr)r=r)rrrrc 3stD]j}|jrzt|j}Wn0tk rN}z|j|dfVW5d}~XYqX|j|t|jrf|jndfVqdS)z$Iterate over attr fields and values.N)rfrUnamer>callable)attrrGrZ attr_fieldsr/r5r6 iter_attrss z/traverse.._traverse..iter_attrs)rcss|]}|jr|VqdSr})rf)rfieldr5r5r6rsz.traverse.._traverse..)rrrr)rrz... +T)rrr)1idrr3rr version_infomajorminorrrrrrrr>r rrUlist __class__rjappendrrTrrrrr7r8rrr r?r rrrE_asdictitemsrrr9r_MAPPING_CONTAINERSiterr enumeraterr)#r/rrZobj_idZobj_type py_versionZreached_max_depthrZfake_attributesZrich_repr_resultrargs class_namerrrrrrrZ child_noderrrGZ repr_callablerZcontainer_typerrrZ num_itemsZlast_item_indexZ iter_itemsindexZ iter_values) _traverserNrL pop_visited push_visitedr visited_idsrr6r|s $                                ztraverse.._traverseT)r)Fr)rrsetaddremovermrr)r{rLrMrNrr5)rrNrLrMrrrrr6traverseSs&| rrrvr)r{rrwrLrMrNrOr0c Cs4t|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. max_depth (int, optional): Maximum depth of nested data structure, or None for no depth. 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. )rLrMrN)rrwrO)rTrrr) r{rrwrLrMrNrOrrr5r5r6r{s r)rHrKrLrMrNrO)r{rHrKrLrMrNrOr0c Cs6|dkrtn|}|jt||||||dddddS)aA 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. max_depth (int, optional): Maximum depth for nested data structures, or None for unlimited depth. Defaults to None. indent_guides (bool, optional): Enable indentation guides. Defaults to True. expand_all (bool, optional): Expand all containers. Defaults to False. NrF)rLrMrNrKrOrIT)Z soft_wrap)rrYrZ)r{rHrKrLrMrNrOZ_consoler5r5r6res re__main__c@seZdZedddZdS) BrokenReprrcCs dddS)Nrrzthis will failr5rr5r5r6r9szBrokenRepr.__repr__Nrjrkrlrr9r5r5r5r6r sr ) NamedTuplec@s:eZdZUeed<eed<eed<eed<eeed<dS)StockKeepingUnitr descriptionZpricecategoryZreviewsN)rjrkrlrrnfloatrr5r5r5r6r s r fooz Hello World!gZd;Y@gʡE3t@gcAr)rrrrvrrZcrumbleZappleZrhubarbZbutterZsugarZflourZorangeZpearZkumquatAXduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurian)FTNzSparkling British Spring WaterzCarbonated spring waterg?Zwaterz its amazing!z its terrible!)rbarrcounterZatomic namedtupleZBroken)rYc@seZdZedddZdS)ThingrcCsdS)NzHello World!r5rr5r5r6r9szThing.__repr__Nr r5r5r5r6rsr)NrFFFNNNF)NrFFFNNNF)NNN)ora collectionsr<rAosrrrrrrrrrr r r itertoolsr typesr typingrrrrrrrrrrrrZ rich.reprrrr2r1 ImportErrorrrZ_looprZ_pickrabcr cellsr"r|r#Zjupyterr$r%Zmeasurer&rr'rHr(r)r*r+r,r-rWrmr7r8objectr?rr@rErr\r3rTrurZrrr_Environdict frozensetrrrrrnkeysrrrrrrrrrerjr r r ddatarrprYrr5r5r5r6s    8             L  Og"(  TC - + %     !