OwlCyberSecurity - MANAGER
Edit File: syntax.cpython-39.pyc
a �F�f�� � 7 @ s| U d dl Zd dlZd dlZd dlZd dlZd dlmZmZ d dl m Z d dl m Z m Z mZmZmZmZmZmZmZmZmZ d dlmZ d dlmZmZ d dl m Z d dl m Z d d l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ d d l,m-Z- d d l.m/Z/ d d l0m1Z1m2Z2 d dl3m4Z4 d dl5m6Z6 d dl7m8Z8m9Z9 d dl:m;Z;m<Z<m=Z=m>Z> d dl?m@Z@ d dlAmBZB d dlCmDZDmEZE d dlFm Z mGZG d dlHmIZI eeJdf ZKe�L� dkZMdZNe*e � e+e dd�e"e d d �e"jOe d d�e%e d d�e%je d d�e(jPe d d�e&jQe d d�e&jRe d!d�e&jSe d d d"�e&jTe d!d d"�e&jUe d d�e&jVe d d d#�e&jWe d$d�e&jXe d$d�e&jYe d d�e&jZe d%d�e)e d&d�e'e d d�e$j[e d'd�e$j\e d!d�e$j]e d d(�e$j^e d d d#�e$j_e d d(�e$j#e d'd�e#e d$d d"�iZ`e eKe f ead)< e*e � e+e d*d�e"e d d �e"jOe d+d�e%e d%d�e%je d+d�e(jPe d,d�e&jQe d+d�e&jRe d-d�e&jSe d+d d"�e&jTe d-d d"�e&jUe d+d�e&jVe d,d d#�e&jWe d'd�e&jXe d'd�e&jYe d+d�e&jZe d%d�e)e d&d�e'e d%d�e$j[e d'd�e$j\e d-d�e$j]e d d(�e$j^e d,d d#�e$j_e d d(�e$j#e d'd�e#e d$d d"�iZbe eKe f ead.< e`ebd/�Zcd0ZdG d1d2� d2e�ZeG d3d4� d4ee�ZfG d5d6� d6ee�Zgeehehf ZiG d7d8� d8e�ZjG d9d:� d:e@�Zkeeh eieeh d;�d<d=�Zlemd>k�rxd dlnZnd dlZenjod?d@�ZpepjqdAdBdCdD� epjqdEdFdGdHddIdJ� epjqdKdLdMdHdNdOdJ� epjqdPdQdRdHdSdT� epjqdUdVehdWddXdY� epjqdZd[d\dHdNd]dJ� epjqd^d_dHd`dNdadb� epjqdcdddeddfdg� epjqdhdidjddkdg� epjqdldmddndodp� epjqdqdrehd dsdtdu� epjqdvehddwdxdu� ep�r� Zsd dyltm;Z; e;esjuesjvdz�Z:esjwd{k�r8ejx�y� Zzekezesj{esj|esj}esj~esjesj�esj�esj�hd|� Z�n0ekj�esjwesj{esj|esj}esj~esjesj�esj�esj�hd}� Z�e:j�e�esj�d~� dS )� N)�ABC�abstractmethod)�Path) �Any�Dict�Iterable�List� NamedTuple�Optional�Sequence�Set�Tuple�Type�Union)�Lexer)�get_lexer_by_name�guess_lexer_for_filename)�Style)�get_style_by_name) �Comment�Error�Generic�Keyword�Name�Number�Operator�String�Token� Whitespace)� ClassNotFound)�Lines)�Padding�PaddingDimensions� )� loop_first��cell_len)�Color� blend_rgb)�Console�ConsoleOptions� JustifyMethod� RenderResult)� JupyterMixin)� Measurement)�Segment�Segments)r � StyleType)�Text.�WindowsZmonokai�white��colorT��dim�cyan�blue�magenta�green)r6 � underline)r6 �bold�red� bright_blue�yellow� bright_red)r> � ANSI_LIGHT� bright_black� bright_cyan�bright_magenta� bright_green� ANSI_DARK)Z ansi_light� ansi_dark� c @ s6 e Zd ZdZeeed�dd��Zeed�dd��ZdS ) � SyntaxThemez Base class for a syntax theme.�� token_type�returnc C s t �dS )z'Get a style for a given Pygments token.N��NotImplementedError)�selfrM � rR �Y/home/gouroczh/virtualenv/test/3.9/lib/python3.9/site-packages/pip/_vendor/rich/syntax.py�get_style_for_token~ s z SyntaxTheme.get_style_for_token�rN c C s t �dS )zGet the background color.NrO �rQ rR rR rS �get_background_style� s z SyntaxTheme.get_background_styleN) �__name__� __module__� __qualname__�__doc__r � TokenTyper rT rW rR rR rR rS rK { s rK c @ sJ e Zd ZdZeeee f dd�dd�Ze e d�dd�Z e d �d d �Z dS ) �PygmentsSyntaxThemez.Syntax theme that delegates to Pygments theme.N)�themerN c C s` i | _ t|t�r>zt|�| _W qD ty: td�| _Y qD0 n|| _| jj| _t| jd�| _ d S )N�default��bgcolor) � _style_cache� isinstance�strr �_pygments_style_classr �background_color�_background_colorr �_background_style)rQ r^ rR rR rS �__init__� s z PygmentsSyntaxTheme.__init__rL c C s� z | j | W S ty� z| j�|�}W n tyD t�� }Y nJ0 |d }|d }t|rdd| nd|rrd| n| j|d |d |d d�}|| j |< Y n0 |S ) z"Get a style from a Pygments class.r6 ra �#z#000000r> �italicr= )r6 ra r> rk r= )rb �KeyErrorre Zstyle_for_tokenr �nullrg )rQ rM Zpygments_style�styler6 ra rR rR rS rT � s$ �z'PygmentsSyntaxTheme.get_style_for_tokenrU c C s | j S �N�rh rV rR rR rS rW � s z(PygmentsSyntaxTheme.get_background_style) rX rY rZ r[ r rd r � PygmentsStyleri r\ r rT rW rR rR rR rS r] � s r] c @ sF e Zd ZdZeeef dd�dd�Zeed�dd�Zed �d d �Z dS ) �ANSISyntaxThemez$Syntax theme to use standard colors.N)� style_maprN c C s$ || _ t�� | _t�� | _i | _d S ro )rs r rm �_missing_stylerh rb )rQ rs rR rR rS ri � s zANSISyntaxTheme.__init__rL c C sp z | j | W S tyj | jj}t|�}| j}|rX||�}|durJ|}qX|dd� }q0|| j |< | Y S 0 dS )z Look up style in the style map.N���)rb rl rs �get�tuplert )rQ rM � get_style�tokenrn �_stylerR rR rS rT � s z#ANSISyntaxTheme.get_style_for_tokenrU c C s | j S ro rp rV rR rR rS rW � s z$ANSISyntaxTheme.get_background_style) rX rY rZ r[ r r\ r ri rT rW rR rR rR rS rr � s rr c @ s* e Zd ZU dZeed< eed< eed< dS )�_SyntaxHighlightRangez� A range to highlight in a Syntax object. `start` and `end` are 2-integers tuples, where the first integer is the line number (starting from 1) and the second integer is the column index (starting from 0). rn �start�endN)rX rY rZ r[ r1 �__annotations__�SyntaxPositionrR rR rR rS r{ � s r{ c @ s| e Zd ZU dZee ed< eed< ee e ef ed�dd��Z e dddd d d d dd dd d � e e e e f e e ef eeeeeee ee f eee ee eeee eed d �dd�Zedd e ddd dd d d dd dd fe e ee e e f e e ef eeeeeef eeee ee eeee eed d�dd��Zed@e ee e d�dd��Zed�dd�Zeee d�dd �Ze ee d�d d ��Z e e d�d d ��Z dAe eeee ee f e d!�d"d#�Z e!e"e"d d$�d%d&�Z#dBe$ed(�d)d*�Z%e ed�d+d,��Z&e'eeeef d-�d.d/�Z(d0d1d2d3�d4d5�Z)e'e*e+d3�d6d7�Z,e'e*e-e. d3�d8d9�Z/e d d:�d;d<�Z0e eee f d=�d>d?�Z1d S )C�Syntaxa� Construct a Syntax object to render syntax highlighted code. Args: code (str): Code to highlight. lexer (Lexer | str): Lexer to use (see https://pygments.org/docs/lexers/) theme (str, optional): Color theme, aka Pygments style (see https://pygments.org/docs/styles/#getting-a-list-of-available-styles). Defaults to "monokai". dedent (bool, optional): Enable stripping of initial whitespace. Defaults to False. line_numbers (bool, optional): Enable rendering of line numbers. Defaults to False. start_line (int, optional): Starting number for line numbers. Defaults to 1. line_range (Tuple[int | None, int | None], optional): If given should be a tuple of the start and end line to render. A value of None in the tuple indicates the range is open in that direction. highlight_lines (Set[int]): A set of line numbers to highlight. code_width: Width of code to render (not including line numbers), or ``None`` to use all available width. tab_size (int, optional): Size of tabs. Defaults to 4. word_wrap (bool, optional): Enable word wrapping. background_color (str, optional): Optional background color, or None to use theme color. Defaults to None. indent_guides (bool, optional): Show indent guides. Defaults to False. padding (PaddingDimensions): Padding to apply around the syntax. Defaults to 0 (no padding). re �_theme)�namerN c C s0 t |t�r|S |tv r$tt| �}nt|�}|S )z Get a syntax theme instance.)rc rK �RICH_SYNTAX_THEMESrr r] )�clsr� r^ rR rR rS � get_theme� s zSyntax.get_themeFr# N� r ) r^ �dedent� line_numbers� start_line� line_range�highlight_lines� code_width�tab_size� word_wraprf � indent_guides�padding)�code�lexerr^ r� r� r� r� r� r� r� r� rf r� r� rN c C s� || _ || _|| _|| _|| _|| _|p,t� | _| | _| | _ | | _ | | _ | rVt | d�nt � | _ | | _|| _| �|�| _g | _d S )Nr` )r� �_lexerr� r� r� r� �setr� r� r� r� rf r �background_styler� r� r� r� �_stylized_ranges)rQ r� r� r^ r� r� r� r� r� r� r� r� rf r� r� rR rR rS ri s" � zSyntax.__init__zutf-8)�path�encodingr� r^ r� r� r� r� r� r� r� r� rf r� r� rN c C sF t |�j|d�}|s"| j||d�}| |||||||| | | | | ||d�S )a Construct a Syntax object from a file. Args: path (str): Path to file to highlight. encoding (str): Encoding of file. lexer (str | Lexer, optional): Lexer to use. If None, lexer will be auto-detected from path/file content. theme (str, optional): Color theme, aka Pygments style (see https://pygments.org/docs/styles/#getting-a-list-of-available-styles). Defaults to "emacs". dedent (bool, optional): Enable stripping of initial whitespace. Defaults to True. line_numbers (bool, optional): Enable rendering of line numbers. Defaults to False. start_line (int, optional): Starting number for line numbers. Defaults to 1. line_range (Tuple[int, int], optional): If given should be a tuple of the start and end line to render. highlight_lines (Set[int]): A set of line numbers to highlight. code_width: Width of code to render (not including line numbers), or ``None`` to use all available width. tab_size (int, optional): Size of tabs. Defaults to 4. word_wrap (bool, optional): Enable word wrapping of code. background_color (str, optional): Optional background color, or None to use theme color. Defaults to None. indent_guides (bool, optional): Show indent guides. Defaults to False. padding (PaddingDimensions): Padding to apply around the syntax. Defaults to 0 (no padding). Returns: [Syntax]: A Syntax object that may be printed to the console )r� )r� ) r^ r� r� r� r� r� r� r� r� rf r� r� )r � read_text� guess_lexer)r� r� r� r� r^ r� r� r� r� r� r� r� r� rf r� r� r� rR rR rS � from_path- s& )�zSyntax.from_path)r� r� rN c C s� d}d}|r.zt ||�}W n ty, Y n0 |stz.tj�|�\}}|r^|�d��� }t|�}W n tyr Y n0 |r�|jr�|jd }n|j }|S )a� Guess the alias of the Pygments lexer to use based on a path and an optional string of code. If code is supplied, it will use a combination of the code and the filename to determine the best lexer to use. For example, if the file is ``index.html`` and the file contains Django templating syntax, then "html+django" will be returned. If the file is ``index.html``, and no templating language is used, the "html" lexer will be used. If no string of code is supplied, the lexer will be chosen based on the file extension.. Args: path (AnyStr): The path to the file containing the code you wish to know the lexer for. code (str, optional): Optional string of code that will be used as a fallback if no lexer is found for the supplied path. Returns: str: The name of the Pygments lexer that best matches the supplied path/code. Nr_ �.r ) r r �osr� �splitext�lstrip�lowerr �aliasesr� )r� r� r� r� � lexer_name�_�ext� extensionrR rR rS r� l s( zSyntax.guess_lexerrU c C s | j �� | j }|S )zGet the base style.)r� rW r� )rQ � default_stylerR rR rS �_get_base_style� s zSyntax._get_base_stylerL c C s | j �|�}|jS )z�Get a color (if any) for the given token. Args: token_type (TokenType): A token type tuple from Pygments. Returns: Optional[Color]: Color from theme, or None for no color. )r� rT r6 )rQ rM rn rR rR rS �_get_token_color� s zSyntax._get_token_colorc C sB t | jt�r| jS zt| jdd| jd�W S ty< Y dS 0 dS )z�The lexer for this syntax, or None if no lexer was found. Tries to find the lexer by name if a string was passed to the constructor. FT�ZstripnlZensurenl�tabsizeN)rc r� r r r� r rV rR rR rS r� � s � z Syntax.lexerc C s t ddd| jd�S )z;A Pygments Lexer to use if one is not specified or invalid.�textFTr� )r r� rV rR rR rS � default_lexer� s �zSyntax.default_lexer)r� r� rN c s | � � }|jrdnd}t||| j| j d�}| jj� | jp>| j��du rT|� �� n�|r�|\��t t t t f d���fdd� �t t t tt f d�� ���fdd � }|�|� � n |�� fd d �����D �� | jdur�|�d | j� �� | j�r| �|� |S ) a Highlight code and return a Text instance. Args: code (str): Code to highlight. line_range(Tuple[int, int], optional): Optional line range to highlight. Returns: Text: A text instance containing highlighted syntax. r_ �left)�justifyrn r� �no_wrapNrU c 3 sD �sJ ��� � �D ],\} }|r|�d�\}}}| || fV qqdS )z Split tokens to one per line.� N)� get_tokens� partition)rM ry Z line_token�new_line)r� r� rR rS � line_tokenize� s z'Syntax.highlight.<locals>.line_tokenizec 3 s� t �� �} d}�r�d nd}||k rlzt| �\}}W n tyL Y qlY n0 |dfV |�d�r |d7 }q | D ]8\}}|� |�fV |�d�rp|d7 }�rp|�krp q�qpdS )zConvert tokens to spans.r r# Nr� )�iter�next� StopIteration�endswith)�tokens�line_noZ _line_startZ _token_typery rM )�_get_theme_style�line_end� line_startr� rR rS �tokens_to_spans� s" z)Syntax.highlight.<locals>.tokens_to_spansc 3 s | ]\}}|� |�fV qd S ro rR )�.0rM ry )r� rR rS � <genexpr> s �z#Syntax.highlight.<locals>.<genexpr>zon )r� �transparent_backgroundr2 r� r� r� rT r� r� �appendr r r rd r r � append_tokensr� rf �stylizer� �_apply_stylized_ranges)rQ r� r� � base_styler� r� r� rR )r� r� r� r� r� r� rS � highlight� s4 �� (� zSyntax.highlight)rn r| r} rN c C s | j �t|||�� dS )a� Adds a custom style on a part of the code, that will be applied to the syntax display when it's rendered. Line numbers are 1-based, while column indexes are 0-based. Args: style (StyleType): The style to apply. start (Tuple[int, int]): The start of the range, in the form `[line number, column index]`. end (Tuple[int, int]): The end of the range, in the form `[line number, column index]`. N)r� r� r{ )rQ rn r| r} rR rR rS � stylize_range s zSyntax.stylize_range�333333�?)�blendrN c C sr | j �� | j }|j}|d u s$|jr,t�� S | �tj �}|d u sF|jrR|pPt�� S t |� � |� � |d�}t� |�S )N)� cross_fade) r� rW r� ra �is_system_definedr' r_ r� r r2 r( � get_truecolor� from_triplet)rQ r� r� rf �foreground_color� new_colorrR rR rS �_get_line_numbers_color' s �z Syntax._get_line_numbers_colorc C s, d}| j r(tt| j| j�d� ��t }|S )z?Get the number of characters used to render the numbers column.r r� )r� �lenrd r� r� �count� NUMBERS_COLUMN_DEFAULT_PADDING)rQ Z column_widthrR rR rS �_numbers_column_width6 s ��z Syntax._numbers_column_width)�consolerN c C s� | � � }|jr&t�� tdd�t�� fS |jdv r�t�|| j�tj �t| � � d�| j �}t�|| j�tj �td| � d�d�| j �}n |tdd� }|tdd� }|||fS )z>Get background, number, and highlight styles for line numbers.Tr7 )�256� truecolorr5 g�������?)r> r6 F) r� r� r rm � color_system�chainr� rT r r2 r� r� )rQ r� r� � number_style�highlight_number_stylerR rR rS �_get_number_stylesA s& � �zSyntax._get_number_stylesr) r* r. )r� �optionsrN c C s� t �| j�\}}}}|| }| jd urF| j| j | d }t| j|�S | j�� }| j| |rntdd� |D ��nd }| j r�|d7 }t| j|�S )Nr# c s s | ]}t |�V qd S ro r% )r� �linerR rR rS r� d � z*Syntax.__rich_measure__.<locals>.<genexpr>r ) r! �unpackr� r� r� r. r� � splitlines�maxr� ) rQ r� r� r� �rightr� r� �width�linesrR rR rS �__rich_measure__X s ���zSyntax.__rich_measure__c c s: t | �||��}| jr0t|| j�� | jd�V n|V d S )N�rn �pad)r0 � _get_syntaxr� r! r� rW )rQ r� r� �segmentsrR rR rS �__rich_console__j s � zSyntax.__rich_console__c c sj | � � j}| jdu r0| jr*|j| j d q4|jn| j}| �| j�\}}| �|| j �}| j�s | j �s | j �s |sz|� d� | � � | j � t� tdd� | j }| jr�|js�|j| j|d�}d|_|jr�|j||j|d�d �E dH n8|j||j|dd d �| jddd �} | D ]} | E dH �q dS | j �p*d \} } d} | �rHtd| d �} |jd|d�}| j �r|| t|�k�rpdS || | � }| j�r�|j�s�| � � | j � t� tdd� | j }t d�� |�j| j|tdd� d�jddd�}| j}|j|d�}| j j }t }|d�}|j!�rdnd}| �"|�\}}}t#|| j$| �D �].\}}| j �rf|j||jdd d�|| d�}n:t%|j|dd��}|j&�r�|g}n|j'||j|| d�g}| j�rF|d| d |�} t(|�D ]�\} } | �r*t)|��*|d �d } ||��r||tdd��V || |�V n|d |�V || |�V n| V | E dH |V �q�n |D ]} | E dH |V �qJ�q4dS ) zc Get the Segments for the Syntax object, excluding any vertical/horizontal padding Nr# r� Tr7 )rn �crop)r� )r� r� )r� �heightr� )rn r� � new_lines)NNr )� allow_blankF)rk z> u ❱ )r� r� r� � )r} � rJ r? r5 z )+r� r� r� r� � max_widthr� � _process_coder� r� r� r� � remove_suffixr� rT r r r� r� � ascii_only�with_indent_guidesr� �overflow�render�update� render_linesr� �splitr� r2 �joinr� � __contains__r/ �legacy_windowsr� � enumerater� �listr� �adjust_line_lengthr$ rd �rjust) rQ r� r� r� r� � ends_on_nl�processed_coder� rn Z syntax_linesZ syntax_liner� Zend_line� line_offsetr� Znumbers_column_width�render_options�highlight_line�_Segmentr� Z line_pointerr� r� r� r� r� Z wrapped_linesr� Zwrapped_line_left_pad�firstZ wrapped_lineZ line_columnrR rR rS r� u s� ���� ���� �� �������� � ��� � zSyntax._get_syntax)r� rN c C s| |j }dgdd� tjd|tjd�D ��t|�d �}| jD ]<}t||j�}t||j�}|dur:|dur:|� |j ||� q:dS )z� Apply stylized ranges to a text instance, using the given code to determine the right portion to apply the style to. Args: text (Text): Text instance to apply the style to. r c S s g | ]}|� � d �qS )r# )r| )r� �matchrR rR rS � <listcomp> s �z1Syntax._apply_stylized_ranges.<locals>.<listcomp>r� )�flagsr# N) �plain�re�finditer� MULTILINEr� r� �#_get_code_index_for_syntax_positionr| r} r� rn )rQ r� r� �newlines_offsetsZstylized_ranger| r} rR rR rS r� � s$ ��� � ��z Syntax._apply_stylized_ranges)r� rN c C sB |� d�}|r|n|d }| jr*t�|�n|}|�| j�}||fS )a� Applies various processing to a raw code string (normalises it so it always ends with a line return, dedents it if necessary, etc.) Args: code (str): The raw code string to process Returns: Tuple[bool, str]: the boolean indicates whether the raw code ends with a line return, while the string is the processed code. r� )r� r� �textwrap� expandtabsr� )rQ r� r r rR rR rS r� s � zSyntax._process_code)N)N)r� )2rX rY rZ r[ r rq r~ rK � classmethodr rd r� � DEFAULT_THEMEr �bool�intr r r r" ri r� r� r r� r\ r' r� �propertyr� r� r2 r� r1 r r� �floatr� r� r) r� r� r* r, r� r r/ r� r� r� rR rR rR rS r� � s� � � &� �>) �� V� � � � r� )r �positionrN c C sb t | �}|\}}||ks(t | �|d k r,dS |d }| |d | | d }t||�}| | | S )a Returns the index of the code string for the given positions. Args: newlines_offsets (Sequence[int]): The offset of each newline character found in the code snippet. position (SyntaxPosition): The position to search for. Returns: Optional[int]: The index of the code string for this position, or `None` if the given position's line number is out of range (if it's the column that is out of range we silently clamp its value so that it reaches the end of the line) r# N)r� �min)r r Z lines_count� line_number� column_index� line_index� line_lengthrR rR rS r . s r �__main__z&Render syntax to the console with Rich)� descriptionr� �PATHz path to file, or - for stdin)�metavar�helpz-cz --force-color� force_color� store_truez force color for non-terminals)�dest�actionr_ r& z-iz--indent-guidesr� Fzdisplay indent guidesz-lz--line-numbersr� zrender line numbers)r) r* r& z-wz--widthr� z*width of output (default will auto-detect))�typer) r_ r& z-rz--wrapr� zword wrap long linesz-sz --soft-wrap� soft_wrapzenable soft wrapping mode)r* r) r_ r& z-tz--themer^ zpygments theme)r) r_ r& z-bz--background-colorrf zOverride background colorz-xz--lexerr� z Lexer name)r_ r) r& z-pz --paddingr� r! )r+ r_ r) r& z--highlight-liner z)The line number (not index!) to highlight)r) )�force_terminalr� �-) r� r� r� r� r^ rf r� r� r� )r� r� r� r^ rf r� r� r� )r, )�Zos.pathr� �platformr �sysr �abcr r �pathlibr �typingr r r r r r r r r r r �pip._vendor.pygments.lexerr �pip._vendor.pygments.lexersr r Zpip._vendor.pygments.styler rq Zpip._vendor.pygments.stylesr �pip._vendor.pygments.tokenr r r r r r r r r r �pip._vendor.pygments.utilr Zpip._vendor.rich.containersr Zpip._vendor.rich.paddingr! r" �_loopr$ �cellsr& r6 r' r( r� r) r* r+ r, �jupyterr- �measurer. �segmentr/ r0 rn r1 r� r2 rd r\ �system�WINDOWSr �Preproc�Word�Builtin�Function� Namespace�Class� Exception� Decorator�Variable�Constant� Attribute�Tag�Deleted�Inserted�Heading� Subheading�PromptrC r~ rH r� r� rK r] rr r r r{ r� r rX �argparse�ArgumentParser�parser� add_argument� parse_args�args�pip._vendor.rich.consoler' r� r� �stdin�readr� r� r� r� r^ rf r� r� r �syntaxr� �printr, rR rR rR rS �<module> s� 4 0 � � *! P� �������� ��� �� � �