helpers.pyc
1 o 2 )��c&J � @ sz d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z ddl 3 m Z ddl m Z dd lmZ dd 4 lmZ dd lmZ dd lmZ dd lmZ ddlmZmZ ddlm Z edg d��Z dd� Z dd� Z d=dd�Z dd� Z!dd � Z"d d � Z#G d d � d e$�Z%d!d"� Z&d#d$� Z'd%d&� Z(d=d'd(�Z)G d)d*� d*�Z*d+d,� Z+d-d.� Z,d/d0� Z-d1d2� Z.ed3�d4d5� �Z/d6d7� Z0d>d9d:�Z1d;d<� Z2dS )?z 5 Helpers for the API 6 � N)� 7 namedtuple)�dedent)�chain��wraps)� Parameter)�Parser)�tree)� NO_VALUES)� 8 infer_atom)�infer_call_of_leaf)�get_string_value_set)�signature_time_cache�memoize_method)�get_parent_scope�CompletionParts)�pathZhas_dot�namec C s 9 | � |�S �N)� 10 startswith)�string� like_name� r ��C:\Users\Jacks.GUTTSPC\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.10_qbz5n2kfra8p0\LocalCache\local-packages\Python310\site-packages\jedi\api\helpers.py� _start_match s 11 r c C sL t |�dkr 12 || v S | �|d �}|dkr$t| |d d � |dd � �S dS )N� r F)�len�find� _fuzzy_match)r r �posr r r r s r Fc C s |rt | |�S t| |�S r )r r )r r �fuzzyr r r �match& s 13 14 r! c C s t | dd� d�S )Nc S s$ t | jpd�| jp 15 d| jpd| jfS )N� r )�str� module_path�line�columnr )�xr r r �<lambda>/ s 16 �z$sorted_definitions.<locals>.<lambda>)�key)�sorted)�defsr r r �sorted_definitions- s r, c C st | � |�}|d u s|jdv r%||d d }t�d|d |d � ��d�S |jdvr,dS |jd |d |jd � S )N)r � 17 error_leafr r z (?!\d)\w+$|$)r �keywordr" )�get_leaf_for_position�type�re�search�group�value� start_pos)� module_node�lines�position�leafr% r r r �get_on_completion_name5 s 18 19 r: c C sR | |d d |d � }|d d |d � |d< |d |d d � |d< d� |�S )Nr r �����r" )�join)� 20 code_linesr5 �end_posr7 r r r � _get_codeC s 21 r? c @ s e Zd Zedd� �ZdS )� OnErrorLeafc C s 22 | j d S �Nr )�args��selfr r r r- N s 23 zOnErrorLeaf.error_leafN)�__name__� 24 __module__� __qualname__�propertyr- r r r r r@ M s r@ c C s� |j |kr|�� }|d u rdS |jdk}|jdkr(|�� }|d u r#dS |jdks|jdks2|jdkrA|j d |d k r=dS t|��|} |jjdv rKn|j}qD|r\|j d |d kr\dS t| |�� |�S ) Nr" �newliner- r r T)� 25 file_input�suite� simple_stmtr )r5 �get_previous_leafr0 r@ �parentr? �get_start_pos_of_prefix)r= r9 r8 Zis_after_newlineZ user_stmtr r r �_get_code_for_stackS s0 26 27 28 29 � �rP c s� G dd� dt �� � ��fdd�}tt|||��}d�|d � }t�jdd�}z 30 |j||�d � W n � y= |j Y S w td 31 t||�� ��) zU 32 Returns the possible node names (e.g. import_from, xor_test or yield_stmt). 33 c @ s e Zd ZdS )z/get_stack_at_position.<locals>.EndMarkerReachedN)rE rF rG r r r r �EndMarkerReached s rQ c 3 sZ � �� | �}|D ]"}|j�kr� � �|j���r� � �|j���r'|V � � �|V qd S r )� _tokenizer �prefix�endswith)�code�tokens�token�rQ �grammarZsafewordr r �tokenize_without_endmarker� s � 34 35 �z9get_stack_at_position.<locals>.tokenize_without_endmarkerZ)ZZZ_USER_WANTS_TO_COMPLETE_HERE_WITH_JEDI� T)�error_recovery)rV z7This really shouldn't happen. There's a bug in Jedi: 36 %s) � Exceptionr rP r � _pgen_grammar�parse�stack� SystemError�list)rY r= r9 r rZ rU �pr rX r �get_stack_at_position{ s 37 � 38 ��rd c C s~ |j dkr | �||�S |j}t}|j dkr |�|j�}|S |j dkr)t||�}|S t|tj�r4t ||�S |j dv r=t 39 | �S |S )Nr �atom�trailer)�fstring_string� fstring_start� fstring_end) r0 �inferrN r 40 � 41 infer_noder � 42 isinstancer �Literalr r )�inference_state�contextr9 rN � definitionsr r r rj � s 43 44 45 � 46 � 47 48 rj c c sj � | D ]/}|� � r/tt|�� |d��}d}|r#|D ] }|jd u r"d}q|r)|V q|E d H q|V qd S )N)�follow_builtin_importsFT)� is_importrb �filter_follow_imports�gotor5 )�namesrq r � new_namesZ found_builtin�new_namer r r rs � s$ �� 49 � �rs c @ sP e Zd Zdd� Zedd� �Zedd� �Zedd� �Zd d 50 � Z d d � Z 51 d d� Z dS )� CallDetailsc C s || _ || _|| _d S r )� bracket_leaf� _children� _position)rD ry �childrenr8 r r r �__init__� s 52 zCallDetails.__init__c C � t | j| j�d S rA ��_get_index_and_keyrz r{ rC r r r �index� � zCallDetails.indexc C r~ )Nr r rC r r r �keyword_name_str� r� z CallDetails.keyword_name_strc C s t t| j| j��S r )rb �_iter_argumentsrz r{ rC r r r �_list_arguments� r� zCallDetails._list_argumentsc C s\ d}t � }d}| �� }|s|rdS d S d}t|�D ]&\}\}}} || |dkB O }|r+q|d t|�kr?| r;|�|� q|d7 }qt|�D ]g\}} 53 | 54 �� } |sg| tjkrW| S | tjtj fv rg||krg| S |d uro|dkrs|dkr�| 55 j 56 |vr�| tj ks�| tjkr�||kr�|r�| S | r�| 57 j 58 |kr�| S n 59 | 60 j 61 � |�r�| S | tj kr�| S qDd S )Nr r; F� r )�setr� � enumerater �add�get_kindr �VAR_POSITIONAL�POSITIONAL_OR_KEYWORD�POSITIONAL_ONLY� string_name� KEYWORD_ONLYr � VAR_KEYWORD) rD � param_names�positional_count� 62 used_names� 63 star_countrB Zis_kwarg�i� key_start� had_equal� 64 param_name�kindr r r �calculate_index� sV � 65 66 67 68 69 � 70 �zCallDetails.calculate_indexc c s, � t | �� �D ] \}}}|r|r|V qd S r )rb r� )rD r� r� r� r r r �iter_used_keyword_arguments s ���z'CallDetails.iter_used_keyword_argumentsc C s: d}| � � d d� D ]\}}}|s|r |S |d7 }q 71 |S )Nr r; r )r� )rD �countr� r� r� r r r �count_positional_arguments s 72 �z&CallDetails.count_positional_argumentsN) rE rF rG r} rH r� r� r r� r� r� r� r r r r rx � s 73 74 75 3 rx c # sZ � � fdd�}� fdd�| D �}|d j dkr$t|d j� �E d H d S d}d}t|�D ]�\}}|j d kr�d 76 }|jd }|jd } | d kr`| j� k rW|j d krWd|jd 77 fV n3d||�dfV n*|dv rpt|j�|| �dfV n|�� } 78 | 79 j d kr�| 80 j� kr�d|| 81 �dfV ndV d}q,|j dkr�|jd d d� D ]} | j dkr�d }| jd } ||| �dfV d}q�tt|j�d �}q,t |t 82 j �r�|jdkr�|s�|ddfV d}d}q,t |t 83 j �r�|jdv r�t|j�}q,|d k�r |d �r d 84 }||d } | j d k�rd| jd 85 fV ndV d}q,|�s+|d j d k�r#|||d �dfV d S |ddfV d S d S )Nc s* | j dkrd S | jd � d | jd � S )Nr r )r0 r4 r5 )r �r8 r r �remove_after_pos" s 86 z)_iter_arguments.<locals>.remove_after_posc � g | ] }|j � k r|�qS r �r5 ��.0�cr� r r � 87 <listcomp>( � z#_iter_arguments.<locals>.<listcomp>r; �arglistFr �argumentTr �=r )�*z**)r NF�testlist_star_exprr� � star_expr�,r" ) r0 r� r| r� r5 r4 r �get_first_leaf�boolrl r � 88 PythonLeaf) �nodesr8 r� � nodes_beforeZprevious_node_yieldedZ 89 stars_seenr� �node�first�second� 90 first_leaf�n�beforer r� r r� ! sh � 91 92 93 94 95 96 ��r� c s� � fdd�| D �}|d j dkrt|d j� �S d}|d }|j dkr9|jd dkr9|jd j� kr9|jd j}n |dkrB|d 97 j}|�d �|fS ) zG 98 Returns the amount of commas and the keyword argument string. 99 c r� r r� r� r� r r r� h r� z&_get_index_and_key.<locals>.<listcomp>r; r� Nr� r r� r �����r� )r0 r� r| r> r4 r� )r� r8 r� Zkey_str�lastr r� r r� d s 100 r� c C s� t tt| j���D ]6\}}|dkr?|j|kr?|dkr?| j|d � }|�� }|d u r*q |jdks5|jjdv r?t||| |� S q d S )N�(r r )rf re ) �reversedrb r� r| r> rM r0 rN rx )r� �additional_childrenr8 r� �elementr| r r r r �&_get_signature_details_from_error_nodey s ��r� c C s> | j |dd�}|j|kr|�� }|d u rd S |j}|d ur�|jdv r$d S g }t|j�D ](}|j|k rS|jdkrMt|||�}|d urE| S |j|dd�< q+|�d|� q+|jdkr`|jd dksl|jdkr�|jd dkr�||jd 101 u rx||j ks�|�� }|d u r�d S t 102 |jdkr�|jd n|jd |j|�S |j}|d us d S ) NT)�include_prefixes)�funcdef�classdef� decorated� 103 async_stmt� 104 error_noder rf r� � decoratorr� r; ) r/ r5 rM rN r0 r� r| r� �insertr> rx )�moduler8 r9 r� r� r� �resultr r r �get_signature_details� sH 105 106 107 108 � � ��'r� �call_signatures_validityc c s� � |d d }|| d|d � }||j d |� }d�||g �}t�d|tj�} |�� �� } 109 | 110 du r8dV n| 111 | |j fV t| ||�� �V dS )z'This function calculates the cache key.r r Nr" z.*\() r5 r< r1 r! �DOTALL�get_root_context� 112 py__file__rj rM ) rn ro ry r= Zuser_pos� 113 line_index� before_cursorZ other_lines�wholeZbefore_bracketr$ r r r �cache_signatures� s � 114 �r� c s t � �d� fdd� �}|S )Nc s� |d u r t t| j�d�n|}d| k r t| j�ks#td�� td��| j|d }t|�}|�d�r8|d8 }n |�d�rA|d8 }|d u rG|n|}d| krS|ks^n td||||f ��� | ||g|�R i |��S )Nr r z)`line` parameter is not in a valid range.z 115 r� � 116 zH`column` parameter (%d) is not in a valid range (0-%d) for line %d (%r).)�maxr � _code_lines� 117 ValueErrorrT )rD r% r& rB �kwargs� line_string�line_len��funcr r �wrapper� s$ � 118 119 120 ��z%validate_line_column.<locals>.wrapper)NNr )r� r� r r� r �validate_line_column� s r� Tc sP � �fdd�}t t���� �� ��}|s#�fdd���fdd�|D �}t||�S )zZ 121 Returns a dictionary with name parts as keys and their call paths as 122 values. 123 c s | � � }� r|p �o | S r )� is_definition)r Zis_def)rp � 124 referencesr r �def_ref_filter� s z(get_module_names.<locals>.def_ref_filterc s( t | �}|r|jdkr|j}|� d fv S )Nr� )r r0 rN )r � parent_scope)r� r r �is_module_scope_name� s z.get_module_names.<locals>.is_module_scope_namec s g | ]}� |�r|�qS r r )r� r� )r� r r r� s z$get_module_names.<locals>.<listcomp>)rb r � from_iterable�get_used_names�values�filter)r� � 125 all_scopesrp r� r� ru r )rp r� r� r� r �get_module_names� s 126 r� c C s* | � d�\}}}|dkrd}||�d�fS )Nr[ �def�function�.)� 127 rpartition�split)r r0 �_Z dotted_namesr r r �split_search_string s r� )F)TF)3�__doc__r1 � collectionsr �textwrapr � itertoolsr � functoolsr �inspectr �parso.python.parserr � parso.pythonr �jedi.inference.base_valuer 128 �jedi.inference.syntax_treer �jedi.inference.helpersr �jedi.inference.compiledr � 129 jedi.cacher r �jedi.parser_utilsr r r r r! r, r: r? r] r@ rP rd rj rs rx r� r� r� r� r� r� r� r� r r r r �<module> sJ 130 131 (* 132 SC7 133 134