Spaces:
Runtime error
Runtime error
Fix non unicode encoding
Browse files- utils/tree_utils.py +3 -3
utils/tree_utils.py
CHANGED
|
@@ -13,7 +13,7 @@ def replace_function(old_func_node, new_func_node):
|
|
| 13 |
tree = give_tree(old_func_node)
|
| 14 |
old_func_start, old_func_end = node_str_idx(old_func_node)
|
| 15 |
# new_func_start, new_func_end = node_str_idx(new_func_node)
|
| 16 |
-
new_code = tree.text.decode()
|
| 17 |
return new_code
|
| 18 |
|
| 19 |
def get_root(node):
|
|
@@ -47,7 +47,7 @@ def parse_functions(in_code):
|
|
| 47 |
returns all functions in the code as their actual nodes.
|
| 48 |
includes any comment made directly after the function definition or diretly after #copilot trigger
|
| 49 |
"""
|
| 50 |
-
tree = parser.parse(bytes(in_code, "
|
| 51 |
funcs = [n for n in tree.root_node.children if n.type == "function_definition"]
|
| 52 |
|
| 53 |
return funcs
|
|
@@ -81,7 +81,7 @@ def full_func_head(func_node) -> str:
|
|
| 81 |
cursor.goto_next_sibling()
|
| 82 |
end = cursor.node.start_point
|
| 83 |
# return "\n".join(func_node.text.decode().split("\n")[:(end[0]-func_node.start_point[0])])[:-(last_char)-1]
|
| 84 |
-
return func_node.text
|
| 85 |
|
| 86 |
def grab_before_comments(func_node):
|
| 87 |
"""
|
|
|
|
| 13 |
tree = give_tree(old_func_node)
|
| 14 |
old_func_start, old_func_end = node_str_idx(old_func_node)
|
| 15 |
# new_func_start, new_func_end = node_str_idx(new_func_node)
|
| 16 |
+
new_code = tree.text[:old_func_start].decode() + new_func_node.text.decode() + tree.text[old_func_end:].decode()
|
| 17 |
return new_code
|
| 18 |
|
| 19 |
def get_root(node):
|
|
|
|
| 47 |
returns all functions in the code as their actual nodes.
|
| 48 |
includes any comment made directly after the function definition or diretly after #copilot trigger
|
| 49 |
"""
|
| 50 |
+
tree = parser.parse(bytes(in_code, encoding="utf-8"))
|
| 51 |
funcs = [n for n in tree.root_node.children if n.type == "function_definition"]
|
| 52 |
|
| 53 |
return funcs
|
|
|
|
| 81 |
cursor.goto_next_sibling()
|
| 82 |
end = cursor.node.start_point
|
| 83 |
# return "\n".join(func_node.text.decode().split("\n")[:(end[0]-func_node.start_point[0])])[:-(last_char)-1]
|
| 84 |
+
return func_node.text[:(last_char - func_node.start_byte)].decode()
|
| 85 |
|
| 86 |
def grab_before_comments(func_node):
|
| 87 |
"""
|