Format: application/x-python-code
1 o
2
3 ��ha1 � @ sV d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddl
4 m
5 Z
6 ddl m Z ddl
7 mZmZ dd lmZ d
8 d lmZ dd lmZ dd
9 lmZ e � Zd
10 ZdZe�dd� dd� eD �� �Z e�d�Z de dee fdd�Z dd� Z!de de"fdd �Z#d d � Z$d/d e dee fd!d"�Z%d#d$� Z&d%d&� Z'd'd(� Z(d)d*� Z)d+d,� Z*d-d.� Z+dS )0z\Implementations for various useful completers.
11
12 These are all loaded by default by IPython.
13 � N)�
14 import_module)� all_suffixes)�time)� zipimporter� )� expand_user�
15 compress_user)�TryNext� )� arg_split)� get_ipython)�List� z?(?P<name>[^\W\d]\w*?)(?P<package>[/\\]__init__)?(?P<suffix>%s)$�|c c � � | ]}t �|�V qd S �N)�re�escape��.0�s� r �Oc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/core/completerlib.py� <genexpr>: � � r z.*(\.ipy|\.ipynb|\.py[w]?)$�path�returnc s� | dkrd} t jj� t j�| �rEg }t j| dd�D ]*\}}}|t| �d d� ��r>|�� �fdd�|D �� g |dd�< q|�|� qnz tt| �j �
16 � �}W n t y[ g }Y nw g }|D ]}t �
17 |�}|rq|�|�d �� q`tt|��S )
18 z`
19 Return the list containing the names of the modules available in the given
20 folder.
21 � �.T)� followlinksr Nc 3 s � | ]}� �|�V qd S r r �r �f��pjoin�subdirr r r X r z module_list.<locals>.<genexpr>�name)�osr �join�isdir�walk�len�extend�listr �_files�keys� Exception� import_re�match�append�group�set)r �files�root�dirs�nondirs�modulesr! �mr r" r � module_listD s0 � �
22 � r; c C s. t � } | du r ttj�S t| jdd�ri }n| j�di �}ttj�}t� }d}tjD ]]}z|| }W nO t y� t
23 |�}z|� d� W n t yM Y nw |dvrV|||< t� | t
24 krn|snd}td� td � tj�� t� | tkrtd
25 � g Y S Y nw |�|� q*|r�|| jd< tt|��}|S ) z�
26 Returns a list containing the names of all the modules available in the
27 folders of the pythonpath.
28
29 ip.db['rootmodules_cache'] maps sys.path entries to list of modules.
30 NZ_mockF�rootmodules_cache�__init__)r r Tz/
31 Caching the list of root modules, please wait!z@(This will only be done once - type '%rehashx' to reset cache!)
32 z%This is taking too long, we give up.
33 )r r, �sys�builtin_module_names�getattr�db�getr r �KeyErrorr; �remove�
34 ValueError�TIMEOUT_STORAGE�print�stdout�flush�TIMEOUT_GIVEUPr+ r4 )�ipr< Z rootmodules�
35 start_time�storer r9 r r r �get_root_modulesl sH
36
37
38 �
39 ��
40 rN �attrc C sT |rzt | |�}W n
41 ty Y dS w t�|�S |d d� dko(|dd � dk S )NFr
42 �__�����)r@ �ModuleNotFoundError�inspect�ismodule)�modulerO � only_modules�modr r r �
43 is_importable� s �
44 "rX c C s@ zt | |�}W n ty Y dS ty Y dS w t�|�S )NTF)r@ �AttributeError� TypeErrorrS rT )rU rO �objr r r �is_possible_submodule� s �
45 r\ FrW c s� | � d�} zt| �� W n g Y S dt� dd�pdv }g }t� d�r(�r(|r7|�� �fdd�t� �D �� t� dg �}�rL|�� fdd �|D �� n|�|� |rg� j}tj� |�}|d
46 urg|�t
47 |�� d d � |D �}|� d� t |�S )
48 zN
49 Try to import given module and return list of potential completions.
50 r r= �__file__r c s g | ]
51 }t � |��r|�qS r )rX �r rO �r: rV r r �
52 <listcomp>� s
53
54 �z try_import.<locals>.<listcomp>�__all__c 3 s � | ]
55 }t � |�r|V qd S r )r\ r^ )r: r r r � s � z try_import.<locals>.<genexpr>Nc S s h | ] }t |t�r|�qS r )�
56 isinstance�str�r �cr r r � <setcomp>� s z try_import.<locals>.<setcomp>)
57 �rstripr r@ �hasattrr+ �dirr] r& r �dirnamer; �discardr, )rW rV Z m_is_init� completionsZm_all�file_� file_pathZcompletions_setr r_ r �
58 try_import� s*
59
60
61 ro c s4 t � t�r � �� � � fdd�}t� jd|| d� dS )a� Easily create a trivial completer for a command.
62
63 Takes either a list of completions, or all completions in string (that will
64 be split on whitespace).
65
66 Example::
67
68 [d:\ipython]|1> import ipy_completers
69 [d:\ipython]|2> ipy_completers.quick_completer('foo', ['bar','baz'])
70 [d:\ipython]|3> foo b<TAB>
71 bar baz
72 [d:\ipython]|3> foo ba
73 c s � S r r ��self�event�rl r r � do_complete� s z$quick_completer.<locals>.do_completeZcomplete_command)Zstr_keyN)rb rc �splitr Zset_hook)�cmdrl rt r rs r �quick_completer� s
74 rw c s� | � d�}t|�}|dkr|d dkrdgS |dk rL|d dv rL|dkr't� S |d � d�� t� �d k r7t� S td�� d
75 d � �d �}� fd
76 d�|D �S |dkr^|d dkr`|d � t� �S d
77 S d
78 S )z�
79 Returns a list containing the completion possibilities for an import line.
80
81 The line looks like this :
82 'import xml.d'
83 'from xml.dom import'
84 � � r �fromzimport > z%aimportrz �importr r r
85 N�����Tc s$ g | ]}d � � dd� |g ��qS )r Nr| )r' )r �el�rW r r r` � $ z%module_completion.<locals>.<listcomp>)ru r* rN ro r' )�line�wordsZnwordsZcompletion_listr r~ r �module_completion� s
86 �r� c C s
87 t |j�S )z@Give completions after user has typed 'import ...' or 'from ...')r� r� rp r r r �module_completer s
88 r� c s� t |jdd�}t|�dkr|j�d�s|d �d�}nd}tj}tjj� t |�\}��t
89 dd � |D ��rB� fd
90 d �||d �D �}n.� fd
91 d �||d �D �}dd � ||d �||d � ||d � ||d � D �}|| }��fdd �|D �S )zKComplete files that end in .py or .ipy or .ipynb for the %run command.
92 F)�strictr rx r| z'"r c s r r )� magic_run_rer1 rd r r r r 7 r z&magic_run_completer.<locals>.<genexpr>c s( g | ]}|� d d�� |�rdnd �qS )�\�/r ��replacer �r( r r r` 8 s �z'magic_run_completer.<locals>.<listcomp>�*c s$ g | ]}� |�r|� d d�d �qS �r� r� r� r r� r r r` ; r c S s g | ]}|� d d��qS r� r� r r r r r` < s �z*.pyz*.ipyz*.ipynbz*.pywc � g | ]}t |� ���qS r �r �r �p�� tilde_expand� tilde_valr r r` C � ) r r� r* �endswith�strip�globr& r r( r �any)rq rr �comps�relpathZlglob�matchesr7 Zpysr )r( r� r� r �magic_run_completer! s*
93
94 �
95 �
96 ��r� c st t � }� j}� j�d�sd� jv r | j�dd�}|r |�� S g S � jdkrRtttt|j d �d ���}d| d ��fd
97 d �t
98 |j d �D �}t|�dkrP|S g S � j� d �rbd
99 d � |j d D �S t |�\}��|�
100 dd�}g }dd � t�|d �D �D ]
101 }d|v r�t�|�|� q}|s�tj�|�r�t|���gS | j�di �} � fdd �| D �}
102 |
103 r�|
104 S t���fdd �|D �S )z:Completer function for cd, which only returns directories.z-bz -b Z bookmarksN�-Z_dhr z-%0zd [%s]c s g | ]
105 \}}� ||f �qS r r )r �ir )�fmtr r r` X � z cd_completer.<locals>.<listcomp>�--c S s g | ]
106 }d t j�|� �qS )r� )r& r �basename)r �dr r r r` ^ r� r� r� c S s( g | ]}t j�|�r|�d d�d �qS r� )r& r r( r� r r r r r` e s
107 �r� rx c s g | ]
108 }|� � j�r|�qS r )�
109 startswith�symbolr )rr r r r` t r� c r� r r� r� r� r r r` z r� )r r� r� r� rA rB r. rc r* Zuser_ns� enumerater� r r� r� r r2 r& r r( r ) rq rr rK r� ZbkmsZwidth_dhZents�foundr� ZbksZ bkmatchesr )rr r� r� r� r � cd_completerF s@
110 r� c C s d� � S )z A completer for %reset magicz-f -s in out array dhist)ru rp r r r �reset_completer| s r� )F),�__doc__r� rS r& r r> � importlibr �importlib.machineryr r � zipimportr Z completerr r �errorr Zutils._process_commonr ZIPythonr �typingr
111 Z _suffixesrF rJ �compiler' r0 r� rc r; rN �boolrX r\ ro rw r� r� r� r� r� r r r r �<module> sB �
112 (/ %%% 6