o
    ZhW                     @   s  U d dl Z d dlmZmZmZmZmZmZmZ ddl	m
Z
mZmZ er-ddlmZmZmZ g dZeed< dd	 eD Zeed
< ddddddZeed< i ejdd ejdd ejdd ejdd ejdd ejdd ejdd ejdd ejdd ejdd ejdd ej dd ej!dd ej"dd ej#d d ej$d!d Z%ee&ed"e'f f ed#< G d$d% d%Z(efd&e'd'ee&df d(e'fd)d*Z)efd&e'd'ee&e'f d(e'fd+d,Z*e+d-krd d.l,mZ e Ze-d/ e.d0D ]Z/e0d1d2e/   e 1d3 qdS dS )4    N)TYPE_CHECKINGCallableDictIterableListUnionFinal   )ControlCodeControlTypeSegment)ConsoleConsoleOptionsRenderResult)               STRIP_CONTROL_CODESc                 C   s   i | ]}|d qS N ).0Z
_codepointr   r   G/var/www/auris/lib/python3.10/site-packages/pip/_vendor/rich/control.py
<dictcomp>   s    r   _CONTROL_STRIP_TRANSLATEz\az\bz\vz\fz\rCONTROL_ESCAPEc                   C      dS )Nr   r   r   r   r   <lambda>       r   c                   C   r   )Nr   r   r   r   r   r      r    c                   C   r   )Nz[Hr   r   r   r   r   r      r    c                   C   r   )Nz[2Jr   r   r   r   r   r       r    c                   C   r   )Nz[?1049hr   r   r   r   r   r   !   r    c                   C   r   )Nz[?1049lr   r   r   r   r   r   "   r    c                   C   r   )Nz[?25hr   r   r   r   r   r   #   r    c                   C   r   )Nz[?25lr   r   r   r   r   r   $   r    c                 C      d|  dS )N[Ar   paramr   r   r   r   %       c                 C   r"   )Nr#   Br   r%   r   r   r   r   &   r'   c                 C   r"   )Nr#   Cr   r%   r   r   r   r   '   r'   c                 C   r"   )Nr#   Dr   r%   r   r   r   r   (   r'   c                 C   s   d| d  dS )Nr#   r	   Gr   r%   r   r   r   r   )   s    c                 C   r"   )Nr#   Kr   r%   r   r   r   r   *   r'   c                 C   s   d|d  d| d  dS )Nr#   r	   ;Hr   xyr   r   r   r   +   s    c                 C   r"   )Nz]0;r   r   )titler   r   r   r   ,   r'   .CONTROL_CODES_FORMATc                   @   s  e Zd ZdZdgZdeeef ddfddZe	d)dd	Z
e	d)d
dZe	d*dededd fddZe	d+dededd fddZe	dededd fddZe	d)ddZe	dedd fddZe	dedd fddZe	dedd fddZdefd d!Z	"	#	$	%		&d,d'd(ZdS )-ControlzA renderable that inserts a control code (non printable but may move cursor).

    Args:
        *codes (str): Positional arguments are either a :class:`~rich.segment.ControlType` enum or a
            tuple of ControlType and an integer parameter
    segmentcodesreturnNc                    s<   dd |D }t  d fdd|D }t|d || _d S )Nc                 S   s    g | ]}t |tr|fn|qS r   )
isinstancer   )r   coder   r   r   
<listcomp>;   s    z$Control.__init__.<locals>.<listcomp> c                 3   s     | ]^}} | | V  qd S r   r   )r   r9   
parameters_format_mapr   r   	<genexpr>?   s    
z#Control.__init__.<locals>.<genexpr>)r3   joinr   r5   )selfr6   control_codesrendered_codesr   r=   r   __init__:   s   zControl.__init__c                 C   
   | t jS )zRing the 'bell'.)r   BELLclsr   r   r   bellD      
