Format: application/x-python-code
1 o
2
3 �hm6 � @ s d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z m
4 Z
5 d dl m Z d dl
6 mZmZ dZe
7 d �� Zd
8 d � Zd d
9 � Zdd� Z d1dd�Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Z d d � Z d!d"� Z d#d$� Z d%d&� Z d'd(� Z!d)d*� Z"d+d,� Z#d-d.� Z$d/d0� Z%dS )2� )�dedent)� split_lines)�debug)�RefactoringError)� Refactoring�EXPRESSION_PARTS)� indent_block)�function_is_classmethod�function_is_staticmethod�Zsuite�
10 file_inputz_atom testlist_star_expr testlist test lambdef lambdef_nocond keyword name number string fstringc C s\ t |||�}t�d|� t|�\}}|st|��|d t|� } |t||| |�i}
11 t| |
12 �S )NzExtracting nodes: %s� = )� _find_nodesr Zdbg�_is_expression_with_errorr �_expression_nodes_to_string�_replacer ) �inference_state�path� module_node�name�pos� until_pos�nodes�
13 is_expression�messageZgenerated_code�file_to_node_changes� r �Rc:\Users\shubh\Videos\agent\nova\Lib\site-packages\jedi/api/refactoring/extract.py�extract_variable s
14 r c C s: t dd� | D ��r dS | d jtvrdd| d j fS dS )z8
15 Returns a tuple (is_expression, error_string).
16 c s s" � | ] }|j d ko |�� V qdS )r N)�type�
17 is_definition)�.0�noder r r � <genexpr># s � z,_is_expression_with_error.<locals>.<genexpr>)Fz,Cannot extract a name that defines somethingr FzCannot extract a "%s")T� )�anyr �_VARIABLE_EXCTRACTABLE�r r r r r s
18 r c C s: | j |dd�}|du rC|jdkr |�� }|dur |j|kr |}t|�r&|j}|jjdkr0|jj}|jjtv r?|j}|jjtv s6|g}nE|j|krR|�� }|durR|}t|�rY|j}| j |dd�}|j|krq|�� }|du rqt d��|}|j|jk r�|j}|j|jk syt
19 |||�}t |�dkr�|jdv r�|d j d gS |S )
20 zd
21 Looks up a module and tries to find the appropriate amount of nodes that
22 are in there.
23 T)Zinclude_prefixesN�operator�trailerz!Cannot extract anything from that� )� return_stmtZ
24 yield_exprr )
25 Zget_leaf_for_positionr �
26 get_next_leaf� start_pos�_is_not_extractable_syntax�parentr �end_posZget_previous_leafr �!_remove_unwanted_expression_nodes�len�children)r r r Z
27 start_nodeZ next_leafr Zend_leaf� parent_noder r r r + sB
28 �
29
30 � r Nc
31 C s� t | d �}|d u r|�� }| d �� }t|jdd�}||u r)|d ur)||d d�< t||d �d g|dd�< d�|�} i }
32 ||u rK| | |
33 | d < n |d u rS|j} n|t| d � } | | |
34 | d < | |j |
35 |< | dd � D ]} d|
36 | < qp|
37 S )Nr T��keepends������
38 r$ r* )�_get_parent_definition�get_first_leafr �prefixr �join�_get_indentation�value)
39 r Zexpression_replacementZ extractedr �insert_before_leaf�remaining_prefixZ
40 definitionZfirst_node_leaf�linesZextracted_prefix�replacement_dct�pr" r r r r a s*
41
42 r c C s d� dd� t| �D ��S )Nr$ c s s$ � | ]
43 \}}|j |d kd�V qdS )r �Zinclude_prefixN��get_code)r! �i�nr r r r# � s �" z._expression_nodes_to_string.<locals>.<genexpr>)r< � enumerater' r r r r � s r c C sV | d }t |�� |d d �\}}||jdd� d�dd� | dd � D �� }||fS )Nr r* FrD r$ c s s � | ]}|� � V qd S )NrE )r! rH r r r r# � s � z)_suite_nodes_to_string.<locals>.<genexpr>)�_split_prefix_atr: rF r< )r r rH r; Z part_of_code�coder r r �_suite_nodes_to_string� s �rL c C sD | j d | }t| jdd�}d�|d| � �d�|| d� �fS )zU
44 Returns a tuple of the leaf's prefix, split at the until_line
45 position.
46 r Tr5 r$ N)r- r r; r< )�leafZ
47 until_lineZsecond_line_countrA r r r rJ � s (rJ c C s t | �� j�d S )Nr7 )r r: r; �r" r r r r= � s r= c C s. | dur| j jtv r | S | j } | dustd��)z8
48 Returns the statement where a node is defined.
49 Nz We should never even get here)r/ r �_DEFINITION_SCOPES�NotImplementedErrorrN r r r r9 � s �r9 c C s | j }|dv }|tv s
50 |r�| j}t|�D ]\}}|j|kr*|}|j dkr(|d8 } nqttt|���D ](\}}|j|k r[|} |j dkrG| d7 } ||d� D ] }
51 t|
52 �rX| d7 } qM nq3||| d � }|s~t |d ||�|dd�< t |d ||�|dd�< |S | gS )z~
53 This function makes it so for `1 * 2 + 3` you can extract `2 + 3`, even
54 though it is not part of the expression.
55 r r( r* Nr r7 )
56 r r r3 rI r0 �reversed�listr- r. r1 ) r4 r r �typZ
57 is_suite_partr rG rH Z start_indexZ end_indexZn2r r r r1 � s8
58
59 �
60
61
62 � r1 c C s | j dkp| j dko| jdvS )Nr( �keyword)�None�True�False)r r> rN r r r r. � s
63 �r. c sX t |j||�}t|�s
64 J �t|�\}}|�|d �} | �� }
65 tt|| |��\} } | �� r/d }
66 n
67 t | j|
68 �}|�
69 � }
70 |rHdt |� d }d }d}nMt |d �}|sh| rftt
71 | |d j|d j| ��pe| d gng } t||�\}}|d �� }t||d �\}}||7 }t|�}|s�d�| �}|d| d 7 }t|r�|d d� n|� d}d � |
72 r�t| j�s�| �� �� }t|�r�|d j� � fdd �| D �} t| j�r�d
73 }n|d7 }d ||d�� d u r�| n� g| �t|�f }d � d u r�dn� d
74 | d�| �f }|r�|}n|�rd| d }n|d | d }t |||||
75 |�}|�s#||j ||< ||i} t | | �S )Nr zreturn r8 Fr7 z, r$ c s g | ]}|� kr|�qS r r )r! rC �Z
76 self_paramr r �
77 <listcomp> s z$extract_function.<locals>.<listcomp>z
78 @classmethod
79 z%sdef %s(%s):
80 %sz%s(%s)�.r
81 ) r Z tree_noder2 r Zcreate_context�is_bound_methodrR �_find_inputs_and_outputs� is_module�_get_code_insertion_noder: r �_is_node_ending_return_stmt� _find_needed_output_variablesr/ r0 rL r, rJ r r< �_check_for_non_extractablesr
82 � get_valueZget_param_namesZ string_namer r r r> r ) r r �module_contextr r r r r �_�contextr[ �params�return_variablesr? r" Z
83 code_blockr@ Zhas_ending_return_stmtZ
84 after_leaf�first�secondZoutput_var_str� decoratorZfunction_param_namesZ
85 function_codeZ
86 function_call� replacementrB r r rX r �extract_function� s� ��� �
87
88
89
90 ���
91 �
92 rl c C sX | D ]'}z|j }W n ty$ |jdkrtd��|jdkr"td��Y qw t|� qd S )N�returnz:Can only extract return statements if they are at the end.�yieldz Cannot extract yield statements.)r3 �AttributeErrorr> r ra )r rH r3 r r r ra ( s
93
94 �
95 ��
96 �ra c C s` |D ]+}|j dks|j�� s-|�� | ur dS |jd u s*||j kr)|k s- dS dS qdS )N�paramTF)Zapi_typeZparent_contextr] Zget_root_contextr- )rc �namesrh �lastr r r r �_is_name_input6 s ��rs c C s� |d j }|d j}g }g }t|�D ],}|�� r#||vr"|�|j� q|j|vr>|�||j �}|r8t| |||�r>|�|j� q||fS )Nr r7 )r- r0 �_find_non_global_namesr �appendr> Zgotors ) rc re r rh rr ZinputsZoutputsr Zname_definitionsr r r r\ @ s"
97
98 �
99 � �r\ c c sf � | D ]-}z|j }W n ty |jdkr|V Y qw |jdkr)|j d dkr)qt|�E d H qd S )Nr r) r rZ )r3 ro r rt )r r" r3 r r r rt U s �
100
101 ���rt c C sL |rt | �r| jjdkr| j} | jjdks | jjdv r$| j} | jjdv s| S )Nr )Z
102 async_funcdefZ decoratedZ
103 async_stmt)r
104 r/ r )r" r[ r r r r^ d s � �r^ c c s\ � |j D ]'}|j|k r qt|�}t|g�D ]}|�� s*|j|v r*|�|j� |jV qqdS )z�
105 Searches everything after at_least_pos in a node and checks if any of the
106 return_variables are used in there and returns those.
107 N)r3 r- �setrt r r> �remove)re Z search_nodeZ at_least_posrg r" r r r r r` n s �
108
109 ���r` c C s$ | j }|dkrt| jd �S |dkS )NZ simple_stmtr r+ )r r_ r3 )r" �tr r r r_ ~ s r_ )NN)&�textwrapr Zparsor Zjedir Zjedi.api.exceptionsr Zjedi.api.refactoringr r Z jedi.commonr Zjedi.parser_utilsr r
110 rO �splitr& r r r r r rL rJ r= r9 r1 r. rl ra rs r\ rt r^ r` r_ r r r r �<module> s@ �
111 7
112 �! $W
113
114