def _get_call_parameters_str(call_node):
"""Get parameters string for a call node."""
if not isinstance(call_node, ast.Call):
raise NodeTypeNotSupport('It is not ast.Call node type.')
parameters_str = ''
call_str = pasta.dump(call_node)
call_name = pasta.dump(call_node.func)
last_parameter_str = ''
if call_node.args:
last_parameter_str = pasta.dump(call_node.args[-1])
if call_node.keywords:
last_parameter_str = pasta.dump(call_node.keywords[-1])
if last_parameter_str:
left_parenthesis_pos = call_str.find(call_name) + len(call_name)
# call is like abc.call(a, b,), last parameter is b,
# but parameters string must have last ',' character after the last parameter b.
last_parameter_pos = call_str.rfind(last_parameter_str) + len(
last_parameter_str)
right_parenthesis_pos = call_str.find(')', last_parameter_pos)
# parameters start pos must skip '(' character for calling.
parameters_str = call_str[left_parenthesis_pos +
1:right_parenthesis_pos]
return parameters_str
def _modify_function_name(func_def_node, new_func_name):
"""Modify function name"""
if not isinstance(func_def_node, ast.FunctionDef):
raise NodeTypeNotSupport('It is not ast.FunctionDef node type.')
old_func_name = func_def_node.name
func_def_node.name = new_func_name
# Modify formatting information stored by pasta
old_function_def = fmt.get(func_def_node, 'function_def')
if old_function_def:
new_function_def = old_function_def.replace(
old_func_name, new_func_name)
fmt.set(func_def_node, 'function_def', new_function_def)
fmt.set(func_def_node, 'name__src', new_func_name)
def mapping_api(self, call_node, check_context=True):
"""
Convert api_name in code to MindSpore api, if api_name is a python api, code will not convert.
If do not check context of the script, the code represented by the node must be written in the standard way.
Args:
call_node (ast.Call): The ast node to convert.
check_context (boolean): If True, the code context will be checked. Default is True.
Returns:
str, the converted code.
"""
if not isinstance(call_node, ast.Call):
raise NodeTypeNotSupport("It is not ast.Call node.")
code = pasta.dump(call_node)
api_call_name = pasta.dump(call_node.func)
if api_call_name.startswith('self.'):
return code
new_code = self._mapping_api(call_node, check_context)
return new_code