def deduceArguments(self):
lines = self.fn.getLinesNotIncludingThoseBelongingToChildScopes()
# strip off comments
lines = [commentRE.sub(self.linesep,line) for line in lines]
extractedLines = maskStringsAndRemoveComments("".join(self.extractedLines)).splitlines(1)
linesbefore = lines[:(self.startline - self.fn.getStartLine())]
linesafter = lines[(self.endline - self.fn.getStartLine()) + 1:]
# split into logical lines
linesbefore = [line for line in generateLogicalLines(linesbefore)]
extractedLines = [line for line in generateLogicalLines(extractedLines)]
linesafter = [line for line in generateLogicalLines(linesafter)]
if self.startline == self.endline:
# need to include the line code is extracted from
line = generateLogicalLines(lines[self.startline - self.fn.getStartLine():]).next()
linesbefore.append(line[:self.startcol] + "dummyFn()" + line[self.endcol:])
assigns = getAssignments(linesbefore)
fnargs = getFunctionArgs(linesbefore)
candidateArgs = assigns + fnargs
refs = getVariableReferencesInLines(extractedLines)
self.newfn.args = [ref for ref in refs if ref in candidateArgs]
assignsInExtractedBlock = getAssignments(extractedLines)
usesAfterNewFunctionCall = getVariableReferencesInLines(linesafter)
usesInPreceedingLoop = getVariableReferencesInLines(
self.getPreceedingLinesInLoop(linesbefore,line))
self.newfn.retvals = [ref for ref in usesInPreceedingLoop+usesAfterNewFunctionCall
if ref in assignsInExtractedBlock]
def getRegionToInline(sourcenode, defn):
line, linecount = getLineAndContinues(sourcenode, defn.lineno)
start, end = findRegionToInline(maskStringsAndRemoveComments(line))
return line[start:end], linecount
