o L@sdZddlmZddlmZddlmZddlmZddlmZddlm Z ddl Z Gd d d e Z Gd d d e Z d dZdddZdS)z2Stylized printing using ANSI codes utility module.)absolute_import)division)unicode_literals) properties)mappings)textNc@s,eZdZdZddZeddZddZdS) _StyleContextzGContains style information used in recursive parsingin TypedTextParser.cCs||_||_dSNcolorattrs)selfr r rE/tmp/google-cloud-sdk/lib/googlecloudsdk/core/console/style/parser.py__init__s z_StyleContext.__init__cCs |s|dgS||j|jpgSr r )clstext_attributesrrrFromTextAttributes#s z _StyleContext.FromTextAttributescCs2|s|S|jp |j}t|dg|j}|||S)Nr )r getattrr __class__)r r new_color new_attrsrrrUpdateFromTextAttributes)s   z&_StyleContext.UpdateFromTextAttributesN)__name__ __module__ __qualname____doc__r classmethodrrrrrrrs   rc@sBeZdZdZdZdZdZdZdZddZ d d Z dd dZ d S)TypedTextParserz%Logger used to styled text to stderr.z[mz38;5;{}z39;0 cCs||_||_dS)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)r r!r"rrrr=s zTypedTextParser.__init__c sd}d}tt|dgtt|dgB}|r4|dtdd|D7}|dtfdd|D7}t|ddp?t|dd}|r\|rH|d7}|j|j7}|rW|d7}|j7}d \}}|rjj|j }|rtj|j }||fS) zCReturns the ANSI start and reset sequences for the text_attributes.r ;cSsg|]}t|jqSr)six text_typevalue.0attrrrr Qs z@TypedTextParser._GetAnsiSequenceForAttribute..cs"g|] }td|jjqS)z%02x)r%r&r'ATTR_OFFr(r rrr+Tsr N)r#r#) setrjoinsortedSET_FOREGROUNDformatr'RESETCSISGR) r r style_contextstyle_sequencereset_sequencer r begin_style end_stylerr-r_GetAnsiSequenceForAttributeJs<     z,TypedTextParser._GetAnsiSequenceForAttributeNTc Cst|tjr|S|o |j}g}|j|j}|||\}}|r%||}nt |}|j D] } t| t j rH| |j| ||d|rG| |q-| | q-d|} |r^|jr^|j| } |rhdj|| |d} | 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. )r6stylizer#z{begin_style}{text}{end_style})r9rr:) isinstancer% string_typesr"r!r&r;rrrtextsr TypedTextappendParseTypedTextToStringr/ format_strr2) r typed_textr6r< parsed_chunksrr9r:new_style_contextchunk parsed_textrrrrBhsD           z&TypedTextParser.ParseTypedTextToString)NT) rrrrr4r5r1r3r,rr;rBrrrrr2s rcCsJtjjj}t|tjjjjjktjjj dktjjj gr#dSdS)z)Returns true if colors should be enabled.neverFT) rVALUEScoreinteractive_ux_styleGetanyInteractiveUXStylesOFFnameshow_structured_logs disable_colorGetBool)stylerrr_ColorsEnableds rVTcCs$t}|o|}|p t}t||S)z=Returns a typed text parser, creating a new one if necessary.)rVrGetStyleMappingsr)enabledr! can_enablerrrGetTypedTextParsers  rZ)TN)r __future__rrrgooglecloudsdk.corer!googlecloudsdk.core.console.stylerrr%objectrrrVrZrrrrs      m