Fix a function extract indentation issue

This commit is contained in:
Dave Halter
2020-02-25 23:52:23 +01:00
parent bc3e1ada03
commit b65c1c26aa
2 changed files with 32 additions and 1 deletions

View File

@@ -266,7 +266,7 @@ def extract_function(inference_state, path, module_context, name, pos, until_pos
if is_expression:
replacement = function_call
else:
replacement = _get_indentation(nodes[0]) + output_var_str + ' = ' + function_call + '\n'
replacement = output_var_str + ' = ' + function_call + '\n'
replacement_dct = _replace(nodes, replacement, function_code, pos,
insert_before_leaf, remaining_prefix)

View File

@@ -211,3 +211,34 @@ def a(test, v3):
y = a(test, v3)
#raaaa
y
# -------------------------------------------------- with-range-func-1
import os
# comment1
@dec
# comment2
def x(v1):
#foo
#? 2 text {'new_name': 'a', 'until_line': 9, 'until_column': 5}
v2 = 2
if 1:
x, y = os.listdir(v1 + v2 * v3)
#bar
return x, y
# ++++++++++++++++++++++++++++++++++++++++++++++++++
import os
# comment1
def a(v1, v2, v3):
v2 = 2
if 1:
x, y = os.listdir(v1 + v2 * v3)
return x, y
@dec
# comment2
def x(v1):
#foo
#? 2 text {'new_name': 'a', 'until_line': 9, 'until_column': 5}
x, y = a(v1, v2, v3)
#bar
return x, y