a )ji@sddlZddlmZddlmZddlmZddlmZm Z ddl m Z ddl m Z mZmZmZmZmZmZmZdd lmZdd lmZmZmZmZdd lmZmZdd lm Z m!Z!ed dddddZ"ee#dfZ$ee dZ%GdddZ&eGdddZ'e'Z(GdddZ)dS)N) lru_cache)count) attrgetter)dumpsloads) getrandbits)AnyDictIterableListOptionalTypeUnioncast)errors)ColorColorParseError ColorSystem blend_rgb)Result rich_repr)DEFAULT_TERMINAL_THEME TerminalTheme_color_bgcolor _attributes_set_attributes_link_metaStylec@s@eZdZdZdgZeddddZdedee dd d Z dS) _Bitz.A descriptor to get/set a style attribute bit.bitN)bit_noreturncCsd|>|_dS)Nr)r#)selfr$r'0E:\soccer\ml_env\Lib\site-packages\rich/style.py__init__sz _Bit.__init__r )objobjtyper%cCs |j|j@r|j|j@dkSdS)Nr)rr#r)r&r*r+r'r'r(__get__!s z _Bit.__get__) __name__ __module__ __qualname____doc__ __slots__intr)r r boolr,r'r'r'r(r"sr"c@seZdZUdZeeed<eeed<eed<eed<eeed<eed<ee ed<gd Z d d d d dddddddddd Z ddddddddddddddd d d!d!d"d#d$d$d%Z d&d&d&d&d&d&d&d&d&d&d&d&d&d&d&d&d&d'ee eefee eefeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeefd'd(d)Zedd*d+d,Zedeeeedd-d.d/Zeeeeefdd0d1d2Zedeeeefedd3d4d5Zed6Zed7Zed8Zed9Zed:Zed;Zed<Zed=Zed>Z ed?Z!ed@Z"edAZ#edBZ$e%ed*dCdDZ&ed*dEdFZ'ed*dGdHZ(e)edIdJdKZ*ee+dLdMeedNdOdPZ,eee-e-dQdRdSZ.e/d*dTdUZ0eedVdWdXZ1eedVdYdZZ2ed*d[d\Z3e%eed*d]d^Z4e%eed*d_d`Z5e%eed*dadbZ6e%ed*dcddZ7e%dd*dedfZ8e%eeefd*dgdhZ9e%dd*didjZ:ee+dkdMeddldmdnZ;e+dLdMdee<edodpdqZ=ee>dddrdsdtZ?edddrdudvZ@dd*dwdxZAe+dydMdd*dzd{ZBdeedd|d}d~ZCde)jDddeee)eedddZEdeed&dddZFe+dLdMedddNddZGedddNddZHd&S)r aA terminal style. A terminal style consists of a color (`color`), a background color (`bgcolor`), and a number of attributes, such as bold, italic etc. The attributes have 3 states: they can either be on (``True``), off (``False``), or not set (``None``). Args: color (Union[Color, str], optional): Color of terminal text. Defaults to None. bgcolor (Union[Color, str], optional): Color of terminal background. Defaults to None. bold (bool, optional): Enable bold text. Defaults to None. dim (bool, optional): Enable dim text. Defaults to None. italic (bool, optional): Enable italic text. Defaults to None. underline (bool, optional): Enable underlined text. Defaults to None. blink (bool, optional): Enabled blinking text. Defaults to None. blink2 (bool, optional): Enable fast blinking text. Defaults to None. reverse (bool, optional): Enabled reverse text. Defaults to None. conceal (bool, optional): Enable concealed text. Defaults to None. strike (bool, optional): Enable strikethrough text. Defaults to None. underline2 (bool, optional): Enable doubly underlined text. Defaults to None. frame (bool, optional): Enable framed text. Defaults to None. encircle (bool, optional): Enable encircled text. Defaults to None. overline (bool, optional): Enable overlined text. Defaults to None. link (str, link): Link URL. Defaults to None. rrrr_hash_nullr) rrrrr_link_id_ansi_style_definitionr4r5r12345678921515253) rr dimbolditalic underlineblinkblink2reverseconcealstrike underline2frameencircleoverline)rQdrRbrSirTurUrVrWrrXcrYsrZuur[r\r]oN)colorbgcolorrRrQrSrTrUrVrWrXrYrZr[r\r]linkmetacCs d|_d|_tttftddd}|dur0dn|||_|durFdn|||_t|du|duobd|duold|duovd|duod|duod| duod | duod | duod | duod | duod |duod|duodf |_|jrt|rdpd|rdpd|odpd|rdpd|rdpd|r*dp,d| r8d p:d| rFd pHd| rTd pVd| rbd pdd| rpd prd|r~dpd|rdpdf nd|_ ||_ |durdnt ||_ |s|rt tt|j nd|_d|_|jp|p|p|p| |_dS)N)rgr%cSst|tr|St|SN) isinstancerparse)rgr'r'r( _make_colorsz#Style.__init__.._make_colorrFrHrL @rr)r7r8rrstrrrsumrrrrrnext _id_generatorhashr6r4r5)r&rgrhrRrQrSrTrUrVrWrXrYrZr[r\r]rirjrnr'r'r(r)s\            #   $zStyle.__init__r%cCstS)zCCreate an 'null' style, equivalent to Style(), but more performant.) NULL_STYLE)clsr'r'r(nullsz Style.null)rgrhr%cCsV|t}d|_d|_||_||_d|_d|_d|_d|_ d|_ |pF| |_ d|_ |S)aCreate a new style with colors and no attributes. Returns: color (Optional[Color]): A (foreground) color, or None for no color. Defaults to None. bgcolor (Optional[Color]): A (background) color, or None for no color. Defaults to None. Nrrx) __new__r r7r8rrrrrr6rr5r4)rrgrhstyler'r'r( from_colors  zStyle.from_color)rjr%cCsh|t}d|_d|_d|_d|_d|_d|_d|_t ||_ t t t |j |_d|_| |_|S)zCreate a new style with meta data. Returns: meta (Optional[Dict[str, Any]]): A dictionary of meta data. Defaults to None. Nr)rr r7r8rrrrrrrr{r|r}r6r4r5)rrjrr'r'r( from_metas  zStyle.from_meta)rjhandlersr%cKs2|dur in|}|dd|D||S)aCreate a blank style with meta information. Example: style = Style.on(click=self.on_click) Args: meta (Optional[Dict[str, Any]], optional): An optional dict of meta information. **handlers (Any): Keyword arguments are translated in to handlers. Returns: Style: A Style with meta information attached. NcSsi|]\}}d||qS)@r').0keyvaluer'r'r( zStyle.on..)updateitemsr)rrjrr'r'r(onszStyle.onrrrFrGrHrIrJrKrLrMrNrOrPcCs|jS)z+Get a link id, used in ansi code for links.)r6r&r'r'r(link_id!sz Style.link_idcCs|jdurg}|j}|j}|d@r|d@r>||jr8dnd|d@rX||jrRdnd|d @rr||jrld nd |d @r||jrd nd|d@r|d@r||jrdnd|d@r||jrdnd|d@r||j rdnd|d@r||j rdnd|d@r||j rdnd|d@r|d @rD||j r>d!nd"|d#@rb||j r\d$nd%|d&@r||jrzd'nd(|d)@r||jrd*nd+|jdur||jj|jdur|d,||jj|jr|d-||jd.|pd/|_|jS)0z-Re-generate style definition from attributes.NrrRznot boldrFrQznot dimrHrSz not italicrLrTz not underlinerorUz not blinkrprVz not blink2rqrWz not reverserrrXz not concealrsrYz not strikertrZznot underline2rur[z not framervr\z not encirclerwr]z not overlinerri none)r8appendrrRrQrSrTrUrVrWrXrYrZr[r\r]rnamerrjoin)r& attributesrbitsr'r'r(__str__&sV             z Style.__str__cCs|j S)z;A Style is false if it has no attributes, colors, or links.)r5rr'r'r(__bool__TszStyle.__bool__) color_systemr%cCs4|jdur.g}|j}|j}|j|j@}|r|d@r@||d|d@rT||d|d@rh||d|d@r|||d|d@rtdd D]}|d|>@r|||q|d @rtd d D]}|d|>@r|||q|jdur||j| |j dur"||j |j d d d ||_|jS)zGenerate ANSI codes for this style. Args: color_system (ColorSystem): Color system. Returns: str: String containing codes. NrrrFrHrLrGrrMr F foreground;) r7r _style_maprrrangerextend downgradeget_ansi_codesrr)r&rsgrrrrr#r'r'r(_make_ansi_codesXs@            zStyle._make_ansi_codesru)maxsizerr%cCs6zt||WStjy0|YS0dS)zNormalize a style definition so that styles with the same effect have the same string representation. Args: style (str): A style definition. Returns: str: Normal form of style definition. N)ryrmrStyleSyntaxErrorstriplower)rrr'r'r( normalizes zStyle.normalize)valuesr%cGs&|D]}|dur|SqtddS)zPick first non-None style.Nz$expected at least one non-None style) ValueError)rrrr'r'r( pick_firsts zStyle.pick_firstccsd|jdfVd|jdfVd|jdfVd|jdfVd|jdfVd|jdfVd|jdfVd|jdfVd |jdfVd |j dfVd |j dfVd |j dfVd |j dfVd|j dfVd|jdfV|jrd|jfVdS)NrgrhrRrQrSrTrUrVrWrXrYrZr[r\rirj)rgrhrRrQrSrTrUrVrWrXrYrZr[r\rirrjrr'r'r( __rich_repr__s"zStyle.__rich_repr__)otherr%cCst|tstS||kSrkrlr NotImplemented__hash__r&rr'r'r(__eq__s z Style.__eq__cCst|tstS||kSrkrrr'r'r(__ne__s z Style.__ne__cCs$|jdur|jStt||_|jSrk)r4r} _hash_getterrr'r'r(rs zStyle.__hash__cCs|jS)z.The foreground color or None if it is not set.)rrr'r'r(rgsz Style.colorcCs|jS)z.The background color or None if it is not set.)rrr'r'r(rhsz Style.bgcolorcCs|jS)zLink text, if set.)rrr'r'r(risz Style.linkcCs|jdup|jjS)z6Check if the style specified a transparent background.N)rh is_defaultrr'r'r(transparent_backgroundszStyle.transparent_backgroundcCs t|jdS)zA Style with background only.)rh)r rhrr'r'r(background_styleszStyle.background_stylecCs&|jduriSttttft|jS)z=Get meta information (can not be changed after construction).N)rrr ryrrrr'r'r(rjsz Style.metacCsp|jr tS|t}d|_d|_d|_d|_|j|_|j |_ |j |_ |j rTt t nd|_ d|_d|_d|_|S)z+Get a copy of the style with color removed.NrxF)r5rrr r7r8rrrrrr{r|r6rr4r&rr'r'r( without_colors zStyle.without_colorrw)style_definitionr%c Cs|dks|s|S|j}d}d}i}d}t|}|D]P}|} | dkrt|d} | sltdzt | Wn<t y} z$td| d| dWYd} ~ n d} ~ 00| }q>| dkrt|d} | | } | durtd | d || <q>| d kr&t|d} | s td | }q>| |vr>d ||| <q>zt | Wn>t y} z$td| d| dWYd} ~ n d} ~ 00| }q>t f|||d|} | S)a Parse a style definition. Args: style_definition (str): A string containing a style. Raises: errors.StyleSyntaxError: If the style definition syntax is invalid. Returns: `Style`: A Style instance. rNrrxzcolor expected after 'on'zunable to parse z as background color; notz,expected style attribute after 'not', found FrizURL expected after 'link'Tz as color; )rgrhri)rrSTYLE_ATTRIBUTESitersplitrr{rrrrmrgetr ) rrrrgrhrriwords original_wordworderror attributerr'r'r(rmsb           z Style.parse)themer%cCs |pt}g}|j}|j}|j}|jr.||}}|jr`|durB|jn||}t t ||j d}|dur||}|d|j |d|j |dur|j|dd}|d|j |j r|d|jr|d |jr|d |jr|d |jr|d d |S)zGet a CSS style rule.Ng?zcolor: ztext-decoration-color: Frzbackground-color: zfont-weight: boldzfont-style: italicztext-decoration: underlineztext-decoration: line-throughztext-decoration: overlinez; )rrrgrhrWrQforeground_color get_truecolorr from_tripletrbackground_colorhexrRrSrTrYr]r)r&rcssrrgrhr theme_colorr'r'r(get_html_style3s>   zStyle.get_html_style)stylesr%cCst|}t|t|S)zCombine styles and get result. Args: styles (Iterable[Style]): Styles to combine. Returns: Style: A new style instance. rrzr{rr iter_stylesr'r'r(combineXs z Style.combinecGst|}t|t|S)zCombine styles from positional argument in to a single style. Args: *styles (Iterable[Style]): Styles to combine. Returns: Style: A new style instance. rrr'r'r(chaines z Style.chaincCs||jr tS|t}|j|_|j|_|j|_|j|_|j|_|j |_ |j |_ |j r\t t nd|_ |j|_d|_|j|_|S)zwGet a copy of this style. Returns: Style: A new Style instance with identical attributes. rxF)r5rrr r7r8rrrrrr{r|r6r4rrr'r'r(copyrs z Style.copyrrcCsf|jr tS|t}|j|_|j|_|j|_|j|_|j|_|j |_ d|_ d|_ d|_ d|_d|_ |S)zGet a copy of this style with link and meta information removed. Returns: Style: New style object. NrxF)r5rrr r7r8rrrrrr6r4rrr'r'r(clear_meta_and_linkss zStyle.clear_meta_and_links)rir%cCsl|t}|j|_|j|_|j|_|j|_|j|_|j|_||_|rNt t nd|_ d|_ d|_ |j|_|S)zGet a copy with a different value for link. Args: link (str, optional): New value for link. Defaults to None. Returns: Style: A new Style instance. rxNF)rr r7r8rrrrrr{r|r6r4r5r)r&rirr'r'r( update_links zStyle.update_linkrxF)rlegacy_windows)textrrr%cCsd|r |dur|S|jp||}|r6d|d|dn|}|jr`|s`d|jd|jd|d}|S) aDRender the ANSI codes for the style. Args: text (str, optional): A string to style. Defaults to "". color_system (Optional[ColorSystem], optional): Color system to render to. Defaults to ColorSystem.TRUECOLOR. Returns: str: A string containing ANSI style codes. Nz[mzz]8;id=rz\z]8;;\)r7rrr6)r&rrrattrsrenderedr'r'r(renders  z Style.render)rr%cCs(|p t|}tj||ddS)zWrite text with style directly to terminal. This method is for testing purposes only. Args: text (Optional[str], optional): Text to style or None for style name.  N)rysysstdoutwriter)r&rr'r'r(tests z Style.testcCs|dus|jr|S|jr|S|t}d|_d|_|jp<|j|_|jpJ|j|_|j|j@|j|j@B|_|j|jB|_|j p|j |_ |j p|j |_ |j|_|j r|j rt i|j |j |_ n|j p|j |_ d|_|Srk)r5rr r7r8rrrrrr6rrrjr4)r&r new_styler'r'r(_adds*    z Style._addcCs||}|jr|S|Srk)rrir)r&rcombined_styler'r'r(__add__s z Style.__add__)NN)N)N)N)rx)N)Ir-r.r/r0r r__annotations__r2r3bytesr1rrrryr rr) classmethodrrrrr"rRrQrSrTrUrVrWrXrYrZr[r\r]propertyrrrrrrr StyleTyperrrrrrrgrhrirrrjrrmrrr rrrrr TRUECOLORrrrrr'r'r'r(r 'sD      L ".+A$    c@sdeZdZdZdgZdddddZedd d Zee dd d Z e dd ddZ e dddZ dS) StyleStackzA stack of styles._stackr N) default_styler%cCs |g|_dSrkr)r&rr'r'r(r)szStyleStack.__init__r~cCsd|jdS)Nz rrr'r'r(__repr__szStyleStack.__repr__cCs |jdS)z&Get the Style at the top of the stack.rrr'r'r(currentszStyleStack.currentrcCs|j|jd|dS)z}Push a new style on to the stack. Args: style (Style): New style to combine with current style. rN)rrrr'r'r(push szStyleStack.pushcCs|j|jdS)z}Pop last style and discard. Returns: Style: New current style (also available as stack.current) r)rpoprr'r'r(rs zStyleStack.pop) r-r.r/r0r1r)ryrrr rrrr'r'r'r(rsr)*r functoolsr itertoolsroperatorrpicklerrrandomrtypingrr r r r r rrrxrrgrrrrreprrrterminal_themerrrryrr|r"r rrr'r'r'r(s2    (    W