D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
lib
/
python3.9
/
site-packages
/
argcomplete
/
__pycache__
/
Filename :
__init__.cpython-39.pyc
back
Copy
a �R _ | � @ sN d dl mZmZmZmZ d dlZd dlZd dlZd dlZddl m Z m Z ddl mZmZmZmZmZ ddl mZmZ ddlmZmZmZmZ ddlmZ d ej v Z ej a d d � Z d Z!d Z"ej#ej$ej%ej&ej'ej(ej)fZ*ej+dd� �Z,ej+dd� �Z-G dd� de.�Z/d dd�Z0dd� Z1G dd� de2�Z3G dd� de3�Z4e3� Z5d e5_6d d � Z7dS ) � )�absolute_import�division�print_function�unicode_literalsN� )� completers�my_shlex)� USING_PYTHON2�str� sys_encoding� ensure_str� ensure_bytes)�FilesCompleter�SuppressCompleter)�IntrospectiveArgumentParser�action_is_satisfied�action_is_open�action_is_greedy)� shellcodeZ _ARC_DEBUGc G s( t r$trdd� | D �} t| dti� d S )Nc S s" g | ]}t |t�rt|�n|�qS � )� isinstancer r ��.0�xr r �8/usr/lib/python3.9/site-packages/argcomplete/__init__.py� <listcomp> � zdebug.<locals>.<listcomp>�file)�_DEBUGr �print� debug_stream��argsr r r �debug s r# �O �P c c s2 t j} ttjd�t _zd V W | t _n| t _0 d S �N�w)�sys�stdout�open�os�devnull)r) r r r � mute_stdout% s r- c c sF t j} ttjd�t _zd V W t j�� | t _nt j�� | t _0 d S r& )r( �stderrr* r+ r, �close)r. r r r � mute_stderr. s � r0 c @ s e Zd ZdS )�ArgcompleteExceptionN)�__name__� __module__� __qualname__r r r r r1 8 s r1 c s �d u rt | ��tj| dd�� d� _tj�dd�� _g �� ��fdd�}z^� �� }|� jkrlddd�d fW S � j � � �kr�t d|dj � j d �� ||�W S ��|� W qH ty� t d� jd j � j d �� � j � � �kr�|� j� Y S d }t|��Y qH0 qHd S ) NT)�posixZ _ARGCOMPLETE_COMP_WORDBREAKS� c s� t | �� � j�� }t� jttf�r:� j� jv r:|d7 }|t | �kr\td� �� | � d} | d |� | |d � }}d}� jd ur�� j� j v r�� j}|||�� j fS )Nr zIn trailing whitespacer6 ) �len�instream�tellr �stater �bytesZ whitespacer# �appendZquotes�last_wordbreak_pos)�wordZ point_in_word�prefix�suffixZprequote�Zlexer�point�wordsr r � split_wordC s z split_line.<locals>.split_wordr> zsplit, lexer state: '{s}')�sz#split (lexer stopped, state: '{s}')zcUnexpected internal state. Please report this bug at https://github.com/kislyuk/argcomplete/issues.)r7 �shlexZwhitespace_splitr+ �environ�getZ wordbreaksZ get_token�eofr8 r9 r# �formatr: r< � ValueError�tokenr1 )�linerB rD r> �msgr rA r � split_line; s* rO c C s | � |�S �N�� startswith)� completionr? r r r �default_validatork s rT c @ s� e Zd ZdZddddde� dfdd�Zdejddddde� fdd�Zd d � Z d d � Z d d� Z dd� Z dd� Z edd� �Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� ZdS )#�CompletionFindera Inherit from this class if you wish to override any of the stages below. Otherwise, use ``argcomplete.autocomplete()`` directly (it's a convenience instance of this class). It has the same signature as :meth:`CompletionFinder.__call__()`. NTFc C s^ || _ || _|| _|d u r t}|| _|| _d| _i | _|| _|d u rTt j � d�dk}|| _ d S )NFZ_ARGCOMPLETE_SUPPRESS_SPACE�1) �_parser�always_complete_options�excluderT � validator�print_suppressed� completing�_display_completions�default_completerr+ rG rH � append_space)�self�argument_parserrX rY rZ r[ r^ r_ r r r �__init__t s zCompletionFinder.__init__c s^ | j ||||||| d� dtjvr&dS zt�dd�aW n tjaY n0 t� |du r�tj�d�} | dur�td� | �� t | d�}|du r�zt�d d�}W n td � |d � Y n0 tj�d d ��t ��d kr�tdj �d�� |d � tj�d�� � �r$t � �d k�r$tdj � d�� |d � tjd } t tjd �} t | �} t| | �\} }}}}t tjd �d }||d� }|�r�|d | jjv �r�d|v �r�|�|�dd �d � td� | �d� | �d� | �d� |�d� |�d|� | �||| |�}� �r"�fd d �| j�� D ��� �fd d �|D �}td |� |���|��t�� |�� t�� |d� dS )!a� :param argument_parser: The argument parser to autocomplete on :type argument_parser: :class:`argparse.ArgumentParser` :param always_complete_options: Controls the autocompletion of option strings if an option string opening character (normally ``-``) has not been entered. If ``True`` (default), both short (``-x``) and long (``--x``) option strings will be suggested. If ``False``, no option strings will be suggested. If ``long``, long options and short options with no long variant will be suggested. If ``short``, short options and long options with no short variant will be suggested. :type always_complete_options: boolean or string :param exit_method: Method used to stop the program after printing completions. Defaults to :meth:`os._exit`. If you want to perform a normal exit that calls exit handlers, use :meth:`sys.exit`. :type exit_method: callable :param exclude: List of strings representing options to be omitted from autocompletion :type exclude: iterable :param validator: Function to filter all completions through before returning (called with two string arguments, completion and prefix; return value is evaluated as a boolean) :type validator: callable :param print_suppressed: Whether or not to autocomplete options that have the ``help=argparse.SUPPRESS`` keyword argument set. :type print_suppressed: boolean :param append_space: Whether to append a space to unique matches. The default is ``True``. :type append_space: boolean .. note:: If you are not subclassing CompletionFinder to override its behaviors, use ``argcomplete.autocomplete()`` directly. It has the same signature as this method. Produces tab completions for ``argument_parser``. See module docs for more info. Argcomplete only executes actions if their class is known not to have side effects. Custom action classes can be added to argcomplete.safe_actions, if their values are wanted in the ``parsed_args`` completer argument, or their execution is otherwise desirable. )rX rY rZ r[ r_ r^ Z _ARGCOMPLETEN� r' Z _ARGCOMPLETE_STDOUT_FILENAMEzUsing output file {}�wb� z)Unable to open fd 8 for writing, quittingr Z_ARGCOMPLETE_IFS� z%Invalid value for IFS, quitting [{v}])�vZ_ARGCOMPLETE_DFSz%Invalid value for DFS, quitting [{v}]Z COMP_LINEZ COMP_POINTr �=z LINE: {!r}z POINT: {!r}z PREQUOTE: {!r}z PREFIX: {!r}z SUFFIX: {!r}z WORDS:c s0 i | ](\}}|D ]}||r&|� � d �nd�qqS )� r6 ��replace)r �key�valueZkey_part)�ifsr r � <dictcomp>� s �z-CompletionFinder.__call__.<locals>.<dictcomp>c s$ g | ] }� � |��|�pd f��qS �r6 )�joinrH )r rl )�dfs�display_completionsr r r � r z-CompletionFinder.__call__.<locals>.<listcomp>z Returning completions:)rb r+ rG �fdopenr r( r. r# rH rJ r* r7 �intr rO rW � prefix_charsr< �split�_get_completionsr] �items�writerq �encoder �flush)r` ra rX � exit_methodZ output_streamrY rZ r[ r_ r^ �filenameZ comp_lineZ comp_point�cword_prequote� cword_prefix� cword_suffix� comp_wordsr= �start� completionsr )rr rs rn r �__call__� sp ( � " � � zCompletionFinder.__call__c C s | � � }t�� }d| _tr(dd� |D �}z^td|dd � � t� �( | jj|dd � |d�}W d � n1 sp0 Y td|� W n: t y� } z"tdt |�t |�d � W Y d }~n d }~0 0 d | _d |v r�d | _ | � |||t�} | �| �} | �| ||�} | S ) NTc S s g | ] }t |��qS r )r )r r> r r r r r z5CompletionFinder._get_completions.<locals>.<listcomp>zinvoking parser withr )� namespacez parsed args:z exceptionzwhile parsing argsFz--)�_patch_argument_parser�argparse� Namespacer\ r r# r0 rW �parse_known_args� BaseException�typer rX �collect_completions�filter_completions�quote_completions) r` r� r� r r= �active_parsers� parsed_args�a�er� r r r rx � s& 6, z!CompletionFinder._get_completionsc sF g | _ g | _| � � �fdd���| j� td| j � td| j� | j S )a� Since argparse doesn't support much introspection, we monkey-patch it to replace the parse_known_args method and all actions with hooks that tell us which action was last taken or about to be taken, and let us have the parser figure out which subparsers need to be activated (then recursively monkey-patch those). We save all active ArgumentParsers to extract all their possible option names later. c s� � j �| � � j�| � t| t�r&d S d}tr6t|�}t|t| jfi �| _| j D ]>}t |d�r`qPG � �fdd�d|j�}|j|_ |j |_ ||_qPd S )NZ(MonkeyPatchedIntrospectiveArgumentParser� _orig_classc s e Zd Zd� �fdd� ZdS )zPCompletionFinder._patch_argument_parser.<locals>.patch.<locals>.IntrospectActionNc s� t d| � t d||||� t d| j� t d| j� � jsL| j||||d� njt| jtj�r�t d� �| j|d � | j||||d� n.| jtv r�| j s�� j � | � | j||||d� d S )NzAction stub called onz args:z orig class:z orig callable:)� option_stringz:orig class is a subparsers action: patching and running itr ) r# r� �_orig_callabler\ � issubclassr� �_SubParsersAction�_name_parser_map� safe_actions�option_strings�visited_positionalsr< )r` �parserr� �valuesr� �� completer�patchr r r� 8 s zYCompletionFinder._patch_argument_parser.<locals>.patch.<locals>.IntrospectAction.__call__)N)r2 r3 r4 r� r r� r r �IntrospectAction7 s r� )r� r< r� r r r r; r� � __class__�_actions�hasattrr� r� r� )r� Z classname�actionr� r� r r r� % s z6CompletionFinder._patch_argument_parser.<locals>.patchzActive parsers:zVisited positionals:)r� r� rW r# �r` r r� r r� s ) z'CompletionFinder._patch_argument_parserc s� dd� }i }|j �� D ] }|j | }|�|g ��|� q|�� D ]*}|||j |j � �}|r@|j| j|< q@� fdd�|j �� D �} | S )Nc s t � fdd�| D ��S )Nc 3 s | ]}|� � �r|V qd S rP rQ r �r? r r � <genexpr>W r zVCompletionFinder._get_subparser_completions.<locals>.filter_aliases.<locals>.<genexpr>)�tuple)�aliasesr? r r� r �filter_aliasesV s zCCompletionFinder._get_subparser_completions.<locals>.filter_aliasesc s g | ]}|� � �r|�qS r rQ )r Zsubcmd�r� r r r c r z?CompletionFinder._get_subparser_completions.<locals>.<listcomp>)�choices�keys� setdefaultr< �_get_subactions�dest�helpr] ) r` r� r� r� Zaliases_by_parserrl �pr� Zsubcmd_with_aliasesr� r r� r �_get_subparser_completionsU s z+CompletionFinder._get_subparser_completionsc sz t � �dks| jdu r*� fdd�|jD �S dd� |jD �}dd� |jD �}| jdkr`|r\|S |S | jdkrv|rr|S |S g S ) Nr Tc s" g | ]}t |��� �rt |��qS r �r rR �r �optr� r r r h r z5CompletionFinder._include_options.<locals>.<listcomp>c S s g | ]}t |�d krt|��qS �� �r7 r r� r r r r i r c S s g | ]}t |�d krt|��qS r� r� r� r r r r j r ZlongZshort)r7 rX r� )r` r� r� Z long_optsZ short_optsr r� r �_include_optionsf s z!CompletionFinder._include_optionsc s� | j �� fdd�|jD �� g }|jD ]b}| js^t|dd �}t|t�rP|�� rPq&|jt j kr^q&| � ||�slq&t|t j �s&|| � |� �7 }q&|S )Nc s0 g | ](}|j rt� fd d�|j D ��|jg�qS )c 3 s$ | ] }t |��� �rt |�V qd S rP r� r r� r r r� s s �zFCompletionFinder._get_option_completions.<locals>.<listcomp>.<genexpr>)r� r� r� )r r� r� r r r s s ��z<CompletionFinder._get_option_completions.<locals>.<listcomp>r� )r] �updater� r[ �getattrr r �suppressr� r� �SUPPRESS�_action_allowedr� r� )r` r� r� Zoption_completionsr� r� r r� r �_get_option_completionsq s$ �� z(CompletionFinder._get_option_completionsc C s( |j �| g �D ]}||jv r dS qdS �NFT)Z_action_conflictsrH �_seen_non_default_actions)r� r� �conflict_actionr r r r� � s z CompletionFinder._action_allowedc s\ t djt|j�|jd�� �o*�d |jv �d��fdd�|jD �}|rxt|�dks\J d��t d |d d � i �_g }n"�r�d �v r���d �\�}�n|S d }|p�|jD �]�� � js�t� �r�d }|s�t � �r�t� �s�t d� � q�t d� � j � t � dd �} | d u �r@� j d u�r,t � tj��s,t�� j �} nt � tj��s@�j} | r�t | t��r\| �� �r\q�t| ��rև�fdd�| �� ||d�D �} | �r4|| 7 }t | tj��r��j�� fdd�| D �� n�j�dd� | D �� n^t d� td�D ]L} | ��| �} | d u �r �q4��| ���r�j�| fdi� |�| � �q��rL�fdd�|D �}t d|� q�|S )NzActive actions (L={l}): {a})�lr� r r6 c s g | ]}t |� �r|�qS r )r r )� isoptionalr r r � r z<CompletionFinder._complete_active_option.<locals>.<listcomp>r z expect at most 1 greedy actionz Resetting completions becausez must consume the next argumentrh FTZSkippingzActivating completion forr� c s g | ]}�� |� �r|�qS r )rZ �r �c)r� r` r r r � s �)r? r� r� r� c s g | ]}|f� j g�qS r )r� r )� active_actionr r r � r c S s g | ]}|fd g�qS rp r r r r r r � r zICompleter is not callable, trying the readline completer protocol insteadi' c s g | ]}� d | �qS )rh r )r rS )�optional_prefixr r r � r z Completions:)r# rJ r7 Zactive_actionsrv r] � partitionr� r r r� r� r� r r� r� r ZChoicesCompleterr^ r r� �callabler� �rangeZcompleterZ r< ) r` r� �next_positionalr� r� r� Zgreedy_actions�_Z complete_remaining_positionalsr� Zcompletions_from_callable�iZnext_completionr )r� r� r� r� r` r �_complete_active_option� sr � � � z(CompletionFinder._complete_active_optionc C s� g }|d|� |d }|d|� | j s@t|�dkrP|d |jv rP|| �||�7 }|d|� | �� }|d|� t|tj�r�|| �||�7 }| � |||||�}|d|� |d| j � |S ) a0 Visits the active parsers and their actions, executes their completers or introspects them to collect their option strings. Returns the resulting completions as a list of strings. This method is exposed for overriding in subclasses; there is no need to use it directly. zall active parsers:���zactive_parser:r zoptional options:znext_positional:zactive options:zdisplay completions:) rX r7 rv r� �_get_next_positionalr r� r� r� r� r] )r` r� r� r� r# r� � active_parserr� r r r r� � s" � z$CompletionFinder.collect_completionsc C sz | j d }| jd }|�� }|s$dS ||kr4|d S d}tt|��D ]}|| |krD qZqD|d t|�k rv||d S dS )z> Get the next positional action if it exists. r� Nr r )r� r� �_get_positional_actionsr� r7 )r` r� Zlast_positionalZall_positionalsr� r r r r� � s z%CompletionFinder._get_next_positionalc s< dd� |D �}| j du r t� | _ t| j �� � fdd�|D �S )a Ensures collected completions are Unicode text, de-duplicates them, and excludes those specified by ``exclude``. Returns the filtered completions as an iterable. This method is exposed for overriding in subclasses; there is no need to use it directly. c S s g | ] }t |��qS r )r r� r r r r r z7CompletionFinder.filter_completions.<locals>.<listcomp>Nc s" g | ]}|� vr� � |�s|�qS r )�addr� ��seenr r r $ r )rY �set)r` r� r r� r r� s z#CompletionFinder.filter_completionsc s� d}|dkr,�r"�fdd�|D �}|d7 }n|dkr<|d7 }t j�d�d v rRd}n|d krld}d d� |D �}|D ]� � fd d�|D �}qp| jr�d }t|�dkr�|d d |vr�|dkr�|d d7 < |S )a� If the word under the cursor started with a quote (as indicated by a nonempty ``cword_prequote``), escapes occurrences of that quote character in the completions, and adds the quote to the beginning of each completion. Otherwise, escapes all characters that bash splits words on (``COMP_WORDBREAKS``), and removes portions of completions before the first colon if (``COMP_WORDBREAKS``) contains a colon. If there is only one completion, and it doesn't end with a **continuation character** (``/``, ``:``, or ``=``), adds a space after the completion. This method is exposed for overriding in subclasses; there is no need to use it directly. �\r6 c s g | ]}|� d d� �qS )r Nr r� )r= r r r : r z6CompletionFinder.quote_completions.<locals>.<listcomp>z();<>|&!`$* "'�"z"`$!Z_ARGCOMPLETE_SHELL)ZtcshZfish�'c S s g | ]}|� d d��qS )r� z'\''rj r� r r r r F r c s g | ]}|� � d � ��qS )r� rj r� )�charr r r I r z=/:r r r� ri )r+ rG rH r_ r7 )r` r� r r= Z special_charsZcontinuation_charsr )r� r= r r� & s( z"CompletionFinder.quote_completionsc sr |dkrRt ��\}� }}}|�dtjd � | �|� ||�}� �fdd�|D �| _|t| j�k rj| j| S dS dS )a� Alternate entry point for using the argcomplete completer in a readline-based REPL. See also `rlcompleter <https://docs.python.org/2/library/rlcompleter.html#completer-objects>`_. Usage: .. code-block:: python import argcomplete, argparse, readline parser = argparse.ArgumentParser() ... completer = argcomplete.CompletionFinder(parser) readline.set_completer_delims("") readline.set_completer(completer.rl_complete) readline.parse_and_bind("tab: complete") result = input("prompt> ") (Use ``raw_input`` instead of ``input`` on Python 2, or use `eight <https://github.com/kislyuk/eight>`_). r c s g | ]}�|t � �d � �qS rP )r7 )r �match�r� �textr r r m r z0CompletionFinder.rl_complete.<locals>.<listcomp>N)rO �insertr( �argvrx Z _rl_matchesr7 )r` r� r: r r� r� Zfirst_colon_pos�matchesr r� r � rl_completeV s z CompletionFinder.rl_completec C s dd� | j �� D �S )a� This function returns a mapping of option names to their help strings for displaying to the user Usage: .. code-block:: python def display_completions(substitution, matches, longest_match_length): _display_completions = argcomplete.autocomplete.get_display_completions() print("") if _display_completions: help_len = [len(x) for x in _display_completions.values() if x] if help_len: maxlen = max([len(x) for x in _display_completions]) print("\n".join("{0:{2}} -- {1}".format(k, v, maxlen) for k, v in sorted(_display_completions.items()))) else: print(" ".join(k for k in sorted(_display_completions))) else: print(" ".join(x for x in sorted(matches))) import readline print("cli /> {0}".format(readline.get_line_buffer()), end="") readline.redisplay() ... readline.set_completion_display_matches_hook(display_completions) c S s i | ]\}}d � |�|�qS )ri )rq )r �krg r r r ro � r z<CompletionFinder.get_display_completions.<locals>.<dictcomp>)r] ry r� r r r �get_display_completionst s z(CompletionFinder.get_display_completions)r2 r3 r4 �__doc__r rb r+ �_exitr� rx r� r� r� r� � staticmethodr� r� r� r� r� r� r� r� r r r r rU n s, � � y < P 0 rU c @ s e Zd Zedd� �ZdS )�ExclusiveCompletionFinderc C s< t �| |�sdS tjtjf}| j|v r*dS | |jvr8dS dS r� )rU r� r� � _AppendAction�_AppendConstActionr� r� )r� r� Zappend_classesr r r r� � s z)ExclusiveCompletionFinder._action_allowedN)r2 r3 r4 r� r� r r r r r� � s r� zV Use this to access argcomplete. See :meth:`argcomplete.CompletionFinder.__call__()`. c G s t dg| �R dti� dS )z� Prints **args** to standard error when running completions. This will interrupt the user's command line interaction; use it to indicate an error condition that is preventing your completer from working. � r N)r r r! r r r �warn� s r� )N)8Z __future__r r r r r+ r( r� � contextlibr6 r r rF �compatr r r r r r r Z my_argparser r r r Zshell_integrationr rG r r. r r# Z BASH_FILE_COMPLETION_FALLBACKZ BASH_DIR_COMPLETION_FALLBACK� _StoreAction�_StoreConstAction�_StoreTrueAction�_StoreFalseActionr� r� � _CountActionr� �contextmanagerr- r0 � Exceptionr1 rO rT �objectrU r� Z autocompleter� r� r r r r �<module> sF � 0 +