1
0
forked from VimPlug/jedi

Separate methods in RefactoringCase

This commit is contained in:
Takafumi Arakaki
2013-03-10 22:55:28 +01:00
parent 0a13b8f114
commit 2362e290d8
+11 -10
View File
@@ -31,20 +31,21 @@ class RefactoringCase(object):
self.start_line_test = start_line_test self.start_line_test = start_line_test
self.desired = desired self.desired = desired
def run(self): def refactor(self):
(source, line_nr, index, path, f_name, new_name, start_line_test) \ script = jedi.Script(self.source, self.line_nr, self.index, self.path)
= (self.source, self.line_nr, self.index, self.path, self.f_name, f_name = os.path.basename(self.path)
self.new_name, self.start_line_test)
script = jedi.Script(source, line_nr, index, path)
refactor_func = getattr(refactoring, f_name.replace('.py', '')) refactor_func = getattr(refactoring, f_name.replace('.py', ''))
args = (script, new_name) if new_name else (script,) args = (self.new_name,) if self.new_name else ()
refactor_object = refactor_func(*args) return refactor_func(script, *args)
def run(self):
refactor_object = self.refactor()
# try to get the right excerpt of the newfile # try to get the right excerpt of the newfile
f = refactor_object.new_files()[path] f = refactor_object.new_files()[self.path]
lines = f.splitlines()[start_line_test:] lines = f.splitlines()[self.start_line_test:]
end = start_line_test + len(lines) end = self.start_line_test + len(lines)
pop_start = None pop_start = None
for i, l in enumerate(lines): for i, l in enumerate(lines):
if l.startswith('# +++'): if l.startswith('# +++'):