hh\ZdZddlmZddlmZddlmZddlZddlZddlZddlm Z ddlm Z ddlm Z Gd d e ZGd d eZGd deZGddeZGdde ZGddeZGddeZGdde ZGdde Zd!dZd"dZdZd#dZdZd"d Zy)$aA module for console attributes, special characters and functions. The target architectures {linux, macos, windows} support inline encoding for all attributes except color. Windows requires win32 calls to manipulate the console color state. Usage: # Get the console attribute state. out = log.out con = console_attr.GetConsoleAttr(out=out) # Get the ISO 8879:1986//ENTITIES Box and Line Drawing characters. box = con.GetBoxLineCharacters() # Print an X inside a box. out.write(box.dr) out.write(box.h) out.write(box.dl) out.write('\n') out.write(box.v) out.write('X') out.write(box.v) out.write('\n') out.write(box.ur) out.write(box.h) out.write(box.ul) out.write('\n') # Print the bullet characters. for c in con.GetBullets(): out.write(c) out.write('\n') # Print FAIL in red. out.write('Epic ') con.Colorize('FAIL', 'red') out.write(', my first.') # Print italic and bold text. bold = con.GetFontCode(bold=True) italic = con.GetFontCode(italic=True) normal = con.GetFontCode() out.write('This is {bold}bold{normal}, this is {italic}italic{normal},' ' and this is normal.\n'.format(bold=bold, italic=italic, normal=normal)) # Read one character from stdin with echo disabled. c = con.GetRawKey() if c is None: print 'EOF\n' # Return the display width of a string that may contain FontCode() chars. display_width = con.DisplayWidth(string) # Reset the memoized state. con = console_attr.ResetConsoleAttr() # Print the console width and height in characters. width, height = con.GetTermSize() print 'width={width}, height={height}'.format(width=width, height=height) # Colorize table data cells. fail = console_attr.Colorizer('FAIL', 'red') pass = console_attr.Colorizer('PASS', 'green') cells = ['label', fail, 'more text', pass, 'end'] for cell in cells; if isinstance(cell, console_attr.Colorizer): cell.Render() else: out.write(cell) )absolute_import)division)unicode_literalsN)console_attr_osencoding)textceZdZdZy)BoxLineCharacterszBox/line drawing characters. The element names are from ISO 8879:1986//ENTITIES Box and Line Drawing//EN: http://www.w3.org/2003/entities/iso8879doc/isobox.html N__name__ __module__ __qualname____doc__W/var/www/html/dev/engine/venv/lib/python3.12/site-packages/fire/console/console_attr.pyr r isrr cheZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZy)BoxLineCharactersUnicodez?unicode Box/line drawing characters (cp437 compatible unicode).u┐u┌u─u┬u┴u┘u└u│u┼u┤u├u╗u╔u═u╦u╩u╝u╚u║u╬u╣u╠Nr rrrdldrhhdhuulurvvhvlvrd_dld_drd_hd_hdd_hud_uld_urd_vd_vhd_vld_vrrrrrrqssG " " ! " " " " ! " " " $ $ # $ $ $ $ # $ $ $rrcheZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZdZdZdZdZdZdZdZdZdZdZdZy)BoxLineCharactersAsciiz"ASCII Box/line drawing characters.+-|#=Nrrrrr.r.ss* " " ! " " " " ! " " " $ $ # $ $ $ $ # $ $ $rr.c0eZdZdZdZdZdZdZdZdZ dZ dZ y)BoxLineCharactersScreenReader N) r rrrrrrrrrr r!rrrr5r5s/ " " " " " " " " "rr5ceZdZdZy)ProgressTrackerSymbols%Characters used by progress trackers.Nr rrrr8r8s-rr8ceZdZdZedZejdgejjZ ejdgejjZ dZ dZdZy ) ProgressTrackerSymbolsUnicoder9c gdS)N)u⠏u⠛u⠹u⠼u⠶u⠧rselfs r spin_marksz(ProgressTrackerSymbolsUnicode.spin_markss 55ru✓) text_typeXr0.N)r rrrpropertyr?r TypedText TextTypes PT_SUCCESSsuccess PT_FAILUREfailed interrupted not_started prefix_lengthrrrr;r;sd- 6 6 DNNE7dnn.G.G H' 4>>3%4>>+D+D E&++-rr;c4eZdZdZedZdZdZdZdZ dZ y) ProgressTrackerSymbolsAsciir9c gdS)N)r1/r0\rr=s rr?z&ProgressTrackerSymbolsAscii.spin_markss !!rOKrAr0rBN) r rrrrDr?rHrJrKrLrMrrrrOrOs2- " " ' &++-rrOceZdZdZdZdddddZdZd Zd Zd Z dd Z d Z ddZ dZ dZdZdZdZdZdZd dZdZdZdZdZdZdZdZy)! ConsoleAttra;Console attribute and special drawing characters and functions accessor. Use GetConsoleAttr() to get a global ConsoleAttr object shared by all callers. Use ConsoleAttr() for abstracting multiple consoles. If _out is not associated with a console, or if the console properties cannot be determined, the default behavior is ASCII art with no attributes. Attributes: _ANSI_COLOR: The ANSI color control sequence dict. _ANSI_COLOR_RESET: The ANSI color reset control sequence string. _csi: The ANSI Control Sequence indicator string, '' if not supported. _encoding: The character encoding. ascii: ASCII art. This is the default. utf8: UTF-8 unicode. win: Windows code page 437. _font_bold: The ANSI bold font embellishment code string. _font_italic: The ANSI italic font embellishment code string. _get_raw_key: A function that reads one keypress from stdin with no echo. _out: The console output file stream. _term: TERM environment variable value. _term_size: The terminal (x, y) dimensions in characters. Nz31;1mz33;1m32mz34;1m)redyellowgreenbluez39;0m)u▪u◆u▸u▫u◇u▹)u■u≡u∞uΦ·)o*r/r0c"|s|j}n|dk(rd}|xsd|_|rdn#tjddj |_|j rd|_d|_d|_ nd |_d|_d|_ d }|jd k(r2|s0t|_ |j|_ t|_n|jdk(r2|s0t|_ |j |_ t#|_n@t%|_ |rt'|_ |j(|_ t#|_t+j,g|_|rd nt+j0|_i|_y ) aConstructor. Args: encoding: Encoding override. ascii -- ASCII art. This is the default. utf8 -- UTF-8 unicode. win -- Windows code page 437. suppress_output: True to create a ConsoleAttr that doesn't want to output anything. wincp437asciiTERMz[14NFutf8)rr)_GetConsoleEncoding _encodingosgetenvlower_term SupportsAnsi_csi _font_bold _font_italicr_box_line_characters_BULLETS_UNICODE_bulletsr;_progress_tracker_symbols_BULLETS_WINDOWSrOr.r5_BULLETS_ASCIIrGetRawKeyFunction _get_raw_key GetTermSize _term_size_display_width_cache)r>rsuppress_outputis_screen_readers r__init__zConsoleAttr.__init__s] ))+h U h(DN&BIIfb,A,G,G,IDJ didoddidod ~~(8":"console_encodings rrhzConsoleAttr._GetConsoleEncoding1sG szz:t<  '--/""  $ $  rc|r||}|jrD||jvr6dj|j|j||j|S|S)ahGenerates a colorized string, optionally justified. Args: string: The string to write. color: The color name -- must be in _ANSI_COLOR. justify: The justification function, no justification if None. For example, justify=lambda s: s.center(10) Returns: str, The colorized string that can be printed to the console. z*{csi}{color_code}{string}{csi}{reset_code})csi color_code reset_codestring)ro _ANSI_COLORformat_ANSI_COLOR_RESETr>rcolorjustifys rColorizezConsoleAttr.ColorizeAsevf yyUd... 9 @ @ii%%e,++ A Mrct|tr|j|j}t||jdS)a8Converts a console output string buf to unicode. Mainly used for testing. Allows test comparisons in unicode while ensuring that unicode => encoding => unicode works. Args: buf: The console output string to convert. Returns: The console output string buf converted to unicode. replace) isinstancestrencoderi)r>bufs rConvertOutputToUnicodez"ConsoleAttr.ConvertOutputToUnicodeXs3#s JJt~~ &c sDNNI ..rc|jS)aReturns the box/line drawing characters object. The element names are from ISO 8879:1986//ENTITIES Box and Line Drawing//EN: http://www.w3.org/2003/entities/iso8879doc/isobox.html Returns: A BoxLineCharacters object for the console output device. )rrr=s rGetBoxLineCharactersz ConsoleAttr.GetBoxLineCharactershs  $ $$rc|jS)aReturns the bullet characters list. Use the list elements in order for best appearance in nested bullet lists, wrapping back to the first element for deep nesting. The list size depends on the console implementation. Returns: A tuple of bullet characters. )rtr=s r GetBulletszConsoleAttr.GetBulletsss ==rc|jS)zReturns the progress tracker characters object. Returns: A ProgressTrackerSymbols object for the console output device. )rur=s rGetProgressTrackerSymbolsz%ConsoleAttr.GetProgressTrackerSymbolss  ) ))rc|jS)zReturns the control sequence indicator string. Returns: The control sequence indicator string or None if control sequences are not supported. )ror=s rGetControlSequenceIndicatorz'ConsoleAttr.GetControlSequenceIndicators 99rc|jr|j|jsyd}|D]}|dz }|js|S|S)aVReturns the control sequence length at the beginning of buf. Used in display width computations. Control sequences have display width 0. Args: buf: The string to check for a control sequence. Returns: The control sequence length at the beginning of buf or 0 if buf does not start with a control sequence. r)ro startswithisalpha)r>rncs rGetControlSequenceLenz!ConsoleAttr.GetControlSequenceLensR 99CNN4995  A 1fa  H  Hrc|jS)zReturns the current encoding.)rir=s r GetEncodingzConsoleAttr.GetEncodings >>rc|jsyg}|r|j|j|r|j|jdj |jdj |S)a}Returns a font code string for 0 or more embellishments. GetFontCode() with no args returns the default font code string. Args: bold: True for bold embellishment. italic: True for italic embellishment. Returns: The font code string for the requested embellishments. Write this string to the console output to control the font settings. rcz {csi}{codes}m;)rcodes)roappendrprqrjoin)r>bolditalicrs r GetFontCodezConsoleAttr.GetFontCodes] 99  E  ll4??#  ll4$$%  ! !diisxx ! GGrc*|jdS)zReads one key press from stdin with no echo. Returns: The key name, None for EOF, for function keys, otherwise a character. r)ryr=s r GetRawKeyzConsoleAttr.GetRawKeys 4  Q  !!rc|jS)zReturns the TERM environment variable for the console. Returns: str: A str that describes the console's text capabilities )rmr=s rGetTermIdentifierzConsoleAttr.GetTermIdentifiers ::rc|jS)zReturns the terminal (x, y) dimensions in characters. Returns: (x, y): A tuple of the terminal x and y dimensions. )r{r=s rrzzConsoleAttr.GetTermSizes ??rct|ts t|S|jj |d}||Sd}d}d}|t|kr|j r6||dj |j r||j||dz }n2||dk(rt||}d}|dz }n|t||z }|dz }|t|krt||}||j|<|S)zReturns the display width of buf, handling unicode and ANSI controls. Args: buf: The string to count from. Returns: The display width of buf, handling unicode and ANSI controls. Nr r) rrlenr|getrorrmaxGetCharacterDisplayWidth)r>rcachedwidth max_widthis r DisplayWidthzConsoleAttr.DisplayWidths c3  Xo  & & * *3 5F  m EI A c#h, s12w))$))4 T ' 'AB 00 q6T>y)  Q )#a&11 Q c#h,E9%I%.Dc" rcF|jr|s|dfgSg}d}|t|kr{|j|j|}|dkr|j||ddf |S|||}||j ||dz}|j||||f|t|kr{|S)a Returns a list of (normal_string, control_sequence) tuples from buf. Args: buf: The input string containing one or more control sequences interspersed with normal strings. Returns: A list of (normal_string, control_sequence) tuples. rcrN)rorfindrr)r>rseqrrnormals rSplitIntoNormalAndControlz%ConsoleAttr.SplitIntoNormalAndControls 99CBi[ C A c#h, ((499a a Q CGR=!  J1Qxf d((QR1 1a jj&#a(#$ c#h, Jrcg}d}d}d}|j|D]L\}}d} ||z } |t|z }||krn#|j||d| zd}d}d}|| d}<|||zz }N|s|r|j||S)aSplits line into width length chunks. Args: line: The line to split. width: The width of each chunk except the last which could be smaller than width. Returns: A list of chunks, all but the last with display width == width. rcrFTN)rrr) r>linerlineschunkwkeeprcontrolrs r SplitLinezConsoleAttr.SplitLines E E A D99$?   d  AI S[ :  UVBQZ'(   ve    ll5 Lrc`|jdk7xrd|jvxsd|jvS)Nrbscreenxterm)rirmr=s rrnzConsoleAttr.SupportsAnsi5s2 NNg % >  # others r__eq__zColorizer.__eq__Ss <<3u: %%rc||k( Srrrs r__ne__zColorizer.__ne__Vsu} rc2|jt|kDSrrrs r__gt__zColorizer.__gt__Y <<#e* $$rc2|jt|kSrrrs r__lt__zColorizer.__lt__\rrc||k Srrrs r__ge__zColorizer.__ge___e| rc||kD Srrrs r__le__zColorizer.__le__brrcL|jj|jSr)rrrr=s r__len__zColorizer.__len__es 99 ! !$,, //rc|jSr)rr=s r__str__zColorizer.__str__hs <<rc|j|jj|j|j|xs |j y)aRenders the string as self._color on the console. Args: stream: The stream to render the string to. The stream given here *must* have the same encoding as sys.stdout for this to work properly. justify: The justification function, self._justify if None. N)writerrrrr)r>streamrs rRenderzColorizer.Renderks8 LL 4<<g6NOQrr)r rrrrrrrrrrrrrrrrrr:s: &%%0 Qrrctj}|s|sd}n|r||jk7rd}|rt|}|t_|S)aGets the console attribute state. If this is the first call or reset is True or encoding is not None and does not match the current encoding or out is not None and does not match the current out then the state is (re)initialized. Otherwise the current state is returned. This call associates the out file stream with the console. All console related output should go to the same stream. Args: encoding: Encoding override. ascii -- ASCII. This is the default. utf8 -- UTF-8 unicode. win -- Windows code page 437. reset: Force re-initialization if True. Returns: The global ConsoleAttr state object. Tr)rVrr)rresetattrs rrrwsL*  ( ($  e h$"2"2"44e  )D&*K# +rct|dS)a$Resets the console attribute state to the console default. Args: encoding: Reset to this encoding instead of the default. ascii -- ASCII. This is the default. utf8 -- UTF-8 unicode. win -- Windows code page 437. Returns: The global ConsoleAttr state object. T)rr)rrs rResetConsoleAttrrs  66rct|tsytjd|}tj|dk7rytj |dk(rytj |dvryy)atReturns the monospaced terminal display width of char. Assumptions: - monospaced display - ambiguous or unknown chars default to width 1 - ASCII control char width is 1 => don't use this for control chars Args: char: The character to determine the display width of. Returns: The monospaced terminal display width of char: either 0, 1, or 2. rNFCrCfFWrC)rr unicodedata normalize combiningcategoryeast_asian_width)chars rrrsl D#     ud +$4 A% D!T) ##D)T1  rc |y|xstj}tj||} |j ||S#t $r(|j ||rdndj |cYSwxYw)NNonerbackslashreplacer)rr encoding_utilDecoder UnicodeErrordecode)datarescapers rSafeTextrs0 \   7)557(   x 8&  MM( M   VHF0 J VH  sA .A=<A=c|yt|tr|St|} |jdS#t$rYnwxYw |jt j S#t$rYnwxYw|jdS)aEncode data to bytes. The primary use case is for base64/mime style 7-bit ascii encoding where the encoder input must be bytes. "safe" means that the conversion always returns bytes and will not raise codec exceptions. If data is text then an 8-bit ascii encoding is attempted, then the console encoding, and finally utf-8. Args: data: Any bytes, string, or object that has str() or unicode() methods. Returns: A bytes string representation of the data. rz iso-8859-1r)rbytesrrUnicodeEncodeErrorrr)r ss r EncodeToBytesrs  \ e K $i! 88L !!    88N$002 33    ' s3 ??&A** A65A6ch|xstj}tj||S)atConverts the given string, bytes, or object to a text string. Args: data: Any bytes, string, or object that has str() or unicode() methods. encoding: A suggesting encoding used to decode. If this encoding doesn't work, other defaults are tried. Defaults to GetConsoleAttr().GetEncoding(). Returns: A text string representation of the data. r)rrrr )r rs rr r s- 7)557(   dX 66rrr)NT)r __future__rrrrjrr fire.consolerrrr objectr rr.r5r8r;rOrVrrrrrrr rrrrs"FR'' )204 . 4  $:  .V. $:  "8 f?&f?R :Q:QzB 7  F(V&R 7r