import ast

import glob

import os

import toml

patchright_version = os.environ.get('playwright_version')

patchright_version = "1.52.4"

def patch_file(file_path: str, patched_tree: ast.AST) -> None:

with open(file_path, "w") as f:

f.write(ast.unparse(ast.fix_missing_locations(patched_tree)))

with open("playwright-python/playwright/_repo_version.py", "w") as f:

f.write(f"version = '{patchright_version}'")

with open("playwright-python/pyproject.toml", "r") as f:

pyproject_source = toml.load(f)

pyproject_source["project"]["name"] = "patchright"

pyproject_source["project"]["description"] = "Undetected Python version of the Playwright testing and automation library."

pyproject_source["project"]["authors"] = [{'name': 'Microsoft Corporation, patched by github.com/Kaliiiiiiiiii-Vinyzu/'}]

pyproject_source["project"]["urls"]["homepage"] = "https://github.com/Kaliiiiiiiiii-Vinyzu/patchright-python"

pyproject_source["project"]["urls"]["Release notes"] = "https://github.com/Kaliiiiiiiiii-Vinyzu/patchright-python/releases"

pyproject_source["project"]["urls"]["Bug Reports"] = "https://github.com/Kaliiiiiiiiii-Vinyzu/patchright-python/issues"

pyproject_source["project"]["urls"]["homeSource Codepage"] = "https://github.com/Kaliiiiiiiiii-Vinyzu/patchright-python"

del pyproject_source["project"]["scripts"]["playwright"]

pyproject_source["project"]["scripts"]["patchright"] = "patchright.__main__:main"

pyproject_source["project"]["entry-points"]["pyinstaller40"]["hook-dirs"] = "patchright._impl.__pyinstaller:get_hook_dirs"

pyproject_source["tool"]["setuptools"]["packages"] = ['patchright', 'patchright.async_api', 'patchright.sync_api', 'patchright._impl', 'patchright._impl.__pyinstaller']

pyproject_source["tool"]["setuptools_scm"] = {'version_file': 'patchright/_repo_version.py'}

with open("playwright-python/pyproject.toml", "w") as f:

toml.dump(pyproject_source, f)

with open("playwright-python/setup.py") as f:

setup_source = f.read()

setup_tree = ast.parse(setup_source)

for node in ast.walk(setup_tree):

# Modify driver_version

if isinstance(node, ast.Assign) and isinstance(node.value, ast.Constant) and isinstance(node.targets[0], ast.Name):

if node.targets[0].id == "driver_version" and node.value.value.startswith("1."):

node.value.value = node.value.value.split("-")[0]

# Modify url

if isinstance(node, ast.Assign) and isinstance(node.value, ast.Constant) and isinstance(node.targets[0], ast.Name):

if node.targets[0].id == "url" and node.value.value == "https://playwright.azureedge.net/builds/driver/":

node.value = ast.JoinedStr(

values=[

ast.Constant(value='https://github.com/Kaliiiiiiiiii-Vinyzu/patchright/releases/download/v'),

ast.FormattedValue(value=ast.Name(id='driver_version', ctx=ast.Load()), conversion=-1),

ast.Constant(value='/')

]

)

# Modify Curl Call

if isinstance(node, ast.Call) and isinstance(node.func, ast.Attribute) and isinstance(node.func.value, ast.Name) and len(node.args) >= 1 and isinstance(node.args[0], ast.List) and len(node.args[0].elts) == 4:

if node.func.value.id == "subprocess" and node.func.attr == "check_call" and node.args[0].elts[0].value == "curl":

node.args[0].elts.insert(1, ast.Constant(value="-L"))

# Modify Shutil Call

if isinstance(node, ast.Call) and isinstance(node.func, ast.Attribute) and isinstance(node.func.value, ast.Name) and len(node.args) >= 1 and isinstance(node.args[0], ast.Constant):

if node.func.value.id == "shutil" and node.func.attr == "rmtree" and node.args[0].value == "playwright.egg-info":

