diff --git a/pydelta/ids_refactor.py b/pydelta/ids_refactor.py index 7d1256b..d114f56 100644 --- a/pydelta/ids_refactor.py +++ b/pydelta/ids_refactor.py @@ -128,6 +128,11 @@ def _replace_identifiers(self, node): for target in stmt.targets: if isinstance(target, ast.Name) and target.id in self.mapping: target.id = self.mapping[target.id] + elif isinstance(node, ast.Call): + for keyword in node.keywords: + if keyword.arg in self.mapping: + keyword.arg = self.mapping[keyword.arg] + for child in ast.iter_child_nodes(node): self._replace_identifiers(child) - return ast.unparse(node) \ No newline at end of file + return ast.unparse(node)