File: //usr/lib/python3.9/site-packages/argcomplete/__pycache__/__init__.cpython-39.pyc
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 +