node.args[0].value = "patchright.egg-info"

# Modify Os Makedirs Call

if isinstance(node, ast.Call) and isinstance(node.func, ast.Attribute) and isinstance(node.func.value, ast.Name) and len(node.args) >= 1 and isinstance(node.args[0], ast.Constant):

if node.func.value.id == "os" and node.func.attr == "makedirs" and node.args[0].value == "playwright/driver":

node.args[0].value = "patchright/driver"

# Modify Zip Write Call

if isinstance(node, ast.Call) and isinstance(node.func, ast.Attribute) and isinstance(node.func.value, ast.Name) and len(node.args) >= 2 and isinstance(node.args[1], ast.JoinedStr):

if node.func.value.id == "zip" and node.func.attr == "write" and node.args[1].values[0].value == "playwright/driver/":

node.args[1].values[0].value = "patchright/driver/"

# Modify Zip Writestr Call

if isinstance(node, ast.Call) and isinstance(node.func, ast.Attribute) and isinstance(node.func.value, ast.Name) and len(node.args) >= 1 and isinstance(node.args[0], ast.Constant):

if node.func.value.id == "zip" and node.func.attr == "writestr" and node.args[0].value == "playwright/driver/README.md":

node.args[0].value = "patchright/driver/README.md"

# Modify Extractall Call

if isinstance(node, ast.Call) and isinstance(node.func, ast.Name) and len(node.args) >= 2 and isinstance(node.args[1], ast.Constant):

if node.func.id == "extractall" and node.args[1].value == "playwright/driver":

node.args[1].value = "patchright/driver"

# Modify Setup Call

if isinstance(node, ast.Call) and isinstance(node.func, ast.Name):

if node.func.id == "setup":

node.keywords.append(ast.keyword(

arg="version",

value=ast.Constant(value=patchright_version)

))

patch_file("playwright-python/setup.py", setup_tree)

with open("playwright-python/playwright/_impl/__pyinstaller/hook-playwright.async_api.py") as f:

async_api_source = f.read()

async_api_tree = ast.parse(async_api_source)

for node in ast.walk(async_api_tree):

if isinstance(node, ast.Call) and isinstance(node.func, ast.Name) and len(node.args) == 1 and isinstance(node.args[0], ast.Constant):

if node.func.id == "collect_data_files" and node.args[0].value == "playwright":

node.args[0].value = "patchright"

patch_file("playwright-python/playwright/_impl/__pyinstaller/hook-playwright.async_api.py", async_api_tree)

with open("playwright-python/playwright/_impl/__pyinstaller/hook-playwright.sync_api.py") as f:

async_api_source = f.read()

async_api_tree = ast.parse(async_api_source)

for node in ast.walk(async_api_tree):

if isinstance(node, ast.Call) and isinstance(node.func, ast.Name) and len(node.args) == 1 and isinstance(node.args[0], ast.Constant):

if node.func.id == "collect_data_files" and node.args[0].value == "playwright":

node.args[0].value = "patchright"

patch_file("playwright-python/playwright/_impl/__pyinstaller/hook-playwright.sync_api.py", async_api_tree)

with open("playwright-python/playwright/_impl/_driver.py") as f:

driver_source = f.read()

driver_tree = ast.parse(driver_source)

for node in ast.walk(driver_tree):

if isinstance(node, ast.Call) and isinstance(node.func, ast.Attribute) and isinstance(node.func.value, ast.Name) and len(node.args) >= 1 and isinstance(node.args[0], ast.Name):

if node.func.value.id == "inspect" and node.func.attr == "getfile" and node.args[0].id == "playwright":

node.args[0].id = "patchright"

patch_file("playwright-python/playwright/_impl/_driver.py", driver_tree)

with open("playwright-python/playwright/_impl/_connection.py") as f:

connection_source = f.read()

connection_source_tree = ast.parse(connection_source)

