Edit File: term.cpython-36.pyc
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