3
�ft`f9 � @ sx d dl mZ d dl mZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dd�Z d dd�Z 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)�fcntlZioctl�termiosZ
TIOCGWINSZ�struct�unpack�IOError)�fdZbuf�ret� 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
ddddd�Z d
d d d
ddddd�Z
ddddddd�Z dd d d d d d!d"d�Z d#d$d%d&d'd(d)d*d�Z
d+d,� ZdFd/d0�ZdGd1d2�Zd3d4� Zd5d6� Zd7d8� ZdHd:d;�Zd<d=� Zd>d?� Zd@dA� ZdBdC� ZdDdE� Zd-S )I�Termz>A class to provide some terminal "UI" helpers based on curses.Tc C s t � S )N)r )�selfr r r
�<lambda>@ s z
Term.<lambda>c C s t � S )N)r )r r r r
r A s ZsmulZrevZsgr0)� underline�reverse�normalr r � � � � � � )�black�blue�green�cyan�red�magenta�yellow�white)r r" r r$ r r# r! r% z[1mz[5m� z[7mz[4mz(B[m)�bold�blink�dimr 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_MODE�MODE�_Term__ansi_forced_FG_COLOR�FG_COLOR�_Term__ansi_forced_BG_COLOR�BG_COLOR)r r r r
Z
__forced_initz s zTerm.__forced_initN�autoc C sX d| _ d| _|dkr | j� dS ddddddd�| _ddddddddd�| _ddddddddd�| _|dkrvd | _ dS |d
ks�t�|s�tj}|j � s�d | _ dS yt
j |j � d � W n t
k
r� d | _ dS X t
j| _t
jd �| _x8| jD ].}|}|| jk�r| j| }| j|�| j|< q�W | jd
�jd�}|�rhx4| jj� D ]&\}}t
j||�j� �pZd| j|< �q>W | jd�jd�}|�r�x8| jj� D ]*\}}t
j||�j� �p�d}|| j|< �q�W | jd�jd�} | �rx4| jj� D ]&\}}t
j| |�j� �p�d| j|< �q�W | jd�jd�}
|
�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/ �AssertionError�sys�stdout�isatty�cursesZ setupterm�fileno� ExceptionZtigetstr�
_ctigetstrZtigetnum�_Term__cap_names� _tigetstr�encode�
_Term__colors�itemsZtparm�decode�_Term__ansi_colors) r � term_stream�color�cap_name�modeZset_fg�valZ 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 �re�sub)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)�beg�endr r
r s zTerm.sub.<locals>.<lambda>)r4 rM �escape�template�IrN )
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_columns�columnsr> 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
�<module> s