for node in ast.walk(connection_source_tree):

if isinstance(node, ast.Call) and isinstance(node.func, ast.Name) and len(node.args) >= 1 and isinstance(node.args[0], ast.Attribute):

if node.func.id == "Path" and node.args[0].value.id == "playwright":

node.args[0].value.id = "patchright"

elif isinstance(node, ast.Attribute) and isinstance(node.value, ast.Attribute) and isinstance(node.value.value, ast.Attribute) and isinstance(node.value.value.value, ast.Name):

if node.value.value.value.id == "playwright" and node.value.value.attr == "_impl" and node.value.attr == "_impl_to_api_mapping":

node.value.value.value.id = "patchright"

patch_file("playwright-python/playwright/_impl/_connection.py", connection_source_tree)

with open("playwright-python/playwright/_impl/_js_handle.py") as f:

js_handle_source = f.read()

js_handle_tree = ast.parse(js_handle_source)

for node in ast.walk(js_handle_tree):

if isinstance(node, ast.FunctionDef) and node.name == "add_source_url_to_script":

for function_node in node.body:

if isinstance(function_node, ast.Return):

function_node.value = ast.Name(id="source", ctx=ast.Load())

if isinstance(node, ast.AsyncFunctionDef) and node.name in ["evaluate", "evaluate_handle"]:

node.args.kwonlyargs.append(ast.arg(

arg="isolatedContext",

annotation=ast.Subscript(

value=ast.Name(id="Optional", ctx=ast.Load()),

slice=ast.Name(id="bool", ctx=ast.Load()),

ctx=ast.Load(),

),

))

node.args.kw_defaults.append(ast.Constant(value=True))

for subnode in ast.walk(node):

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":

for i, arg in enumerate(inner_call.args):

if isinstance(arg, ast.Call) and arg.func.id == "dict":

arg.keywords.append(ast.keyword(

arg="isolatedContext",

value=ast.Name(id="isolatedContext", ctx=ast.Load())

))

patch_file("playwright-python/playwright/_impl/_js_handle.py", js_handle_tree)

with open("playwright-python/playwright/_impl/_frame.py") as f:

frame_source = f.read()

frame_tree = ast.parse(frame_source)

for node in ast.walk(frame_tree):

if isinstance(node, ast.AsyncFunctionDef) and node.name in ["evaluate", "evaluate_handle"]:

node.args.kwonlyargs.append(ast.arg(

arg="isolatedContext",

annotation=ast.Subscript(

value=ast.Name(id="Optional", ctx=ast.Load()),

slice=ast.Name(id="bool", ctx=ast.Load()),

ctx=ast.Load(),

),

))

node.args.kw_defaults.append(ast.Constant(value=True))

for subnode in ast.walk(node):

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":

for i, arg in enumerate(inner_call.args):

if isinstance(arg, ast.Call) and arg.func.id == "dict":

arg.keywords.append(ast.keyword(

arg="isolatedContext",

value=ast.Name(id="isolatedContext", ctx=ast.Load())

))

patch_file("playwright-python/playwright/_impl/_frame.py", frame_tree)

with open("playwright-python/playwright/_impl/_locator.py") as f:

frame_source = f.read()

frame_tree = ast.parse(frame_source)

for node in ast.walk(frame_tree):

if isinstance(node, ast.AsyncFunctionDef) and node.name in ["evaluate", "evaluate_handle"]:

node.args.kwonlyargs.append(ast.arg(

arg="isolatedContext",

annotation=ast.Subscript(

value=ast.Name(id="Optional", ctx=ast.Load()),

slice=ast.Name(id="bool", ctx=ast.Load()),

ctx=ast.Load(),

),

))

node.args.kw_defaults.append(ast.Constant(value=True))

for subnode in ast.walk(node):

if isinstance(subnode, ast.Return) and isinstance(subnode.value, ast.Await):

call_expr = subnode.value.value

if isinstance(call_expr, ast.Call):

