Fix function attributes

Author: Vinyzu

patch_python_package.py

@@ -211,7 +211,7 @@ def patch_file(file_path: str, patched_tree: ast.AST) -> None:
                 if isinstance(subnode, ast.Return) and isinstance(subnode.value, ast.Call):
                     if subnode.value.args and isinstance(subnode.value.args[0], ast.Await):
                         inner_call = subnode.value.args[0].value
-                        if isinstance(inner_call, ast.Call) and inner_call.func.attr == "send":
+                        if isinstance(inner_call, ast.Call) and isinstance(inner_call.func, ast.Attribute) and inner_call.func.attr == "send":
                             for i, arg in enumerate(inner_call.args):
                                 if isinstance(arg, ast.Call) and arg.func.id == "dict":
                                     arg.keywords.append(ast.keyword(
@@ -257,7 +257,7 @@ def patch_file(file_path: str, patched_tree: ast.AST) -> None:
                 if isinstance(subnode, ast.Return) and isinstance(subnode.value, ast.Call):
                     if subnode.value.args and isinstance(subnode.value.args[0], ast.Await):
                         inner_call = subnode.value.args[0].value
-                        if isinstance(inner_call, ast.Call) and inner_call.func.attr == "send":
+                        if isinstance(inner_call, ast.Call) and isinstance(inner_call.func, ast.Attribute) and inner_call.func.attr == "send":
                             for i, arg in enumerate(inner_call.args):
                                 if isinstance(arg, ast.Call) and arg.func.id == "dict":
                                     arg.keywords.append(ast.keyword(
@@ -559,7 +559,7 @@ async def route_handler(route: Route) -> None:
                     if isinstance(subnode, ast.Return) and isinstance(subnode.value, ast.Call):
                         if subnode.value.args and isinstance(subnode.value.args[0], ast.Await):
                             inner_call = subnode.value.args[0].value
-                            if isinstance(inner_call, ast.Call) and inner_call.func.attr == "send":
+                            if isinstance(inner_call, ast.Call) and isinstance(inner_call.func, ast.Attribute) and inner_call.func.attr == "send":
                                 for i, arg in enumerate(inner_call.args):
                                     if isinstance(arg, ast.Call) and arg.func.id == "dict":
                                         arg.keywords.append(ast.keyword(
@@ -621,7 +621,7 @@ async def route_handler(route: Route) -> None:
 
                     # Modify the inner function call inside return statement
                     for subnode in ast.walk(node):
-                        if isinstance(subnode, ast.Call) and subnode.func.attr == node.name:
+                        if isinstance(subnode, ast.Call) and isinstance(subnode.func, ast.Attribute) and subnode.func.attr == node.name:
                             subnode.keywords.append(
                                 ast.keyword(arg="isolatedContext",
                                             value=ast.Name(id="isolated_context",
@@ -641,7 +641,7 @@ async def route_handler(route: Route) -> None:
                     node.args.kw_defaults.append(ast.Constant(value=None))
 
                     for subnode in ast.walk(node):
-                        if isinstance(subnode, ast.Call) and subnode.func.attr == node.name:
+                        if isinstance(subnode, ast.Call) and isinstance(subnode.func, ast.Attribute) and subnode.func.attr == node.name:
                             subnode.keywords.append(
                                 ast.keyword(arg="focusControl", value=ast.Name(id="focus_control", ctx=ast.Load()))
                             )
@@ -692,7 +692,7 @@ async def route_handler(route: Route) -> None:
 
                     # Modify the inner function call inside return statement
                     for subnode in ast.walk(node):
-                        if isinstance(subnode, ast.Call) and subnode.func.attr == node.name:
+                        if isinstance(subnode, ast.Call) and isinstance(subnode.func, ast.Attribute) and subnode.func.attr == node.name:
                             subnode.keywords.append(
                                 ast.keyword(arg="isolatedContext",
                                             value=ast.Name(id="isolated_context",
@@ -715,7 +715,7 @@ async def route_handler(route: Route) -> None:
 
                     if class_node.name != "BrowserContext" and node.name != "new_page":
                         for subnode in ast.walk(node):
-                            if isinstance(subnode, ast.Call) and subnode.func.attr == node.name:
+                            if isinstance(subnode, ast.Call) and isinstance(subnode.func, ast.Attribute) and subnode.func.attr == node.name:
                                 subnode.keywords.append(
                                     ast.keyword(arg="focusControl", value=ast.Name(id="focus_control", ctx=ast.Load()))
                                 )