3
ft`f9 @ sx d dl mZ d dl mZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d ddZ d ddZ G d d
d
e
ZdS )
)absolute_import)unicode_literalsN c C sB y(d}t j| tj|}tjd|d }|S tk
r< dS X dS )z Get the real terminal width Zabcdefghs hhhhr N)fcntlZioctltermiosZ
TIOCGWINSZstructunpackIOError)fdZbufret r /usr/lib/python3.6/term.py_real_term_width s r c C s& t dd}|sdS |dk r dS |S dS )z@ Compute terminal width falling to default 80 in case of troubler )r
P N)r )r
Ztwr r r
_term_width) s
r c @ s e Zd ZdZdZedd Zedd Zdddd Zd
d d d
dddddZ d
d d d
dddddZ
dddddddZ dd d d d d d!d"dZ d#d$d%d&d'd(d)d*dZ
d+d, ZdFd/d0ZdGd1d2Zd3d4 Zd5d6 Zd7d8 ZdHd:d;Zdd? Zd@dA ZdBdC ZdDdE Zd-S )ITermz>A class to provide some terminal "UI" helpers based on curses.Tc C s t S )N)r )selfr r r
@ s z
Term.c C s t S )N)r )r r r r
r A s ZsmulZrevZsgr0) underlinereversenormalr r )blackbluegreencyanredmagentayellowwhite)r r" r r$ r r# r! r% z[1mz[5m z[7mz[4mz(B[m)boldblinkdimr r r z[30mz[31mz[32mz[33mz[34mz[35mz[36mz[37mz[40mz[41mz[42mz[43mz[44mz[45mz[46mz[47mc C s | j | _| j| _| j| _d S )N)_Term__ansi_forced_MODEMODE_Term__ansi_forced_FG_COLORFG_COLOR_Term__ansi_forced_BG_COLORBG_COLOR)r r r r
Z
__forced_initz s zTerm.__forced_initNautoc C sX d| _ d| _|dkr | j dS ddddddd| _ddddddddd| _ddddddddd| _|dkrvd | _ dS |d
kst|stj}|j sd | _ dS yt
j |j d W n t
k
r d | _ dS X t
j| _t
jd | _x8| jD ].}|}|| jkr| j| }| j|| j|< qW | jd
jd}|rhx4| jj D ]&\}}t
j||j pZd| j|< q>W | jdjd}|rx8| jj D ]*\}}t
j||j pd}|| j|< qW | jdjd} | rx4| jj D ]&\}}t
j| |j pd| j|< qW | jdjd}
|
rTx8| jj D ]*\}}t
j|
|j pBd} | | j|< q&W dS )a
Reinitializes the :class:`Term`.
:param term_stream: the terminal stream that the
:class:`Term` should be initialized to use. If
*term_stream* is not given, :attr:`sys.stdout` is used.
:param color: when to colorize output. Valid values are
'always', 'auto', and 'never'. 'always' will use ANSI codes
to always colorize output, 'auto' will decide whether do
colorize depending on the terminal, and 'never' will never
colorize.
T alwaysNr& )r' r( r) r r r )r r r r! r" r# r$ r% ZneverFr0 )r
linesZsetfzutf-8ZsetafZsetbZsetab)_Term__enabledr3 _Term__forced_initr+ r- r/ AssertionErrorsysstdoutisattycursesZ setuptermfileno ExceptionZtigetstr
_ctigetstrZtigetnum_Term__cap_names _tigetstrencode
_Term__colorsitemsZtparmdecode_Term__ansi_colors) r term_streamcolorcap_namemodeZset_fgvalZ set_fg_ansiZfg_colorZset_bgZ set_bg_ansiZbg_colorr r r
reinit s
""z Term.reinitc C s | j || d S )N)rJ )r rE rF r r r
__init__ s z
Term.__init__c C s0 | j |p d}tjj|r"|j }tjdd|S )Nr& z \$<\d+>[/*]?)r= dnfZpycompZ is_py3bytesrC resub)r rG Zcapr r r
r? s zTerm._tigetstrc C s | j | t| | j d S )zColorize string with colorr )r+ str)r rF sr r r
rF s z
Term.colorc C s | j d|S )zMake string bold.r' )rF )r rP r r r
r' s z Term.boldFc
s\ | j s
|S |stj} fdd}x4|D ],}||} |rFtj| tj} tj| ||}q(W |S )a Search the string *haystack* for all occurrences of any
string in the list *needles*. Prefix each occurrence with
*beg*, and postfix each occurrence with *end*, then return the
modified string. For example::
>>> yt = Term()
>>> yt.sub('spam and eggs', 'x', 'z', ['and'])
'spam xandz eggs'
This is particularly useful for emphasizing certain words
in output: for example, calling :func:`sub` with *beg* =
MODE['bold'] and *end* = MODE['normal'] will return a string
that when printed to the terminal will appear to be *haystack*
with each occurrence of the strings in *needles* in bold
face. Note, however, that the :func:`sub_mode`,
:func:`sub_bold`, :func:`sub_fg`, and :func:`sub_bg` methods
provide convenient ways to access this same emphasizing functionality.
:param haystack: the string to be modified
:param beg: the string to be prefixed onto matches
:param end: the string to be postfixed onto matches
:param needles: a list of strings to add the prefixes and
postfixes to
:param escape: a function that accepts a string and returns
the same string with problematic characters escaped. By
default, :func:`re.escape` is used.
:param ignore_case: whether case should be ignored when
searching for matches
:return: *haystack* with *beg* prefixing, and *end*
postfixing, occurrences of the strings in *needles*
c s | j S )N)group)match)begendr r
r s zTerm.sub..)r4 rM escapetemplateIrN )
r haystackrS rT needlesrU Z ignore_caseZrenderZneedleZpatr )rS rT r
rN s
zTerm.subc K s | j ||| jd |f|S )aO Search the string *haystack* for all occurrences of any
string in the list *needles*. Prefix each occurrence with
*beg*, and postfix each occurrence with self.MODE['normal'],
then return the modified string. If *beg* is an ANSI escape
code, such as given by self.MODE['bold'], this method will
return *haystack* with the formatting given by the code only
applied to the strings in *needles*.
:param haystack: the string to be modified
:param beg: the string to be prefixed onto matches
:param end: the string to be postfixed onto matches
:param needles: a list of strings to add the prefixes and
postfixes to
:return: *haystack* with *beg* prefixing, and self.MODE['normal']
postfixing, occurrences of the strings in *needles*
r )rN r+ )r rX rS rY kwdsr r r
sub_norm& s z
Term.sub_normc K s | j || j| |f|S )aT Search the string *haystack* for all occurrences of any
string in the list *needles*. Prefix each occurrence with
self.MODE[*mode*], and postfix each occurrence with
self.MODE['normal'], then return the modified string. This
will return a string that when printed to the terminal will
appear to be *haystack* with each occurrence of the strings in
*needles* in the given *mode*.
:param haystack: the string to be modified
:param mode: the mode to set the matches to be in. Valid
values are given by self.MODE.keys().
:param needles: a list of strings to add the prefixes and
postfixes to
:return: *haystack* with self.MODE[*mode*] prefixing, and
self.MODE['normal'] postfixing, occurrences of the strings
in *needles*
)r[ r+ )r rX rH rY rZ r r r
sub_mode9 s z
Term.sub_modec K s | j |d|f|S )a Search the string *haystack* for all occurrences of any
string in the list *needles*. Prefix each occurrence with
self.MODE['bold'], and postfix each occurrence with
self.MODE['normal'], then return the modified string. This
will return a string that when printed to the terminal will
appear to be *haystack* with each occurrence of the strings in
*needles* in bold face.
:param haystack: the string to be modified
:param needles: a list of strings to add the prefixes and
postfixes to
:return: *haystack* with self.MODE['bold'] prefixing, and
self.MODE['normal'] postfixing, occurrences of the strings
in *needles*
r' )r\ )r rX rY rZ r r r
sub_boldM s z
Term.sub_boldc K s | j || j| |f|S )ac Search the string *haystack* for all occurrences of any
string in the list *needles*. Prefix each occurrence with
self.FG_COLOR[*color*], and postfix each occurrence with
self.MODE['normal'], then return the modified string. This
will return a string that when printed to the terminal will
appear to be *haystack* with each occurrence of the strings in
*needles* in the given color.
:param haystack: the string to be modified
:param color: the color to set the matches to be in. Valid
values are given by self.FG_COLOR.keys().
:param needles: a list of strings to add the prefixes and
postfixes to
:return: *haystack* with self.FG_COLOR[*color*] prefixing, and
self.MODE['normal'] postfixing, occurrences of the strings
in *needles*
)r[ r- )r rX rF rY rZ r r r
sub_fg_ s z Term.sub_fgc K s | j || j| |f|S )a Search the string *haystack* for all occurrences of any
string in the list *needles*. Prefix each occurrence with
self.BG_COLOR[*color*], and postfix each occurrence with
self.MODE['normal'], then return the modified string. This
will return a string that when printed to the terminal will
appear to be *haystack* with each occurrence of the strings in
*needles* highlighted in the given background color.
:param haystack: the string to be modified
:param color: the background color to set the matches to be in. Valid
values are given by self.BG_COLOR.keys().
:param needles: a list of strings to add the prefixes and
postfixes to
:return: *haystack* with self.BG_COLOR[*color*] prefixing, and
self.MODE['normal'] postfixing, occurrences of the strings
in *needles*
)r[ r/ )r rX rF rY rZ r r r
sub_bgs s z Term.sub_bg)Nr0 )Nr0 )NF)__name__
__module__ __qualname____doc__r4 propertyZ real_columnscolumnsr> rA rD r* r, r. r5 rJ rK r? rF r' rN r[ r\ r] r^ r_ r r r r
r 4 sr
f
-r )r )r )Z
__future__r r r: Z
dnf.pycomprL r rM r r7 r r r objectr r r r r
s