if node.name in ["evaluate", "evaluate_handle"] and isinstance(call_expr.func, ast.Attribute):

if call_expr.func.attr == "_with_element":

if call_expr.args and isinstance(call_expr.args[0], ast.Lambda):

lambda_func = call_expr.args[0].body

if isinstance(lambda_func, ast.Call) and isinstance(lambda_func.func, ast.Attribute):

if lambda_func.func.attr == node.name:

lambda_func.keywords.append(ast.keyword(

arg="isolatedContext",

value=ast.Name(id="isolatedContext", ctx=ast.Load())

))

patch_file("playwright-python/playwright/_impl/_locator.py", frame_tree)

with open("playwright-python/playwright/_impl/_browser_context.py") as f:

browser_context_source = f.read()

browser_context_tree = ast.parse(browser_context_source)

for node in ast.walk(browser_context_tree):

if isinstance(node, ast.ClassDef) and node.name == "BrowserContext":

for class_node in node.body:

if isinstance(class_node, ast.AsyncFunctionDef) and class_node.name == "add_init_script":

class_node.body.insert(0, ast.parse("await self.install_inject_route()"))

elif isinstance(class_node, ast.AsyncFunctionDef) and class_node.name == "expose_binding":

class_node.body.insert(0, ast.parse("await self.install_inject_route()"))

node.body.append(

ast.Assign(

targets=[ast.Name(id='route_injecting', ctx=ast.Store())],

value=ast.Constant(value=False))

)

node.body.append(

ast.parse("""\

self.route_injecting = True""").body[0])

patch_file("playwright-python/playwright/_impl/_browser_context.py", browser_context_tree)

with open("playwright-python/playwright/_impl/_page.py") as f:

page_source = f.read()

page_tree = ast.parse(page_source)

for node in ast.walk(page_tree):

if isinstance(node, ast.ClassDef) and node.name == "Page":

for class_node in node.body:

if isinstance(class_node, ast.AsyncFunctionDef) and class_node.name == "add_init_script":

class_node.body.insert(0, ast.parse("await self.install_inject_route()"))

elif isinstance(class_node, ast.AsyncFunctionDef) and class_node.name == "expose_binding":

class_node.body.insert(0, ast.parse("await self.install_inject_route()"))

node.body.append(

ast.Assign(

targets=[ast.Name(id='route_injecting', ctx=ast.Store())],

value=ast.Constant(value=False))

)

node.body.append(

ast.parse("""\

self.route_injecting = True""").body[0])

if isinstance(node, ast.AsyncFunctionDef) and node.name in ["evaluate", "evaluate_handle"]:

node.args.kwonlyargs.append(ast.arg(

arg="isolatedContext",

annotation=ast.Subscript(

value=ast.Name(id="Optional", ctx=ast.Load()),

slice=ast.Name(id="bool", ctx=ast.Load()),

ctx=ast.Load(),

),

))

node.args.kw_defaults.append(ast.Constant(value=True))

if "evaluateExpression" in ast.unparse(node.body[0]):

for subnode in ast.walk(node):

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":

for i, arg in enumerate(inner_call.args):

if isinstance(arg, ast.Call) and arg.func.id == "dict":

arg.keywords.append(ast.keyword(

arg="isolatedContext",

value=ast.Name(id="isolatedContext", ctx=ast.Load())

))

elif "_main_frame" in ast.unparse(node.body[0]):

for subnode in ast.walk(node):

if isinstance(subnode, ast.Return) and isinstance(subnode.value, ast.Await) and isinstance(subnode.value.value, ast.Call):

subnode.value.value.keywords.append(ast.keyword(

arg="isolatedContext",

value=ast.Name(id="isolatedContext", ctx=ast.Load())

))

patch_file("playwright-python/playwright/_impl/_page.py", page_tree)

with open("playwright-python/playwright/_impl/_clock.py") as f:

clock_source = f.read()

