U
‰d"& ã @ sò d dl mZ d dlmZ d dlmZmZmZ d dlm Z m
Z
d dlm Z ddl
mZ dd gZG d
d„ dƒZd d „ e
¡ D ƒZd
d „ e ¡ D ƒZi Z ee jƒD ]*\Z\ZZZded›ed›ed›e e< qœdddœdd „ZG dd„ deƒZeƒ Z dS )é )Ú annotations)Ú Formatter)Ú GeneratorÚListÚOptional)ÚBG_ANSI_COLORSÚFG_ANSI_COLORS)Ú _256_colorsé )ÚStyleAndTextTuplesÚANSIÚ ansi_escapec @ sŠ e Zd ZdZdddœdd„Zddœd d
„Zd dd œd
d„Zddœdd„Zddœdd„Zddœdd„Z ddd dœdd„Z
dd dœdd„Z d S ) r aÒ
ANSI formatted text.
Take something ANSI escaped text, for use as a formatted string. E.g.
::
ANSI('\x1b[31mhello \x1b[32mworld')
Characters between ``\001`` and ``\002`` are supposed to have a zero width
when printed, but these are literally sent to the terminal output. This can
be used for instance, for inserting Final Term prompt commands. They will
be translated into a prompt_toolkit '[ZeroWidthEscape]' fragment.
ÚstrÚNone)ÚvalueÚreturnc C sl || _ g | _d | _d | _d| _d| _d| _d| _d| _d| _ d| _
| ¡ }| d ¡ |D ]}| |¡ qXd S )NF)
r Ú_formatted_textÚ_colorÚ_bgcolorÚ_boldÚ
_underlineÚ_strikeÚ_italicÚ_blinkÚ_reverseÚ_hiddenÚ _parse_corotÚsend)Úselfr ÚparserÚc© r! úF/tmp/pip-unpacked-wheel-4x_7prb2/prompt_toolkit/formatted_text/ansi.pyÚ__init__ s
z
ANSI.__init__zGenerator[None, str, None])r c c s> d}| j }d}dV }|dkrXd}|dkrXdV }|dkrN| d|f¡ dV }qXq ||7 }q |dkrxdV }|dkr
d }q„q
n |d
kr„d }|r*d}g }dV } | ¡ rª|| 7 }q’| tt|p¸d ƒd ƒ¡ | d
krÐd}q’| dkrð| |¡ | ¡ }q8q’| dkr(t|d ƒD ]}
| |df¡ qq8q’q8q’q
| ||f¡ q
dS )zB
Coroutine that parses the ANSI escape sequences.
Ú FNúúz[ZeroWidthEscape]úú[Tõ ›r i' ú;ÚmÚCú )r ÚappendÚisdigitÚminÚintÚ_select_graphic_renditionÚ_create_style_stringÚrange) r ÚstyleZformatted_textZcsir Z escaped_textZsquare_bracketÚcurrentÚparamsÚcharÚir! r! r" r 5 sN
zANSI._parse_corotz list[int])Úattrsr c C sp |s dg}nt |ddd… ƒ}|rl| ¡ }|tkr@t| | _q |tkrTt| | _q |dkrdd| _q |dkrtd| _q |dkr„d| _q |dkr”d| _ q |d kr¤d| _ q |d
kr´d| _
q |d krÄd| _ q |d krÔd| _ q |d
kräd| _q |dk röd| _q |dkrd| _q |dkrd| _ q |dkr,d| _
q |dkr>d| _ q |dkrPd| _ q |sŽd| _d| _d| _d| _d| _ d| _d| _ d| _
d| _ q |dkr t
|ƒdkr | ¡ }|dkr t
|ƒdkr |dkrâ| ¡ }t |¡| _n |dkr | ¡ }t |¡| _|dkr t
|ƒdkr z d | ¡ | ¡ | ¡ ¡}W n tk
rH Y q X |dkr\|| _q |dkr || _q dS )zH
Taken a list of graphics attributes and apply changes.
r Néÿÿÿÿr
Té é é é é é é é Fé é é é é é )é& é0 rJ rK é z#{:02x}{:02x}{:02x})ÚlistÚpopÚ
_fg_colorsr Ú
_bg_colorsr r r r r r r r Úlenr ÚgetÚformatÚ
IndexError)r r: ÚattrÚnr+ Z color_strr! r! r" r2 Š sŽ
ý
z ANSI._select_graphic_renditionc C s¦ g }| j r| | j ¡ | jr,| d| j ¡ | jr<| d¡ | jrL| d¡ | jr\| d¡ | jrl| d¡ | jr|| d¡ | jrŒ| d¡ | j rœ| d¡ d
|¡S )
zW
Turn current style flags into a string for usage in a formatted text.
zbg:ZboldZ underlineÚstrikeZitalicZblinkÚreverseÚhiddenr- ) r r. r r r r r r r r Újoin)r Úresultr! r! r" r3 â s(
zANSI._create_style_stringc C s d| j ›dS )NzANSI(ú))r ©r r! r! r" Ú__repr__ü s z
ANSI.__repr__r c C s | j S ©N)r r] r! r! r" Ú__pt_formatted_text__ÿ s zANSI.__pt_formatted_text__)ÚargsÚkwargsr c O s t t | j||¡ƒS )z†
Like `str.format`, but make sure that the arguments are properly
escaped. (No ANSI escapes can be injected.)
)r Ú FORMATTERÚvformatr )r ra rb r! r! r" rS s z ANSI.formatÚobjectc C s0 t |tƒs|f}tdd„ |D ƒƒ}t| j| ƒS )z+
ANSI('%s') % value
c s s | ]}t |ƒV qd S r_ )r
)Ú.0r9 r! r! r" Ú s z ANSI.__mod__..)Ú
isinstanceÚtupler r )r r r! r! r" Ú__mod__ s
z ANSI.__mod__N) Ú__name__Ú
__module__Ú __qualname__Ú__doc__r# r r2 r3 r^ r` rS rj r! r! r! r" r s UXc C s i | ]\}}||“qS r! r! ©rf ÚkÚvr! r! r" Ú
s rr c C s i | ]\}}||“qS r! r! ro r! r! r" rr s ú#Z02xre r )Útextr c C s t | ƒ dd¡ dd¡S )z4
Replace characters with a special meaning.
r' ú?ú)r Úreplace)rt r! r! r" r
s c @ s e Zd Zddddœdd„ZdS )Ú
ANSIFormatterre r )r Ú format_specr c C s t t||ƒƒS r_ )r
rS )r r ry r! r! r" Ú format_field' s zANSIFormatter.format_fieldN)rk rl rm rz r! r! r! r" rx & s rx N) Ú
__future__r Ústringr Útypingr r r Zprompt_toolkit.output.vt100r r r Z_256_colors_tableÚbaser Ú__all__r ÚitemsrO rP Ú enumerateÚcolorsr9 ÚrÚgÚbr
rx rc r! r! r! r" Ú s& þ