U
Òb‡hÏ% ã @ sê d dl mZ d dlmZ d dlmZ d dlmZmZ d dlm Z
ddl m Z dd gZ
G 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ƒ ZdS )é )Ú annotations)Ú Formatter)Ú Generator)Ú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-9s95tbmv/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 sz |s dg}nt |ddd… ƒ}|rv| ¡ }|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| ¡ d›| ¡ d›| ¡ d›}W n tk
rR Y q X |dkrf|| _q |dkr || _q dS )zH
Taken a list of graphics attributes and apply changes.
r Néÿÿÿÿr Té é é é é é é é Fé é é é é é )é& é0 rH rI é ú#Ú02x)ÚlistÚpopÚ
_fg_colorsr Ú
_bg_colorsr r r r r r r r Úlenr ÚgetÚ
IndexError)r r8 ÚattrÚnr) Z color_strr r r r0 Š 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 r1 à 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 r` ra r r r Úformat 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 )Ú.0r7 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 r0 r1 r] r_ rd rj r r r r r
s UVc 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 rK rL re r )Útextr c C s t | ƒ dd¡ dd¡S )z4
Replace characters with a special meaning.
r% ú?ú)r Úreplace)rs 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 rd )r r rx r r r Ú format_field% s zANSIFormatter.format_fieldN)rk rl rm ry r r r r rw $ s rw N)Ú
__future__r Ústringr Útypingr Zprompt_toolkit.output.vt100r r r Z_256_colors_tableÚbaser Ú__all__r
ÚitemsrO rP Ú enumerateÚcolorsr7 ÚrÚgÚbr rw rb r r r r Ú s& þ