zControl.bellc                 C   rE   )zMove cursor to 'home' position.)r   HOMErG   r   r   r   homeI   rJ   zControl.homer   r0   r1   c                    s&   dt t f fdd}| |  }|S )zMove cursor relative to current position.

        Args:
            x (int): X offset.
            y (int): Y offset.

        Returns:
            ~Control: Control object.

        r7   c                  3   sR    t }  r dkr| jn| jt fV  r'dkr| jn| jtfV  d S d S )Nr   )r   CURSOR_FORWARDCURSOR_BACKWARDabsCURSOR_DOWN	CURSOR_UP)controlr/   r   r   	get_codes[   s   
zControl.move.<locals>.get_codes)r   r
   )rH   r0   r1   rS   rR   r   r/   r   moveN   s   
zControl.movec                 C   s:   |r| t j|f|dkrt jnt jt|fS | t j|fS )zMove to the given column, optionally add offset to row.

        Returns:
            x (int): absolute x (column)
            y (int): optional y offset (row)

        Returns:
            ~Control: Control object.
        r   )r   CURSOR_MOVE_TO_COLUMNrP   rQ   rO   rH   r0   r1   r   r   r   move_to_columnk   s   	zControl.move_to_columnc                 C   s   | t j||fS )zMove cursor to absolute position.

        Args:
            x (int): x offset (column)
            y (int): y offset (row)

        Returns:
            ~Control: Control object.
        )r   CURSOR_MOVE_TOrV   r   r   r   move_to   s   zControl.move_toc                 C   rE   )zClear the screen.)r   CLEARrG   r   r   r   clear   rJ   zControl.clearshowc                 C   s   | |rt jS t jS )zShow or hide the cursor.)r   SHOW_CURSORHIDE_CURSOR)rH   r\   r   r   r   show_cursor   s   zControl.show_cursorenablec                 C   s   |r	| t jt jS | t jS )zEnable or disable alt screen.)r   ENABLE_ALT_SCREENrK   DISABLE_ALT_SCREEN)rH   r`   r   r   r   
alt_screen   s   
zControl.alt_screenr2   c                 C   s   | t j|fS )zlSet the terminal window title

        Args:
            title (str): The new terminal window title
        )r   SET_WINDOW_TITLE)rH   r2   r   r   r   r2      s   zControl.titlec                 C   s   | j jS r   r5   text)rA   r   r   r   __str__   s   zControl.__str__consoler   optionsr   r   c                 c   s    | j jr| j V  d S d S r   re   )rA   rh   ri   r   r   r   __rich_console__   s   zControl.__rich_console__)r7   r4   )r   r   )r   )rh   r   ri   r   r7   r   )__name__
__module____qualname____doc__	__slots__r   r   r
   rD   classmethodrI   rL   intrT   rW   rY   r[   boolr_   rc   strr2   rg   rj   r   r   r   r   r4   0   s<    
r4   rf   _translate_tabler7   c                 C   
   |  |S )zRemove control codes from text.

    Args:
        text (str): A string possibly contain control codes.

    Returns:
        str: String with control codes removed.
    	translaterf   rt   r   r   r   strip_control_codes   s   
ry   c                 C   ru   )a  Replace control codes with their "escaped" equivalent in the given text.
    (e.g. "" becomes "\b")

    Args:
        text (str): A string possibly containing control codes.

    Returns:
        str: String with control codes replaced with their escaped version.
    rv   rx   r   r   r   escape_control_codes   s   
rz   __main__)r   z+Look at the title of your terminal window ^
   u   🚀 Loading.g      ?)2timetypingr   r   r   r   r   r   r   r5   r
   r   r   rh   r   r   r   r   __annotations__r   r   rF   CARRIAGE_RETURNrK   rZ   ra   rb   r]   r^   rQ   rP   rM   rN   rU   ERASE_IN_LINErX   rd   r3   rq   rs   r4   ry   rz   rk   pip._vendor.rich.consoleprintrangeiset_window_titlesleepr   r   r   r   <module>   s   
 $








	







 





