Skip to content

Commit

Permalink
chore: better naming
Browse files Browse the repository at this point in the history
  • Loading branch information
Mattwmaster58 committed Oct 23, 2024
1 parent 2626dd3 commit 4e49e0f
Show file tree
Hide file tree
Showing 18 changed files with 22 additions and 22 deletions.
38 changes: 19 additions & 19 deletions playwright_stealth/stealth.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,22 +20,22 @@ def from_file(name) -> str:
SCRIPTS: Dict[str, str] = {
"generate_magic_arrays": from_file("generate.magic.arrays.js"),
"utils": from_file("utils.js"),
"chrome_app": from_file("mitigations/chrome.app.js"),
"chrome_csi": from_file("mitigations/chrome.csi.js"),
"chrome_hairline": from_file("mitigations/chrome.hairline.js"),
"chrome_load_times": from_file("mitigations/chrome.load.times.js"),
"chrome_runtime": from_file("mitigations/chrome.runtime.js"),
"iframe_content_window": from_file("mitigations/iframe.contentWindow.js"),
"media_codecs": from_file("mitigations/media.codecs.js"),
"navigator_hardware_concurrency": from_file("mitigations/navigator.hardwareConcurrency.js"),
"navigator_languages": from_file("mitigations/navigator.languages.js"),
"navigator_permissions": from_file("mitigations/navigator.permissions.js"),
"navigator_platform": from_file("mitigations/navigator.platform.js"),
"navigator_plugins": from_file("mitigations/navigator.plugins.js"),
"navigator_user_agent": from_file("mitigations/navigator.userAgent.js"),
"navigator_vendor": from_file("mitigations/navigator.vendor.js"),
"navigator_webdriver": from_file("mitigations/navigator.webdriver.js"),
"webgl_vendor": from_file("mitigations/webgl.vendor.js"),
"chrome_app": from_file("evasions/chrome.app.js"),
"chrome_csi": from_file("evasions/chrome.csi.js"),
"chrome_hairline": from_file("evasions/chrome.hairline.js"),
"chrome_load_times": from_file("evasions/chrome.load.times.js"),
"chrome_runtime": from_file("evasions/chrome.runtime.js"),
"iframe_content_window": from_file("evasions/iframe.contentWindow.js"),
"media_codecs": from_file("evasions/media.codecs.js"),
"navigator_hardware_concurrency": from_file("evasions/navigator.hardwareConcurrency.js"),
"navigator_languages": from_file("evasions/navigator.languages.js"),
"navigator_permissions": from_file("evasions/navigator.permissions.js"),
"navigator_platform": from_file("evasions/navigator.platform.js"),
"navigator_plugins": from_file("evasions/navigator.plugins.js"),
"navigator_user_agent": from_file("evasions/navigator.userAgent.js"),
"navigator_vendor": from_file("evasions/navigator.vendor.js"),
"navigator_webdriver": from_file("evasions/navigator.webdriver.js"),
"webgl_vendor": from_file("evasions/webgl.vendor.js"),
}

from typing import Tuple, Optional
Expand Down Expand Up @@ -190,7 +190,7 @@ def use_async(self, ctx: async_api.PlaywrightContextManager) -> AsyncWrappingCon
"""
Instruments the playwright context manager.
Any browser connected to or any page created with any method from
the patched context should have stealth mitigations applied automatically.
the patched context should have stealth evasions applied automatically.
async with Stealth().use_async(async_playwright()) as p:
...
Expand All @@ -201,7 +201,7 @@ def use_sync(self, ctx: sync_api.PlaywrightContextManager) -> SyncWrappingContex
"""
Instruments the playwright context manager.
Any browser connected to or any page created with any method from
the patched context should have stealth mitigations applied automatically.
the patched context should have stealth evasions applied automatically.
with Stealth().use_sync(sync_playwright()) as p:
...
Expand Down Expand Up @@ -356,7 +356,7 @@ def _patch_cli_arg(existing_args: List[str], flag: str) -> List[str]:
return new_args


ALL_DISABLED_KWARGS = {
ALL_EVASIONS_DISABLED_KWARGS = {
"navigator_webdriver": False,
"webgl_vendor": False,
"chrome_app": False,
Expand Down
6 changes: 3 additions & 3 deletions tests/test_basics.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from playwright.async_api import async_playwright
from playwright.sync_api import sync_playwright, Browser

from playwright_stealth.stealth import Stealth, ALL_DISABLED_KWARGS
from playwright_stealth.stealth import Stealth, ALL_EVASIONS_DISABLED_KWARGS


@pytest.mark.parametrize("browser_type", ["chromium", "firefox"])
Expand Down Expand Up @@ -41,7 +41,7 @@ def test_sync_navigator_webdriver_smoketest(hooked_sync_browser):


def test_payload_is_empty_when_no_evasions_active():
assert len(Stealth(**ALL_DISABLED_KWARGS).script_payload) == 0
assert len(Stealth(**ALL_EVASIONS_DISABLED_KWARGS).script_payload) == 0

def test_empty_payload_not_injected():
init_script_added = False
Expand All @@ -52,7 +52,7 @@ def add_init_script(self, *args, **kwargs):
init_script_added = True

# noinspection PyTypeChecker
Stealth(**ALL_DISABLED_KWARGS).apply_stealth_sync(MockBrowser())
Stealth(**ALL_EVASIONS_DISABLED_KWARGS).apply_stealth_sync(MockBrowser())
assert not init_script_added


0 comments on commit 4e49e0f

Please sign in to comment.