def _handle_buildin(self, node, method_name):
if method_name == "open":
arglist = node.children[1].children[1]
first_arg, remaining_args = self._split_arguments(arglist)
x = Call(Name("Path"), first_arg, prefix=node.prefix)
x.append_child(Call(Name("open"), remaining_args, prefix="."))
return x
else:
raise ValueError("Method not found")
def transform(self, node: Node, results):
if "imp" in results:
return self._handle_import(node, results)
else:
if "buildin" in results:
method_name = results["buildin"].value
return self._handle_buildin(node, method_name)
if isinstance(results['method'], Leaf):
method_name = results["method"].value
else:
method_name = results["method"][0].value
if "obj" not in results:
# Prefix: Path.
return self._handle_no_args(node, method_name)
else:
obj = results["obj"]
argname = obj.clone()
if "submod" in results:
if method_name == "join" and len(argname.children) >= 1:
first_arg, remaining_args = self._split_arguments(
argname)
x = Call(Name("Path"), first_arg, prefix=node.prefix)
if len(remaining_args) > 0 and all(
a.type in [token.COMMA, token.STRING]
for a in remaining_args):
if str(remaining_args[0].value).startswith('*'):
x.append_child(
Call(Name('joinpath', prefix=""),
remaining_args,
prefix="."))
return x
x.append_child(Slash)
for i, e in enumerate(remaining_args):
if isinstance(e.value, Node):
val = e.value
elif isinstance(e.value, Leaf):
val = e.value.value
else:
continue
if e.type == token.STRING and val != ",":
# if self.split_strings and "/" in e.value:
# # TODO: get more robust e.value without quotes
# parts = re.split('(/|\\\\)', e.value[1:-1])
# for part in parts:
# if part in ["/", "\\"]:
# x.append_child(Slash)
# else:
# x.append_child(String('"{}"'.format(part), prefix=" "))
# else:
if i < 1:
p = " "
else:
p = ""
x.append_child(String(e, prefix=p))
else:
x.append_child(Slash)
return x
else:
first_arg, remaining_args = self._split_arguments(
argname)
x = Call(Name("Path"), first_arg, prefix=node.prefix)
new_names = {
"isabs": "is_absolute",
"isdir": "is_dir",
"isfile": "is_file",
"islink": "is_symlink",
"abspath": "resolve",
"realpath": "resolve",
"normpath": "resolve",
"same": "samefile",
}
new_attribs = {
"basename": "name",
"dirname": "parent",
"getsize": "stat().st_size",
}
if method_name in new_names:
x.append_child(
Call(Name(new_names[method_name], prefix=""),
remaining_args,
prefix="."))
elif method_name in new_attribs:
x.append_child(
String(new_attribs[method_name], prefix="."))
else:
x.append_child(
Call(Name(method_name, prefix=""),
remaining_args,
prefix="."))
return x
else:
arglist = argname
first_arg, remaining_args = self._split_arguments(arglist)
x = Call(Name("Path"), first_arg, prefix=node.prefix)
if method_name == "remove":
method_name = "unlink"
if method_name == "listdir":
x.append_child(String('glob("*")', prefix="."))
x.prefix = ""
return Call(Name('list'), [x], prefix=node.prefix)
elif method_name == 'makedirs':
if len(remaining_args) > 0:
children = [
Leaf(12, ','),
Leaf(1, 'parents', prefix=' '),
Leaf(22, '='),
Leaf(3, 'True')
]
remaining_args += [
Node(type=260, children=children)
]
else:
remaining_args = [
Leaf(1, 'parents', prefix=''),
Leaf(22, '='),
Leaf(3, 'True')
]
x.append_child(
Call(Name("mkdir"), remaining_args, prefix="."))
else:
x.append_child(
Call(Name(method_name), remaining_args,
prefix="."))
return x
