def _getNativePythonFunctionDefFromWithBlockDescription(self,
withBlockDescription,
objectIdToObjectDefinition):
sourceText = objectIdToObjectDefinition[withBlockDescription.sourceFileId].text
sourceLineOffsets = PythonAstConverter.computeLineOffsets(sourceText)
sourceTree = ast.parse(sourceText)
withTree = PyAstUtil.withBlockAtLineNumber(
sourceTree,
withBlockDescription.lineNumber
)
withBodyAsFunctionAst = ast.FunctionDef(
name="__withBodyFunction",
lineno=withTree.lineno,
col_offset=withTree.col_offset,
args=ast.arguments(
args=[],
vararg=None,
kwarg=None,
defaults=[]),
body=withTree.body,
decorator_list=[]
)
nativeWithBodyAst = PythonAstConverter.convertPythonAstToForaPythonAst(
withBodyAsFunctionAst,
sourceLineOffsets
)
return nativeWithBodyAst
