def var_id(var_node): if isinstance(var_node, ast.Name): return var_node.id elif isinstance(var_node, ast.Subscript): return var_node.value.id