U EZh'@sfUddlZddlZddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z ddlm Z mZmZddlmZddlmZddlmZddlmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"dd l#m$Z$zddl%Z&e'e&d Z(Wne)k rd Z(YnXd d l*m+Z+d dl,m-Z-d dl.m/Z/d dl0m1Z1d dl2m3Z3d dl4m5Z5d dl6m7Z7m8Z8d dl9m:Z:d dl;mZ>m?Z?m@Z@mAZAmBZBmCZCeeDdddZEeeddddZFeGeDdddZHeIdgZJeGeDdd d!ZKdeed#d$eDeDeeLeeLeeLeDe"eMdfd% d&d'ZNeGe"eOe!eOd(ffeDd)d*d+ZPded#d$eDeDeeLeeLeeLeDdd, d-d.ZQGd/d0d0e7ZReeefe!eMeMeMfd1d2d3ZSeee!eMeMeMfd1d4d5ZTd6e!eMeMeMfd1d7d8ZUejVd9d:eeUe eSed;d:e eTeWdd:eYd?d:e d@d:eZdAd:e[dBd:edCd:i Z\eeOeege!eMeMeMfffe]dD<e[e\^Z_eWejVee fZ`eeDddEdFZae GdGdHdHZbe GdIdJdJZceeDddKdLZddeeeLeeLeeLebdMdNdOZedPdQdddd dReeLeLeeLeeLeeLeDeMdSdTdUZfddVdddd dWeed#eDeeLeeLeeLeDddXdYdZZgehd[krbGd\d]d]Zidd^lmjZjGd_d`d`ejZke eLZldaeldb<d dcdddedfdadgdhdidjhgeXd dkdlhe eYdmdndodpdqdrgiedndsdtdududvgdwekdxdydzd{d|d}geid~ZmemdbnemddlompZpepeRemdVddGdddZqepeReqdS)N)array)CounterUserDictUserList defaultdictdeque) dataclassfields is_dataclass)isclass)islice)MappingProxyType) TYPE_CHECKINGAnyCallable DefaultDictDequeDictIterableListOptionalSequenceSetTupleUnion)RichReprResultibF get_console) loop_last) pick_bool)RichRenderablecell_len)ReprHighlighter) JupyterMixinJupyterRenderable) Measurement)Text)ConsoleConsoleOptionsHighlighterType JustifyMethodOverflowMethod RenderResultobjreturncCstott|S)z1Check if an object was created with attrs module.) _has_attrs _attr_moduleZhastyper1r7;/tmp/pip-unpacked-wheel-36xvkur8/pip/_vendor/rich/pretty.py_is_attr_object<sr9z_attr_module.Attribute[Any]cCstrtt|SgS)zGet fields for an attrs object.)r3r4r r5r6r7r7r8_get_attr_fieldsAsr:cCs4z|jjjtjtjfkWStk 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__reprlib Exceptionr6r7r7r8_is_dataclass_reprFs rB_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)inspectgetfiler;OSError TypeErrorrC)r1Zobj_fileZdefault_repr_filer7r7r8_has_default_namedtuple_repr]s  rHignorer*r.) valueconsoleoverflowcrop indent_guides max_length max_string max_depth expand_allr2c Csddlm} t|ts|dkr"dS|p*t}|L} t|| rH||jt|trZ|nt |||||||dd|dddW5QRX| dS) Nr)ConsoleRenderable )rLrNrOrPrQrRmarginT)rMZnew_line_startend ) rKrS_safe_isinstancer'rcapturelineprintr"Prettygetrstrip) rJrKrLrMrNrOrPrQrRrSrZr7r7r8_ipy_display_hookqs2     r`.)r1class_or_tupler2cCs(z t||WStk r"YdSXdS)zFisinstance can fail in rare cases, for example types with no __class__FN) isinstancerA)r1rar7r7r8rYs rY) rKrLrMrNrOrPrQrRr2c sddlmpdk s"ttddfdd }z t} Wntk rh|t_Yn@Xddlm } Gfdd d | } | } | | j j d <dS) 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. rrNrJr2c sR|dk rNdk stdt_jt|tr,|nt|dd|t_dS)z?Replacement sys.displayhook which prettifies objects with Rich.N)rLrNrOrPrQrR)rM)AssertionErrorbuiltins_r\rYr"r])rJ)rKrMrRrNrQrOrPrLr7r8 display_hooks$  zinstall..display_hook) BaseFormattercs:eZdZUdZeed<eedfdd ZdS)zinstall..RichFormatterTpprintrcc s,|jr t|dSt|SdS)N)rKrLrNrOrPrQrR)rir`repr)selfrJrRrrNrQrOrPrLr7r8__call__s z'install..RichFormatter.__call__N)__name__ __module__ __qualname__ribool__annotations__rrmr7rlr7r8 RichFormatters  rsz text/plain) pip._vendor.richrrdrZ get_ipython NameErrorsys displayhookZIPython.core.formattersrhZdisplay_formatter formatters) rKrLrMrNrOrPrQrRrgiprhrsZrich_formatterr7) rKrMrRrrNrQrOrPrLr8installs   "    rzc@seZdZdZddddddddddddd eedeeded eeeeeeeeeeeedd d d Zd dddddZ d dddddZ dS)r]akA 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_sizejustifyrLno_wraprNrOrPrQrRrU insert_liner,r-r.)_object highlighterr|r}rLr~rNrOrPrQrRrUrr2c CsX||_|pt|_||_||_||_||_||_||_| |_ | |_ | |_ | |_ | |_ dSN)rr%rr|r}rLr~rNrOrPrQrRrUr)rkrrr|r}rLr~rNrOrPrQrRrUrr7r7r8__init__s zPretty.__init__r*r+r/)rKoptionsr2c 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_widthr|rOrPrQrRpretty)r}rLr~stylez.__repr__ returned empty stringz dim italic)rz repr.indentrXrV) pretty_reprrrrUr|rOrPrQrRr)Z from_ansir}rLr!r~rr5rNZ ascii_onlyZwith_indent_guidesr)rkrKr pretty_strZ pretty_textr7r7r8__rich_console__0s>        zPretty.__rich_console__r(c CsLt|j|j|j|j|j|j|jd}|r>tdd| Dnd}t ||S)Nrcss|]}t|VqdSrr#.0r[r7r7r8 `sz*Pretty.__rich_measure__..r) rrrr|rOrPrQrRmax splitlinesr()rkrKrr text_widthr7r7r8__rich_measure__Ss zPretty.__rich_measure__)N) rnrorp__doc__rrintrqrrrr7r7r7r8r]sL  $r])rr2cCsd|jddd|jdfS)Nz defaultdict(z, {})z, {}))default_factoryrr7r7r8_get_braces_for_defaultdictes  rcCs,|jdkrdSdd|jdd|jdfS)N)deque([])zdeque()rz ], maxlen=)z deque(maxlen=)maxlenrr7r7r8_get_braces_for_dequems    rz array[Any]cCsd|jddd|jdfS)Nzarray(z, [rr)typecoderr7r7r8_get_braces_for_arraywsrcCsdS)N)z environ({rz environ({})r7rr7r7r8|rcCsdS)N)z Counter({rz Counter()r7rr7r7r8rrcCsdSN){}z{}r7rr7r7r8rrcCsdSrr7rr7r7r8rrcCsdS)N)z frozenset({rz frozenset()r7rr7r7r8rrcCsdSN)[]z[]r7rr7r7r8rrcCsdSrr7rr7r7r8rrcCsdS)N)rrzset()r7rr7r7r8rrcCsdS)N)(r()r7rr7r7r8rrcCsdS)N)zmappingproxy({rzmappingproxy({})r7rr7r7r8rr_BRACEScCs.t|ts$t|s$t|ds$t|o,t| S)z3Check if an object may be expanded by pretty print. __rich_repr__)rY _CONTAINERSr hasattrr9r r6r7r7r8 is_expandables rc@seZdZUdZdZeed<dZeed<dZeed<dZ eed<dZ eed<dZ e ed <dZ e ed <dZe ed <d Zeeded <dZeed<dZeed<eedddZeee dddZedddZdeee edddZd S) Nodez4A node in a repr tree. May be atomic or a container.rVkey_repr value_repr open_brace close_braceemptyFlastis_tuple is_namedtupleNchildrenz: key_separatorz, separatorr2ccs|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,) rrrrrrrlen iter_tokensrrrr)rkchildr7r7r8rs"     zNode.iter_tokens) start_lengthrOr2cCs0|}|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$)rkrrO total_lengthtokenr7r7r8 check_lengths   zNode.check_lengthcCsd|}|SNrV)joinr)rkZ repr_textr7r7r8__str__sz Node.__str__Pr{)rr|rRr2cCsxt|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_rootrrrXcss|]}t|VqdSr)strrr7r7r8rszNode.render..)_Liner expandableexpandedrexpandr)rkrr|rRlinesZline_nor[repr_strr7r7r8renders    z Node.render)rr{F)rnrorprrrrrrrrrrrqrrrrrrrrrrrrrr7r7r7r8rs0           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.NparentFrrrVtextsuffix whitespacerrrcCst|jdk o|jjS)z"Check if the line may be expanded.N)rqrrrkr7r7r8rsz_Line.expandable)rOr2cCs:t|jt|jt|j}|jdk s,t|j||S)z4Check this line fits within a given number of cells.N)rrr$rrrrdr)rkrOrr7r7r8rsz_Line.check_length)r|r2c 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)rrdrrrrrrrrr rrrr) rkr|rrnew_lineZchild_whitespaceZ tuple_of_onerrrr[r7r7r8rs8   z _Line.expandcCsH|jr |j|j|jpdS|j|j|jp4d|jSdSr)rrrrrr_rr7r7r8r(s"z _Line.__str__)rnrorprrrrrrrqrrrrrrrrpropertyrrrrrrr7r7r7r8rs       !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)getattrrArbtuple)r1r r7r7r8_is_namedtuple1s r)rrOrPrQr2c 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. r0c 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 )rYbytesrrrjrA)r1Z truncatedZobj_reprerror)rPr7r8to_reprYs    "ztraverse..to_reprFr)r1rootdepthr2c"st}|krtddSt}dk o0|k}tttttttffddd}ztd}Wnt k r|d}YnXd}|sztd rt s }Wnt k rYnX|dk r|t j d d} t ||} jj} | rg} | j} |r0| rtd | d d}nt| d d}n| rRtd | dd| |dd}nt| dd| |d}t| D]h\}}t|tr|\}}||dd}||_||_d|_| |n||dd}||_| |qpn&t| rd | dn| dg|d}|ntr*|s*|g} | j} tr|rZtjjd d}ntjjdd| |d}tttttttgtffdfdd }t|D]T\}\}}}|rtt||d}n||dd}||_||_d|_| |qntjjdg|d}|nptr tts |s tr |g} | j} |r~tjjd d}ntjjdd| |jjdd}tfddtDD]>\}}t |j|dd}|j|_||_d|_| |q|ntrt rˆ|jj} |rFt| d d}npg} | j} t| dd| | dd}t!"D]8\}\}}||dd}||_||_d|_| |q||ntt#rt#D]}t|r|}qq҈|t$|\}}}|r(t|d|d}nX|j%tj%krNt|d }n2rrg} t||| |d}| j} t&}|d}tt'rt("}dk rt)|}t*|D]:\} \}}||dd}||_| |k|_| |qnRt(}!dk r t)|!}!t*|!D],\} }||dd}| |k|_| |qdk r|kr| td!|d"d nt|g|d#}|nt|d }ttk|_+t|_,|S)$zWalk the object depth first.z...)rN) rich_argsr2css|D]x}t|trvt|dkr>|\}}}||kr2q||fVq|t|dkr^|\}}||fVq|t|dkr||dVq|VqdS)Nrr)rYrr)rargkeyrdefaultr7r7r8iter_rich_argsys        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)rjrnamerAcallable)attrrJrZ attr_fieldsr1r7r8 iter_attrss z/traverse.._traverse..iter_attrs)rrrrrc3s$|]}|jrt|jr|VqdSr)rjrr)rfieldr6r7r8rs z.traverse.._traverse..)rrrr)rrz... +T)rrr)-idrr5rrrrrrrAr rrlist __class__rnappendr rYrrrrr9r:rrr rBr rrrH_asdictitemsrrr;r_MAPPING_CONTAINERSiterr enumeraterr)"r1rrZobj_idZobj_typeZreached_max_depthrZfake_attributesZrich_repr_resultrargs class_namerrrrrrrZ child_noderrrJZ repr_callablerZcontainer_typerrrZ num_itemsZlast_item_indexZ iter_itemsindexZ iter_values) _traverserQrO pop_visited push_visitedr visited_idsrr8rms| $                                 ztraverse.._traverseT)r)Fr)rrsetaddremoverqrr)rrOrPrQrr7)rrQrOrPrrrrr8traverseDs&~ rrr{r)rrr|rOrPrQrRr2c 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. )rOrPrQ)rr|rR)rYrrr) rrr|rOrPrQrRrrr7r7r8rns rT)rKrNrOrPrQrR)rrKrNrOrPrQrRr2c 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. NrI)rOrPrQrNrRrLT)Z soft_wrap)rr\r])rrKrNrOrPrQrRZ_consoler7r7r8ris ri__main__c@seZdZedddZdS) BrokenReprrcCs dddS)Nrrzthis will failr7rr7r7r8r;szBrokenRepr.__repr__Nrnrorprr;r7r7r7r8r sr ) NamedTuplec@s:eZdZUeed<eed<eed<eed<eeed<dS)StockKeepingUnitr descriptionZpricecategoryZreviewsN)rnrorprrrfloatrr7r7r7r8r s r fooz Hello World!gZd;Y@gʡE3t@gcA)rrrr{rrZcrumbleZappleZrhubarbZbutterZsugarZflourZorangeZpearZkumquatAXduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurianduriandurian)FTNzSparkling British Spring WaterzCarbonated spring waterg?Zwaterz its amazing!z its terrible!)rbarrcounterZatomic namedtupleZBroken)r\)rNrPc@seZdZedddZdS)ThingrcCsdS)NzHello World!r7rr7r7r8r;szThing.__repr__Nr r7r7r7r8rsr)NrIFFNNNF)NrIFFNNNF)NNN)rre collectionsr>rDosr@rvrrrrrrrr r r itertoolsr typesr typingrrrrrrrrrrrrrZpip._vendor.rich.reprrrr4rr3 ImportErrorrVrZ_loopr Z_pickr!abcr"cellsr$rr%Zjupyterr&r'Zmeasurer(rr)rKr*r+r,r-r.r/rqr9r:objectrBrrCrHrrr`r5rYrzr]rrr_Environdict frozensetrrrrrrkeysrrrrrrrrrirnr r r ddatarrtr\rr7r7r7r8s    <             1  Rh" (  TC / + %     !