LdZddlmZddlmZddlmZddlmZddlmZddlm Z ddl Z Gd d e Z Gd d e Z d ZddZy)z2Stylized printing using ANSI codes utility module.)absolute_import)division)unicode_literals) properties)mappings)textNc,eZdZdZdZedZdZy) _StyleContextzGContains style information used in recursive parsingin TypedTextParser.c ||_||_yNcolorattrs)selfrrs /lib/googlecloudsdk/core/console/style/parser.py__init__z_StyleContext.__init__sDJDJcZ|s |dgS||j|jxsgSr r )clstext_attributess rFromTextAttributesz _StyleContext.FromTextAttributes#s/  r] $$o&;&;&Ar BBrc|s|S|jxs |j}t|dg|jz}|j||S)Nr)rgetattrr __class__)rr new_color new_attrss rUpdateFromTextAttributesz&_StyleContext.UpdateFromTextAttributes)sF  k%%3I"5 BI >>)Y //rN)__name__ __module__ __qualname____doc__r classmethodrrrrr r s%OCC 0rr c<eZdZdZdZdZdZdZdZdZ dZ d d Z y ) TypedTextParserz%Logger used to styled text to stderr.z[mz38;5;{}z39;0 c ||_||_y)amCreates a styled logger used to print styled text to stdout. Args: style_mappings: (StyleMapping), A mapping from TextTypes to mappings.TextAttributes used to stylize the output. If the map does not contain a TextAttribute object, plain text will bef logged. style_enabled: (bool), whether logged text should be styled. N)style_mappings style_enabled)rr)r*s rrzTypedTextParser.__init__=s)D&Drc  d}d}tt|dgtt|dgz}|r|djt|Dcgc]!}t j |j #c}z }|djt|Dcgc]1}t j d|j |jzz3c}z }t|ddxs t|dd}|rE|r|dz }||jj|j z }|r|dz }||jz }d\}} |r|j|z|jz}|r|j|z|jz} || fScc}wcc}w)zCReturns the ANSI start and reset sequences for the text_attributes.r;z%02xrN)r,r,) setrjoinsortedsix text_typevalueATTR_OFFSET_FOREGROUNDformatRESETCSISGR) rr style_contextstyle_sequencereset_sequencerattrr begin_style end_styles r_GetAnsiSequenceForAttributez,TypedTextParser._GetAnsiSequenceForAttributeJsNN 4 5 '26 78E 05)705#-- #)7"899n  $mmFdjj4==&@AB  n _gt 43 ]GT 2  #++225;;??n # "n#KHH~-8k((^+dhh6i  !!+)7 s &E<6FNct|tjr|S|xr |j}g}|j|j }|j ||\}}|r|j|}ntj|}|jD]e} t| tjr8|j|j| |||sC|j|U|j| gdj|} |r'|j r|j j#| } |rdj#|| |} | S)a4Parses a TypedText object into plain and ansi-annotated unicode. The reason this returns both the plain and ansi-annotated strings is to support file logging. Args: typed_text: mappings.TypedText, typed text to be converted to unicode. style_context: _StyleContext, argument used for recursive calls to preserve text attributes and colors. Recursive calls are made when a TypedText object contains TypedText objects. stylize: bool, Whether or not to stylize the string. Returns: str, the parsed text. )r:stylizer,z{begin_style}{text}{end_style})r>rr?) isinstancer1 string_typesr*r)r2r@rr rtextsr TypedTextappendParseTypedTextToStringr/ format_strr6) r typed_textr:rB parsed_chunksrr>r?new_style_contextchunk parsed_texts rrHz&TypedTextParser.ParseTypedTextToStringhsM"*c../ ,$,,GM))**>*>?O!>>(K'@@ (::?K!! E4>> *T88 +9     { +U#"''-(K?55#..55kBk4;;!<k r)NT) rrr r!r8r9r5r7r4rr@rHr#rrr%r%2s8-# #. %( '"<>B%)4rr%ctjjjj }t |tjjj jjk(tjjjj dk7tjjjjgryy)z)Returns true if colors should be enabled.neverFT) rVALUEScoreinteractive_ux_styleGetanyInteractiveUXStylesOFFnameshow_structured_logs disable_colorGetBool)styles r_ColorsEnabledr]s    5 5 9 9 ;% z  %%99==BBB115577B**224 67  rcjt}|xr|}|xstj}t||S)z=Returns a typed text parser, creating a new one if necessary.)r]rGetStyleMappingsr%)enabledr) can_enables rGetTypedTextParserrbs4*  "7'!@X%>%>%@.  11r)TN)r! __future__rrrgooglecloudsdk.corer!googlecloudsdk.core.console.stylerrr1objectr r%r]rbr#rrrgsD9&'*62 0F0,jfjZ2r