U
ʗRe$Y @ s U d Z ddlZddlZddlmZ dZeed< ejdkrFeej Zne
e dddl Z ddlm
Z
mZmZ ddlmZmZmZmZ dd lmZ dd
lmZ d Zd ZejZG d
d de Z G dd deZ G dd de
Z G dd dej
Z ej!j"Z#ej$ge#_%ej&e#_'efe(ej&dddZ"ej!j)Z*ej&ej+ge*_%ej,e*_'ej&e(dddZ)ej!j-Z.ej&ej/ej$eee e e0ej$ge._%ej,e._'ej&e1e(e e(dd d Z2ej!j3Z4ej&ej5ej$eee e e0ej$ge4_%ej,e4_'ej&e(e(e e(d d d Z3ej!j6Z7ej&ej5ge7_%ej,e7_'ej&ej5e8d!d"d#Z6ej!j9Z:ej&e0e ge:_%ej,e:_'ej&e dd$d%Z9ej!j;Zej&e0e ge>_%ej,e>_'ej&e e8d)d*d+Z=ej!j?Z@ej&e0e ge@_%ej,e@_'ej&e e8d)d,d-Z?ej!jAZBejCgeB_%ej,eB_'e1e8d.d/d0ZDG d1d2 d2ZEe d3kre" ZFdd4lGmHZH eH ZIeEejJZKeKLd5 ed6d7d8ZMeNd9ZOeIPd: eIQd; eIQd< eIQd= eIQd> eIQd? eIQd@ eIQdA eIQdB eIQdC eIQdD eIQdE eIQdF eIQdG eIQdH eIQdI eIPdJ eIQ eKR eKR eKSdK e TdL eKU eKSdM e TdL eKV eKSdN e TdL eKU eKR eKR eKSdO e TdL eKV eKR eKR eKSdP e TdL eKW eIQ eIPdQ eIQdR eKSdS e TdL eKXdT eKYdUeNdV eKR e TdL eKZ e TdL eIQdW eKSdS e TdL eKXdT eKYdXeNdV e TdL eK[ e TdL eIQdY eKjYdZeNd[d\ e TdL eK\ eK] eQd] dS )^zLight wrapper around the Win32 Console API - this module should only be imported on Windows
The API that this module wraps is documented at https://docs.microsoft.com/en-us/windows/console/console-functions
N)Anywindllwin32z can only be imported on Windows) Structurebyrefwintypes)IO
NamedTupleTypecast) ColorSystem)Stylei c @ s e Zd ZdS )LegacyWindowsErrorN)__name__
__module__ __qualname__ r r /builddir/build/BUILDROOT/alt-python38-pip-22.2.1-2.el8.x86_64/opt/alt/python38/lib/python3.8/site-packages/pip/_vendor/rich/_win32_console.pyr s r c @ s6 e Zd ZU dZeed< eed< ed edddZdS )WindowsCoordinateszCoordinates in the Windows Console API are (y, x), not (x, y).
This class is intended to prevent that confusion.
Rows and columns are indexed from 0.
This class can be used in place of wintypes._COORD in arguments and argtypes.
rowcol)valuereturnc C s t |j|jS )aE Converts a WindowsCoordinates into a wintypes _COORD structure.
This classmethod is internally called by ctypes to perform the conversion.
Args:
value (WindowsCoordinates): The input coordinates to convert.
Returns:
wintypes._COORD: The converted coordinates struct.
)COORDr r )clsr r r r
from_param* s z WindowsCoordinates.from_paramN) r r r __doc__int__annotations__ classmethodr r r r r r r s
r c @ s2 e Zd ZdefdefdejfdejfdefgZdS )CONSOLE_SCREEN_BUFFER_INFOdwSizedwCursorPosition wAttributesZsrWindowZdwMaximumWindowSizeN)r r r r r WORDZ
SMALL_RECT_fields_r r r r r! 8 s r! c @ s e Zd ZdejfdejfgZdS )CONSOLE_CURSOR_INFOr" bVisibleN)r r r r DWORDBOOLr& r r r r r' B s r' )handler c C s t tjt| S )zRetrieves a handle to the specified standard device (standard input, standard output, or standard error).
Args:
handle (int): Integer identifier for the handle. Defaults to -11 (stdout).
Returns:
wintypes.HANDLE: The handle
)r r HANDLE
_GetStdHandle)r+ r r r GetStdHandleM s r. )
std_handler c C s( t }tt| |}|s"td|jS )a Retrieves the current input mode of a console's input buffer
or the current output mode of a console screen buffer.
Args:
std_handle (wintypes.HANDLE): A handle to the console input buffer or the console screen buffer.
Raises:
LegacyWindowsError: If any error occurs while calling the Windows console API.
Returns:
int: Value representing the current console mode as documented at
https://docs.microsoft.com/en-us/windows/console/getconsolemode#parameters
z)Unable to get legacy Windows Console Mode)r r) bool_GetConsoleModer r )r/ Z console_modesuccessr r r GetConsoleMode^ s
r3 )r/ charlengthstartr c C s< t | }t|}td}t| |||t| |jS )a Writes a character to the console screen buffer a specified number of times, beginning at the specified coordinates.
Args:
std_handle (wintypes.HANDLE): A handle to the console input buffer or the console screen buffer.
char (str): The character to write. Must be a string of length 1.
length (int): The number of times to write the character.
start (WindowsCoordinates): The coordinates to start writing at.
Returns:
int: The number of characters written.
r )ctypesc_charencoder r) _FillConsoleOutputCharacterWr r )r/ r4 r5 r6 characterZnum_characters num_writtenr r r FillConsoleOutputCharacter s
r= )r/
attributesr5 r6 r c C s8 t |}t |}t d}t| |||t| |jS )al Sets the character attributes for a specified number of character cells,
beginning at the specified coordinates in a screen buffer.
Args:
std_handle (wintypes.HANDLE): A handle to the console input buffer or the console screen buffer.
attributes (int): Integer value representing the foreground and background colours of the cells.
length (int): The number of cells to set the output attribute of.
start (WindowsCoordinates): The coordinates of the first cell whose attributes are to be set.
Returns:
int: The number of cells whose attributes were actually set.
r )r r) r% _FillConsoleOutputAttributer r )r/ r> r5 r6 Z num_cellsZ style_attrsr< r r r FillConsoleOutputAttribute s
r@ )r/ r> r c C s t t| |S )a| Set the colour attributes for all text written after this function is called.
Args:
std_handle (wintypes.HANDLE): A handle to the console input buffer or the console screen buffer.
attributes (int): Integer value representing the foreground and background colours.
Returns:
bool: True if the attribute was set successfully, otherwise False.
)r0 _SetConsoleTextAttribute)r/ r> r r r SetConsoleTextAttribute s
rB c C s t }t| t| |S )aq Retrieves information about the specified console screen buffer.
Args:
std_handle (wintypes.HANDLE): A handle to the console input buffer or the console screen buffer.
Returns:
CONSOLE_SCREEN_BUFFER_INFO: A CONSOLE_SCREEN_BUFFER_INFO ctype struct contain information about
screen size, cursor position, colour attributes, and more.)r! _GetConsoleScreenBufferInfor )r/ Zconsole_screen_buffer_infor r r GetConsoleScreenBufferInfo s rD )r/ coordsr c C s t t| |S )aE Set the position of the cursor in the console screen
Args:
std_handle (wintypes.HANDLE): A handle to the console input buffer or the console screen buffer.
coords (WindowsCoordinates): The coordinates to move the cursor to.
Returns:
bool: True if the function succeeds, otherwise False.
)r0 _SetConsoleCursorPosition)r/ rE r r r SetConsoleCursorPosition s rG )r/ cursor_infor c C s t t| t|S )a Get the cursor info - used to get cursor visibility and width
Args:
std_handle (wintypes.HANDLE): A handle to the console input buffer or the console screen buffer.
cursor_info (CONSOLE_CURSOR_INFO): CONSOLE_CURSOR_INFO ctype struct that receives information
about the console's cursor.
Returns:
bool: True if the function succeeds, otherwise False.
)r0 _GetConsoleCursorInfor r/ rH r r r GetConsoleCursorInfo s
rK c C s t t| t|S )aw Set the cursor info - used for adjusting cursor visibility and width
Args:
std_handle (wintypes.HANDLE): A handle to the console input buffer or the console screen buffer.
cursor_info (CONSOLE_CURSOR_INFO): CONSOLE_CURSOR_INFO ctype struct containing the new cursor info.
Returns:
bool: True if the function succeeds, otherwise False.
)r0 _SetConsoleCursorInfor rJ r r r SetConsoleCursorInfo* s rM titler c C s t t| S )zSets the title of the current console window
Args:
title (str): The new title of the console window.
Returns:
bool: True if the function succeeds, otherwise False.
)r0 _SetConsoleTitle)rO r r r SetConsoleTitle> s rQ c @ sJ e Zd ZdZdZddddddd d
dd d d
ddddgZdddddZeedddZ eedddZ
e dd d d Z e e
dd d d!Zedd"d#d$Zddd%d&Zddd'd(Zddd)d*Zddd+d,Zddd-d.Zddd/d0Zedd1d2d3Zddd4d5Zddd6d7Zddd8d9Ze dd:d;d<Zedd=d>Z dS )?LegacyWindowsTerma This class allows interaction with the legacy Windows Console API. It should only be used in the context
of environments where virtual terminal processing is not available. However, if it is used in a Windows environment,
the entire API should work.
Args:
file (IO[str]): The file which the Windows Console API HANDLE is retrieved from, defaults to sys.stdout.
r r
zIO[str]N)filer c C sb t t}|| _t|j}|| _|d@ | _|d? d@ | _| j| jd> B | _|| _ |j
| _
|j | _ d S )NrY r ) r. STDOUT_handlerD r$
_default_text
_default_fore
_default_back_default_attrs_filewriteflush)selfra r+ Z default_textr r r __init__i s
zLegacyWindowsTerm.__init__)r c C s( t | jj}ttt|jtt|jdS )zReturns the current position of the cursor (0-based)
Returns:
WindowsCoordinates: The current cursor position.
r r )rD rc r# r r r YX)rk Zcoordr r r cursor_positionw s z!LegacyWindowsTerm.cursor_positionc C s( t | jj}ttt|jtt|jdS )zReturns the current size of the console screen buffer, in character columns and rows
Returns:
WindowsCoordinates: The width and height of the screen as WindowsCoordinates.
rm )rD rc r" r r r rn ro )rk screen_sizer r r rq s
z LegacyWindowsTerm.screen_size)textr c C s | | | dS )zWrite text directly to the terminal without any modification of styles
Args:
text (str): The text to write to the console
N)ri rj )rk rr r r r
write_text s
z LegacyWindowsTerm.write_text)rr styler c C s |j }|j}|jr || }}|rl|tjj}|dk r:|nd}|jrN|| jB }|j r`|| j @ }| j
| }n| j }|r|tjj}|dk r|nd}| j
| }n| j }|dk st
|dk st
t| jt||d> B d | | t| j| jd dS )zWrite styled text to the terminal.
Args:
text (str): The text to write
style (Style): The style of the text
NrY r r )r> )colorbgcolorreverse downgrader WINDOWSnumberbold
BRIGHT_BITdimANSI_TO_WINDOWSre rf AssertionErrorrB rc r7 Zc_ushortrs rd )rk rr rt ru rv forebackr r r write_styled s4
z LegacyWindowsTerm.write_styled) new_positionr c C s* |j dk s|jdk rdS t| j|d dS )zSet the position of the cursor
Args:
new_position (WindowsCoordinates): The WindowsCoordinates representing the new position of the cursor.
r NrE )r r rG rc )rk r r r r move_cursor_to s z LegacyWindowsTerm.move_cursor_toc C sJ | j }| j}|j}t|jdd}t| jd||d t| j| j||d dS )z@Erase all content on the line the cursor is currently located atr rm r5 r6 N) rq rp r r r r= rc r@ rg )rk rq rp cells_to_eraseZstart_coordinatesr r r
erase_line s z LegacyWindowsTerm.erase_linec C s> | j }| jj|j }t| jd||d t| j| j||d dS )zBErase all content from the cursor position to the end of that liner r N)rp rq r r= rc r@ rg )rk rp r r r r erase_end_of_line s z#LegacyWindowsTerm.erase_end_of_linec C s> | j \}}t|d}t| jd||d t| j| j||d dS )zDErase all content from the cursor position to the start of that liner r r N)rp r r= rc r@ rg )rk r r r6 r r r erase_start_of_line s
z%LegacyWindowsTerm.erase_start_of_linec C s( | j }t| jt|jd |jdd dS )z Move the cursor up a single cellrV rm r Nrp rG rc r r r rk rp r r r move_cursor_up s z LegacyWindowsTerm.move_cursor_upc C s( | j }t| jt|jd |jdd dS )z"Move the cursor down a single cellrV rm r Nr r r r r move_cursor_down s z"LegacyWindowsTerm.move_cursor_downc C sJ | j \}}|| jjd kr(|d7 }d}n|d7 }t| jt||dd dS )zIMove the cursor forward a single cell. Wrap to the next line if required.rV r rm r Nrp rq r rG rc r rk r r r r r move_cursor_forward s
z%LegacyWindowsTerm.move_cursor_forward)columnr c C s" | j \}}t| jt||d dS )zMove cursor to the column specified by the zero-based column index, staying on the same row
Args:
column (int): The zero-based column index to move the cursor to.
r N)rp rG rc r )rk r r _r r r move_cursor_to_column s
z'LegacyWindowsTerm.move_cursor_to_columnc C sJ | j \}}|dkr(|d8 }| jjd }n|d8 }t| jt||dd dS )zNMove the cursor backward a single cell. Wrap to the previous line if required.r rV rm r Nr r r r r move_cursor_backward s
z&LegacyWindowsTerm.move_cursor_backwardc C s& | }t|dd}t| j|d dS )zHide the cursorr r" r( rH N_get_cursor_sizer' rM rc )rk current_cursor_sizeZinvisible_cursorr r r hide_cursor$ s z LegacyWindowsTerm.hide_cursorc C s& | }t|dd}t| j|d dS )zShow the cursorrV r r Nr )rk r Zvisible_cursorr r r show_cursor* s z LegacyWindowsTerm.show_cursorrN c C s t |dk stdt| dS )zySet the title of the terminal window
Args:
title (str): The new title of the console window
z.Console title must be less than 255 charactersN)lenr rQ )rk rO r r r set_title0 s zLegacyWindowsTerm.set_titlec C s t }t| j|d t|jS )zEGet the percentage of the character cell that is filled by the cursorr )r' rK rc r r" )rk rH r r r r 9 s z"LegacyWindowsTerm._get_cursor_size) r r r r r| r~ rl propertyr rp rq strrs r
r r r r r r r r r r r r r r r r r r r rR J sN '
rR __main__)ConsolezWin32 Console Examplesblackred)ru rv zblack on greenzChecking colour outputz[on red]on red!z [blue]blue!z[yellow]yellow!z[bold yellow]bold yellow!z [bright_yellow]bright_yellow!z%[dim bright_yellow]dim bright_yellow!z[italic cyan]italic cyan!z'[bold white on blue]bold white on blue!z7[reverse bold white on blue]reverse bold white on blue!z'[bold black on cyan]bold black on cyan!z [black on green]black on green!z [blue on green]blue on green!z [white on black]white on black!z [black on white]black on white!z'[#1BB152 on #DA812D]#1BB152 on #DA812D!zChecking cursor movementz"went back and wrapped to prev linerV zwe go upzand downzwe went up and back 2zwe went down and back 2zChecking line erasingz(
...Deleting to the start of the line...z?The red arrow shows the cursor location, and direction of erase z)
...Now the whole line will be erased...zI'm going to disappear!z
black on cyan)rt
)^r r7 systypingr r r platformZ
LibraryLoaderZWinDLL ImportErrorr timer r r r r r
r pip._vendor.rich.colorr pip._vendor.rich.styler
rb Z"ENABLE_VIRTUAL_TERMINAL_PROCESSINGZ_COORDr Exceptionr r r! r' kernel32r. r- r) Zargtypesr, restyper r3 r1 ZLPDWORDr* ZFillConsoleOutputCharacterWr: r8 ZPOINTERr r= r@ r? r% rB rA r0 rD rC rG rF rK rI rM rL ZSetConsoleTitleWrP ZLPCWSTRrQ rR r+ pip._vendor.rich.consoler consolestdouttermr rt parseheadingruleprintr rs sleepr r r r r r r r r r r r r sd
w