clock_tree = ast.parse(clock_source)

for node in ast.walk(clock_tree):

if isinstance(node, ast.ClassDef) and node.name == "Clock":

for class_node in node.body:

if isinstance(class_node, ast.AsyncFunctionDef) and class_node.name == "install":

class_node.body.insert(0, ast.parse("await self._browser_context.install_inject_route()"))

patch_file("playwright-python/playwright/_impl/_clock.py", clock_tree)

with open("playwright-python/playwright/async_api/_generated.py") as f:

async_generated_source = f.read()

async_generated_tree = ast.parse(async_generated_source)

for class_node in ast.walk(async_generated_tree):

if isinstance(class_node, ast.ClassDef) and class_node.name in ["Page", "Frame", "Worker", "Locator"]:

for node in class_node.body:

if isinstance(node, ast.AsyncFunctionDef) and node.name in ["evaluate", "evaluate_handle"]: # , "evaluate_all"

new_arg = ast.arg(arg="isolated_context", annotation=ast.Subscript(

value=ast.Name(id="typing.Optional", ctx=ast.Load()),

slice=ast.Name(id="bool", ctx=ast.Load()),

ctx=ast.Load(),

))

# Append the argument to kwonlyargs

node.args.kwonlyargs.append(new_arg)

node.args.kw_defaults.append(ast.Constant(value=True))

# 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:

subnode.keywords.append(

ast.keyword(arg="isolatedContext",

value=ast.Name(id="isolated_context",

ctx=ast.Load())

)

)

patch_file("playwright-python/playwright/async_api/_generated.py", async_generated_tree)

with open("playwright-python/playwright/sync_api/_generated.py") as f:

async_generated_source = f.read()

async_generated_tree = ast.parse(async_generated_source)

for class_node in ast.walk(async_generated_tree):

if isinstance(class_node, ast.ClassDef) and class_node.name in ["Page", "Frame", "Worker", "Locator"]:

for node in class_node.body:

if isinstance(node, ast.FunctionDef) and node.name in ["evaluate", "evaluate_handle"]: # , "evaluate_all"

new_arg = ast.arg(arg="isolated_context", annotation=ast.Subscript(

value=ast.Name(id="typing.Optional", ctx=ast.Load()),

slice=ast.Name(id="bool", ctx=ast.Load()),

ctx=ast.Load(),

))

# Append the argument to kwonlyargs

node.args.kwonlyargs.append(new_arg)

node.args.kw_defaults.append(ast.Constant(value=True))

# 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:

subnode.keywords.append(

ast.keyword(arg="isolatedContext",

value=ast.Name(id="isolated_context",

ctx=ast.Load())

)

)

patch_file("playwright-python/playwright/sync_api/_generated.py", async_generated_tree)

for python_file in glob.glob("playwright-python/playwright/**.py") + glob.glob("playwright-python/playwright/**/**.py"):

with open(python_file) as f:

file_source = f.read()

file_tree = ast.parse(file_source)

renamed_attributes = []

for node in ast.walk(file_tree):

if isinstance(node, ast.Import):

for alias in node.names:

if alias.name.startswith("playwright"):

if "__init__" in python_file:

renamed_attributes.append(alias.name)

alias.name = alias.name.replace("playwright", "patchright", 1)

if isinstance(node, ast.ImportFrom) and node.module.startswith("playwright"):

node.module = node.module.replace("playwright", "patchright", 1)

if renamed_attributes and isinstance(node, ast.Attribute):

unparsed_attribute = ast.unparse(node.value)

if unparsed_attribute in renamed_attributes:

node.value = ast.parse(unparsed_attribute.replace("playwright", "patchright", 1)).body[0].value

patch_file(python_file, file_tree)

os.rename("playwright-python/playwright", "playwright-python/patchright")

with open("README.md", 'r') as src:

with open("playwright-python/README.md", 'w') as dst:

# Read from the source readme and write to the destination readme

dst.write(src.read())