def extend_subscript_for_input(node, extension):
if isinstance(node.slice, ast.Index):
node = copy.deepcopy(node)
idx = node.slice.value
if isinstance(idx, ast.Tuple):
new_idx = ast.Tuple([extension] + idx.elts, ast.Load())
else:
new_idx = ast.Tuple([extension, idx], ast.Load())
node.slice.value = new_idx
else:
raise Exception("Unhandled node indexing: '%s'" % (unparse(node).rstrip()))
return node
评论列表
文章目录